diff --git a/libs/prebuilt/langgraph/prebuilt/tool_node.py b/libs/prebuilt/langgraph/prebuilt/tool_node.py index 8ebb77abf..3634031a4 100644 --- a/libs/prebuilt/langgraph/prebuilt/tool_node.py +++ b/libs/prebuilt/langgraph/prebuilt/tool_node.py @@ -584,7 +584,7 @@ class ToolNode(RunnableCallable): raise ValueError("No AIMessage found in input") tool_calls = [ - self.inject_tool_args(call, input, store) + self.inject_tool_args(call, input, store, config) for call in latest_ai_message.tool_calls ] return tool_calls, input_type @@ -1268,3 +1268,4 @@ def _get_runtime_arg(tool: BaseTool) -> Optional[str]: return None +