Update networks/message_streamer.py

networks/message_streamer.py

@@ -106,7 +106,6 @@ class MessageStreamer:
         return stream_response
 
     def chat_return_dict(self, stream_response):
-        # https://platform.openai.com/docs/guides/text-generation/chat-completions-response-format
         final_output = self.message_outputer.default_data.copy()
         final_output["choices"] = [
             {
@@ -118,24 +117,28 @@ class MessageStreamer:
                 },
             }
         ]
+    
         logger.back(final_output)
-
+    
         final_content = ""
         for line in stream_response.iter_lines():
             if not line:
                 continue
+    
             content = self.parse_line(line)
-
-            if content.strip() == self.stop_sequences:
-                logger.success("\n[Finished]")
-                break
-            else:
-                logger.back(content, end="")
-                final_content += content
-
+            
+            # Check if content is None before calling strip()
+            if content is not None:
+                if content.strip() == self.stop_sequences:
+                    logger.success("\n[Finished]")
+                    break
+                else:
+                    logger.back(content, end="")
+                    final_content += content
+    
         if self.model in self.STOP_SEQUENCES_MAP.keys():
             final_content = final_content.replace(self.stop_sequences, "")
-
+    
         final_output["choices"][0]["message"]["content"] = final_content
         return final_output