root
22fdb625a4
Some checks failed
构建并部署 AI Agent 服务 / deploy (push) Failing after 6m36s
主要变更： - 迁移到极简 LangGraph 标准架构（START → init_state → 记忆 → Agent ⇄ Tools → finalize → END） - 添加 Baosi API 支持，配置 ops4.7 模型 - 保留本地模型作为默认首选，Baosi 作为备选 - 新架构使用 LangGraph 原生 ToolNode 和 bind_tools - 移除旧的混合路由、JSON 解析等复杂逻辑 - 把旧代码移到 deprecated/ 目录 - 添加新的 Agent 节点和 Tools 模块 - 添加测试脚本验证新架构 - 所有测试通过 ✓
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""
|
|
完成事件节点模块（新架构版本）
|
|
负责发送完成事件
|
|
"""
|
|
|
|
from typing import Any, Dict
|
|
from datetime import datetime
|
|
|
|
# 本地模块
|
|
from .state import AgentState
|
|
from backend.app.logger import info, warning
|
|
|
|
from langchain_core.runnables.config import RunnableConfig
|
|
|
|
|
|
async def finalize_node(state: AgentState, config: RunnableConfig) -> Dict[str, Any]:
|
|
"""
|
|
完成事件节点（新架构版本）
|
|
|
|
Args:
|
|
state: 当前对话状态
|
|
config: 运行时配置
|
|
|
|
Returns:
|
|
空（不修改状态）
|
|
"""
|
|
info("[Finalize] 进入完成节点")
|
|
|
|
try:
|
|
# 获取流式写入器并发送完成事件
|
|
from backend.app.main_graph.config import get_stream_writer
|
|
writer = get_stream_writer()
|
|
|
|
# 提取最后的回复
|
|
final_reply = ""
|
|
if state.messages:
|
|
last_msg = state.messages[-1]
|
|
final_reply = last_msg.content if hasattr(last_msg, 'content') else str(last_msg)
|
|
|
|
# 只在 writer 存在且不是 noop 时才发送
|
|
if writer and hasattr(writer, '__call__'):
|
|
try:
|
|
writer({
|
|
"type": "custom",
|
|
"data": {
|
|
"type": "done",
|
|
"token_usage": state.last_token_usage,
|
|
"elapsed_time": state.last_elapsed_time,
|
|
"final_result": final_reply
|
|
}
|
|
})
|
|
info("🏁 [完成事件] 已发送完成事件")
|
|
except Exception as e:
|
|
warning(f"⚠️ [完成事件] 发送完成事件失败 (非致命): {e}")
|
|
except Exception as e:
|
|
warning(f"⚠️ [完成事件] 处理失败 (非致命): {e}")
|
|
|
|
info("[Finalize] 离开完成节点")
|
|
return {}
|