Update app.py

app.py

@@ -6,9 +6,9 @@ from flask import Flask, request, Response, jsonify
 import socketio
 import requests
 import logging
+import traceback
 from threading import Event
 import re
-import traceback  # 用于捕获详细的堆栈信息
 
 app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
@@ -49,12 +49,15 @@ sio_opts = {
     }
 }
 
-def log_request(ip, route, status):
+def log_request(ip, route, status, data_size=None):
     """
-    记录请求的关键信息：IP地址、路由、状态码。
+    记录请求的关键信息：IP地址、路由、状态码，以及请求数据大小（如果提供）。
     """
     timestamp = datetime.now().isoformat()
-    logging.info(f"{timestamp} - {ip} - {route} - {status}")
+    log_message = f"{timestamp} - {ip} - {route} - {status}"
+    if data_size is not None:
+        log_message += f" - Data size: {data_size} bytes"
+    logging.info(log_message)
 
 def validate_api_key():
     api_key = request.headers.get('x-api-key')
@@ -124,13 +127,11 @@ def messages():
         return auth_error
 
     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")
+        # 记录请求数据大小
+        data_size = len(request.data)
+        log_request(request.remote_addr, request.path, 200, data_size)
 
+        json_body = request.json
         model = json_body.get('model', 'claude-3-opus-20240229')  # 动态获取模型，默认 claude-3-opus-20240229
         stream = json_body.get('stream', True)  # 默认为True
 
@@ -148,9 +149,6 @@ def messages():
             # 处理 stream 为 false 的情况
             return handle_non_stream(previous_messages, msg_id, model, input_tokens)
 
-        # 记录请求的关键信息：IP地址、路由、状态码
-        log_request(request.remote_addr, request.path, 200)
-
         def generate():
             yield create_event("message_start", {
                 "type": "message_start",
@@ -230,9 +228,7 @@ 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,
@@ -256,9 +252,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())  # 输出堆栈信息
+        logging.error(traceback.format_exc())  # 记录堆栈信息
         log_request(request.remote_addr, request.path, 400)
         return jsonify({"error": str(e)}), 400
 
@@ -339,10 +335,11 @@ 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())  # 输出堆栈信息
+        logging.error(traceback.format_exc())  # 记录堆栈信息
         return jsonify({"error": str(e)}), 500
+
     finally:
         if sio.connected:
             sio.disconnect()
@@ -355,6 +352,8 @@ def not_found(error):
 @app.errorhandler(500)
 def server_error(error):
     logging.error(f"Server error: {str(error)}")
+        # 记录详细的错误堆栈信息
+    logging.error(traceback.format_exc())  # 记录堆栈信息
     log_request(request.remote_addr, request.path, 500)
     return "Something broke!", 500