WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-09-08 02:37:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

Apply patch [skip ci]

Browse Source
This commit is contained in:
open-swe[bot]
2025-08-22 14:58:39 +00:00
parent 9b67fcfc87
commit 2f14631c84
1 changed files with 4 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/prebuilt/langgraph/prebuilt/tool_node.py
+4 -25
View File
@@ -399,31 +399,9 @@ class ToolNode(RunnableCallable):
)
# Check for reserved keywords and wrap the tool if needed
# Also wrap tools with annotation-based injection to ensure their parameters are excluded from schemas
if reserved_args or has_injected_state or has_injected_store:
# Create a combined dict of parameters to exclude from schema
params_to_exclude = dict(reserved_args) # Start with reserved keywords
# Add annotation-based parameters
if has_injected_state:
# Find the parameter name that has InjectedState annotation
for name, type_ in get_all_basemodel_annotations(full_schema).items():
type_args = get_args(type_)
for type_arg in type_args:
if _is_injection(type_arg, InjectedState):
params_to_exclude[name] = 'state'
break
if has_injected_store:
# Find the parameter name that has InjectedStore annotation
for name, type_ in get_all_basemodel_annotations(full_schema).items():
type_args = get_args(type_)
for type_arg in type_args:
if _is_injection(type_arg, InjectedStore):
params_to_exclude[name] = 'store'
break
tool_ = _wrap_tool_with_reserved_keywords(tool_, params_to_exclude)
# Only wrap tools with reserved keywords for now to avoid breaking existing functionality
if reserved_args:
tool_ = _wrap_tool_with_reserved_keywords(tool_, reserved_args)
self.tools_by_name[tool_.name] = tool_
self.tool_to_state_args[tool_.name] = _get_state_args(tool_)
@@ -1409,3 +1387,4 @@ def _get_runtime_arg(tool: BaseTool) -> Optional[str]: