diff --git a/libs/langgraph/langgraph/prebuilt/tool_node.py b/libs/langgraph/langgraph/prebuilt/tool_node.py index d3d0751e2..e2ac50b8e 100644 --- a/libs/langgraph/langgraph/prebuilt/tool_node.py +++ b/libs/langgraph/langgraph/prebuilt/tool_node.py @@ -297,7 +297,7 @@ class ToolNode(RunnableCallable): try: input = {**call, **{"type": "tool_call"}} - response = self.tools_by_name[call["name"]].invoke(input) + response = self.tools_by_name[call["name"]].invoke(input, config) # GraphInterrupt is a special exception that will always be raised. # It can be triggered in the following scenarios: @@ -352,7 +352,7 @@ class ToolNode(RunnableCallable): try: input = {**call, **{"type": "tool_call"}} - response = await self.tools_by_name[call["name"]].ainvoke(input) + response = await self.tools_by_name[call["name"]].ainvoke(input, config) # GraphInterrupt is a special exception that will always be raised. # It can be triggered in the following scenarios: