Update app.py

app.py

@@ -57,22 +57,15 @@ def validate_api_key():
         return jsonify({"error": "Invalid API key"}), 401
     return None
 
-def normalize_content(content):
+def is_valid_json(response_text):
     """
-    递归处理 msg['content']，确保其为字符串。
-    如果 content 是字典或列表，将其转换为字符串。
+    检查响应是否为有效的 JSON 格式。
     """
-    if isinstance(content, str):
-        return content
-    elif isinstance(content, dict):
-        # 将字典转化为 JSON 字符串
-        return json.dumps(content, ensure_ascii=False)
-    elif isinstance(content, list):
-        # 对于列表，递归处理每个元素
-        return " ".join([normalize_content(item) for item in content])
-    else:
-        # 如果是其他类型，返回空字符串
-        return ""
+    try:
+        json.loads(response_text)
+        return True
+    except ValueError:
+        return False
 
 @app.route('/')
 def root():
@@ -159,10 +152,14 @@ def messages():
             def on_query_progress(data):
                 nonlocal response_text
                 if 'text' in data:
-                    text = json.loads(data['text'])
-                    chunk = text['chunks'][-1] if text['chunks'] else None
-                    if chunk:
-                        response_text.append(chunk)
+                    text = data['text']  # 不再使用 json.loads() 解析
+                    if is_valid_json(text):
+                        text = json.loads(text)  # 确保是有效的 JSON
+                        chunk = text['chunks'][-1] if text['chunks'] else None
+                        if chunk:
+                            response_text.append(chunk)
+                    else:
+                        logging.error(f"Received non-JSON response: {text}")
 
                 # 检查是否是最终响应
                 if data.get('final', False):
@@ -227,7 +224,7 @@ def messages():
 
 def handle_non_stream(previous_messages, msg_id, model):
     """
-    处理 stream 为 false 的情况，返回符合 Claude API 格式的完整响应。
+    处理 stream 为 false 的情况，返回完整的响应。
     """
     try:
         response_event = Event()
@@ -254,10 +251,14 @@ def handle_non_stream(previous_messages, msg_id, model):
         def on_query_progress(data):
             nonlocal response_text
             if 'text' in data:
-                text = json.loads(data['text'])
-                chunk = text['chunks'][-1] if text['chunks'] else None
-                if chunk:
-                    response_text.append(chunk)
+                text = data['text']  # 不再使用 json.loads() 解析
+                if is_valid_json(text):
+                    text = json.loads(text)  # 确保是有效的 JSON
+                    chunk = text['chunks'][-1] if text['chunks'] else None
+                    if chunk:
+                        response_text.append(chunk)
+                else:
+                    logging.error(f"Received non-JSON response: {text}")
 
             # 检查是否是最终响应
             if data.get('final', False):
@@ -282,18 +283,13 @@ def handle_non_stream(previous_messages, msg_id, model):
         # 等待响应完成
         response_event.wait(timeout=30)
 
-        # 生成符合 Claude API 格式的完整响应
-        if not response_text:
-            response_text = [""]
-
+        # 生成完整的响应
         full_response = {
             "id": msg_id,
-            "content": [{"text": ''.join(response_text), "type": "text"}],  # 合并所有文本块，并设置类型
+            "content": [{"text": ''.join(response_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)),