From bf6ad199635a34240bf634cdddbe522c64d1cb3b Mon Sep 17 00:00:00 2001
From: gbaian10 <34255899+gbaian10@users.noreply.github.com>
Date: Thu, 12 Sep 2024 01:11:39 +0800
Subject: [PATCH] refactor: Type narrowing for ToolNode.name (#1678)

Avoid type incompatibility when using `add_node` to add a `ToolNode`.
---
 libs/langgraph/langgraph/prebuilt/tool_node.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/libs/langgraph/langgraph/prebuilt/tool_node.py b/libs/langgraph/langgraph/prebuilt/tool_node.py
index c5636145b..a596e829f 100644
--- a/libs/langgraph/langgraph/prebuilt/tool_node.py
+++ b/libs/langgraph/langgraph/prebuilt/tool_node.py
@@ -79,6 +79,8 @@ class ToolNode(RunnableCallable):
         - The `AIMessage` MUST have `tool_calls` populated.
     """
 
+    name: str = "ToolNode"
+
     def __init__(
         self,
         tools: Sequence[Union[BaseTool, Callable]],