lint

Author: twishabansal

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

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Callable, Optional, Union, Mapping, Awaitable
+from typing import Any, Awaitable, Callable, Mapping, Optional, Union
 from warnings import warn
 
 from aiohttp import ClientSession
@@ -41,7 +41,9 @@ def __init__(
             url: The base URL of the Toolbox service.
             session: An HTTP client session.
         """
-        self.__core_client = ToolboxCoreClient(url=url, session=session, client_headers=client_headers)
+        self.__core_client = ToolboxCoreClient(
+            url=url, session=session, client_headers=client_headers
+        )
 
     async def aload_tool(
         self,
@@ -188,7 +190,7 @@ def load_toolset(
         strict: bool = False,
     ) -> list[AsyncToolboxTool]:
         raise NotImplementedError("Synchronous methods not supported by async client.")
-    
+
     def add_headers(
         self,
         headers: Mapping[str, Union[Callable[[], str], Callable[[], Awaitable[str]]]],

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

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 from asyncio import to_thread
-from typing import Any, Callable, Optional, Union, Mapping, Awaitable
+from typing import Any, Awaitable, Callable, Mapping, Optional, Union
 from warnings import warn
 
 from toolbox_core.sync_client import ToolboxSyncClient as ToolboxCoreSyncClient
@@ -36,7 +36,9 @@ def __init__(
         Args:
             url: The base URL of the Toolbox service.
         """
-        self.__core_client = ToolboxCoreSyncClient(url=url, client_headers=client_headers)
+        self.__core_client = ToolboxCoreSyncClient(
+            url=url, client_headers=client_headers
+        )
 
     async def aload_tool(
         self,
@@ -289,7 +291,7 @@ def load_toolset(
         for core_sync_tool in core_sync_tools:
             tools.append(ToolboxTool(core_tool=core_sync_tool))
         return tools
-    
+
     def add_headers(
         self,
         headers: Mapping[str, Union[Callable[[], str], Callable[[], Awaitable[str]]]],

packages/toolbox-langchain/tests/test_async_client.py

@@ -339,7 +339,7 @@ async def test_load_toolset_not_implemented(self, mock_client):
         assert "Synchronous methods not supported by async client." in str(
             excinfo.value
         )
-    
+
     @patch("toolbox_langchain.async_client.ToolboxCoreClient")
     async def test_init_with_client_headers(
         self, mock_core_client_constructor, mock_session
@@ -357,4 +357,4 @@ async def test_add_headers(self, mock_client):
         mock_client.add_headers(headers)
         mock_client._AsyncToolboxClient__core_client.add_headers.assert_called_once_with(
             headers
-        )
+        )

packages/toolbox-langchain/tests/test_client.py

@@ -437,4 +437,4 @@ def test_add_headers(self, mock_core_client_constructor):
         client = ToolboxClient(URL)
         headers = {"X-Another-Header": "dynamic_value"}
         client.add_headers(headers)
-        mock_core_instance.add_headers.assert_called_once_with(headers)
+        mock_core_instance.add_headers.assert_called_once_with(headers)