From 5396887801ebf5db9ad59bbe1fc7b2d5e03fc585 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Fri, 17 May 2024 18:40:34 +0200 Subject: Fix You.com model selection --- g4f/Provider/You.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'g4f') 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: -- cgit v1.2.3