Upload router_proxy.py

router_proxy.py

@@ -107,22 +107,24 @@ async def proxy_download(req_data: ProxyDownloadRequest, db: Session = Depends(g
     if hf_token and "huggingface.co" in target_url:
         headers["Authorization"] = f"Bearer {hf_token}"
     
-    # 2. 使用 urllib 拉取真实 JSON 数据（保持原始中文文件名，不自动编码）
+    # 🚀 核心修复：使用异步 httpx 替代同步 urllib，避免阻塞事件循环
     try:
-        # 🚀 核心修复：创建 SSL 上下文，禁用证书验证（解决 Hugging Face SSL 问题）
-        import ssl
-        ssl_context = ssl.create_default_context()
-        ssl_context.check_hostname = False
-        ssl_context.verify_mode = ssl.CERT_NONE
-        
-        req = urllib.request.Request(target_url, headers=headers)
-        
-        # 🚀 使用自定义 SSL 上下文，超时 120 秒
-        with urllib.request.urlopen(req, timeout=120, context=ssl_context) as resp:
-            content = resp.read()
-            # 🚀 调试日志：打印响应大小
+        async with httpx.AsyncClient(follow_redirects=True, verify=False, timeout=120.0) as client:
+            print(f"🔍 开始下载资源 [{req_data.item_id}]")
+            print(f"🔗 目标地址：{target_url[:80]}...")
+            
+            response = await client.get(target_url, headers=headers)
+            
+            if response.status_code != 200:
+                print(f"❌ 源文件拉取失败 [HTTP {response.status_code}]")
+                return JSONResponse(
+                    content={"error": f"源文件拉取失败，HTTP 状态码：{response.status_code}"}, 
+                    status_code=response.status_code
+                )
+            
+            content = response.content
             print(f"✅ 成功下载资源 [{req_data.item_id}], 大小：{len(content)} bytes")
-            # 🚀 核心修复：使用正确的 Response 参数名，并明确指定状态码
+            
             return Response(
                 content=content, 
                 media_type="application/json", 
@@ -130,10 +132,9 @@ async def proxy_download(req_data: ProxyDownloadRequest, db: Session = Depends(g
                 headers={"Content-Disposition": f"attachment; filename={req_data.item_id}.json"}
             )
             
-    except urllib.error.HTTPError as e:
-        error_detail = e.read().decode('utf-8', errors='ignore') if hasattr(e, 'read') else str(e)
-        print(f"❌ 源文件拉取失败 [HTTP {e.code}]: {error_detail}")
-        return JSONResponse(content={"error": f"源文件拉取失败，HTTP 状态码：{e.code}"}, status_code=e.code)
+    except httpx.TimeoutException as e:
+        print(f"❌ 下载超时：{str(e)}")
+        return JSONResponse(content={"error": "下载超时，请稍后重试"}, status_code=504)
     except Exception as e:
         import traceback
         print(f"❌ 代理下载异常：{str(e)}")