chore: correct arg name

Author: kurtisvg

packages/toolbox-core/src/toolbox_core/client.py

@@ -119,7 +119,7 @@ async def close(self):
     async def load_tool(
         self,
         name: str,
-        auth_service_tokens: dict[str, Callable[[], str]] = {},
+        auth_token_getters: dict[str, Callable[[], str]] = {},
     ) -> ToolboxTool:
         """
         Asynchronously loads a tool from the server.
@@ -130,6 +130,8 @@ async def load_tool(
 
         Args:
             name: The unique name or identifier of the tool to load.
+            auth_token_getters: A mapping of authentication service names to
+                callables that return the corresponding authentication token.
 
         Returns:
             ToolboxTool: A callable object representing the loaded tool, ready
@@ -148,7 +150,7 @@ async def load_tool(
         if name not in manifest.tools:
             # TODO: Better exception
             raise Exception(f"Tool '{name}' not found!")
-        tool = self.__parse_tool(name, manifest.tools[name], auth_service_tokens)
+        tool = self.__parse_tool(name, manifest.tools[name], auth_token_getters)
 
         return tool
 
@@ -162,6 +164,9 @@ async def load_toolset(
 
         Args:
             name: Name of the toolset to load tools.
+            auth_token_getters: A mapping of authentication service names to
+                callables that return the corresponding authentication token.
+
 
         Returns:
             list[ToolboxTool]: A list of callables, one for each tool defined

packages/toolbox-core/tests/test_client.py

@@ -18,7 +18,6 @@
 import pytest
 import pytest_asyncio
 from aioresponses import CallbackResult
-
 from toolbox_core import ToolboxClient
 from toolbox_core.protocol import ManifestSchema, ParameterSchema, ToolSchema
 
@@ -150,7 +149,7 @@ def token_handler():
             return expected_header
 
         tool = await client.load_tool(
-            tool_name, auth_service_tokens={"my-auth-service": token_handler}
+            tool_name, auth_token_getters={"my-auth-service": token_handler}
         )
         res = await tool(5)