diff --git a/packages/toolbox-langchain/src/toolbox_langchain/async_client.py b/packages/toolbox-langchain/src/toolbox_langchain/async_client.py index aacbc5af..c58bbfdf 100644 --- a/packages/toolbox-langchain/src/toolbox_langchain/async_client.py +++ b/packages/toolbox-langchain/src/toolbox_langchain/async_client.py @@ -68,31 +68,31 @@ async def aload_tool( Returns: A tool loaded from the Toolbox. """ - if auth_headers: + if auth_tokens: if auth_token_getters: warn( - "Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.", + "Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.", DeprecationWarning, ) else: warn( - "Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.", + "Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.", DeprecationWarning, ) - auth_token_getters = auth_headers + auth_token_getters = auth_tokens - if auth_tokens: + if auth_headers: if auth_token_getters: warn( - "Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.", + "Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.", DeprecationWarning, ) else: warn( - "Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.", + "Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.", DeprecationWarning, ) - auth_token_getters = auth_tokens + auth_token_getters = auth_headers url = f"{self.__url}/api/tool/{tool_name}" manifest: ManifestSchema = await _load_manifest(url, self.__session) @@ -136,31 +136,31 @@ async def aload_toolset( Returns: A list of all tools loaded from the Toolbox. """ - if auth_headers: + if auth_tokens: if auth_token_getters: warn( - "Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.", + "Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.", DeprecationWarning, ) else: warn( - "Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.", + "Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.", DeprecationWarning, ) - auth_token_getters = auth_headers + auth_token_getters = auth_tokens - if auth_tokens: + if auth_headers: if auth_token_getters: warn( - "Both `auth_token_getters` and `auth_tokens` are provided. `auth_tokens` is deprecated, and `auth_token_getters` will be used.", + "Both `auth_token_getters` and `auth_headers` are provided. `auth_headers` is deprecated, and `auth_token_getters` will be used.", DeprecationWarning, ) else: warn( - "Argument `auth_tokens` is deprecated. Use `auth_token_getters` instead.", + "Argument `auth_headers` is deprecated. Use `auth_token_getters` instead.", DeprecationWarning, ) - auth_token_getters = auth_tokens + auth_token_getters = auth_headers url = f"{self.__url}/api/toolset/{toolset_name or ''}" manifest: ManifestSchema = await _load_manifest(url, self.__session)