Update networks/message_streamer.py
Browse files
networks/message_streamer.py
CHANGED
@@ -26,9 +26,9 @@ class MessageStreamer:
|
|
26 |
self.model = model
|
27 |
else:
|
28 |
self.model = "default"
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
|
33 |
def parse_line(self, line):
|
34 |
try:
|
|
|
26 |
self.model = model
|
27 |
else:
|
28 |
self.model = "default"
|
29 |
+
self.model_fullname = self.MODEL_MAP[self.model]
|
30 |
+
self.message_outputer = OpenaiStreamOutputer()
|
31 |
+
self.stop_sequences = self.STOP_SEQUENCES_MAP.get(self.model, "")
|
32 |
|
33 |
def parse_line(self, line):
|
34 |
try:
|