Update app.py

app.py

@@ -14,6 +14,8 @@ import logging
 import os
 from dotenv import load_dotenv
 
+from apscheduler.schedulers.background import BackgroundScheduler
+  
 # 加载环境变量
 load_dotenv()
 
@@ -129,6 +131,14 @@ class ChatCompletionRequest(BaseModel):
     frequency_penalty: Optional[float] = 0
     user: Optional[str] = None
 
+def reset_task():
+    try:
+        for key,value in token_status.items():
+            token_status[key]["total"] = value["count"]
+            token_status[key]["active"] = True
+        print(f"执行重置任务... 当前时间: {datetime.now()}")
+    except Exception as e:
+        print(f"任务执行出错: {e}")
 
 # 初始化token状态
 async def initialize_token_status():
@@ -152,6 +162,7 @@ async def initialize_token_status():
             active = False
             quota_info = {}
             count = 0
+            total = 0
 
             if response.status_code == 200:
                 data = response.json()
@@ -163,6 +174,7 @@ async def initialize_token_status():
                         item_type = item.get('type', '')
                         available = item.get('available', 0)
                         count += available
+                        total += item.get('total', 0)
                         if available > 0:
                             active = True
 
@@ -177,7 +189,8 @@ async def initialize_token_status():
                 "quota": quota_info,
                 "last_checked": datetime.now(),
                 "failed_count": 0,
-                "count": count
+                "count": count,
+                'total': total
             }
 
             logger.info(f"Token {token[:8]}... 状态：{'活跃' if active else '无效'}")
@@ -202,7 +215,7 @@ def verify_api_key(api_key: str = Header(..., alias="Authorization")):
 
 def map_openai_to_deepsider_model(model: str) -> str:
     """将OpenAI模型名称映射到DeepSider模型名称"""
-    return MODEL_MAPPING.get(model, "anthropic/claude-3.7-sonnet")
+    return MODEL_MAPPING.get(model, model)
 
 
 def format_messages_for_deepsider(messages: List[ChatMessage]) -> str:
@@ -582,7 +595,7 @@ async def refresh_token_status(admin_key: str = Header(None, alias="X-Admin-Key"
     if not admin_key or admin_key != expected_admin_key:
         raise HTTPException(status_code=403, detail="Unauthorized")
 
-    await initialize_token_status()
+    await ()
     return {"message": "所有token状态已刷新", "active_tokens": sum(1 for s in token_status.values() if s["active"])}
 
 
@@ -618,9 +631,15 @@ async def startup_event():
         active_tokens = sum(1 for s in token_status.values() if s["active"])
         logger.info(f"初始化完成 活跃token: {active_tokens}/{len(DEEPSIDER_TOKEN)}")
 
-
 # 主程序
 if __name__ == "__main__":
+    scheduler = BackgroundScheduler()
+
+    # 添加任务，每天0点执行
+    scheduler.add_job(reset_task, 'cron', hour=0, minute=0)
+    # 启动调度器
+    scheduler.start()
+
     # 启动服务器
     port = int(os.getenv("PORT", "3000"))
     logger.info(f"启动OpenAI API代理服务 端口: {port}")