diff options
| -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: |