yangtb24 commited on
Commit
7a2a875
·
verified ·
1 Parent(s): c31239d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -19
app.py CHANGED
@@ -472,28 +472,29 @@ def handsome_chat_completions():
472
  completion_tokens = 0
473
  response_content = ""
474
 
475
- user_content = ""
476
- messages = data.get("messages", [])
477
- for message in messages:
478
- if message["role"] == "user":
479
- if isinstance(message["content"], str):
480
- user_content += message["content"] + " "
481
- elif isinstance(message["content"], list):
482
- for item in message["content"]:
483
- if isinstance(item, dict) and item.get("type") == "text":
484
- user_content += item.get("text", "") + " "
485
-
486
- user_content = user_content.strip()
487
-
488
- user_content_replaced = user_content.replace('\n', '\\n').replace('\r', '\\n')
489
- response_content_replaced = response_content.replace('\n', '\\n').replace('\r', '\\n')
490
-
491
- logging.info(
492
- f"使用的key: {api_key}, 提示token: {prompt_tokens}, 输出token: {completion_tokens}, 首字用时: 0, 总共用时: {total_time:.4f}秒, 使用的模型: {model_name}, 用户的内容: {user_content_replaced}, 输出的内容: {response_content_replaced}"
493
- )
494
  return jsonify(response_json)
495
 
496
  except requests.exceptions.RequestException as e:
 
497
  return jsonify({"error": str(e)}), 500
498
 
499
  @app.route('/handsome/v1/models', methods=['GET'])
 
472
  completion_tokens = 0
473
  response_content = ""
474
 
475
+ user_content = ""
476
+ messages = data.get("messages", [])
477
+ for message in messages:
478
+ if message["role"] == "user":
479
+ if isinstance(message["content"], str):
480
+ user_content += message["content"] + " "
481
+ elif isinstance(message["content"], list):
482
+ for item in message["content"]:
483
+ if isinstance(item, dict) and item.get("type") == "text":
484
+ user_content += item.get("text", "") + " "
485
+
486
+ user_content = user_content.strip()
487
+
488
+ user_content_replaced = user_content.replace('\n', '\\n').replace('\r', '\\n')
489
+ response_content_replaced = response_content.replace('\n', '\\n').replace('\r', '\\n')
490
+
491
+ logging.info(
492
+ f"使用的key: {api_key}, 提示token: {prompt_tokens}, 输出token: {completion_tokens}, 首字用时: 0, 总共用时: {total_time:.4f}秒, 使用的模型: {model_name}, 用户的内容: {user_content_replaced}, 输出的内容: {response_content_replaced}"
493
+ )
494
  return jsonify(response_json)
495
 
496
  except requests.exceptions.RequestException as e:
497
+ logging.error(f"请求转发异常: {e}")
498
  return jsonify({"error": str(e)}), 500
499
 
500
  @app.route('/handsome/v1/models', methods=['GET'])