Update app.py

app.py

@@ -96,19 +96,24 @@ def chat_completions():
                 
                 elif '"final":true' in event.data:
                     final_data = json.loads(event.data)
-                    finish_reason = final_data.get('responseMessage', {}).get('finish_reason', 'stop')
+                    response_message = final_data.get('responseMessage', {})
+                    finish_reason = response_message.get('finish_reason', 'stop')
+                    
                     if finish_reason == 'length':
                         messages.append({"role": "assistant", "content": full_response})
                         messages.append({"role": "user", "content": "请继续你的输出，不要重复之前的内容"})
                         break  # 跳出当前循环，继续下一次请求
                     else:
+                        # 正常结束，发送最后的内容（如果有的话）
+                        last_content = response_message.get('text', '')
+                        if last_content and last_content != full_response:
+                            yield f"data: {json.dumps(format_openai_response(last_content[len(full_response):]))}\n\n"
+                        
                         yield f"data: {json.dumps(format_openai_response('', finish_reason))}\n\n"
                         yield "data: [DONE]\n\n"
                         return  # 完全结束生成
-            
-            # 如果因为 length 而跳出内部循环，则继续外部循环
 
-        # 如果所有请求都因 length 而结束，发送一个最终的停止信号
+        # 如果因为多次长度限制而最终结束，发送一个停止信号
         yield f"data: {json.dumps(format_openai_response('', 'stop'))}\n\n"
         yield "data: [DONE]\n\n"