smgc commited on
Commit
5c71812
1 Parent(s): 0feefe3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -23,12 +23,12 @@ def chat():
23
  """
24
  Handle chat completion requests.
25
  """
26
- logger.info("Received chat completion request")
27
 
28
  # 记录请求信息
29
  logger.info(f"Request method: {request.method}")
30
  logger.info(f"Request URL: {request.url}")
31
- logger.info(f"Request headers: {dict(request.headers)}")
32
 
33
  # Get the payload from the request
34
  payload = request.json
@@ -46,7 +46,8 @@ def chat():
46
 
47
  try:
48
  github_username, zed_user_id = auth_header[7:].split(',')
49
- logger.info(f"GitHub username: {github_username}, Zed user ID: {zed_user_id}")
 
50
  except ValueError:
51
  logger.error("Invalid Authorization header format")
52
  return Response('Invalid Authorization header format', status=401)
@@ -79,7 +80,7 @@ def chat():
79
  'authorization': f'Bearer {jwt}',
80
  'user-agent': 'Zed/0.149.3 (macos; aarch64)'
81
  }
82
- logger.info(f"Request headers: {json.dumps(headers, indent=2)}")
83
 
84
  # Get proxy from environment variable
85
  proxy = os.environ.get('HTTP_PROXY', None)
@@ -91,7 +92,7 @@ def chat():
91
  logger.info("Sending request to LLM API")
92
  with requests.post(url, headers=headers, json=llm_payload, stream=True, proxies=proxies, allow_redirects=True) as response:
93
  logger.info(f"LLM API response status: {response.status_code}")
94
- logger.info(f"LLM API response headers: {dict(response.headers)}")
95
 
96
  if response.status_code == 301:
97
  new_location = response.headers.get('Location')
@@ -111,6 +112,7 @@ def chat():
111
 
112
  @app.route('/', methods=['GET'])
113
  def home():
 
114
  return "Welcome to the Chat Completion API", 200
115
 
116
  if __name__ == '__main__':
 
23
  """
24
  Handle chat completion requests.
25
  """
26
+ logger.info("=== Received chat completion request ===")
27
 
28
  # 记录请求信息
29
  logger.info(f"Request method: {request.method}")
30
  logger.info(f"Request URL: {request.url}")
31
+ logger.info(f"Request headers: {json.dumps(dict(request.headers), indent=2)}")
32
 
33
  # Get the payload from the request
34
  payload = request.json
 
46
 
47
  try:
48
  github_username, zed_user_id = auth_header[7:].split(',')
49
+ logger.info(f"Extracted GitHub username: {github_username}")
50
+ logger.info(f"Extracted Zed user ID: {zed_user_id}")
51
  except ValueError:
52
  logger.error("Invalid Authorization header format")
53
  return Response('Invalid Authorization header format', status=401)
 
80
  'authorization': f'Bearer {jwt}',
81
  'user-agent': 'Zed/0.149.3 (macos; aarch64)'
82
  }
83
+ logger.info(f"Request headers for LLM API: {json.dumps(headers, indent=2)}")
84
 
85
  # Get proxy from environment variable
86
  proxy = os.environ.get('HTTP_PROXY', None)
 
92
  logger.info("Sending request to LLM API")
93
  with requests.post(url, headers=headers, json=llm_payload, stream=True, proxies=proxies, allow_redirects=True) as response:
94
  logger.info(f"LLM API response status: {response.status_code}")
95
+ logger.info(f"LLM API response headers: {json.dumps(dict(response.headers), indent=2)}")
96
 
97
  if response.status_code == 301:
98
  new_location = response.headers.get('Location')
 
112
 
113
  @app.route('/', methods=['GET'])
114
  def home():
115
+ logger.info("Received request to home page")
116
  return "Welcome to the Chat Completion API", 200
117
 
118
  if __name__ == '__main__':