Update app.py
Browse files
app.py
CHANGED
|
@@ -472,28 +472,29 @@ def handsome_chat_completions():
|
|
| 472 |
completion_tokens = 0
|
| 473 |
response_content = ""
|
| 474 |
|
| 475 |
-
|
| 476 |
-
|
| 477 |
-
|
| 478 |
-
|
| 479 |
-
|
| 480 |
-
|
| 481 |
-
|
| 482 |
-
|
| 483 |
-
|
| 484 |
-
|
| 485 |
-
|
| 486 |
-
|
| 487 |
-
|
| 488 |
-
|
| 489 |
-
|
| 490 |
-
|
| 491 |
-
|
| 492 |
-
|
| 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'])
|