chore: Delint

Author: anubhav756

packages/toolbox-langchain/src/toolbox_langchain/async_client.py

@@ -16,9 +16,9 @@
 from warnings import warn
 
 from aiohttp import ClientSession
+from toolbox_core.client import ToolboxClient as ToolboxCoreClient
 
 from .async_tools import AsyncToolboxTool
-from toolbox_core.client import ToolboxClient as ToolboxCoreClient
 
 
 # This class is an internal implementation detail and is not exposed to the
@@ -92,7 +92,7 @@ async def aload_tool(
         core_tool = await self.__core_client.load_tool(
             name=tool_name,
             auth_token_getters=auth_token_getters,
-            bound_params=bound_params
+            bound_params=bound_params,
         )
         return AsyncToolboxTool(core_tool=core_tool)
 
@@ -157,7 +157,7 @@ async def aload_toolset(
             name=toolset_name,
             auth_token_getters=auth_token_getters,
             bound_params=bound_params,
-            strict=strict
+            strict=strict,
         )
 
         tools = []

packages/toolbox-langchain/src/toolbox_langchain/async_tools.py

@@ -18,7 +18,6 @@
 from toolbox_core.tool import ToolboxTool as ToolboxCoreTool
 
 
-
 # This class is an internal implementation detail and is not exposed to the
 # end-user. It should not be used directly by external code. Changes to this
 # class will not be considered breaking changes to the public API.
@@ -64,8 +63,6 @@ async def _arun(self, **kwargs: Any) -> str:
         """
         return await self.__core_tool(**kwargs)
 
-
-
     def add_auth_token_getters(
         self, auth_token_getters: dict[str, Callable[[], str]]
     ) -> "AsyncToolboxTool":

packages/toolbox-langchain/src/toolbox_langchain/client.py

@@ -13,13 +13,13 @@
 # limitations under the License.
 
 import asyncio
-from warnings import warn
 from typing import Any, Callable, Optional, Union
+from warnings import warn
 
-from .tools import ToolboxTool
 from toolbox_core.sync_client import ToolboxSyncClient as ToolboxCoreSyncClient
 from toolbox_core.sync_tool import ToolboxSyncTool
 
+from .tools import ToolboxTool
 
 
 class ToolboxClient:
@@ -88,7 +88,7 @@ async def aload_tool(
         coro = self.__core_sync_client._ToolboxSyncClient__async_client.load_tool(
             name=tool_name,
             auth_token_getters=auth_token_getters,
-            bound_params=bound_params
+            bound_params=bound_params,
         )
 
         if not self.__core_sync_client._ToolboxSyncClient__loop:
@@ -97,10 +97,16 @@ async def aload_tool(
         else:
             # Otherwise, run in the background thread.
             core_tool = await asyncio.wrap_future(
-                asyncio.run_coroutine_threadsafe(coro, self.__core_sync_client._ToolboxSyncClient__loop)
+                asyncio.run_coroutine_threadsafe(
+                    coro, self.__core_sync_client._ToolboxSyncClient__loop
+                )
             )
 
-        core_sync_tool = ToolboxSyncTool(core_tool, self.__core_sync_client._ToolboxSyncClient__loop, self.__core_sync_client._ToolboxSyncClient__thread)
+        core_sync_tool = ToolboxSyncTool(
+            core_tool,
+            self.__core_sync_client._ToolboxSyncClient__loop,
+            self.__core_sync_client._ToolboxSyncClient__thread,
+        )
         return ToolboxTool(core_sync_tool=core_sync_tool)
 
     async def aload_toolset(
@@ -164,7 +170,7 @@ async def aload_toolset(
             name=toolset_name,
             auth_token_getters=auth_token_getters,
             bound_params=bound_params,
-            strict=strict
+            strict=strict,
         )
 
         if not self.__core_sync_client._ToolboxSyncClient__loop:
@@ -173,11 +179,17 @@ async def aload_toolset(
         else:
             # Otherwise, run in the background thread.
             core_tools = await asyncio.wrap_future(
-                asyncio.run_coroutine_threadsafe(coro, self.__core_sync_client._ToolboxSyncClient__loop)
+                asyncio.run_coroutine_threadsafe(
+                    coro, self.__core_sync_client._ToolboxSyncClient__loop
+                )
             )
 
         core_sync_tools = [
-            ToolboxSyncTool(core_tool, self.__core_sync_client._ToolboxSyncClient__loop, self.__core_sync_client._ToolboxSyncClient__thread)
+            ToolboxSyncTool(
+                core_tool,
+                self.__core_sync_client._ToolboxSyncClient__loop,
+                self.__core_sync_client._ToolboxSyncClient__thread,
+            )
             for core_tool in core_tools
         ]
         tools = []
@@ -237,7 +249,7 @@ def load_tool(
         core_sync_tool = self.__core_sync_client.load_tool(
             name=tool_name,
             auth_token_getters=auth_token_getters,
-            bound_params=bound_params
+            bound_params=bound_params,
         )
         return ToolboxTool(core_sync_tool=core_sync_tool)
 
@@ -302,7 +314,7 @@ def load_toolset(
             name=toolset_name,
             auth_token_getters=auth_token_getters,
             bound_params=bound_params,
-            strict=strict
+            strict=strict,
         )
 
         tools = []

packages/toolbox-langchain/src/toolbox_langchain/tools.py

@@ -19,7 +19,6 @@
 from toolbox_core.sync_tool import ToolboxSyncTool as ToolboxCoreSyncTool
 
 
-
 class ToolboxTool(BaseTool):
     """
     A subclass of LangChain's BaseTool that supports features specific to
@@ -58,10 +57,11 @@ async def _arun(self, **kwargs: Any) -> dict[str, Any]:
 
         # Otherwise, run in the background thread.
         return await asyncio.wrap_future(
-            asyncio.run_coroutine_threadsafe(coro, self.__core_sync_tool._ToolboxSyncTool__loop)
+            asyncio.run_coroutine_threadsafe(
+                coro, self.__core_sync_tool._ToolboxSyncTool__loop
+            )
         )
 
-
     def add_auth_token_getters(
         self, auth_token_getters: dict[str, Callable[[], str]]
     ) -> "ToolboxTool":
@@ -81,10 +81,11 @@ def add_auth_token_getters(
             ValueError: If any of the provided auth parameters is already
                 registered.
         """
-        new_core_sync_tool = self.__core_sync_tool.add_auth_token_getters(auth_token_getters)
+        new_core_sync_tool = self.__core_sync_tool.add_auth_token_getters(
+            auth_token_getters
+        )
         return ToolboxTool(core_sync_tool=new_core_sync_tool)
 
-
     def add_auth_token_getter(
         self, auth_source: str, get_id_token: Callable[[], str]
     ) -> "ToolboxTool":

packages/toolbox-langchain/tests/test_e2e.py

@@ -119,7 +119,8 @@ async def test_run_tool_unauth_with_auth(self, toolbox, auth_token2):
             match="Validation failed for tool 'get-row-by-id': unused auth tokens: my-test-auth.",
         ):
             await toolbox.aload_tool(
-                "get-row-by-id", auth_token_getters={"my-test-auth": lambda: auth_token2}
+                "get-row-by-id",
+                auth_token_getters={"my-test-auth": lambda: auth_token2},
             )
 
     async def test_run_tool_no_auth(self, toolbox):
@@ -182,10 +183,11 @@ async def test_run_tool_param_auth_no_field(self, toolbox, auth_token1):
         )
         with pytest.raises(
             Exception,
-            match="provided parameters were invalid: error parsing authenticated parameter \"data\": no field named row_data in claims"
+            match='provided parameters were invalid: error parsing authenticated parameter "data": no field named row_data in claims',
         ):
             await tool.ainvoke({})
 
+
 @pytest.mark.usefixtures("toolbox_server")
 class TestE2EClientSync:
     @pytest.fixture(scope="session")
@@ -262,7 +264,8 @@ def test_run_tool_unauth_with_auth(self, toolbox, auth_token2):
             match="Validation failed for tool 'get-row-by-id': unused auth tokens: my-test-auth.",
         ):
             toolbox.load_tool(
-                "get-row-by-id", auth_token_getters={"my-test-auth": lambda: auth_token2}
+                "get-row-by-id",
+                auth_token_getters={"my-test-auth": lambda: auth_token2},
             )
 
     def test_run_tool_no_auth(self, toolbox):
@@ -325,6 +328,6 @@ def test_run_tool_param_auth_no_field(self, toolbox, auth_token1):
         )
         with pytest.raises(
             ToolException,
-            match="provided parameters were invalid: error parsing authenticated parameter \"data\": no field named row_data in claims",
+            match='provided parameters were invalid: error parsing authenticated parameter "data": no field named row_data in claims',
         ):
             tool.invoke({})