添加下载进度显示

router_proxy.py

@@ -85,19 +85,36 @@ async def proxy_github_zip(req_data: ProxyGithubZipRequest, db: Session = Depend
         headers["Authorization"] = f"Bearer {active_token}"
 
     # 3. 异步请求 GitHub API 并以流形式透传回客户端 (防内存打爆)
-    async def stream_generator():
-        async with httpx.AsyncClient(follow_redirects=True) as client:
+    client = httpx.AsyncClient(follow_redirects=True)
+    try:
+        response = await client.send(
+            client.build_request("GET", github_zip_api, headers=headers),
+            stream=True
+        )
+
+        # 获取 Content-Length 并透传给客户端，使本地端能计算下载进度
+        content_length = response.headers.get('content-length', '')
+        resp_headers = {}
+        if content_length:
+            resp_headers["Content-Length"] = content_length
+
+        async def stream_generator():
             try:
-                async with client.stream("GET", github_zip_api, headers=headers, timeout=120.0) as response:
-                    if response.status_code != 200:
-                        yield b"GITHUB_DOWNLOAD_FAILED"
-                        return
-                    async for chunk in response.aiter_bytes():
-                        yield chunk
-            except Exception as e:
+                if response.status_code != 200:
+                    yield b"GITHUB_DOWNLOAD_FAILED"
+                    return
+                async for chunk in response.aiter_bytes():
+                    yield chunk
+            except Exception:
                 yield b"GITHUB_DOWNLOAD_FAILED"
+            finally:
+                await response.aclose()
+                await client.aclose()
 
-    return StreamingResponse(stream_generator(), media_type="application/zip")
+        return StreamingResponse(stream_generator(), media_type="application/zip", headers=resp_headers)
+    except Exception as e:
+        await client.aclose()
+        return JSONResponse(content={"error": f"代理下载时发生网络异常：{str(e)}"}, status_code=500)
 
 # ==========================================
 # 新增：工作流/应用 (App) JSON 代理下载接口