Update networks/huggingface_streamer.py

networks/huggingface_streamer.py

@@ -209,7 +209,7 @@ class HuggingfaceStreamer:
  output['repsonse'] = output['choices'][-1]['delta'].get('content')
  output['done'] = output['choices'][-1].get('finish_reason')
  output['created_at'] = output.get('created')
- yield output
+ yield json.dumps(output)
 
  if not is_finished:
  final_output = self.message_outputer.output(content="", content_type="Finished")
@@ -217,5 +217,5 @@ class HuggingfaceStreamer:
  final_output = final_output['choices'][-1]['delta'].get('content')
  final_output['done'] = output['choices'][-1].get('finish_reason')
  final_output['created_at'] = final_output.get('created')
- yield final_output
+ yield json.dumps(final_output)
  # yield self.message_outputer.output(content="", content_type="Finished").split('data: ')[1]