again 修改

app.py

@@ -31,13 +31,14 @@ def chat_stream_completion(messages, history, system_prompt):
         messages=messages_prompts,
         stream=True
     )
-    partial_message = ""
+    partial_message = []
     for chunk in response:
         print(json.dumps(chunk, ensure_ascii=False, indent=2))
-        if len(chunk['choices'][0]) != 0:
-            partial_message = partial_message + \
-                chunk['choices'][0]['content']
-            yield partial_message
+        partial_message.append(chunk)
+        # if len(chunk['choices'][0]) != 0:
+        #     partial_message = partial_message + \
+        #         chunk['choices'][0]['content']
+        yield json.dumps(partial_message, ensure_ascii=False, indent=2)
 
 
 gr.ChatInterface(chat_stream_completion,