Update app.py

app.py

@@ -8,6 +8,7 @@ import requests
 import logging
 from threading import Event
 import re
+import traceback  # 用于捕获详细的堆栈信息
 
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
@@ -124,6 +125,12 @@ def messages():
 
     try:
         json_body = request.json
+
+        # 检查请求数据是否过大
+        request_size = len(json.dumps(json_body))
+        if request_size > 1 * 1024 * 1024:  # 超过 1MB 的请求大小
+            logging.warning(f"Request size is too large: {request_size / 1024:.2f} KB")
+
         model = json_body.get('model', 'claude-3-opus-20240229')  # 动态获取模型，默认 claude-3-opus-20240229
         stream = json_body.get('stream', True)  # 默认为True
 
@@ -223,7 +230,9 @@ def messages():
                         })
                 
             except Exception as e:
+                # 捕获并记录详细的堆栈信息
                 logging.error(f"Error during WebSocket communication: {str(e)}")
+                logging.error(traceback.format_exc())  # 输出堆栈信息
                 yield create_event("content_block_delta", {
                     "type": "content_block_delta",
                     "index": 0,
@@ -247,7 +256,9 @@ def messages():
         return Response(generate(), content_type='text/event-stream')
 
     except Exception as e:
+        # 捕获并记录详细的堆栈信息
         logging.error(f"Request error: {str(e)}")
+        logging.error(traceback.format_exc())  # 输出堆栈信息
         log_request(request.remote_addr, request.path, 400)
         return jsonify({"error": str(e)}), 400
 
@@ -328,7 +339,9 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
         return Response(json.dumps(full_response, ensure_ascii=False), content_type='application/json')
 
     except Exception as e:
+        # 捕获并记录详细的堆栈信息
         logging.error(f"Request error: {str(e)}")
+        logging.error(traceback.format_exc())  # 输出堆栈信息
         return jsonify({"error": str(e)}), 500
     finally:
         if sio.connected: