From 5396887801ebf5db9ad59bbe1fc7b2d5e03fc585 Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
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