bug fix

2025-12-12 14:12:42 +08:00
parent 3dccddede2
commit a478f7d601
1 changed files with 1 additions and 1 deletions
--- 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(