添加长期存储，流式检查

app/graph_builder.py

@@ -16,6 +16,7 @@ from app.nodes import (
     should_continue
 )
 from app.memory import Mem0Client
+from app.nodes.finalize import finalize_node
 
 
 class GraphBuilder:
@@ -57,6 +58,7 @@ class GraphBuilder:
         builder.add_node("llm_call", llm_call_node)
         builder.add_node("tool_node", tool_call_node)
         builder.add_node("summarize", summarize_node)
+        builder.add_node("finalize", finalize_node)
 
         # 添加边
         builder.add_edge(START, "retrieve_memory")
@@ -67,10 +69,11 @@ class GraphBuilder:
             {
                 "tool_node": "tool_node",
                 "summarize": "summarize",
-                'END': END
+                "finalize": "finalize"
             }
         )
         builder.add_edge("tool_node", "llm_call")
-        builder.add_edge("summarize", END)
+        builder.add_edge("summarize", "finalize")
+        builder.add_edge("finalize", END)
         
-        return builder
+        return builder