update

api_key_sb.py

@@ -12,7 +12,7 @@ SUPABASE_KEY = os.getenv("SUPABASE_KEY")
 
 # 初始化 Supabase 客户端
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
-async def get_api_key(model: str = None):
+async def get_api_key_info(model: str = None):
     try:
         # 根据用户提供的SQL语句修改，从 'airs_model_api_keys_view' 视图中获取 'api_key'
         if model:
@@ -41,7 +41,7 @@ async def get_api_key(model: str = None):
 
 
 
-            return api_key
+            return api_key_info
         else:
             return None
     except Exception as e:

proxy.py

@@ -1,7 +1,21 @@
 import httpx
 import asyncio
 from starlette.responses import Response
-from api_key_sb import get_api_key
+from api_key_sb import get_api_key_info
+from supabase import create_client, Client
+from dotenv import load_dotenv
+import os
+from datetime import timezone, timedelta, datetime
+from fastapi import FastAPI, Request, HTTPException, Depends, status
+
+load_dotenv()
+
+# Supabase 配置
+SUPABASE_URL = os.getenv("SUPABASE_URL")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY")
+
+# 初始化 Supabase 客户端
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 
 async def do_proxy(url: str, method: str, headers: dict, content: str, max_retries: int = 3):
     print("Proxy service started.", url)
@@ -11,8 +25,9 @@ async def do_proxy(url: str, method: str, headers: dict, content: str, max_retri
             for attempt in range(max_retries):
 
                 # 获取API密钥信息
-                api_key = await get_api_key('gemini-2.5-flash')
-                # api_key = api_key_info.get('api_key')
+                api_key_info = await get_api_key_info('gemini-2.5-flash')
+                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:]
                 print(f"使用API密钥：{api_key_show}")
                 headers["Authorization"] = f"Bearer {api_key}"
@@ -30,6 +45,17 @@ async def do_proxy(url: str, method: str, headers: dict, content: str, max_retri
                     wait_time = int(retry_after) + attempt * 2  # 指数退避基础值
                     print(f"⚠️ 429错误！{wait_time}秒后重试 (尝试:{attempt+1})")
                     await asyncio.sleep(wait_time)
+
+                    try:
+                        # 创建北京时间时区对象（UTC+8）
+                        beijing_tz = timezone(timedelta(hours=8))
+                        now_beijing = datetime.now(beijing_tz)  # 带时区的datetime对象
+                        future_time = now_beijing + timedelta(days=1)
+                        future_timestamp = future_time.timestamp()
+                        supabase.table("airs_api_keys").update({"ran_at": future_timestamp}).eq("id", api_key_id).execute()
+                        print('更新成功')
+                    except Exception as e:
+                        raise HTTPException(status_code=500, detail="更新API密钥运行时间失败！")
                     continue
                     
                 response.raise_for_status()  # 触发其他4xx/5xx异常