:boom: [Fix] Value of (delta_)content or content_type not assigned
Browse files
networks/huggingchat_streamer.py
CHANGED
@@ -224,6 +224,9 @@ class HuggingchatStreamer:
|
|
224 |
line = line.strip()
|
225 |
if not line:
|
226 |
continue
|
|
|
|
|
|
|
227 |
try:
|
228 |
data = json.loads(line, strict=False)
|
229 |
msg_type = data.get("type")
|
|
|
224 |
line = line.strip()
|
225 |
if not line:
|
226 |
continue
|
227 |
+
|
228 |
+
content = ""
|
229 |
+
content_type = "Completions"
|
230 |
try:
|
231 |
data = json.loads(line, strict=False)
|
232 |
msg_type = data.get("type")
|
networks/huggingface_streamer.py
CHANGED
@@ -22,6 +22,7 @@ class HuggingfaceStreamer:
|
|
22 |
line = line.decode("utf-8")
|
23 |
line = re.sub(r"data:\s*", "", line)
|
24 |
data = json.loads(line)
|
|
|
25 |
try:
|
26 |
content = data["token"]["text"]
|
27 |
except:
|
|
|
22 |
line = line.decode("utf-8")
|
23 |
line = re.sub(r"data:\s*", "", line)
|
24 |
data = json.loads(line)
|
25 |
+
content = ""
|
26 |
try:
|
27 |
content = data["token"]["text"]
|
28 |
except:
|
networks/openai_streamer.py
CHANGED
@@ -203,6 +203,7 @@ class OpenaiStreamer:
|
|
203 |
is_finished = True
|
204 |
else:
|
205 |
content_type = "Completions"
|
|
|
206 |
try:
|
207 |
data = json.loads(line, strict=False)
|
208 |
message_role = data["message"]["author"]["role"]
|
|
|
203 |
is_finished = True
|
204 |
else:
|
205 |
content_type = "Completions"
|
206 |
+
delta_content = ""
|
207 |
try:
|
208 |
data = json.loads(line, strict=False)
|
209 |
message_role = data["message"]["author"]["role"]
|