Update networks/message_streamer.py

networks/message_streamer.py

@@ -34,16 +34,16 @@ class MessageStreamer:
             line = line.decode("utf-8")
             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
+    
+            # Check if 'token' key exists and return its 'text' value
+            if 'token' in data:
+                return data['token'].get('text', '')  # Return an empty string if 'text' is not found
+            else:
+                return None  # Return None if 'token' key is not present
         except Exception as e:
             print(f"An error occurred: {e}")
             return None
 
-
     def chat_response(
         self,
         prompt: str = None,