update

app.py

@@ -7,6 +7,7 @@ import httpx # 使用httpx替代requests，因为requests是同步的，而FastA
 import os, json
 from dotenv import load_dotenv
 from enum import Enum
+from supabase import create_client, Client
 
 # 加载环境变量
 load_dotenv()
@@ -14,6 +15,13 @@ load_dotenv()
 # 从环境变量获取代理自身的API密钥
 proxy_api_key = os.getenv("PROXY_API_KEY")
 
+# Supabase 配置
+SUPABASE_URL = os.getenv("SUPABASE_URL")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY")
+
+# 初始化 Supabase 客户端
+supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+
 # 定义HTTPBearer安全方案
 security = HTTPBearer()
 
@@ -43,23 +51,9 @@ def greet_json():
 async def health_check():
     return {"status": "ok", "message": "Proxy service is running."}
 
-gemini_api_keys_str = os.getenv("GEMINI_API_KEYS_STR")
-try:
-    gemini_api_keys_arr = json.loads(gemini_api_keys_str)
-except json.JSONDecodeError as e:
-    print(f"JSON 解析错误：{e}")
-    gemini_api_keys_arr = []  # 解析失败时设置默认值
-except Exception as e:
-    print(f"其他错误：{e}")
-    gemini_api_keys_arr = []
-
-key_index = 0
-keys_count = len(gemini_api_keys_arr)
-
 @app.api_route("/v1/{protocol}/{host}/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
 # @app.api_route("/v1/{protocol}/{host}/{path:path}", methods=["POST"])
-async def proxy_gemini(request: Request, protocol: ProtocolType, host:str, path: str, proxy_api_key: str = Depends(verify_proxy_api_key)): # 添加代理认证依赖
-    global key_index # 声明使用全局变量
+async def proxy(request: Request, protocol: ProtocolType, host:str, path: str, proxy_api_key: str = Depends(verify_proxy_api_key)): # 添加代理认证依赖
     real_url = f"{protocol.value}://{host}/{path}"
     
     # 提取客户端请求的headers和body，透传给Gemini
@@ -67,26 +61,42 @@ async def proxy_gemini(request: Request, protocol: ProtocolType, host:str, path:
     
     # 移除FastAPI自带的Host头，避免Gemini校验报错
     client_headers.pop("host", None)
+    client_headers.pop("authorization", None)
+    client_headers.pop("user-agent", None)
+    client_headers.pop("content-length", None)
     
     # 将User-Agent改为curl/8.7.1，以模拟curl请求
     client_headers["User-Agent"] = "curl/8.7.1"
-    
-    # 添加Authorization头，用于Gemini的OpenAI兼容API
-    if gemini_api_keys_arr: # 确保有可用的API密钥
-        print(f"key_index: {key_index}")
-        gemini_api_key = gemini_api_keys_arr[key_index]
-        key_index = (key_index + 1) % keys_count
-        print(f"next_key_index: {key_index}")
-        client_headers["Authorization"] = f"Bearer {gemini_api_key}"
-    elif gemini_api_key: # 如果没有API密钥列表，使用单个API密钥
-        client_headers["Authorization"] = f"Bearer {gemini_api_key}"
-    else:
-        raise HTTPException(status_code=500, detail="No Gemini API keys configured.")
-    
+
+    # 获取 api_key
+    api_key = await get_api_key('glm-4-flash')
+    print('\n\napi_key', api_key)
+    if not api_key:
+        raise HTTPException(status_code=500, detail="没有可用的API密钥！")
+
+     # 添加Authorization头
+    client_headers["Authorization"] = f"Bearer {api_key}"
+
     try:
         # 读取客户端请求体
         client_body = await request.body()
         
+        # 尝试从请求体中获取model
+        model_from_body = None
+        try:
+            if client_body:
+                body_json = json.loads(client_body)
+                model_from_body = body_json.get("model")
+        except json.JSONDecodeError:
+            # 如果请求体不是JSON，则忽略
+            pass
+
+        # 获取 api_key
+        api_key = await get_api_key(model_from_body if model_from_body else 'glm-4-flash')
+        print('\n\napi_key', api_key)
+        if not api_key:
+            raise HTTPException(status_code=500, detail="没有可用的API密钥！")
+
         # 使用httpx异步客户端发起请求
         async with httpx.AsyncClient() as client:
             response = await client.request(
@@ -98,6 +108,7 @@ async def proxy_gemini(request: Request, protocol: ProtocolType, host:str, path:
             )
         
         # 将Gemini的响应透传给客户端
+        print('444444444444')
         return Response(
             content=response.content,
             status_code=response.status_code,
@@ -110,3 +121,26 @@ async def proxy_gemini(request: Request, protocol: ProtocolType, host:str, path:
         raise HTTPException(status_code=e.response.status_code, detail=f"目标API返回错误：{e.response.text}")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"转发失败：{str(e)}")
+
+async def get_api_key(model: str = None):
+    try:
+        # 根据用户提供的SQL语句修改，从 'airs_model_api_keys_view' 视图中获取 'api_key'
+        if model:
+            print('\n\n1111111')
+            response = supabase.from_('airs_model_api_keys_view').select('api_key').eq('model_name', model).execute()
+            print('\n\n2222222')
+            print(response.data)
+        else:
+            print('\n\n3333333')
+            # 如果没有提供model，则获取所有模型的api_key
+            response = supabase.from_('airs_model_api_keys_view').select('api_key').execute()
+        
+        if response.data:
+            # 随机选择一个API密钥
+            import random
+            return random.choice([item['api_key'] for item in response.data])
+        else:
+            return None
+    except Exception as e:
+        print(f"从Supabase获取API密钥失败: {e}")
+        return None

memory-bank/activeContext.md

@@ -15,8 +15,8 @@
 *   **`.clinerules` 更新**：添加了自动激活 `conda` 环境 `airs` 的自定义指令。
 
 ## 下一步
-*   **验证 Postman 兼容性**：用户需要再次尝试使用 Postman 调用部署的代理，以确认 `User-Agent` 头部修改是否解决了“incorrect header check”错误。
-*   **本地测试环境问题**：如果本地测试仍然遇到“User location is not supported for the API use.”错误，需要明确告知用户这是 Gemini API 的地域限制，而非代理代码问题。
+*   **验证 Postman 兼容性**：用户已指示暂时搁置此问题。
+*   **本地测试环境问题**：已向用户明确告知这是 Gemini API 的地域限制，而非代理代码问题。
 
 ## 活跃决策和考虑
 *   **API 密钥传递方式**：从 URL 查询参数改为 `Authorization` 头是符合 OpenAI 兼容 API 规范的正确做法。

memory-bank/progress.md

@@ -12,11 +12,11 @@
 *   **部署环境验证**：通过 `curl` 命令验证了部署在 Hugging Face Spaces 上的代理功能正常。
 
 ## 待完成的工作
-*   **Postman 客户端兼容性验证**：用户需要再次尝试使用 Postman 调用部署的代理，以确认 `User-Agent` 头部修改是否解决了“incorrect header check”错误。
-*   **本地测试地域限制说明**：如果用户在本地测试时仍然遇到“User location is not supported for the API use.”错误，需要明确告知这是 Gemini API 的地域限制，而非代理代码问题。
+*   **Postman 客户端兼容性验证**：用户已指示暂时搁置此问题。
+*   **本地测试地域限制说明**：已向用户明确告知这是 Gemini API 的地域限制，而非代理代码问题。
 
 ## 当前状态
-代理应用程序的核心功能已完成并部署验证。主要剩余任务是解决特定客户端（Postman）的兼容性问题，并向用户明确本地测试可能遇到的地域限制。
+代理应用程序的核心功能已完成并部署验证。所有待完成的任务已处理或暂时搁置。
 
 ## 已知问题
 *   **本地测试地域限制**：在某些地域，直接从本地机器访问 Gemini API 可能会遇到“User location is not supported for the API use.”错误。

read_supabase_data.py

@@ -0,0 +1,64 @@
+import os
+from dotenv import load_dotenv
+from supabase import create_client, Client
+import asyncio
+
+# 加载环境变量
+load_dotenv()
+
+# Supabase 配置
+SUPABASE_URL = os.getenv("SUPABASE_URL")
+SUPABASE_KEY = os.getenv("SUPABASE_KEY")
+
+async def read_api_keys_from_supabase(model_name: str = None, limit: int = None):
+    """
+    从 Supabase 的 airs_model_api_keys_view 视图中读取 API 密钥。
+    """
+    if not SUPABASE_URL or not SUPABASE_KEY:
+        print("错误：SUPABASE_URL 或 SUPABASE_KEY 未在 .env 文件中配置。")
+        return None
+
+    try:
+        supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
+        print(f"已连接到 Supabase: {SUPABASE_URL}")
+        print('supabase', supabase)
+
+        print(f"尝试从 Supabase 获取 API 密钥，模型名称: {model_name if model_name else '所有模型'}{f', 限制: {limit}' if limit else ''}")
+
+        query = supabase.from_('airs_model_api_keys_view').select('api_key,model_name')
+
+        if model_name:
+            query = query.eq('model_name', model_name)
+        
+        if limit:
+            query = query.limit(limit)
+
+        response = query.execute()
+
+        if response.data:
+            print("\n成功从 Supabase 获取数据：")
+            for item in response.data:
+                print(f"  API Key: {item.get('api_key')}, Model Name: {item.get('model_name')}")
+            return response.data
+        else:
+            print("\n从 Supabase 获取数据成功，但未找到匹配的 API 密钥。")
+            return None
+    except Exception as e:
+        print(f"\n从 Supabase 获取 API 密钥失败: {e}")
+        return None
+
+async def main():
+    print("--- 测试获取所有 API 密钥 ---")
+    await read_api_keys_from_supabase()
+
+    print("\n--- 测试获取特定模型 (glm-4-flash) 的 API 密钥 ---")
+    await read_api_keys_from_supabase('glm-4-flash')
+
+    print("\n--- 测试获取特定模型 (gemini-2.5-flash) 的 API 密钥 ---")
+    await read_api_keys_from_supabase('gemini-2.5-flash')
+
+    print("\n--- 测试获取前 5 条 API 密钥 ---")
+    await read_api_keys_from_supabase(limit=5)
+
+if __name__ == "__main__":
+    asyncio.run(main())