Merge pull request #976 from fenglui/master

fix msg.data.split(DELIMITER) exception when msg.data is int
binary-husky · Jul 21, 2023 · dd7a01c · dd7a01c
2 parents 61ba544 + 00a3b91
commit dd7a01c
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