Update app.py

app.py

@@ -122,7 +122,7 @@ def messages():
 
         msg_id = str(uuid.uuid4())
         response_event = Event()
-        response_text = []
+        response_text = []  # 用于存储输出的文本块
 
         if not stream:
             # 处理 stream 为 false 的情况
@@ -220,16 +220,20 @@ def messages():
                 if sio.connected:
                     sio.disconnect()
 
+            # 合并所有的输出块
             output_text = ''.join(response_text)
             # 动态计算 output_tokens
             output_tokens = calculate_tokens(output_text)
 
+            # 生成 content_block_stop 事件
             yield create_event("content_block_stop", {"type": "content_block_stop", "index": 0})
+            # 生成 message_delta 事件，包含 output_tokens
             yield create_event("message_delta", {
                 "type": "message_delta",
                 "delta": {"stop_reason": "end_turn", "stop_sequence": None},
                 "usage": {"output_tokens": output_tokens},  # 动态 output_tokens
             })
+            # 生成 message_stop 事件
             yield create_event("message_stop", {"type": "message_stop"})  # 确保发送 message_stop 事件
 
         return Response(generate(), content_type='text/event-stream')
@@ -296,6 +300,7 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
         # 等待响应完成
         response_event.wait(timeout=30)
 
+        # 合并所有的输出块
         output_text = ''.join(response_text)
         # 动态计算 output_tokens
         output_tokens = calculate_tokens(output_text)