Spaces:
Running
Running
:gem: [Feature] HuggingchatStreamer: Support no-stream mode
Browse files
networks/huggingchat_streamer.py
CHANGED
|
@@ -300,8 +300,27 @@ class HuggingchatStreamer:
|
|
| 300 |
if not is_finished:
|
| 301 |
yield self.message_outputer.output(content="", content_type="Finished")
|
| 302 |
|
| 303 |
-
def chat_return_dict(self, stream_response):
|
| 304 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 305 |
|
| 306 |
|
| 307 |
if __name__ == "__main__":
|
|
|
|
| 300 |
if not is_finished:
|
| 301 |
yield self.message_outputer.output(content="", content_type="Finished")
|
| 302 |
|
| 303 |
+
def chat_return_dict(self, stream_response: requests.Response):
|
| 304 |
+
final_output = self.message_outputer.default_data.copy()
|
| 305 |
+
final_output["choices"] = [
|
| 306 |
+
{
|
| 307 |
+
"index": 0,
|
| 308 |
+
"finish_reason": "stop",
|
| 309 |
+
"message": {"role": "assistant", "content": ""},
|
| 310 |
+
}
|
| 311 |
+
]
|
| 312 |
+
final_content = ""
|
| 313 |
+
for item in self.chat_return_generator(stream_response):
|
| 314 |
+
try:
|
| 315 |
+
data = json.loads(item)
|
| 316 |
+
delta = data["choices"][0]["delta"]
|
| 317 |
+
delta_content = delta.get("content", "")
|
| 318 |
+
if delta_content:
|
| 319 |
+
final_content += delta_content
|
| 320 |
+
except Exception as e:
|
| 321 |
+
logger.warn(e)
|
| 322 |
+
final_output["choices"][0]["message"]["content"] = final_content.strip()
|
| 323 |
+
return final_output
|
| 324 |
|
| 325 |
|
| 326 |
if __name__ == "__main__":
|