diff --git a/CHANGELOG.md b/CHANGELOG.md index d6ab448..3a94562 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -2,9 +2,7 @@ ## 4.4.2 (unreleased) - -- Nothing changed yet. - +- Fix auth validation with async client. ## 4.4.1 (2024-12-19) diff --git a/nuclia/data.py b/nuclia/data.py index 57cb50a..a4da486 100644 --- a/nuclia/data.py +++ b/nuclia/data.py @@ -86,7 +86,7 @@ def get_client(kbid: str) -> NucliaDBClient: return ndb -def get_async_client(kbid: str) -> AsyncNucliaDBClient: +async def get_async_client(kbid: str) -> AsyncNucliaDBClient: auth = get_async_auth() kb_obj = auth._config.get_kb(kbid) @@ -96,7 +96,7 @@ def get_async_client(kbid: str) -> AsyncNucliaDBClient: # OSS ndb = AsyncNucliaDBClient(environment=Environment.OSS, url=kb_obj.url) else: - if kb_obj.token is None and auth._validate_user_token(): + if kb_obj.token is None and await auth._validate_user_token(): # User token auth ndb = AsyncNucliaDBClient( environment=Environment.CLOUD, diff --git a/nuclia/decorators.py b/nuclia/decorators.py index 0d78d7e..557afd1 100644 --- a/nuclia/decorators.py +++ b/nuclia/decorators.py @@ -47,7 +47,7 @@ async def async_wrapper_checkout_kb(*args, **kwargs): kbid = auth._config.get_default_kb() if kbid is None: raise NotDefinedDefault() - ndb = get_async_client(kbid) + ndb = await get_async_client(kbid) elif url.find(BASE_DOMAIN) >= 0: region = url.split(".")[0].split("/")[-1] ndb = AsyncNucliaDBClient( diff --git a/nuclia/sdk/kb.py b/nuclia/sdk/kb.py index 511e09c..b3ec7dd 100644 --- a/nuclia/sdk/kb.py +++ b/nuclia/sdk/kb.py @@ -552,7 +552,7 @@ async def copy( remote_files[file_id] = file.value else: files_to_upload.append({"id": file_id, "data": file.value}) - destination_kb = get_async_client(destination) + destination_kb = await get_async_client(destination) failed = False try: uuid = await self.resource.create(ndb=destination_kb, **data)