yangtb24 commited on
Commit
d9e4cd0
·
verified ·
1 Parent(s): 6001b99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -310,6 +310,9 @@ def check_tokens():
310
 
311
  @app.route('/handsome/v1/chat/completions', methods=['POST'])
312
  def handsome_chat_completions():
 
 
 
313
  if not check_authorization(request):
314
  return jsonify({"error": "Unauthorized"}), 401
315
 
@@ -363,7 +366,8 @@ def handsome_chat_completions():
363
  prompt_tokens = response_json["usage"]["prompt_tokens"]
364
  completion_tokens = response_json["usage"]["completion_tokens"]
365
  response_content = response_json["choices"][0]["message"]["content"]
366
- except (KeyError, ValueError, IndexError):
 
367
  prompt_tokens = 0
368
  completion_tokens = 0
369
  response_content = ""
@@ -392,11 +396,12 @@ def handsome_chat_completions():
392
  prompt_tokens = response_json["usage"]["prompt_tokens"]
393
  completion_tokens = response_json["usage"]["completion_tokens"]
394
  response_content = response_json["choices"][0]["message"]["content"]
395
- except (KeyError, ValueError, IndexError):
 
396
  prompt_tokens = 0
397
  completion_tokens = 0
398
  response_content = ""
399
-
400
  # 提取用户输入的内容
401
  user_content = ""
402
  messages = data.get("messages", [])
 
310
 
311
  @app.route('/handsome/v1/chat/completions', methods=['POST'])
312
  def handsome_chat_completions():
313
+ """
314
+ 处理 /handsome/v1/chat/completions 路由的请求,添加鉴权,实现 KEY 的轮询和重试机制,并记录日志。
315
+ """
316
  if not check_authorization(request):
317
  return jsonify({"error": "Unauthorized"}), 401
318
 
 
366
  prompt_tokens = response_json["usage"]["prompt_tokens"]
367
  completion_tokens = response_json["usage"]["completion_tokens"]
368
  response_content = response_json["choices"][0]["message"]["content"]
369
+ except (KeyError, ValueError, IndexError) as e:
370
+ logging.error(f"解析流式响应 JSON 失败: {e}, 完整响应内容: {full_response_content}")
371
  prompt_tokens = 0
372
  completion_tokens = 0
373
  response_content = ""
 
396
  prompt_tokens = response_json["usage"]["prompt_tokens"]
397
  completion_tokens = response_json["usage"]["completion_tokens"]
398
  response_content = response_json["choices"][0]["message"]["content"]
399
+ except (KeyError, ValueError, IndexError) as e:
400
+ logging.error(f"解析非流式响应 JSON 失败: {e}")
401
  prompt_tokens = 0
402
  completion_tokens = 0
403
  response_content = ""
404
+
405
  # 提取用户输入的内容
406
  user_content = ""
407
  messages = data.get("messages", [])