update

api_key_sb.py

@@ -0,0 +1,27 @@
+from supabase import create_client, Client
+import os
+from fastapi import FastAPI, Request, HTTPException, Depends, status
+
+# 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

app.py

@@ -8,6 +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
 
 # 加载环境变量
 load_dotenv()
@@ -67,8 +68,14 @@ async def proxy_gemini(request: Request, protocol: ProtocolType, host:str, path:
 
     # 移除FastAPI自带的Host头，避免Gemini校验报错
     client_headers.pop("host", None)
+    
     # 将User-Agent改为curl/8.7.1，以模拟curl请求
     client_headers["User-Agent"] = "curl/8.7.1"
+    
+    # 获取API密钥信息
+    api_key_info = await get_api_key_info()
+
+    return api_key_info
 
     # 处理Gemini API密钥轮换和Authorization头
     global key_index # 声明使用全局变量