update

api_key_sb copy.py

@@ -0,0 +1,30 @@
+from supabase import create_client, Client
+import os
+from fastapi import FastAPI, Request, HTTPException, Depends, status
+from dotenv import load_dotenv
+
+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 get_api_key_info(model: str = None):
+    try:
+        # 根据用户提供的SQL语句修改，从 'airs_model_api_keys_view' 视图中获取 'api_key'
+        if model:
+            response = supabase.from_('airs_model_api_keys_view').select('api_key', 'api_key_id').eq('model_name', model).order('api_key_ran_at', desc=False).limit(1).execute()
+        else:
+            # 如果没有提供model，则获取所有模型的api_key
+            raise HTTPException(status_code=400, detail="请提供模型名称！")
+        
+        if response.data:
+            api_key_info = response.data[0]
+            return api_key_info
+        else:
+            return None
+    except Exception as e:
+        print(f"从Supabase获取API密钥失败: {e}")
+        return None

api_key_sb.py

@@ -2,6 +2,7 @@ from supabase import create_client, Client
 import os
 from fastapi import FastAPI, Request, HTTPException, Depends, status
 from dotenv import load_dotenv
+from datetime import timezone, timedelta, datetime
 
 load_dotenv()
 
@@ -11,7 +12,7 @@ SUPABASE_KEY = os.getenv("SUPABASE_KEY")
 
 # 初始化 Supabase 客户端
 supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
-async def get_api_key_info(model: str = None):
+async def get_api_key(model: str = None):
     try:
         # 根据用户提供的SQL语句修改，从 'airs_model_api_keys_view' 视图中获取 'api_key'
         if model:
@@ -22,7 +23,25 @@ async def get_api_key_info(model: str = None):
         
         if response.data:
             api_key_info = response.data[0]
-            return api_key_info
+            api_key = api_key_info.get('api_key')
+            api_key_id = api_key_info.get('api_key_id')
+
+             
+            try:
+                # 创建北京时间时区对象（UTC+8）
+                beijing_tz = timezone(timedelta(hours=8))
+                now_beijing = datetime.now(beijing_tz)  # 带时区的datetime对象
+                current_local_time = now_beijing.isoformat()
+                # current_local_time = int(now_beijing.timestamp())  # 正确的时间戳
+                # current_local_time = datetime.now().isoformat()
+                supabase.table("airs_api_keys").update({"ran_at": current_local_time}).eq("id", send_req.get('api_key_info').get('api_key_id')).execute()
+                print('更新成功')
+            except Exception as e:
+                raise HTTPException(status_code=500, detail="更新API密钥运行时间失败！")
+
+
+
+            return api_key
         else:
             return None
     except Exception as e:

app.py

@@ -8,7 +8,7 @@ import os, json
 from dotenv import load_dotenv
 from enum import Enum
 from proxy import do_proxy
-from api_key_sb import get_api_key_info
+# from api_key_sb import get_api_key
 
 # 加载环境变量
 load_dotenv()

proxy.py

@@ -1,7 +1,7 @@
 import httpx
 import asyncio
 from starlette.responses import Response
-from api_key_sb import get_api_key_info
+from api_key_sb import get_api_key
 
 async def do_proxy(url: str, method: str, headers: dict, content: str, max_retries: int = 3):
     print("Proxy service started.", url)
@@ -11,8 +11,8 @@ async def do_proxy(url: str, method: str, headers: dict, content: str, max_retri
             for attempt in range(max_retries):
 
                 # 获取API密钥信息
-                api_key_info = await get_api_key_info('gemini-2.5-flash')
-                api_key = api_key_info.get('api_key')
+                api_key = await get_api_key('gemini-2.5-flash')
+                # api_key = api_key_info.get('api_key')
                 api_key_show = api_key[:5]+'*****'+api_key[-5:]
                 print(f"使用API密钥：{api_key_show}")
                 headers["Authorization"] = f"Bearer {api_key}"