update

app.py

@@ -73,17 +73,12 @@ async def proxy_gemini(request: Request, protocol: ProtocolType, host:str, path:
     # 将User-Agent改为curl/8.7.1，以模拟curl请求
     client_headers["User-Agent"] = "curl/8.7.1"
     
-    # 获取API密钥信息
-    api_key_info = await get_api_key_info('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}")
-    client_headers["Authorization"] = f"Bearer {api_key}"
+
     # api_key_id = api_key_info.get('api_key_id')
 
     print('\n\n-----------')
     print(client_headers)
     print('-----------\n\n')
 
-    # 调用 do_proxy 并返回其结果
+    # 调用 do_proxy 并返回其结果，此处传的 headers 是不带 api_key 的数据
     return await do_proxy(real_url, request.method, client_headers, client_body)

proxy copy 2.py

@@ -0,0 +1,44 @@
+import httpx
+import asyncio
+from starlette.responses import Response
+
+async def do_proxy(url: str, method: str, headers: dict, content: str, max_retries: int = 3):
+    print("Proxy service started.", url)
+    client = None
+    try:
+        async with httpx.AsyncClient() as client:
+            for attempt in range(max_retries):
+                response = await client.request(
+                    method=method,
+                    url=url,
+                    headers=headers,
+                    content=content,
+                    timeout=30
+                )
+                
+                # 关键：显式检查状态码
+                if response.status_code == 429:
+                    retry_after = response.headers.get("Retry-After", "5")  # 默认5秒
+                    wait_time = int(retry_after) + attempt * 2  # 指数退避基础值
+                    print(f"⚠️ 429错误！{wait_time}秒后重试 (尝试:{attempt+1})")
+                    await asyncio.sleep(wait_time)
+                    continue
+                    
+                response.raise_for_status()  # 触发其他4xx/5xx异常
+                return Response(
+                    content=response.content,
+                    status_code=response.status_code,
+                    headers=dict(response.headers)
+                )
+                
+    except httpx.HTTPStatusError as e:
+        print(f"🚨 服务器错误 {e.response.status_code}: {e.request.url}")
+        return Response(content=str(e), status_code=e.response.status_code)
+        
+    except Exception as e:
+        print(f"🔥 致命错误: {type(e).__name__}: {str(e)}")
+        return Response(content="Internal Server Error", status_code=500)
+        
+    finally:
+        if client: 
+            await client.aclose()  # 确保连接关闭

proxy.py

@@ -1,6 +1,7 @@
 import httpx
 import asyncio
 from starlette.responses import Response
+from api_key_sb import get_api_key_info
 
 async def do_proxy(url: str, method: str, headers: dict, content: str, max_retries: int = 3):
     print("Proxy service started.", url)
@@ -8,6 +9,13 @@ async def do_proxy(url: str, method: str, headers: dict, content: str, max_retri
     try:
         async with httpx.AsyncClient() as client:
             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_show = api_key[:5]+'*****'+api_key[-5:]
+                print(f"使用API密钥：{api_key_show}")
+                headers["Authorization"] = f"Bearer {api_key}"
                 response = await client.request(
                     method=method,
                     url=url,