fix(toolbox-langchain)!: Base toolbox-langchain over toolbox-core

Author: anubhav756

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

@@ -18,7 +18,8 @@
 from aiohttp import ClientSession
 
 from .tools import AsyncToolboxTool
-from .utils import ManifestSchema, _load_manifest
+
+from toolbox_core.client import ToolboxClient as ToolboxCoreClient
 
 
 # This class is an internal implementation detail and is not exposed to the
@@ -38,8 +39,7 @@ def __init__(
             url: The base URL of the Toolbox service.
             session: An HTTP client session.
         """
-        self.__url = url
-        self.__session = session
+        self.__core_client = ToolboxCoreClient(url=url, session=session)
 
     async def aload_tool(
         self,
@@ -48,7 +48,6 @@ async def aload_tool(
         auth_tokens: Optional[dict[str, Callable[[], str]]] = None,
         auth_headers: Optional[dict[str, Callable[[], str]]] = None,
         bound_params: dict[str, Union[Any, Callable[[], Any]]] = {},
-        strict: bool = True,
     ) -> AsyncToolboxTool:
         """
         Loads the tool with the given tool name from the Toolbox service.
@@ -61,9 +60,6 @@ async def aload_tool(
             auth_headers: Deprecated. Use `auth_token_getters` instead.
             bound_params: An optional mapping of parameter names to their
                 bound values.
-            strict: If True, raises a ValueError if any of the given bound
-                parameters are missing from the schema or require
-                authentication. If False, only issues a warning.
 
         Returns:
             A tool loaded from the Toolbox.
@@ -94,18 +90,12 @@ async def aload_tool(
                 )
                 auth_token_getters = auth_tokens
 
-        url = f"{self.__url}/api/tool/{tool_name}"
-        manifest: ManifestSchema = await _load_manifest(url, self.__session)
-
-        return AsyncToolboxTool(
-            tool_name,
-            manifest.tools[tool_name],
-            self.__url,
-            self.__session,
-            auth_token_getters,
-            bound_params,
-            strict,
+        core_tool = await self.__core_client.load_tool(
+            name=tool_name,
+            auth_token_getters=auth_token_getters,
+            bound_params=bound_params
         )
+        return AsyncToolboxTool(core_tool=core_tool)
 
     async def aload_toolset(
         self,
@@ -162,22 +152,16 @@ async def aload_toolset(
                 )
                 auth_token_getters = auth_tokens
 
-        url = f"{self.__url}/api/toolset/{toolset_name or ''}"
-        manifest: ManifestSchema = await _load_manifest(url, self.__session)
-        tools: list[AsyncToolboxTool] = []
-
-        for tool_name, tool_schema in manifest.tools.items():
-            tools.append(
-                AsyncToolboxTool(
-                    tool_name,
-                    tool_schema,
-                    self.__url,
-                    self.__session,
-                    auth_token_getters,
-                    bound_params,
-                    strict,
-                )
-            )
+        core_tools = await self.__core_client.load_toolset(
+            name=toolset_name,
+            auth_token_getters=auth_token_getters,
+            bound_params=bound_params,
+            strict=strict
+        )
+
+        tools = []
+        for core_tool in core_tools:
+            tools.append(AsyncToolboxTool(core_tool_instance=core_tool))
         return tools
 
     def load_tool(