summaryrefslogtreecommitdiffstats
path: root/g4f
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2024-05-17 18:44:45 +0200
committerGitHub <noreply@github.com>2024-05-17 18:44:45 +0200
commit91e7252123aa93fb003307c942d4596016020d6a (patch)
tree85821a6c87f481d7366303ab06c36967b1b650c0 /g4f
parentImprove compatibility of openai-node (#1966) (diff)
parentFix You.com model selection (diff)
downloadgpt4free-0.3.1.3.tar
gpt4free-0.3.1.3.tar.gz
gpt4free-0.3.1.3.tar.bz2
gpt4free-0.3.1.3.tar.lz
gpt4free-0.3.1.3.tar.xz
gpt4free-0.3.1.3.tar.zst
gpt4free-0.3.1.3.zip
Diffstat (limited to '')
-rw-r--r--g4f/Provider/You.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/g4f/Provider/You.py b/g4f/Provider/You.py
index 2b08b38a..d549423e 100644
--- a/g4f/Provider/You.py
+++ b/g4f/Provider/You.py
@@ -100,9 +100,10 @@ class You(AsyncGeneratorProvider, ProviderModelMixin):
"selectedChatMode": chat_mode,
}
if chat_mode == "custom":
- # print(f"You model: {model}")
- params["selectedAIModel"] = model.replace("-", "_")
-
+ if debug.logging:
+ print(f"You model: {model}")
+ params["selectedAiModel"] = model.replace("-", "_")
+
async with (session.post if chat_mode == "default" else session.get)(
f"{cls.url}/api/streamingSearch",
data=data,
@@ -117,9 +118,9 @@ class You(AsyncGeneratorProvider, ProviderModelMixin):
elif line.startswith(b'data: '):
if event in ["youChatUpdate", "youChatToken"]:
data = json.loads(line[6:])
- if event == "youChatToken" and event in data:
+ if event == "youChatToken" and event in data and data[event]:
yield data[event]
- elif event == "youChatUpdate" and "t" in data and data["t"] is not None:
+ elif event == "youChatUpdate" and "t" in data and data["t"]:
if chat_mode == "create":
match = re.search(r"!\[(.+?)\]\((.+?)\)", data["t"])
if match: