diff --git a/libs/langchain_v1/langchain/agents/__init__.py b/libs/langchain_v1/langchain/agents/__init__.py index 3545a18c90a..9f859fd1f02 100644 --- a/libs/langchain_v1/langchain/agents/__init__.py +++ b/libs/langchain_v1/langchain/agents/__init__.py @@ -1,10 +1,8 @@ """langgraph.prebuilt exposes a higher-level API for creating and executing agents and tools.""" from langchain.agents.react_agent import AgentState, create_agent -from langchain.agents.tool_node import ToolNode __all__ = [ "AgentState", - "ToolNode", "create_agent", ] diff --git a/libs/langchain_v1/langchain/tools/__init__.py b/libs/langchain_v1/langchain/tools/__init__.py index f4b2e7c1495..e53f14171c0 100644 --- a/libs/langchain_v1/langchain/tools/__init__.py +++ b/libs/langchain_v1/langchain/tools/__init__.py @@ -8,10 +8,6 @@ from langchain_core.tools import ( tool, ) -__all__ = [ - "BaseTool", - "InjectedToolArg", - "InjectedToolCallId", - "ToolException", - "tool", -] +from langchain.tools.tool_node import ToolNode + +__all__ = ["BaseTool", "InjectedToolArg", "InjectedToolCallId", "ToolException", "ToolNode", "tool"] diff --git a/libs/langchain_v1/langchain/agents/tool_node.py b/libs/langchain_v1/langchain/tools/tool_node.py similarity index 100% rename from libs/langchain_v1/langchain/agents/tool_node.py rename to libs/langchain_v1/langchain/tools/tool_node.py diff --git a/libs/langchain_v1/pyproject.toml b/libs/langchain_v1/pyproject.toml index b5760a8c17f..ba7eeffd733 100644 --- a/libs/langchain_v1/pyproject.toml +++ b/libs/langchain_v1/pyproject.toml @@ -168,6 +168,16 @@ flake8-annotations.allow-star-arg-any = true "N801", # class name should use CapWords convention ] +"langchain/tools/tool_node.py" = [ + "ANN401", # we use Any right now, need to narrow + "A002", # input is shadowing builtin + "A001", # input is shadowing builtin + "B904", # use from for exceptions + "PLR2004", # magic values are fine for this case + "C901", # too complex + "TRY004", # type error exception +] + [tool.coverage.run] omit = ["tests/*"]