Update app.py

app.py

@@ -158,16 +158,11 @@ def messages():
 
             def on_query_progress(data):
                 nonlocal response_text
-                try:
-                    if 'text' in data:
-                        text = json.loads(data['text'])  # 尝试解析为JSON
-                        chunk = text['chunks'][-1] if text['chunks'] else None
-                        if chunk:
-                            response_text.append(chunk)
-                except json.JSONDecodeError:
-                    # 如果解析失败，说明不是有效的JSON，忽略或记录错误
-                    logging.error("Failed to parse response as JSON")
-                    response_text.append("Invalid response format from API.")
+                if 'text' in data:
+                    text = json.loads(data['text'])
+                    chunk = text['chunks'][-1] if text['chunks'] else None
+                    if chunk:
+                        response_text.append(chunk)
 
                 # 检查是否是最终响应
                 if data.get('final', False):
@@ -232,7 +227,7 @@ def messages():
 
 def handle_non_stream(previous_messages, msg_id, model):
     """
-    处理 stream 为 false 的情况，返回完整的响应。
+    处理 stream 为 false 的情况，返回符合 Claude API 格式的完整响应。
     """
     try:
         response_event = Event()
@@ -258,16 +253,11 @@ def handle_non_stream(previous_messages, msg_id, model):
 
         def on_query_progress(data):
             nonlocal response_text
-            try:
-                if 'text' in data:
-                    text = json.loads(data['text'])  # 尝试解析为JSON
-                    chunk = text['chunks'][-1] if text['chunks'] else None
-                    if chunk:
-                        response_text.append(chunk)
-            except json.JSONDecodeError:
-                # 如果解析失败，说明不是有效的JSON，忽略或记录错误
-                logging.error("Failed to parse response as JSON")
-                response_text.append("Invalid response format from API.")
+            if 'text' in data:
+                text = json.loads(data['text'])
+                chunk = text['chunks'][-1] if text['chunks'] else None
+                if chunk:
+                    response_text.append(chunk)
 
             # 检查是否是最终响应
             if data.get('final', False):
@@ -292,13 +282,15 @@ def handle_non_stream(previous_messages, msg_id, model):
         # 等待响应完成
         response_event.wait(timeout=30)
 
-        # 生成完整的响应
+        # 生成符合 Claude API 格式的完整响应
         full_response = {
             "id": msg_id,
-            "content": [{"text": ''.join(response_text)}],  # 合并所有文本块
+            "content": [{"text": ''.join(response_text), "type": "text"}],  # 合并所有文本块，并设置类型
             "model": model,  # 动态模型
+            "role": "assistant",  # Claude 的角色
             "stop_reason": "end_turn",
             "stop_sequence": None,
+            "type": "message",
             "usage": {
                 "input_tokens": 8,
                 "output_tokens": len(''.join(response_text)),