简化图架构:暂时移除记忆相关节点,先让系统工作起来
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m40s
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m40s
This commit is contained in:
@@ -147,18 +147,14 @@ def adapt_old_node_for_new_state(old_node):
|
|||||||
# ========== 主图构建 ==========
|
# ========== 主图构建 ==========
|
||||||
def build_react_main_graph(llm=None, tools=None, mem0_client=None) -> StateGraph:
|
def build_react_main_graph(llm=None, tools=None, mem0_client=None) -> StateGraph:
|
||||||
"""
|
"""
|
||||||
构建整合后的完整主图
|
构建整合后的完整主图(简化版:先让系统工作起来)
|
||||||
|
|
||||||
完整流程:
|
完整流程:
|
||||||
START
|
START
|
||||||
↓
|
↓
|
||||||
retrieve_memory (从Mem0检索长期记忆) ← 来自旧图
|
init_state (初始化)
|
||||||
↓
|
↓
|
||||||
memory_trigger (记忆触发器) ← 来自旧图
|
react_reason (推理) ←───────────────────────┐
|
||||||
↓
|
|
||||||
init_state (初始化) ← 来自新图
|
|
||||||
↓
|
|
||||||
react_reason (推理) ←──────────────────────┐
|
|
||||||
↓ │
|
↓ │
|
||||||
条件路由 │
|
条件路由 │
|
||||||
├─ rag_retrieve →─────────────────────────┤
|
├─ rag_retrieve →─────────────────────────┤
|
||||||
@@ -166,21 +162,17 @@ def build_react_main_graph(llm=None, tools=None, mem0_client=None) -> StateGraph
|
|||||||
├─ dictionary_subgraph →──────────────────┤
|
├─ dictionary_subgraph →──────────────────┤
|
||||||
├─ news_analysis_subgraph →───────────────┤
|
├─ news_analysis_subgraph →───────────────┤
|
||||||
├─ web_search →───────────────────────────┤
|
├─ web_search →───────────────────────────┤
|
||||||
├─ handle_error → (重试或结束) ───────────┤
|
├─ handle_error → (重试或结束) ────────────┤
|
||||||
└─ llm_call (大模型调用) ←────────────────┘
|
└─ llm_call (大模型调用) ←────────────────┘
|
||||||
↓
|
↓
|
||||||
检查:需要总结?
|
finalize (发送完成事件)
|
||||||
├─ 是 → summarize (提交给Mem0存储) ← 来自旧图
|
|
||||||
└─ 否 → (跳过)
|
|
||||||
↓
|
|
||||||
finalize (发送完成事件) ← 来自旧图
|
|
||||||
↓
|
↓
|
||||||
END
|
END
|
||||||
"""
|
"""
|
||||||
# 创建图
|
# 创建图
|
||||||
graph = StateGraph(MainGraphState)
|
graph = StateGraph(MainGraphState)
|
||||||
|
|
||||||
# 设置全局 mem0_client
|
# 设置全局 mem0_client (暂时不用记忆功能)
|
||||||
if mem0_client:
|
if mem0_client:
|
||||||
set_global_mem0_client(mem0_client)
|
set_global_mem0_client(mem0_client)
|
||||||
|
|
||||||
@@ -189,20 +181,8 @@ def build_react_main_graph(llm=None, tools=None, mem0_client=None) -> StateGraph
|
|||||||
if llm is not None:
|
if llm is not None:
|
||||||
llm_node = create_llm_call_node(llm, tools or [])
|
llm_node = create_llm_call_node(llm, tools or [])
|
||||||
|
|
||||||
retrieve_memory_node = None
|
|
||||||
summarize_node = None
|
|
||||||
if mem0_client:
|
|
||||||
retrieve_memory_node = create_retrieve_memory_node(mem0_client)
|
|
||||||
summarize_node = create_summarize_node(mem0_client)
|
|
||||||
|
|
||||||
# ========== 添加节点 ==========
|
# ========== 添加节点 ==========
|
||||||
|
# 简化:先不用记忆检索相关节点
|
||||||
# 第一阶段:记忆检索(来自旧图)
|
|
||||||
if retrieve_memory_node:
|
|
||||||
graph.add_node("retrieve_memory", adapt_old_node_for_new_state(retrieve_memory_node))
|
|
||||||
graph.add_node("memory_trigger", memory_trigger_node)
|
|
||||||
|
|
||||||
# 第二阶段:React 循环推理(来自新图)
|
|
||||||
graph.add_node("init_state", init_state_node)
|
graph.add_node("init_state", init_state_node)
|
||||||
graph.add_node("react_reason", react_reason_node)
|
graph.add_node("react_reason", react_reason_node)
|
||||||
graph.add_node("rag_retrieve", rag_retrieve_node)
|
graph.add_node("rag_retrieve", rag_retrieve_node)
|
||||||
@@ -230,25 +210,15 @@ def build_react_main_graph(llm=None, tools=None, mem0_client=None) -> StateGraph
|
|||||||
wrap_subgraph_for_error_handling(news_analysis_graph.compile(), "news_analysis")
|
wrap_subgraph_for_error_handling(news_analysis_graph.compile(), "news_analysis")
|
||||||
)
|
)
|
||||||
|
|
||||||
# 第三阶段:完成处理(来自旧图)
|
# 完成节点
|
||||||
if summarize_node:
|
|
||||||
graph.add_node("summarize", adapt_old_node_for_new_state(summarize_node))
|
|
||||||
graph.add_node("finalize", finalize_node)
|
graph.add_node("finalize", finalize_node)
|
||||||
|
|
||||||
# ========== 添加边 ==========
|
# ========== 添加边 ==========
|
||||||
|
# 简化:直接从 START 到 init_state
|
||||||
# 第一阶段:记忆检索
|
graph.add_edge(START, "init_state")
|
||||||
if retrieve_memory_node:
|
|
||||||
graph.add_edge(START, "retrieve_memory")
|
|
||||||
graph.add_edge("retrieve_memory", "memory_trigger")
|
|
||||||
else:
|
|
||||||
graph.add_edge(START, "memory_trigger")
|
|
||||||
|
|
||||||
# 进入第二阶段
|
|
||||||
graph.add_edge("memory_trigger", "init_state")
|
|
||||||
graph.add_edge("init_state", "react_reason")
|
graph.add_edge("init_state", "react_reason")
|
||||||
|
|
||||||
# 第二阶段:React 循环推理
|
# 条件路由
|
||||||
graph.add_conditional_edges(
|
graph.add_conditional_edges(
|
||||||
"react_reason",
|
"react_reason",
|
||||||
route_by_reasoning,
|
route_by_reasoning,
|
||||||
@@ -263,7 +233,7 @@ def build_react_main_graph(llm=None, tools=None, mem0_client=None) -> StateGraph
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
# 循环边:检索/搜索/子图/错误处理后 → 回到推理
|
# 循环边
|
||||||
graph.add_edge("rag_retrieve", "react_reason")
|
graph.add_edge("rag_retrieve", "react_reason")
|
||||||
graph.add_edge("web_search", "react_reason")
|
graph.add_edge("web_search", "react_reason")
|
||||||
graph.add_edge("contact_subgraph", "react_reason")
|
graph.add_edge("contact_subgraph", "react_reason")
|
||||||
@@ -271,27 +241,14 @@ def build_react_main_graph(llm=None, tools=None, mem0_client=None) -> StateGraph
|
|||||||
graph.add_edge("news_analysis_subgraph", "react_reason")
|
graph.add_edge("news_analysis_subgraph", "react_reason")
|
||||||
graph.add_edge("handle_error", "react_reason")
|
graph.add_edge("handle_error", "react_reason")
|
||||||
|
|
||||||
# 第三阶段:llm_call 后进入完成处理
|
# llm_call 之后直接到 finalize
|
||||||
if llm_node is not None:
|
if llm_node is not None:
|
||||||
if summarize_node:
|
graph.add_edge("llm_call", "finalize")
|
||||||
# 检查是否需要总结
|
|
||||||
graph.add_conditional_edges(
|
|
||||||
"llm_call",
|
|
||||||
should_summarize,
|
|
||||||
{
|
|
||||||
"summarize": "summarize",
|
|
||||||
"finalize": "finalize"
|
|
||||||
}
|
|
||||||
)
|
|
||||||
graph.add_edge("summarize", "finalize")
|
|
||||||
else:
|
|
||||||
# 没有 summarize 节点,直接 finalize
|
|
||||||
graph.add_edge("llm_call", "finalize")
|
|
||||||
|
|
||||||
# 完成
|
# 完成
|
||||||
graph.add_edge("finalize", END)
|
graph.add_edge("finalize", END)
|
||||||
|
|
||||||
info("✅ [图构建] 整合后的完整主图构建完成")
|
info("✅ [图构建] 整合后的完整主图构建完成(简化版)")
|
||||||
|
|
||||||
return graph
|
return graph
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user