chore: Fix types

Author: anubhav756

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

@@ -43,7 +43,7 @@ def __init__(
         super().__init__(
             name=core_tool.__name__,
             description=core_tool.__doc__,
-            args_schema=core_tool._ToolboxTool__pydantic_model,
+            args_schema=core_tool._pydantic_model,
         )
         self.__core_tool = core_tool
 

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

@@ -45,10 +45,10 @@ def __init__(
         )
         self.__core_sync_tool = core_sync_tool
 
-    def _run(self, **kwargs: Any) -> dict[str, Any]:
+    def _run(self, **kwargs: Any) -> str:
         return self.__core_sync_tool(**kwargs)
 
-    async def _arun(self, **kwargs: Any) -> dict[str, Any]:
+    async def _arun(self, **kwargs: Any) -> str:
         coro = self.__core_sync_tool._async_tool(**kwargs)
 
         # If a loop has not been provided, attempt to run in current thread.