diff options
| author | H Lohaus <hlohaus@users.noreply.github.com> | 2024-05-17 18:44:45 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-17 18:44:45 +0200 |
| commit | 91e7252123aa93fb003307c942d4596016020d6a (patch) | |
| tree | 85821a6c87f481d7366303ab06c36967b1b650c0 | |
| parent | Improve compatibility of openai-node (#1966) (diff) | |
| parent | Fix You.com model selection (diff) | |
| download | gpt4free-main.tar gpt4free-main.tar.gz gpt4free-main.tar.bz2 gpt4free-main.tar.lz gpt4free-main.tar.xz gpt4free-main.tar.zst gpt4free-main.zip | |
| -rw-r--r-- | g4f/Provider/You.py | 11 |
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: |