bug fixes

Author: vaibhavatlan

pyatlan/client/aio/client.py

@@ -132,12 +132,17 @@ class AsyncAtlanClient(AtlanClient):
     _async_user_cache: Optional[AsyncUserCache] = PrivateAttr(default=None)
 
     def __init__(self, **kwargs):
+        # Initialize sync client (handles all validation, env vars, etc.)
         super().__init__(**kwargs)
 
-        if self.oauth_client_id and self.oauth_client_secret:
+        if self.oauth_client_id and self.oauth_client_secret and self.api_key is None:
+            LOGGER.debug(
+                "API Key not provided. Using Async OAuth flow for authentication"
+            )
             from pyatlan.client.aio.oauth import AsyncOAuthTokenManager
 
             if self._oauth_token_manager:
+                LOGGER.debug("Sync oauth flow open. Closing it for Async oauth flow")
                 self._oauth_token_manager.close()
                 self._oauth_token_manager = None
 
@@ -147,6 +152,7 @@ def __init__(self, **kwargs):
                 client_secret=self.oauth_client_secret,
             )
 
+        # Build proxy/SSL configuration (reuse from sync client)
         transport_kwargs = self._build_transport_proxy_config(kwargs)
 
         # Create async session with custom transport that supports retry and proxy
@@ -707,10 +713,6 @@ async def _handle_error_response(
                 and response.status_code
                 == ErrorCode.AUTHENTICATION_PASSTHROUGH.http_error_code
             ):
-                """
-                Async version of token refresh and retry logic.
-                Handles token refresh and retries the API request upon a 401 Unauthorized response.
-                """
                 try:
                     LOGGER.debug("Starting async 401 automatic token refresh.")
                     return await self._handle_401_token_refresh(
@@ -761,6 +763,10 @@ async def _handle_401_token_refresh(
         download_file_path=None,
         text_response=False,
     ):
+        """
+        Async version of token refresh and retry logic.
+        Handles token refresh and retries the API request upon a 401 Unauthorized response.
+        """
         if self._async_oauth_token_manager:
             await self._async_oauth_token_manager.invalidate_token()
             token = await self._async_oauth_token_manager.get_token()

pyatlan/client/aio/oauth.py

@@ -42,6 +42,13 @@ async def get_token(self) -> str:
 
             data = response.json()
             access_token = data.get("accessToken") or data.get("access_token")
+
+            if not access_token:
+                raise ValueError(
+                    f"OAuth token response missing 'accessToken' field. "
+                    f"Response keys: {list(data.keys())}"
+                )
+
             expires_in = data.get("expiresIn") or data.get("expires_in", 600)
 
             self._token = OAuth2Token(

pyatlan/client/atlan.py

@@ -176,7 +176,7 @@ class Config:
     def __init__(self, **data):
         super().__init__(**data)
 
-        if self.oauth_client_id and self.oauth_client_secret:
+        if self.oauth_client_id and self.oauth_client_secret and self.api_key is None:
             from pyatlan.client.oauth import OAuthTokenManager
 
             self._oauth_token_manager = OAuthTokenManager(

pyatlan/client/oauth.py

@@ -42,6 +42,13 @@ def get_token(self) -> str:
 
             data = response.json()
             access_token = data.get("accessToken") or data.get("access_token")
+
+            if not access_token:
+                raise ValueError(
+                    f"OAuth token response missing 'accessToken' field. "
+                    f"Response keys: {list(data.keys())}"
+                )
+
             expires_in = data.get("expiresIn") or data.get("expires_in", 600)
 
             self._token = OAuth2Token(