添加总结节点

Author: LifeCheckpoint

src/main_agent/utils/nodes/__init__.py

@@ -6,5 +6,6 @@
 
 __all__ = [
     "welcome", "finish_interrupt", "agent_execution", 
-    "should_tool", "no_tools_warning", "tool_result_transport", "ask_interrupt"
+    "should_tool", "no_tools_warning", "tool_result_transport", "ask_interrupt",
+    "summarization_node"
 ]

src/main_agent/utils/nodes/execution.py

@@ -2,13 +2,32 @@
 
 from langchain_openai import ChatOpenAI
 from langchain_core.runnables import RunnableConfig
+from langchain_core.messages.utils import count_tokens_approximately
+from langmem.short_term import SummarizationNode
+from pathlib import Path
 from aiopath import AsyncPath
 from typing import Any, Dict
 
 from src.main_agent.utils.state import MainAgentState
 from src.main_agent.utils.tools import tool_list
 
 
+summarization_node = SummarizationNode(
+    token_counter=count_tokens_approximately,
+    model=ChatOpenAI(
+        model="deepseek/deepseek-chat-v3-0324",
+        temperature=0.3,
+        base_url="https://openrouter.ai/api/v1",
+        api_key=(Path(__file__).parent.parent / "api_key").read_text(encoding="utf-8").strip(),
+        max_retries=3,
+        max_tokens=8192,
+    ), # TODO: 更灵活的模型配置
+    max_tokens=16384,
+    max_tokens_before_summary=65536,
+    max_summary_tokens=16384,
+)
+
+
 async def agent_execution(state: MainAgentState, config: RunnableConfig) -> Dict[str, Any]:
     """
     Agent 执行节点，处理用户输入并执行任务

src/main_agent/utils/nodes/routing.py

@@ -39,7 +39,7 @@ def tool_result_transport(state: MainAgentState, config: RunnableConfig) -> str:
             # 检查是否有工具调用
             match tool_call["name"]:
                 case "attempt_completion":
-                    return "finish_interrupt"
+                    return "summarization"
                 case "ask_question":
                     return "ask_interrupt"