From a478f7d6011bd30b32550190bb8adec3131cb97b Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Fri, 12 Dec 2025 14:12:42 +0800
Subject: [PATCH] bug fix

---
 lang_agent/graphs/tool_nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lang_agent/graphs/tool_nodes.py b/lang_agent/graphs/tool_nodes.py
index 7b8d33d..99df0cb 100644
--- a/lang_agent/graphs/tool_nodes.py
+++ b/lang_agent/graphs/tool_nodes.py
@@ -166,7 +166,7 @@ class ChattyToolNode(ToolNodeBase):
 
     def _handoff_node(self, state:ChattyToolState):
         # NOTE: This exists just to stream the thing correctly
-        tool_msgs = state.get("tool_messages")["messages"]
+        tool_msgs = state.get("tool_messages")
         reit_msg = f"{self.tool_key}\n"+tool_msgs[-1].content
         inp = [
                 SystemMessage(