重构:简化流式架构,将 ReAct 循环移入 agent 节点
All checks were successful
构建并部署 AI Agent 服务 / deploy (push) Successful in 5m41s

主要变更:
- 简化 agent_service:移除复杂双协程,只用 stream_mode=["updates"]
- stream_context:提供更清晰的 API (set_stream_queue/get_stream_queue)
- main_graph_builder:简化图结构,移除 tools 节点和条件边
- agent 节点:包含完整 ReAct 循环 + 流式 Tool Calling 拼接
- 前端:适配新的事件格式
- 添加测试文件:test_full_react_streaming.py, test_stream.py
This commit is contained in:
2026-05-07 02:56:35 +08:00
parent eb33203b5c
commit 5b41598d50
8 changed files with 450 additions and 367 deletions

View File

@@ -1,12 +1,12 @@
"""
极简 Agent 主图 - 自己的节点结构,更好控制流式
极简 Agent 主图 - 简化版本!
因为完整的 ReAct 循环已经在 agent.py 里了!
"""
from langgraph.graph import StateGraph, START, END
from langgraph.prebuilt import ToolNode
from ..state import AgentState
from ..nodes.memory_trigger import memory_trigger_node, set_mem0_client
from ..nodes.agent import create_agent_node
from backend.app.main_graph.state import AgentState
from backend.app.main_graph.nodes.memory_trigger import memory_trigger_node, set_mem0_client
from backend.app.main_graph.nodes.agent import create_agent_node
from backend.app.logger import info, warning
from backend.app.tools import ALL_TOOLS
@@ -17,7 +17,7 @@ def build_agent_graph(
max_steps: int = 10
):
"""
构建包含记忆节点的 Agent 图
构建简化的 Agent 图ReAct 循环在 agent 节点内)
Args:
chat_services: 模型服务字典
@@ -51,28 +51,16 @@ def build_agent_graph(
except Exception as e:
info(f"[Graph Builder] 记忆节点初始化失败: {e}")
# ========== 3. 核心节点 ==========
# ========== 3. Agent 节点(包含完整 ReAct 循环) ==========
llm_with_tools = primary_model.bind_tools(ALL_TOOLS)
agent_node_fn = create_agent_node(llm_with_tools, primary_model)
tool_node_fn = ToolNode(ALL_TOOLS)
# ========== 4. 条件边判断函数 ==========
def should_continue(state: AgentState):
"""判断是继续调用工具还是结束"""
messages = state.messages
last_message = messages[-1] if messages else None
if last_message and hasattr(last_message, 'tool_calls') and last_message.tool_calls:
return "tools"
return "finalize"
# ========== 5. 完成节点 ==========
# ========== 4. 完成节点 ==========
async def finalize_node_simple(state: AgentState):
info("[Finalize] 进入完成节点")
return {}
# ========== 6. 构建图 ==========
# ========== 5. 构建图 ==========
graph = StateGraph(AgentState)
graph.add_node("init_state", init_state_node)
@@ -80,10 +68,9 @@ def build_agent_graph(
graph.add_node("retrieve_memory", retrieve_memory_node)
graph.add_node("memory_trigger", memory_trigger_node)
graph.add_node("agent", agent_node_fn)
graph.add_node("tools", tool_node_fn)
graph.add_node("finalize", finalize_node_simple)
# ========== 7. 边的连接 ==========
# ========== 6. 边的连接 ==========
graph.add_edge(START, "init_state")
if retrieve_memory_node:
@@ -93,18 +80,8 @@ def build_agent_graph(
graph.add_edge("init_state", "memory_trigger")
graph.add_edge("memory_trigger", "agent")
graph.add_conditional_edges(
"agent",
should_continue,
{
"tools": "tools",
"finalize": "finalize"
}
)
graph.add_edge("tools", "agent")
graph.add_edge("agent", "finalize")
graph.add_edge("finalize", END)
info("✅ [Graph Builder] 简 Agent 图构建完成")
info("✅ [Graph Builder] 简 Agent 图构建完成ReAct 在节点内)")
return graph