Update networks/message_streamer.py

networks/message_streamer.py

@@ -28,6 +28,7 @@ class MessageStreamer:
             self.model = "default"
         self.model_fullname = self.MODEL_MAP[self.model]
         self.message_outputer = OpenaiStreamOutputer()
+        self.stop_sequences = self.STOP_SEQUENCES_MAP.get(self.model, "")
 
     def parse_line(self, line):
         try:
@@ -149,6 +150,8 @@ class MessageStreamer:
                 continue
     
             content = self.parse_line(line)
+
+            stop_sequence = getattr(self, 'stop_sequences', '')  # Default value if not set
     
             if content is not None and content.strip() == self.stop_sequences:
                 content_type = "Finished"