SuperAPI AI 工具集

lakebase-memory-patterns

智能体与模型

查看项目

更新于 10-15 20:33

技能介绍

用途:为有状态智能体实现两层记忆架构。短期记忆用 CheckpointSaver 在会话内按 thread_id 保持对话连续性;长期记忆用 DatabricksStore 按 user_id 跨会话保存用户偏好与洞察。记忆存储在 Unity Catalog Delta 表中,可治理、可审计、可查询,支持 TTL 清理。

主要能力:提供 ShortTermMemory 与 LongTermMemory 类,通过 .setup() 自动建表;thread_id 按 custom_inputs > conversation_id > 新 UUID 的优先级解析;长期记忆支持嵌入检索(如 databricks-gte-large-en,维度 1024);记忆不可用时静默降级为无状态模式。

用法要点:初始化 ShortTermMemory(instance_name=...) 后调用 setup(),再用 get_checkpointer() 配合 LangGraph 编译;LongTermMemory 需指定 embedding_endpoint 与 embedding_dims,用 save_memory 与 search_memories 读写。返回结果时应在 custom_outputs 中回传 thread_id 供客户端续接。用户 ID 应从请求上下文动态提取,不可硬编码。

依赖与限制:需 databricks-sdk >= 0.61.0(推荐 0.81.0+)、psycopg >= 3.0、SQLAlchemy 2.x 配 postgresql+psycopg 驱动。生产环境 OAuth 令牌 1 小时过期,需实现约每 50 分钟的后台刷新。容量按 CU_1/CU_2/CU_4/CU_8 计。资料未提供具体价格与平台支持范围。