From 944947e889f6fa3973466de19e9b15cca85481bf Mon Sep 17 00:00:00 2001 From: "open-swe[bot]" Date: Fri, 22 Aug 2025 13:20:00 +0000 Subject: [PATCH] Apply patch [skip ci] --- libs/prebuilt/langgraph/prebuilt/tool_node.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/libs/prebuilt/langgraph/prebuilt/tool_node.py b/libs/prebuilt/langgraph/prebuilt/tool_node.py index 0f7621bec..e2021e507 100644 --- a/libs/prebuilt/langgraph/prebuilt/tool_node.py +++ b/libs/prebuilt/langgraph/prebuilt/tool_node.py @@ -404,8 +404,8 @@ class ToolNode(RunnableCallable): tool_ = _wrap_tool_with_reserved_keywords(tool_, reserved_args) self.tools_by_name[tool_.name] = tool_ - self.tool_to_state_args[tool_.name] = state_args - self.tool_to_store_arg[tool_.name] = store_arg + self.tool_to_state_args[tool_.name] = _get_state_args(tool_) + self.tool_to_store_arg[tool_.name] = _get_store_arg(tool_) self.tool_to_runtime_arg[tool_.name] = _get_runtime_arg(tool_) def _func( @@ -1394,5 +1394,6 @@ def _get_runtime_arg(tool: BaseTool) -> Optional[str]: +