网易首页 > 网易号 > 正文 申请入驻

构建有记忆的 AI Agent:SQLite 存储 + 向量检索完整方案示例

0
分享至

现在的 Agent 系统有个很明显的问题 —— 会话一结束,什么都忘了。

这不是个技术缺陷,但是却限制了整个系统的能力边界。Agent 可以做推理、规划、执行复杂任务,但就是记不住之前发生过什么。每次对话都像是第一次见面,这种状态下很难说它真正"理解"了什么。

记忆能力是把 LLM 从简单的问答工具变成真正协作伙伴的关键。一个只能"回答当前问题",另一个能"基于历史经验做决策",这就是增加了记忆能力后的改进。

这篇文章会讲怎么给 Agent 加上记忆、反思和目标跟踪能力。技术栈很简单:

  • SQLite 做结构化存储
  • 向量数据库(Pinecone、FAISS、Chroma 都行)处理语义检索
  • LLM 层负责反思和总结

这套架构可以直接集成到现有框架里,不管你用 LangChain、CrewAI 还是自己写的框架。

记忆为什么这么重要

Agent 的自主性需要记忆支撑,先说跨会话连续性,一个数据质量监控 Agent 如果能记住哪些数据集经常出问题,就能提前预警而不是每次都从头排查。

而且通过记忆还可以增加反思的能力,Agent 可以自己评估"这次任务完成得怎么样"、"推理过程有没有问题",这种自我评估不需要复杂的奖励函数,用自然语言就能实现强化学习的效果。

长期目标跟踪也是一个很大的需求,数据整理、研究辅助这类工作往往跨越多次交互,需要 Agent 记住目标、追踪进度、理解任务之间的依赖关系。

没有记忆的 Agent 永远困在当下,无法积累经验也无法改进。

系统架构

整体设计分两个记忆层面:情景记忆记录发生的事情,语义记忆提炼学到的经验。

这个架构的核心是让 Agent 既能查询"我之前做过什么",也能理解"类似的情况该怎么处理"。

数据库设计

为了演示,我们使用SQLite ,因为它轻量、本地化、跨平台支持好。对于大多数场景够用了,除非你的 Agent 需要处理海量并发。

数据库如下:

CREATE TABLE IF NOT EXISTS memory_events (
id INTEGER PRIMARY KEY AUTOINCREMENT,
agent_name TEXT,
timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
input TEXT,
output TEXT,
summary TEXT,
embedding BLOB
);
CREATE TABLE IF NOT EXISTS goals (
id INTEGER PRIMARY KEY AUTOINCREMENT,
agent_name TEXT,
goal TEXT,
status TEXT DEFAULT 'in_progress',
last_updated DATETIME DEFAULT CURRENT_TIMESTAMP
);

memory_events 表存交互历史和摘要,goals 表追踪目标状态。每个任务执行完就自动记录,这些数据后面会用来做反思分析。

记录情景记忆

每次任务执行完,把关键信息存下来:

def log_memory_event(agent_name, input_text, output_text, summary, embedding):
conn = sqlite3.connect("memory.db")
cur = conn.cursor()
cur.execute("""
INSERT INTO memory_events (agent_name, input, output, summary, embedding)
VALUES (?, ?, ?, ?, ?)
""", (agent_name, input_text, output_text, summary, embedding))
conn.commit()
conn.close()

summary 可以让 LLM 生成:

summary_prompt = f"Summarize this agent interaction:\n\nInput: {input_text}\nOutput: {output_text}\n"
summary = llm.complete(summary_prompt)

然后把摘要转成向量存起来:

embedding = embedding_model.embed(summary)

这样做的好处是,检索时不依赖关键词匹配,而是基于语义相似度。Agent 记住的不是原始文本,是事情的"意思"。

语义检索实现

新查询来的时候,先找出相关的历史记忆:

def recall_related_memories(query, top_k=3):
query_embedding = embedding_model.embed(query)
results = pinecone_index.query(vector=query_embedding, top_k=top_k)
return [r['metadata']['summary'] for r in results]

检索出来的摘要直接注入到 prompt 里:

"以下是一些相关的历史经验,处理当前问题时可以参考……"

这个过程模拟了人类决策前回忆类似经历的思维方式。向量检索能找到语义上相关但表述完全不同的内容,比传统的全文搜索要智能得多。

反思机制

有了记忆以后,还可以让 Agent 学会从记忆中学习。反思循环把被动的存储变成主动的能力提升。每隔几次交互触发一次:

reflection_prompt = f"""
You are reviewing your recent actions. Based on the following summaries, what patterns,
mistakes, or improvements do you notice?
{recent_summaries}
Provide 3 takeaways and 1 improvement plan for your future tasks.
"""
reflection = llm.complete(reflection_prompt)

可以把这个反思结果作为元记忆存起来,也可以生成 embedding 用于后续检索。

更进一步还可以加自我批评机制。每个主要任务完成后,Agent 评估是否达成目标。没达成就写个修正笔记,下次遇到类似情况知道该怎么改进。

这种方式实现了推理层面的强化学习,不需要梯度更新,纯靠自然语言就能调整行为模式。

目标管理

Agent 需要目标感。可以动态定义、更新、评估目标:

def update_goal(agent_name, goal, status):
conn = sqlite3.connect("memory.db")
cur = conn.cursor()
cur.execute("""
UPDATE goals SET status = ?, last_updated = CURRENT_TIMESTAMP
WHERE agent_name = ? AND goal = ?
""", (status, agent_name, goal))
conn.commit()
conn.close()

可以专门跑一个目标跟踪 Agent,定期检查未完成的目标然后提醒相关 Agent:

"目标:提升数据时效性
当前进度:70%
建议行动:检查上周的数据延迟情况"

这样整个系统就有了持续性。Agent 不再是处理单次请求的工具,而是在追求长期目标的过程中持续运作。

完整流程

把这些组件串起来流程就是这样的:

1、用户请求进来,Agent 执行任务,同时把交互记录写入 SQLite 和向量库。

2、处理新请求之前,先做语义检索调出相关记忆,把这些信息加到上下文里。

3、每隔 N 次交互,Agent 总结最近的行为表现,写反思笔记存档。

4、目标独立于单次会话存在,可以跨越多天甚至多周追踪进度。

5、这样构建出来的 Agent 更接近一个能学习、能记忆、能进化的系统。

一些实践经验

存摘要比存完整对话有效得多,既节省空间又便于检索。

定期清理数据很重要。比如说设置个定时任务,合并相似的记忆或者删掉不再有用的旧记录。

语义压缩是个好技巧 —— 把多个相关事件总结成一条元记忆,减少信息冗余的同时保留关键模式。

提示词设计也要引导元认知。给 Agent 一个明确角色比如"反思分析师",会让自我评估的质量明显提升。

如果想直观看到效果,可以搭个 Streamlit 或 React 界面,实时展示记忆聚类、目标进度、反思内容这些信息。可视化对调试和优化很有帮助。

最后

加上记忆、反思、目标追踪,Agent 就从一次性的工具变成了学习型伙伴。他们的区别在于一个只会执行任务,另一个能理解意图的演变并主动适应。

Agent AI 这个方向发展下去,记忆系统会成为基础设施。数字助手需要记住昨天发生的事,反思今天的表现,规划明天的行动。

最简单的记忆功能实现起来并不复杂,SQLite 加个向量库,写几个精心设计的 prompt,就能让 Agent 开始进化了。

https://avoid.overfit.cn/post/44c8d547475340d59aa4480f634ea67f

作者:Kyle knudson

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
弄巧成拙了!网传义乌一饭店:客人白酒啤酒饮料连菜都自带,14个人只花430元,包厢还弄得满地垃圾

弄巧成拙了!网传义乌一饭店:客人白酒啤酒饮料连菜都自带,14个人只花430元,包厢还弄得满地垃圾

王老师你还好吗
2026-09-28 01:39:54
范冰冰亮相 2026 曼谷国际电影节闭幕红毯,状态惊艳

范冰冰亮相 2026 曼谷国际电影节闭幕红毯,状态惊艳

墨薷桃桃
2026-09-29 09:43:16
张继科:我和马龙樊振东是男乒最强3人!复出打奥运会?打一两场还行

张继科:我和马龙樊振东是男乒最强3人!复出打奥运会?打一两场还行

念洲
2026-09-29 12:14:09
普约尔:退役后生活平静,送女儿上学后就运动、带着狗上山

普约尔:退役后生活平静,送女儿上学后就运动、带着狗上山

懂球帝
2026-09-29 17:00:22
联合国秘书长有句话,他曾挑明:打垮日本,中国才是决定性力量;日本被耗到油尽灯枯,最后低头投降,中国立的是根本大功

联合国秘书长有句话,他曾挑明:打垮日本,中国才是决定性力量;日本被耗到油尽灯枯,最后低头投降,中国立的是根本大功

回京历史梦
2026-09-27 17:58:12
已经宣布独立建国,但是中国不肯承认的5个国家

已经宣布独立建国,但是中国不肯承认的5个国家

七号说三国
2026-09-29 19:11:46
美国为何没有贪官?不是没有,是因为美国从制度上消灭了贪官,就连贪污都是合法的

美国为何没有贪官?不是没有,是因为美国从制度上消灭了贪官,就连贪污都是合法的

扶苏聊历史
2026-09-12 17:43:57
美日韩外长发表声明,不许改变台海现状,不到24小时,中方罕见表态

美日韩外长发表声明,不许改变台海现状,不到24小时,中方罕见表态

关权教授聊经济
2026-09-29 10:14:07
发现一个奇怪的现象:你吼了孩子,孩子敢顶嘴,说明你这个家庭还有救;如果你不停对孩子吼叫谩骂,孩子一声不吭沉默寡言,那你家庭没救了

发现一个奇怪的现象:你吼了孩子,孩子敢顶嘴,说明你这个家庭还有救;如果你不停对孩子吼叫谩骂,孩子一声不吭沉默寡言,那你家庭没救了

阿呆爸
2026-09-27 21:04:44
提醒|天津毛毛虫泛滥成灾!小区、街头、学校已大范围出现!网友:比云南虫谷还厉害…

提醒|天津毛毛虫泛滥成灾!小区、街头、学校已大范围出现!网友:比云南虫谷还厉害…

天津人
2026-09-29 21:11:33
嘌呤大户被揪出,是海鲜的15倍?医生提醒:经常吃,关节或“扛不住”

嘌呤大户被揪出,是海鲜的15倍?医生提醒:经常吃,关节或“扛不住”

观星赏月
2026-09-28 16:17:06
TVB小生疑似偷瞄女神低胸照片疯传!拨头发被指变搓耳做性挑逗?网民怒斥肯定是花花公子

TVB小生疑似偷瞄女神低胸照片疯传!拨头发被指变搓耳做性挑逗?网民怒斥肯定是花花公子

TVB资讯台
2026-09-29 00:47:58
有点吓人!4所港校更新2027年学费,涨幅超50%,最贵的读完400万起步

有点吓人!4所港校更新2027年学费,涨幅超50%,最贵的读完400万起步

阅读第一
2026-09-28 08:33:38
严子怡夺冠余波不断:日本电视台破防不知如何解说,北口榛花绝望大哭

严子怡夺冠余波不断:日本电视台破防不知如何解说,北口榛花绝望大哭

一枚野球君
2026-09-29 18:53:43
希音闷声发财,半年净赚150亿

希音闷声发财,半年净赚150亿

跨境派Pro
2026-09-29 14:20:02
今年养老金上调推迟真正原因:看看全国统筹养老金有多麻烦!

今年养老金上调推迟真正原因:看看全国统筹养老金有多麻烦!

云居历史
2026-09-29 00:31:36
视频来了!台湾岛中央山脉清晰可见,专家:显然是对“台独”分裂势力与日菲勾连的警告

视频来了!台湾岛中央山脉清晰可见,专家:显然是对“台独”分裂势力与日菲勾连的警告

环球时报国际
2026-09-29 18:52:11
赢了所有比赛,却可能留不下?林诗栋被调侃去德国,饭圈才是推手

赢了所有比赛,却可能留不下?林诗栋被调侃去德国,饭圈才是推手

曹老师评球
2026-09-28 20:38:42
亚运会:拼到受伤痛失好局!刘圣书/谭宁1-2世界第2,国羽无缘三连冠

亚运会:拼到受伤痛失好局!刘圣书/谭宁1-2世界第2,国羽无缘三连冠

钉钉陌上花开
2026-09-29 13:33:36
张家齐母女相处引热议,锤娜丽莎发长文,称“别再艾特我关于闺女的内容了,我不想再被气第N遍”

张家齐母女相处引热议,锤娜丽莎发长文,称“别再艾特我关于闺女的内容了,我不想再被气第N遍”

韩小娱
2026-09-29 16:28:48
2026-09-29 21:39:00
deephub incentive-icons
deephub
CV NLP和数据挖掘知识
2028文章数 1465关注度
往期回顾 全部

科技要闻

82亿美元收购!李飞飞将与苏姿丰并肩做AI

头条要闻

老夫妻开包子铺28年想退休 儿子纠结是否接班:太辛苦

头条要闻

老夫妻开包子铺28年想退休 儿子纠结是否接班:太辛苦

体育要闻

石雨豪:亚运金牌与背后的十年

娱乐要闻

赌王四房婚礼,何超琼带三房成员现身

财经要闻

央行调整完善若干货币政策工具

汽车要闻

京北"小兰博"/限时售9.99万起 北京现代艾尼氪V上市

态度原创

家居
手机
亲子
房产
数码

家居要闻

2026建博会(广州) 公装联探展交流活动

手机要闻

iQOO 16首发Q-flex技术,操控体验大升级

亲子要闻

宝蓝和爸爸比赛用箱子的零食化妆,快看看谁画的更漂亮~

房产要闻

山海悦·壹号院喜封金顶 第五代住宅重塑自贸港人居新标杆

数码要闻

火影T5A 2026款游戏本发售:酷睿i5-13420H + RTX 3050,5499元起

无障碍浏览 进入关怀版