Update app.py

app.py

@@ -6,15 +6,10 @@ from flask import Flask, request, Response, jsonify
 import socketio
 import requests
 import logging
-import traceback
 from threading import Event
 import re
 
 app = Flask(__name__)
-
-# 设置 Flask 的 MAX_CONTENT_LENGTH，防止请求数据过大导致 Flask 抛出 413 错误
-app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 设置为 16MB
-
 logging.basicConfig(level=logging.INFO)
 
 # 从环境变量中获取API密钥
@@ -34,8 +29,7 @@ if proxy_url:
 else:
     transport = None
 
-# 禁用 socketio 和 engineio 的详细日志
-sio = socketio.Client(http_session=transport, logger=False, engineio_logger=False)
+sio = socketio.Client(http_session=transport, logger=True, engineio_logger=True)
 
 # 连接选项
 connect_opts = {
@@ -53,15 +47,9 @@ sio_opts = {
     }
 }
 
-def log_request(ip, route, status, data_size=None):
-    """
-    记录请求的关键信息：IP地址、路由、状态码，以及请求数据大小（如果提供）。
-    """
+def log_request(ip, route, status):
     timestamp = datetime.now().isoformat()
-    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)
+    logging.info(f"{timestamp} - {ip} - {route} - {status}")
 
 def validate_api_key():
     api_key = request.headers.get('x-api-key')
@@ -131,10 +119,6 @@ def messages():
         return auth_error
 
     try:
-        # 记录请求数据大小
-        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
@@ -153,6 +137,9 @@ def messages():
             # 处理 stream 为 false 的情况
             return handle_non_stream(previous_messages, msg_id, model, input_tokens)
 
+        # 记录日志：此时请求上下文仍然有效
+        log_request(request.remote_addr, request.path, 200)
+
         def generate():
             yield create_event("message_start", {
                 "type": "message_start",
@@ -171,7 +158,7 @@ def messages():
             yield create_event("ping", {"type": "ping"})
 
             def on_connect():
-                logging.info("WebSocket connected to Perplexity AI")
+                logging.info("Connected to Perplexity AI")
                 emit_data = {
                     "version": "2.9",
                     "source": "default",
@@ -204,11 +191,11 @@ def messages():
                 response_event.set()
 
             def on_disconnect():
-                logging.info("WebSocket disconnected from Perplexity AI")
+                logging.info("Disconnected from Perplexity AI")
                 response_event.set()
 
             def on_connect_error(data):
-                logging.error(f"WebSocket connection error: {data}")
+                logging.error(f"Connection error: {data}")
                 response_text.append(f"Error connecting to Perplexity AI: {data}")
                 response_event.set()
 
@@ -232,12 +219,11 @@ def messages():
                         })
                 
             except Exception as e:
-                logging.error(f"Error during WebSocket communication: {str(e)}")
-                logging.error(traceback.format_exc())  # 记录详细的堆栈信息
+                logging.error(f"Error during socket connection: {str(e)}")
                 yield create_event("content_block_delta", {
                     "type": "content_block_delta",
                     "index": 0,
-                    "delta": {"type": "text_delta", "text": f"Error during WebSocket communication: {str(e)}"},
+                    "delta": {"type": "text_delta", "text": f"Error during socket connection: {str(e)}"},
                 })
             finally:
                 if sio.connected:
@@ -257,9 +243,7 @@ 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
 
@@ -272,7 +256,7 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
         response_text = []
 
         def on_connect():
-            logging.info("WebSocket connected to Perplexity AI")
+            logging.info("Connected to Perplexity AI")
             emit_data = {
                 "version": "2.9",
                 "source": "default",
@@ -302,11 +286,11 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
                 response_event.set()
 
         def on_disconnect():
-            logging.info("WebSocket disconnected from Perplexity AI")
+            logging.info("Disconnected from Perplexity AI")
             response_event.set()
 
         def on_connect_error(data):
-            logging.error(f"WebSocket connection error: {data}")
+            logging.error(f"Connection error: {data}")
             response_text.append(f"Error connecting to Perplexity AI: {data}")
             response_event.set()
 
@@ -340,11 +324,8 @@ 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(f"Error during socket connection: {str(e)}")
         return jsonify({"error": str(e)}), 500
-
     finally:
         if sio.connected:
             sio.disconnect()
@@ -357,8 +338,6 @@ 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