ka1kuk commited on
Commit
807a492
·
1 Parent(s): f5ce7b4

Update networks/message_streamer.py

Browse files
Files changed (1) hide show
  1. networks/message_streamer.py +12 -12
networks/message_streamer.py CHANGED
@@ -30,18 +30,18 @@ class MessageStreamer:
30
  self.message_outputer = OpenaiStreamOutputer()
31
 
32
  def parse_line(self, line):
33
- try:
34
- line = line.decode("utf-8")
35
- line = re.sub(r"data:\s*", "", line)
36
- data = json.loads(line)
37
- content = data["token"]["text"]
38
- return content
39
- except KeyError as e:
40
- print(f"KeyError: {e}")
41
- return None
42
- except Exception as e:
43
- print(f"An error occurred: {e}")
44
- return None
45
 
46
 
47
  def chat_response(
 
30
  self.message_outputer = OpenaiStreamOutputer()
31
 
32
  def parse_line(self, line):
33
+ try:
34
+ line = line.decode("utf-8")
35
+ line = re.sub(r"data:\s*", "", line)
36
+ data = json.loads(line)
37
+ content = data["token"]["text"]
38
+ return content
39
+ except KeyError as e:
40
+ print(f"KeyError: {e}")
41
+ return None
42
+ except Exception as e:
43
+ print(f"An error occurred: {e}")
44
+ return None
45
 
46
 
47
  def chat_response(