update

app.py

@@ -55,6 +55,9 @@ async def health_check():
 @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(request: Request, protocol: ProtocolType, host:str, path: str, proxy_api_key: str = Depends(verify_proxy_api_key)): # 添加代理认证依赖
+
+    print('\n\n======================================')
+
     real_url = f"{protocol.value}://{host}/{path}"
     
     # 提取客户端请求的headers和body，透传给Gemini
@@ -114,26 +117,38 @@ async def proxy(request: Request, protocol: ProtocolType, host:str, path: str, p
         )
     
     except httpx.RequestError as e:
+        print('\n\n----------------- 1 ------------------')
+        print(str(e))
+        print('\n\n----------------- 2 ------------------')
+        # return 
+        # raise HTTPException(status_code=e.response.status_code, detail=f"转发请求失败：{str(e)}")
         raise HTTPException(status_code=500, detail=f"转发请求失败：{str(e)}")
     except httpx.HTTPStatusError as e:
+        print('\n\n----------------- 3 ------------------')
+        print(str(e))
+        print('\n\n----------------- 4 ------------------')
         raise HTTPException(status_code=e.response.status_code, detail=f"目标API返回错误：{e.response.text}")
     except Exception as e:
+        print('\n\n----------------- 5 ------------------')
+        print(str(e))
+        print('\n\n----------------- 6 ------------------')
+        # raise HTTPException(status_code=e.response.status_code, detail=f"转发失败：{str(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:
-            response = supabase.from_('airs_model_api_keys_view').select('api_key').eq('model_name', model).execute()
+            response = supabase.from_('airs_model_api_keys_view').select('api_key').eq('model_name', model).order('api_key_ran_at', desc=False).limit(1).execute()
             # print(response.data)
         else:
             # 如果没有提供model，则获取所有模型的api_key
-            response = supabase.from_('airs_model_api_keys_view').select('api_key').execute()
+            raise HTTPException(status_code=400, detail="请提供模型名称！")
+            # response = supabase.from_('airs_model_api_keys_view').select('api_key').order('api_key_ran_at', desc=False).limit(1).execute()
         
         if response.data:
-            # 随机选择一个API密钥
-            import random
-            return random.choice([item['api_key'] for item in response.data])
+            # 由于已经limit(1)，直接返回第一条记录
+            return response.data[0]['api_key']
         else:
             return None
     except Exception as e: