Update app.py

app.py

@@ -195,7 +195,15 @@ def messages():
 
                 # 检查是否是最终响应
                 if data.get('final', False):
-                    response_event.set()
+                    # 在响应结束时发送结束事件
+                    yield create_event("content_block_stop", {"type": "content_block_stop", "index": 0})
+                    yield create_event("message_delta", {
+                        "type": "message_delta",
+                        "delta": {"stop_reason": "end_turn", "stop_sequence": None},
+                        "usage": {"input_tokens": input_tokens, "output_tokens": output_tokens},  # 最终 output_tokens
+                    })
+                    yield create_event("message_stop", {"type": "message_stop"})  # 确保发送 message_stop 事件
+                    response_event.set()  # 触发响应结束
 
             def on_query_complete(data):
                 response_event.set()
@@ -239,18 +247,6 @@ def messages():
                 if sio.connected:
                     sio.disconnect()
 
-            # 在响应结束时发送结束事件
-            yield create_event("content_block_stop", {"type": "content_block_stop", "index": 0})
-            yield create_event("message_delta", {
-                "type": "message_delta",
-                "delta": {"stop_reason": "end_turn", "stop_sequence": None},
-                "usage": {"input_tokens": input_tokens, "output_tokens": output_tokens},  # 最终 output_tokens
-            })
-            yield create_event("message_stop", {"type": "message_stop"})  # 确保发送 message_stop 事件
-
-            # 触发响应结束
-            response_event.set()
-
         return Response(generate(), content_type='text/event-stream')
 
     except Exception as e: