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 模块 - 添加测试脚本验证新架构 - 所有测试通过 ✓
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
"""
|
|
记忆检索节点模块
|
|
负责从 Mem0 检索相关长期记忆
|
|
"""
|
|
|
|
from typing import Any, Dict
|
|
|
|
# 本地模块
|
|
from ...main_graph.state import AgentState
|
|
from ...memory.mem0_client import Mem0Client
|
|
from ...utils.logging import log_state_change
|
|
from backend.app.logger import debug
|
|
|
|
|
|
def create_retrieve_memory_node(mem0_client: Mem0Client):
|
|
"""
|
|
工厂函数：创建记忆检索节点
|
|
|
|
Args:
|
|
mem0_client: Mem0 客户端实例
|
|
|
|
Returns:
|
|
异步节点函数
|
|
"""
|
|
|
|
from langchain_core.runnables.config import RunnableConfig
|
|
|
|
async def retrieve_memory(state: AgentState, config: RunnableConfig) -> Dict[str, Any]:
|
|
"""
|
|
记忆检索节点 - 使用 Mem0
|
|
|
|
Args:
|
|
state: 当前对话状态
|
|
config: 运行时配置
|
|
|
|
Returns:
|
|
包含 memory_context 的状态更新
|
|
"""
|
|
log_state_change("retrieve_memory", state, "进入")
|
|
|
|
# 从 metadata 中获取 user_id
|
|
user_id = config.get("metadata", {}).get("user_id", "default_user")
|
|
|
|
# 获取最后一条消息
|
|
messages = state.messages
|
|
last_msg = messages[-1] if messages else None
|
|
query = ""
|
|
if last_msg:
|
|
if isinstance(last_msg, dict):
|
|
query = str(last_msg.get("content", ""))
|
|
else:
|
|
query = str(last_msg.content)
|
|
|
|
memory_text_parts = []
|
|
|
|
# 确保 Mem0 已初始化（懒加载）
|
|
if not mem0_client._initialized:
|
|
await mem0_client.initialize()
|
|
|
|
if mem0_client.mem0:
|
|
try:
|
|
# 异步调用 Mem0 语义检索
|
|
facts = await mem0_client.search_memories(query, user_id=user_id, limit=5)
|
|
|
|
if facts:
|
|
memory_text_parts.append(f"【相关长期记忆】\n" + "\n".join(f"- {f}" for f in facts))
|
|
else:
|
|
debug("🔍 [记忆检索] 未找到相关记忆")
|
|
except Exception as e:
|
|
from backend.app.logger import warning
|
|
warning(f"⚠️ Mem0 检索失败: {e}")
|
|
else:
|
|
from backend.app.logger import warning
|
|
warning("⚠️ Mem0 未初始化，跳过记忆检索")
|
|
|
|
memory_context = "\n\n".join(memory_text_parts) if memory_text_parts else "暂无用户信息"
|
|
result = {"memory_context": memory_context}
|
|
log_state_change("retrieve_memory", state, "离开")
|
|
return result
|
|
|
|
return retrieve_memory |