github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-17 00:17:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Eugene Yurtsev 2025-06-20 13:16:40 -04:00
parent b9ebae37a4
commit 97db95b426
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
libs/core/tests/unit_tests/test_tools.py
View File

@ -52,7 +52,6 @@ from langchain_core.tools import (
tool, tool,
) )
from langchain_core.tools.base import ( from langchain_core.tools.base import (
ArgsSchema,
InjectedToolArg, InjectedToolArg,
InjectedToolCallId, InjectedToolCallId,
SchemaAnnotationError, SchemaAnnotationError,
@ -2712,11 +2711,7 @@ def test_args_schema_without_injected_arguments() -> None:
"type": "tool_call", "type": "tool_call",
} }
result = broken_tool.invoke(tool_call) # The full input schema should include tool_call_id for validation
assert isinstance(result, ToolMessage)
assert result.content == "Processed 'test data' (call_id: test_123)"
# But the full input schema should include tool_call_id for validation
schema = _schema(broken_tool.get_input_schema()) schema = _schema(broken_tool.get_input_schema())
assert schema == { assert schema == {
"description": "Tool with injected arguments.", "description": "Tool with injected arguments.",
@ -2736,10 +2731,16 @@ def test_args_schema_without_injected_arguments() -> None:
"text", "text",
"tool_call_id", "tool_call_id",
], ],
"title": "broken_tool", "title": "CustomSchema",
"type": "object", "type": "object",
} }
result = broken_tool.invoke(tool_call)
assert isinstance(result, ToolMessage)
assert result.content == "Processed 'test data' (call_id: test_123)"
# The tool schema does not include injectable arguments that are hidden from the
# LLM.
tool_call_schema = _schema(broken_tool.tool_call_schema) tool_call_schema = _schema(broken_tool.tool_call_schema)
assert tool_call_schema == { assert tool_call_schema == {
"description": "Tool that should work with automatic injection", "description": "Tool that should work with automatic injection",