imp(sdk/python): increase the default timeout to 3 secs for Python SDK.

Author: Leechael

sdk/python/src/dstack_sdk/dstack_client.py

@@ -226,7 +226,7 @@ class BaseClient:
 class AsyncDstackClient(BaseClient):
     PATH_PREFIX = "/"
 
-    def __init__(self, endpoint: str | None = None, use_sync_http: bool = False):
+    def __init__(self, endpoint: str | None = None, *, use_sync_http: bool = False, timeout: float = 3):
         """Initialize async client with HTTP or Unix-socket transport.
 
         Args:
@@ -239,6 +239,7 @@ def __init__(self, endpoint: str | None = None, use_sync_http: bool = False):
         self._client: Optional[httpx.AsyncClient] = None
         self._sync_client: Optional[httpx.Client] = None
         self._client_ref_count = 0
+        self._timeout = timeout
 
         if endpoint.startswith("http://") or endpoint.startswith("https://"):
             self.async_transport = httpx.AsyncHTTPTransport()
@@ -255,14 +256,14 @@ def __init__(self, endpoint: str | None = None, use_sync_http: bool = False):
     def _get_client(self) -> httpx.AsyncClient:
         if self._client is None:
             self._client = httpx.AsyncClient(
-                transport=self.async_transport, base_url=self.base_url, timeout=0.5
+                transport=self.async_transport, base_url=self.base_url, timeout=self._timeout
             )
         return self._client
 
     def _get_sync_client(self) -> httpx.Client:
         if self._sync_client is None:
             self._sync_client = httpx.Client(
-                transport=self.sync_transport, base_url=self.base_url, timeout=0.5
+                transport=self.sync_transport, base_url=self.base_url, timeout=self._timeout
             )
         return self._sync_client
 
@@ -392,13 +393,13 @@ async def is_reachable(self) -> bool:
 class DstackClient(BaseClient):
     PATH_PREFIX = "/"
 
-    def __init__(self, endpoint: str | None = None):
+    def __init__(self, endpoint: str | None = None, *, timeout: float = 3):
         """Initialize client with HTTP or Unix-socket transport.
 
         If a non-HTTP(S) endpoint is provided, it is treated as a Unix socket
         path and validated for existence.
         """
-        self.async_client = AsyncDstackClient(endpoint, use_sync_http=True)
+        self.async_client = AsyncDstackClient(endpoint, use_sync_http=True, timeout=timeout)
 
     @call_async
     def get_key(
@@ -463,7 +464,7 @@ class AsyncTappdClient(AsyncDstackClient):
     DEPRECATED: Use ``AsyncDstackClient`` instead.
     """
 
-    def __init__(self, endpoint: str | None = None, use_sync_http: bool = False):
+    def __init__(self, endpoint: str | None = None, *, use_sync_http: bool = False, timeout: float = 3):
         """Initialize deprecated async tappd client wrapper."""
         if not use_sync_http:
             # Already warned in TappdClient.__init__
@@ -472,7 +473,7 @@ def __init__(self, endpoint: str | None = None, use_sync_http: bool = False):
             )
 
         endpoint = get_tappd_endpoint(endpoint)
-        super().__init__(endpoint, use_sync_http=use_sync_http)
+        super().__init__(endpoint, use_sync_http=use_sync_http, timeout=timeout)
         # Set the correct path prefix for tappd
         self.PATH_PREFIX = "/prpc/Tappd."
 
@@ -542,13 +543,13 @@ class TappdClient(DstackClient):
     DEPRECATED: Use ``DstackClient`` instead.
     """
 
-    def __init__(self, endpoint: str | None = None):
+    def __init__(self, endpoint: str | None = None, timeout: float = 3):
         """Initialize deprecated tappd client wrapper."""
         emit_deprecation_warning(
             "TappdClient is deprecated, please use DstackClient instead"
         )
         endpoint = get_tappd_endpoint(endpoint)
-        self.async_client = AsyncTappdClient(endpoint, use_sync_http=True)
+        self.async_client = AsyncTappdClient(endpoint, use_sync_http=True, timeout=timeout)
 
     @call_async
     def derive_key(