add tool_done in state

2025-12-29 22:40:57 +08:00
parent 18bb795dd3
commit 216b41e471
1 changed files with 1 additions and 0 deletions
--- 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