Update app.py

app.py

@@ -3,27 +3,24 @@ from flask import Flask, request, Response
 import requests
 import json
 import os
+import logging
 from helper import create_jwt
 
 app = Flask(__name__)
 
+# 配置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 @app.route('/ai/v1/chat/completions', methods=['POST'])
 async def chat():
     """
     Handle chat completion requests.
-
-    This function processes incoming POST requests to the '/chat/completions' endpoint.
-    It prepares the payload for the LLM API, generates a JWT for authentication,
-    and streams the response from the LLM API back to the client.
-
-    Returns:
-        Response: A streaming response containing the LLM API's output.
-
-    Note:
-        - The function uses environment variables for proxy configuration.
-        - It extracts GitHub username and Zed user ID from the Authorization header.
-        - The LLM model defaults to "claude-3-5-sonnet-20240620" if not specified.
     """
+    # 记录请求信息
+    logger.info(f"Received request: {request.method} {request.url}")
+    logger.info(f"Headers: {request.headers}")
+    
     # Get the payload from the request
     payload = request.json
 
@@ -33,15 +30,18 @@ async def chat():
     # Extract GitHub username and Zed user ID from Authorization header
     auth_header = request.headers.get('Authorization')
     if not auth_header or not auth_header.startswith('Bearer '):
+        logger.error("Invalid Authorization header")
         return Response('Invalid Authorization header', status=401)
 
     try:
         github_username, zed_user_id = auth_header[7:].split(',')
+        logger.info(f"GitHub username: {github_username}, Zed user ID: {zed_user_id}")
     except ValueError:
+        logger.error("Invalid Authorization header format")
         return Response('Invalid Authorization header format', status=401)
 
     # Prepare the request for the LLM API
-    url = "https://llm.zed.dev/completion?"
+    url = "https://llm.zed.dev/completion"
     
     llm_payload = {
         "provider": "anthropic",
@@ -57,6 +57,7 @@ async def chat():
     }
 
     jwt = create_jwt(github_username, int(zed_user_id))
+    logger.info(f"Generated JWT token: {jwt}")
 
     headers = {
         'Host': 'llm.zed.dev',
@@ -71,10 +72,21 @@ async def chat():
     proxies = {'http': proxy, 'https': proxy} if proxy else None
 
     async def generate():
-        with requests.post(url, headers=headers, json=llm_payload, stream=True, proxies=proxies) as response:
-            for chunk in response.iter_content(chunk_size=1024):
-                if chunk:
-                    yield chunk
+        try:
+            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}")
+                
+                if response.status_code == 301:
+                    logger.warning(f"Received 301 redirect. New location: {response.headers.get('Location')}")
+                    # 如果需要，可以在这里处理重定向
+                
+                for chunk in response.iter_content(chunk_size=1024):
+                    if chunk:
+                        yield chunk
+        except Exception as e:
+            logger.error(f"Error during API request: {str(e)}")
+            yield str(e).encode()
 
     return Response(generate(), content_type='application/octet-stream')