From bcbf2fff5fa8dff02ca43f387b98b2634b3aac12 Mon Sep 17 00:00:00 2001 From: Nick Huang Date: Fri, 10 Oct 2025 16:05:00 -0400 Subject: [PATCH] Fix import in test --- .../tests/unit_tests/agents/middleware/test_anthropic_tools.py | 2 +- .../unit_tests/agents/middleware/test_subagent_middleware.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) 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."""