fix: Use correct core client interfaces in langchain client

Author: anubhav756

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

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import asyncio
+from asyncio import wrap_future
 from typing import Any, Callable, Optional, Union
 from warnings import warn
 
@@ -85,30 +85,14 @@ async def aload_tool(
                 )
                 auth_token_getters = auth_headers
 
-        coro = self.__core_client._async_client.load_tool(
-            name=tool_name,
-            auth_token_getters=auth_token_getters,
-            bound_params=bound_params,
-        )
-
-        if not self.__core_client._loop:
-            # If a loop has not been provided, attempt to run in current thread.
-            core_tool = await coro
-        else:
-            # Otherwise, run in the background thread.
-            core_tool = await asyncio.wrap_future(
-                asyncio.run_coroutine_threadsafe(coro, self.__core_client._loop)
+        core_tool = await wrap_future(
+            self.__core_client._load_tool_future(
+                name=tool_name,
+                auth_token_getters=auth_token_getters,
+                bound_params=bound_params,
             )
-
-        if not self.__core_client._loop or not self.__core_client._thread:
-            raise ValueError("Background loop or thread cannot be None.")
-
-        core_sync_tool = ToolboxSyncTool(
-            core_tool,
-            self.__core_client._loop,
-            self.__core_client._thread,
         )
-        return ToolboxTool(core_tool=core_sync_tool)
+        return ToolboxTool(core_tool=core_tool)
 
     async def aload_toolset(
         self,
@@ -167,36 +151,18 @@ async def aload_toolset(
                 )
                 auth_token_getters = auth_headers
 
-        coro = self.__core_client._async_client.load_toolset(
-            name=toolset_name,
-            auth_token_getters=auth_token_getters,
-            bound_params=bound_params,
-            strict=strict,
-        )
-
-        if not self.__core_client._loop:
-            # If a loop has not been provided, attempt to run in current thread.
-            core_tools = await coro
-        else:
-            # Otherwise, run in the background thread.
-            core_tools = await asyncio.wrap_future(
-                asyncio.run_coroutine_threadsafe(coro, self.__core_client._loop)
+        core_tools = await wrap_future(
+            self.__core_client._load_toolset_future(
+                name=toolset_name,
+                auth_token_getters=auth_token_getters,
+                bound_params=bound_params,
+                strict=strict,
             )
+        )
 
-        if not self.__core_client._loop or not self.__core_client._thread:
-            raise ValueError("Background loop or thread cannot be None.")
-
-        core_sync_tools = [
-            ToolboxSyncTool(
-                core_tool,
-                self.__core_client._loop,
-                self.__core_client._thread,
-            )
-            for core_tool in core_tools
-        ]
         tools = []
-        for core_sync_tool in core_sync_tools:
-            tools.append(ToolboxTool(core_tool=core_sync_tool))
+        for core_tool in core_tools:
+            tools.append(ToolboxTool(core_tool=core_tool))
         return tools
 
     def load_tool(