fix msg.data.split(DELIMITER) exception when msg.data is int

binary-husky · Jul 20, 2023 · 00a3b91 · 00a3b91
1 parent 61ba544
commit 00a3b91
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/request_llm/edge_gpt_free.py b/request_llm/edge_gpt_free.py
@@ -519,7 +519,11 @@ async def ask_stream(
         resp_txt_no_link = ""
         while not final:
             msg = await self.wss.receive()
-            objects = msg.data.split(DELIMITER)
+            try:
+                objects = msg.data.split(DELIMITER)
+            except :
+                continue
+
             for obj in objects:
                 if obj is None or not obj:
                     continue