github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-15 23:57:21 +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,
)
from langchain_core.tools.base import (
ArgsSchema,
InjectedToolArg,
InjectedToolCallId,
SchemaAnnotationError,
@ -2712,11 +2711,7 @@ def test_args_schema_without_injected_arguments() -> None:
"type": "tool_call",
}
result = broken_tool.invoke(tool_call)
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
# The full input schema should include tool_call_id for validation
schema = _schema(broken_tool.get_input_schema())
assert schema == {
"description": "Tool with injected arguments.",
@ -2736,10 +2731,16 @@ def test_args_schema_without_injected_arguments() -> None:
"text",
"tool_call_id",
],
"title": "broken_tool",
"title": "CustomSchema",
"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)
assert tool_call_schema == {
"description": "Tool that should work with automatic injection",