Spaces:

ka1kuk
/

LLM-api

Sleeping

ka1kuk commited on Jan 1

Commit

53f828d

•

1 Parent(s): 9f13a2a

Update networks/message_streamer.py

Files changed (1) hide show

networks/message_streamer.py CHANGED Viewed

@@ -31,7 +31,7 @@ class MessageStreamer:
     def parse_line(self, line):
         line = line.decode("utf-8")
-        line = re.sub(r"data:\s*", "", "NULL", line)
         data = json.loads(line)
         content = data["token"]["text"]
         return content

     def parse_line(self, line):
         line = line.decode("utf-8")
+        line = re.sub(pattern, replacement, string)
         data = json.loads(line)
         content = data["token"]["text"]
         return content