update

proxy.py

@@ -8,6 +8,7 @@ import os
 from datetime import timezone, timedelta, datetime
 from fastapi import FastAPI, Request, HTTPException, Depends, status
 import logging # 导入logging模块
+import json # 导入json模块
 
 load_dotenv()
 
@@ -20,10 +21,21 @@ async def do_proxy(url: str, method: str, headers: dict, content: str, max_retri
     try:
         async with httpx.AsyncClient() as client:
             for attempt in range(max_retries):
-                print('content',content)
+                logger.info(f"接收到的内容: {content}")
+                try:
+                    content_json = json.loads(content)
+                    model_name = content_json.get('model')
+                    if model_name:
+                        logger.info(f"从内容中提取的模型名称: {model_name}")
+                    else:
+                        logger.warning("内容中未找到 'model' 键。")
+                except json.JSONDecodeError:
+                    logger.warning("内容不是有效的 JSON 格式，无法提取 'model' 键。")
+                except Exception as e:
+                    logger.error(f"提取模型名称时发生错误: {str(e)}")
 
                 # 获取API密钥信息
-                api_key_info = await get_api_key_info('gemini-2.5-flash')
+                api_key_info = await get_api_key_info(model_name)
                 api_key = api_key_info.get('api_key')
                 api_key_id = api_key_info.get('api_key_id')
                 api_key_show = api_key[:5]+'*****'+api_key[-5:]