Skip to content

Commit 6906393

Browse files
committed
fix: Use correct interface from core tool
1 parent 551b113 commit 6906393

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
from langchain_core.tools import BaseTool
1818
from toolbox_core.tool import ToolboxTool as ToolboxCoreTool
19+
from toolbox_core.utils import params_to_pydantic_model
1920

2021

2122
# This class is an internal implementation detail and is not exposed to the
@@ -43,7 +44,7 @@ def __init__(
4344
super().__init__(
4445
name=core_tool.__name__,
4546
description=core_tool.__doc__,
46-
args_schema=core_tool._pydantic_model,
47+
args_schema=params_to_pydantic_model(core_tool._name, core_tool._params),
4748
)
4849
self.__core_tool = core_tool
4950

0 commit comments

Comments
 (0)