Update app.py

app.py

@@ -3,29 +3,41 @@ from flask import Flask, request, Response
 import requests
 import json
 import os
+import sys
 import logging
 from helper import create_jwt
 
 app = Flask(__name__)
 
 # 配置日志
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s [%(levelname)s] %(message)s',
+    handlers=[
+        logging.StreamHandler(sys.stdout)
+    ]
+)
 logger = logging.getLogger(__name__)
 
-@app.route('/ai/v1/chat/completions', methods=['POST'])
+@app.route('/chat/completions', methods=['POST'])
 async def chat():
     """
     Handle chat completion requests.
     """
+    logger.info("Received chat completion request")
+    
     # 记录请求信息
-    logger.info(f"Received request: {request.method} {request.url}")
-    logger.info(f"Headers: {request.headers}")
+    logger.info(f"Request method: {request.method}")
+    logger.info(f"Request URL: {request.url}")
+    logger.info(f"Request headers: {dict(request.headers)}")
     
     # Get the payload from the request
     payload = request.json
+    logger.info(f"Request payload: {json.dumps(payload, indent=2)}")
 
     # Get the model from the payload, defaulting to "claude-3-5-sonnet-20240620"
     model = payload.get('model', 'claude-3-5-sonnet-20240620')
+    logger.info(f"Using model: {model}")
 
     # Extract GitHub username and Zed user ID from Authorization header
     auth_header = request.headers.get('Authorization')
@@ -42,6 +54,7 @@ async def chat():
 
     # Prepare the request for the LLM API
     url = "https://llm.zed.dev/completion"
+    logger.info(f"LLM API URL: {url}")
     
     llm_payload = {
         "provider": "anthropic",
@@ -55,6 +68,7 @@ async def chat():
             "system": ""
         }
     }
+    logger.info(f"LLM API payload: {json.dumps(llm_payload, indent=2)}")
 
     jwt = create_jwt(github_username, int(zed_user_id))
     logger.info(f"Generated JWT token: {jwt}")
@@ -66,28 +80,34 @@ async def chat():
         'authorization': f'Bearer {jwt}',
         'user-agent': 'Zed/0.149.3 (macos; aarch64)'
     }
+    logger.info(f"Request headers: {json.dumps(headers, indent=2)}")
 
     # Get proxy from environment variable
     proxy = os.environ.get('HTTP_PROXY', None)
     proxies = {'http': proxy, 'https': proxy} if proxy else None
+    logger.info(f"Using proxy: {proxy}")
 
     async def generate():
         try:
+            logger.info("Sending request to LLM API")
             with requests.post(url, headers=headers, json=llm_payload, stream=True, proxies=proxies, allow_redirects=True) as response:
                 logger.info(f"LLM API response status: {response.status_code}")
-                logger.info(f"LLM API response headers: {response.headers}")
+                logger.info(f"LLM API response headers: {dict(response.headers)}")
                 
                 if response.status_code == 301:
-                    logger.warning(f"Received 301 redirect. New location: {response.headers.get('Location')}")
+                    new_location = response.headers.get('Location')
+                    logger.warning(f"Received 301 redirect. New location: {new_location}")
                     # 如果需要，可以在这里处理重定向
                 
                 for chunk in response.iter_content(chunk_size=1024):
                     if chunk:
+                        logger.debug(f"Received chunk of size: {len(chunk)} bytes")
                         yield chunk
         except Exception as e:
             logger.error(f"Error during API request: {str(e)}")
             yield str(e).encode()
 
+    logger.info("Returning streaming response")
     return Response(generate(), content_type='application/octet-stream')
 
 # Convert the Flask app to an ASGI app
@@ -95,4 +115,5 @@ asgi_app = WsgiToAsgi(app)
 
 if __name__ == '__main__':
     import uvicorn
+    logger.info("Starting the application")
     uvicorn.run(asgi_app, host="0.0.0.0", port=8000)