From 216b41e4719d48aa4f351d18fb26cbcb24e58428 Mon Sep 17 00:00:00 2001 From: goulustis Date: Mon, 29 Dec 2025 22:40:57 +0800 Subject: [PATCH] add tool_done in state --- lang_agent/graphs/graph_states.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lang_agent/graphs/graph_states.py b/lang_agent/graphs/graph_states.py index 4ff3963..47f92e2 100644 --- a/lang_agent/graphs/graph_states.py +++ b/lang_agent/graphs/graph_states.py @@ -14,3 +14,4 @@ class ChattyToolState(TypedDict): Dict[str, Dict[str, str|int]]] tool_messages: List[SystemMessage | HumanMessage | AIMessage] chatty_messages: List[SystemMessage | HumanMessage | AIMessage] + tool_done: bool # Flag to signal when tool execution is complete