Increase GatewayClient timeout to 120s

dstackai · Apr 30, 2024 · c3687db · c3687db
1 parent 2761a90
commit c3687db
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/dstack/_internal/server/services/gateways/client.py b/src/dstack/_internal/server/services/gateways/client.py
@@ -33,8 +33,10 @@ def __init__(self, uds: Optional[str] = None, port: Optional[int] = None):
         self.is_server_ready = False
 
         self.base_url = "http://gateway" if uds else f"http://localhost:{port}"
+        # We set large timeout because service registration can take time.
+        # Consider making gateway API async to avoid long-running requests.
         self._client = AsyncClientWrapper(
-            transport=httpx.AsyncHTTPTransport(uds=uds) if uds else None, timeout=30
+            transport=httpx.AsyncHTTPTransport(uds=uds) if uds else None, timeout=120
         )
 
     async def register_service(