Skip to content

Commit 9c08e1f

Browse files
committed
fixes
1 parent 0ff1ca6 commit 9c08e1f

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

pyatlan/pkg/utils.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def _is_valid_type(self, value: Any) -> bool:
6969

7070

7171
def get_client(
72-
impersonate_user_id: str, set_pkg_headers: Optional[bool] = False
72+
impersonate_user_id: Optional[str] = None, set_pkg_headers: Optional[bool] = False
7373
) -> AtlanClient:
7474
"""
7575
Set up the default Atlan client, based on environment variables.
@@ -96,6 +96,11 @@ def get_client(
9696
if set_pkg_headers:
9797
client = set_package_headers(client)
9898
return client
99+
else:
100+
LOGGER.info(
101+
"No OAuth client credentials found. Attempting to use API token or user impersonation."
102+
)
103+
99104
if api_token:
100105
LOGGER.info("Using provided API token for authentication.")
101106
api_key = api_token
@@ -119,7 +124,7 @@ def get_client(
119124

120125

121126
async def get_client_async(
122-
impersonate_user_id: str, set_pkg_headers: Optional[bool] = False
127+
impersonate_user_id: Optional[str] = None, set_pkg_headers: Optional[bool] = False
123128
) -> AsyncAtlanClient:
124129
"""
125130
Set up the default async Atlan client, based on environment variables.
@@ -146,6 +151,10 @@ async def get_client_async(
146151
if set_pkg_headers:
147152
client = set_package_headers(client)
148153
return client
154+
else:
155+
LOGGER.info(
156+
"No OAuth client credentials found. Attempting to use API token or user impersonation."
157+
)
149158

150159
if api_token:
151160
LOGGER.info("Using provided API token for authentication.")

0 commit comments

Comments
 (0)