refactor: Type narrowing for ToolNode.name (#1678)

Avoid type incompatibility when using `add_node` to add a `ToolNode`.
langchain-ai · Sep 11, 2024 · bf6ad19 · bf6ad19
1 parent 4c59a5e
commit bf6ad19
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libs/langgraph/langgraph/prebuilt/tool_node.py b/libs/langgraph/langgraph/prebuilt/tool_node.py
@@ -79,6 +79,8 @@ def tool_node(state: dict):
         - The `AIMessage` MUST have `tool_calls` populated.
     """
 
+    name: str = "ToolNode"
+
     def __init__(
         self,
         tools: Sequence[Union[BaseTool, Callable]],