Add timeout

Clinical-Genomics · Feb 6, 2024 · 9a135f6 · 9a135f6
1 parent fa507ec
commit 9a135f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/schug/load/fetch_resource.py b/schug/load/fetch_resource.py
@@ -65,7 +65,8 @@ def fetch_resource(url: str) -> List[str]:
 
 async def stream_resource(url: str) -> AsyncGenerator:
     """Stream a file from an external service"""
-    async with httpx.AsyncClient(timeout=None) as client:
+    timeout = httpx.Timeout(connect=None, read=None, write=None, pool=None)
+    async with httpx.AsyncClient(timeout=timeout) as client:
         async with client.stream("GET", url) as r:
             async for chunk in r.aiter_bytes():
                 yield chunk