timeout fix

Author: vaibhavatlan

pyatlan/client/aio/client.py

@@ -159,6 +159,8 @@ def __init__(self, **kwargs):
                 base_url=final_base_url,
                 client_id=final_oauth_client_id,
                 client_secret=final_oauth_client_secret,
+                connect_timeout=self.connect_timeout,
+                read_timeout=self.read_timeout,
             )
 
         # Build proxy/SSL configuration (reuse from sync client)

pyatlan/client/aio/oauth.py

@@ -18,13 +18,19 @@ def __init__(
         client_id: str,
         client_secret: str,
         http_client: Optional[httpx.AsyncClient] = None,
+        connect_timeout: float = 30.0,
+        read_timeout: float = 900.0,
     ):
         self.base_url = base_url
         self.client_id = client_id
         self.client_secret = client_secret
         self.token_url = self._create_path(GET_OAUTH_CLIENT)
         self._lock = asyncio.Lock()
-        self._http_client = http_client or httpx.AsyncClient(timeout=30.0)
+        self._http_client = http_client or httpx.AsyncClient(
+            timeout=httpx.Timeout(
+                connect=connect_timeout, read=read_timeout, write=30.0, pool=30.0
+            )
+        )
         self._token: Optional[OAuth2Token] = None
         self._owns_client = http_client is None
 

pyatlan/client/atlan.py

@@ -193,6 +193,8 @@ def __init__(self, **data):
                 base_url=final_base_url,
                 client_id=final_oauth_client_id,
                 client_secret=final_oauth_client_secret,
+                connect_timeout=self.connect_timeout,
+                read_timeout=self.read_timeout,
             )
             self._request_params = {"headers": {}}
         else:

pyatlan/client/oauth.py

@@ -18,13 +18,19 @@ def __init__(
         client_id: str,
         client_secret: str,
         http_client: Optional[httpx.Client] = None,
+        connect_timeout: float = 30.0,
+        read_timeout: float = 900.0,
     ):
         self.base_url = base_url
         self.client_id = client_id
         self.client_secret = client_secret
         self.token_url = self._create_path(GET_OAUTH_CLIENT)
         self._lock = threading.Lock()
-        self._http_client = http_client or httpx.Client(timeout=30.0)
+        self._http_client = http_client or httpx.Client(
+            timeout=httpx.Timeout(
+                connect=connect_timeout, read=read_timeout, write=30.0, pool=30.0
+            )
+        )
         self._token: Optional[OAuth2Token] = None
         self._owns_client = http_client is None