x

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",
 ]

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"]

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/*"]