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 15:03:10 +00:00
parent 2d097ee9d1
commit 4d098c4cb5
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/prebuilt/langgraph/prebuilt/tool_node.py
+12 -7
View File
@@ -1220,7 +1220,8 @@ def _wrap_tool_with_reserved_keywords(
return tool return tool
# Create a wrapper tool that filters the schema # Create a wrapper tool that filters the schema
class FilteredTool(tool.__class__): # Use type: ignore to suppress mypy error for dynamic class creation
class FilteredTool(tool.__class__): # type: ignore[name-defined]
"""Tool wrapper that excludes reserved keywords from schema.""" """Tool wrapper that excludes reserved keywords from schema."""
def get_input_schema( def get_input_schema(
@@ -1239,11 +1240,12 @@ def _wrap_tool_with_reserved_keywords(
# Get fields to keep (exclude reserved keywords) # Get fields to keep (exclude reserved keywords)
if hasattr(original_schema, "model_fields"): if hasattr(original_schema, "model_fields"):
# Pydantic v2 # Pydantic v2
fields_to_keep = { fields_to_keep: dict[str, Any] = {}
name: (field.annotation, field) for name, field in original_schema.model_fields.items():
for name, field in original_schema.model_fields.items() if name not in reserved_args:
if name not in reserved_args # For create_model, we need (type, default) or just type
} field_type = field.annotation if hasattr(field, 'annotation') else Any
fields_to_keep[name] = field_type
else: else:
# Pydantic v1 fallback # Pydantic v1 fallback
fields_to_keep = {} fields_to_keep = {}
@@ -1251,7 +1253,9 @@ def _wrap_tool_with_reserved_keywords(
# Create filtered schema # Create filtered schema
try: try:
filtered_schema = create_model( filtered_schema = create_model(
f"{original_schema.__name__}Filtered", **fields_to_keep f"{original_schema.__name__}Filtered",
__base__=BaseModel,
**fields_to_keep
) )
return filtered_schema return filtered_schema
except Exception: except Exception:
@@ -1393,3 +1397,4 @@ def _get_runtime_arg(tool: BaseTool) -> Optional[str]:
""" """
reserved_args = _get_reserved_keyword_args(tool) reserved_args = _get_reserved_keyword_args(tool)
return "runtime" if "runtime" in reserved_args else None return "runtime" if "runtime" in reserved_args else None