Update app.py

app.py

@@ -48,6 +48,9 @@ sio_opts = {
 }
 
 def log_request(ip, route, status):
+    """
+    记录请求的关键信息：IP地址、路由、状态码。
+    """
     timestamp = datetime.now().isoformat()
     logging.info(f"{timestamp} - {ip} - {route} - {status}")
 
@@ -137,7 +140,7 @@ 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():
@@ -158,7 +161,7 @@ def messages():
             yield create_event("ping", {"type": "ping"})
 
             def on_connect():
-                logging.info("Connected to Perplexity AI")
+                logging.info("Connected to Perplexity AI WebSocket")
                 emit_data = {
                     "version": "2.9",
                     "source": "default",
@@ -191,11 +194,11 @@ def messages():
                 response_event.set()
 
             def on_disconnect():
-                logging.info("Disconnected from Perplexity AI")
+                logging.info("Disconnected from Perplexity AI WebSocket")
                 response_event.set()
 
             def on_connect_error(data):
-                logging.error(f"Connection error: {data}")
+                logging.error(f"WebSocket connection error: {data}")
                 response_text.append(f"Error connecting to Perplexity AI: {data}")
                 response_event.set()
 
@@ -219,11 +222,11 @@ def messages():
                         })
                 
             except Exception as e:
-                logging.error(f"Error during socket connection: {str(e)}")
+                logging.error(f"Error during WebSocket connection: {str(e)}")
                 yield create_event("content_block_delta", {
                     "type": "content_block_delta",
                     "index": 0,
-                    "delta": {"type": "text_delta", "text": f"Error during socket connection: {str(e)}"},
+                    "delta": {"type": "text_delta", "text": f"Error during WebSocket connection: {str(e)}"},
                 })
             finally:
                 if sio.connected:
@@ -256,20 +259,20 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
         response_text = []
 
         def on_connect():
-            logging.info("Connected to Perplexity AI")
+            logging.info("Connected to Perplexity AI WebSocket")
             emit_data = {
                 "version": "2.9",
                 "source": "default",
                 "attachments": [],
-                    "language": "en-GB",
-                    "timezone": "Europe/London",
-                    "mode": "concise",
-                    "is_related_query": False,
-                    "is_default_related_query": False,
-                    "visitor_id": str(uuid.uuid4()),
-                    "frontend_context_uuid": str(uuid.uuid4()),
-                    "prompt_source": "user",
-                    "query_source": "home"
+                "language": "en-GB",
+                "timezone": "Europe/London",
+                "mode": "concise",
+                "is_related_query": False,
+                "is_default_related_query": False,
+                "visitor_id": str(uuid.uuid4()),
+                "frontend_context_uuid": str(uuid.uuid4()),
+                "prompt_source": "user",
+                "query_source": "home"
             }
             sio.emit('perplexity_ask', (previous_messages, emit_data))
 
@@ -286,11 +289,11 @@ def handle_non_stream(previous_messages, msg_id, model, input_tokens):
                 response_event.set()
 
         def on_disconnect():
-            logging.info("Disconnected from Perplexity AI")
+            logging.info("Disconnected from Perplexity AI WebSocket")
             response_event.set()
 
         def on_connect_error(data):
-            logging.error(f"Connection error: {data}")
+            logging.error(f"WebSocket connection error: {data}")
             response_text.append(f"Error connecting to Perplexity AI: {data}")
             response_event.set()
 
@@ -324,7 +327,7 @@ 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"Error during socket connection: {str(e)}")
+        logging.error(f"Request error: {str(e)}")
         return jsonify({"error": str(e)}), 500
     finally:
         if sio.connected: