diff --git a/libs/langchain_v1/tests/unit_tests/agents/middleware/test_anthropic_tools.py b/libs/langchain_v1/tests/unit_tests/agents/middleware/test_anthropic_tools.py index a7cf985c470..7e871c69aad 100644 --- a/libs/langchain_v1/tests/unit_tests/agents/middleware/test_anthropic_tools.py +++ b/libs/langchain_v1/tests/unit_tests/agents/middleware/test_anthropic_tools.py @@ -6,7 +6,7 @@ from langchain.agents.middleware.anthropic_tools import ( StateClaudeMemoryMiddleware, StateClaudeTextEditorMiddleware, ) -from langchain.agents.middleware.file_utils import validate_path +from langchain.agents._internal.file_utils import validate_path from langchain_core.messages import ToolMessage from langgraph.types import Command diff --git a/libs/langchain_v1/tests/unit_tests/agents/middleware/test_subagent_middleware.py b/libs/langchain_v1/tests/unit_tests/agents/middleware/test_subagent_middleware.py index fbc6bee92d7..d7c721a2c9a 100644 --- a/libs/langchain_v1/tests/unit_tests/agents/middleware/test_subagent_middleware.py +++ b/libs/langchain_v1/tests/unit_tests/agents/middleware/test_subagent_middleware.py @@ -1,6 +1,9 @@ from langchain.agents.middleware.subagents import TASK_TOOL_DESCRIPTION, SubAgentMiddleware +import pytest + +@pytest.mark.requires("langchain_openai") class TestSubagentMiddleware: """Test the SubagentMiddleware class."""