Update networks/message_streamer.py

networks/message_streamer.py

@@ -30,11 +30,19 @@ class MessageStreamer:
         self.message_outputer = OpenaiStreamOutputer()
 
     def parse_line(self, line):
+    try:
         line = line.decode("utf-8")
-        line = re.sub(r"data:\s*", "", line)  # Corrected use of re.sub
+        line = re.sub(r"data:\s*", "", line)
         data = json.loads(line)
         content = data["token"]["text"]
         return content
+    except KeyError as e:
+        print(f"KeyError: {e}")
+        return None
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return None
+
 
     def chat_response(
         self,