update

proxy copy.py

@@ -0,0 +1,24 @@
+from fastapi.responses import Response # 导入Response
+import httpx # 使用httpx替代requests，因为requests是同步的，而FastAPI是异步的
+
+async def do_proxy(url:str, method:str, headers:dict, content:str):
+    print("Proxy service started.", url)
+    try:
+        # 使用httpx异步客户端发起请求
+        async with httpx.AsyncClient() as client:
+            response = await client.request(
+            method=method,
+            url=url,
+            headers=headers,
+            content=content, # httpx使用content参数传递请求体
+            timeout=30  # 超时时间，可调整
+        )
+    except Exception as e:
+        print("Error:", e)
+    
+    # 将Gemini的响应透传给客户端
+    return Response(
+        content=response.content,
+        status_code=response.status_code,
+        headers=dict(response.headers)
+    )

proxy.py

@@ -1,24 +1,44 @@
-from fastapi.responses import Response # 导入Response
-import httpx # 使用httpx替代requests，因为requests是同步的，而FastAPI是异步的
+import httpx
+import asyncio
+from starlette.responses import Response
 
-async def do_proxy(url:str, method:str, headers:dict, content:str):
+async def do_proxy(url: str, method: str, headers: dict, content: str, max_retries: int = 3):
     print("Proxy service started.", url)
+    client = None
     try:
-        # 使用httpx异步客户端发起请求
         async with httpx.AsyncClient() as client:
-            response = await client.request(
-            method=method,
-            url=url,
-            headers=headers,
-            content=content, # httpx使用content参数传递请求体
-            timeout=30  # 超时时间，可调整
-        )
+            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("Error:", e)
-    
-    # 将Gemini的响应透传给客户端
-    return Response(
-            content=response.content,
-            status_code=response.status_code,
-        headers=dict(response.headers)
-    )
+        print(f"🔥 致命错误: {type(e).__name__}: {str(e)}")
+        return Response(content="Internal Server Error", status_code=500)
+        
+    finally:
+        if client: 
+            await client.aclose()  # 确保连接关闭