x

libs/langchain_v1/langchain/agents/middleware_agent.py

@@ -33,7 +33,7 @@ from langchain.agents.structured_output import (
     StructuredOutputValidationError,
     ToolStrategy,
 )
-from langchain.agents.tool_node import ToolNode
+from langchain.tools import ToolNode
 from langchain.chat_models import init_chat_model
 
 STRUCTURED_OUTPUT_ERROR_TEMPLATE = "Error: {error}\n Please fix your mistakes."

libs/langchain_v1/langchain/agents/react_agent.py

@@ -54,8 +54,8 @@ from langchain.agents.structured_output import (
     StructuredOutputValidationError,
     ToolStrategy,
 )
-from langchain.agents.tool_node import ToolNode
 from langchain.chat_models import init_chat_model
+from langchain.tools import ToolNode
 
 if TYPE_CHECKING:
     from langchain_core.tools import BaseTool
@@ -66,7 +66,7 @@ if TYPE_CHECKING:
     from langchain.agents._internal._typing import (
         SyncOrAsync,
     )
-    from langchain.agents.types import AgentMiddleware
+    from langchain.agents.middleware import AgentMiddleware
 
 StructuredResponseT = TypeVar("StructuredResponseT", default=None)
 

libs/langchain_v1/tests/unit_tests/tools/test_imports.py

@@ -6,6 +6,7 @@ EXPECTED_ALL = {
     "InjectedToolCallId",
     "ToolException",
     "tool",
+    "ToolNode",
 }