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

200行Rust代码,给AI Agent装上永久记忆

0
分享至

如果你的 AI 助手每次对话都像失忆一样从零开始,这篇文章就是为你写的。
一、ChatGPT 那么强,为什么它还是记不住你?

试过让 GPT 帮你连续写代码吗?第二天打开,它又问你"请提供项目背景"。这不是 GPT 不聪明,而是所有大语言模型(LLM)都有一个先天性缺陷每次对话都是孤立会话,没有记忆

市面上的解决方案五花八门——LangChain 的 Memory 模块、Mem0、Zep、Letta——但它们大多笨重,不是强依赖 Python 生态就是绑定了特定框架。做 Rust 技术栈的 Tauri 应用、嵌入式 Agent,或者需要一个零重依赖、即时可用的记忆系统?选择很少。

于是我写了个开源库:ai-memory-core。核心代码不到 800 行,仅依赖 serde 几项 Rust 标配,却能提供完整的持久化记忆能力。


rust 开源库 ai-memory-core 发布


二、核心理念:热记忆 + 冷记忆

我把记忆拆成两层:

┌────────────────────────────────┐│       热记忆 (Hot Memory)       │  ← 每次对话自动注入 system prompt│   Token 预算:2000 tokens       │     按优先级排序,高价值记忆优先├────────────────────────────────┤│       冷记忆 (Cold Memory)      │  ← 全部历史,持久化到磁盘│   格式:memories.md (纯 Markdown) │     人类可读、grep 友好、可手动编辑└────────────────────────────────┘

热记忆丢进 LLM 上下文窗口,让 AI 每次对话都知道你的偏好和项目背景。冷记忆落盘,永远不会丢。

冷记忆的存储用了纯 Markdown 文件——对,就是 .md 文件。不是 SQLite,不是向量数据库,不是 Redis。你可以用 VS Code 打开 memories.md 直接看、直接改。极致简单。

三、三段伪代码看懂怎么用 第一步:保存记忆

use ai_memory_core::handle_memory_save;// 空间路径 + 内容,就这么简单handle_memory_save("Preferences/programming-languages", "偏好 TypeScript 和 Rust")?;handle_memory_save("Decisions/database-choice", "生产环境使用 PostgreSQL")?;

路径结构是 Wing/Hall/Room 三层分类——想象成图书馆的"馆/厅/室"——让记忆有组织地生长,而不是扔进一个扁平列表。

第二步:注入到 System Prompt

use ai_memory_core::inject_memories_into_system_prompt;let prompt = "You are a helpful coding assistant.";let enriched = inject_memories_into_system_prompt(prompt);// 输出变成了:// "You are a helpful coding assistant.//  [USER_MEMORY]//  偏好 TypeScript 和 Rust (高优先级, 被访问 8 次)//  生产环境使用 PostgreSQL//  [/USER_MEMORY]"

系统自动做了三件事:① 加载全部记忆 ② 按"访问频次×是否为高价值记忆"排序 ③ 控制在 2000 token 预算内。你不需要手动截断。

第三步:会话结束后自动提取新记忆

use ai_memory_core::extract_memories_with_llm;// 传 API Key + 对话记录 + 模型提供商let count = extract_memories_with_llm(    &api_key,    &conversation_history,    "deepseek"  // 也支持 anthropic / openai).await?;println!("本次会话提取了 {count} 条新记忆");

LLM 会自动从对话中识别偏好、决策、知识,并写进对应的 Memory Hall。你可以在 prompts/memory/extract.md 里自定义提取规则,修改后立即生效,不需要重新编译。

四、三个容易被忽视的"魔鬼细节" ① MD5 指纹去重

相同内容的记忆不会被重复写入,只更新日期。不是靠标题或 ID,是靠内容哈希。你写"偏好 TypeScript"和上次一样,它就只刷时间戳。

② 跨会话"热度"算法

每条记忆背后有一个 MetadataStore,记录访问次数。如果一个偏好被引用了 5 次以上,系统自动标记为"高价值",下次注入更靠前。这本质上是一个无监督的强化信号——不是你告诉系统什么重要,而是系统从使用模式中自己学会

③ 零重依赖,编译即用

核心依赖列表:serde、serde_json、chrono、dirs、md-5。没有 async runtime(除非你开 llm-extraction feature)、没有数据库 driver、没有网络库。编译时间以秒计,二进制增量极小。

五、适合谁用?

场景

为什么合适

Tauri 桌面 AI 应用

Rust 原生,直接嵌入 sidecar

自建 AI CLI 工具

会话记忆自动持久化,跨重启保留

LLM Agent 框架

作为 Memory Backend 接入 Agent 生命周期

嵌入式/边缘设备

零数据库依赖,Markdown 文件即可

六、开源 & 一行命令开始

cargo add ai-memory-core# 如果需要 LLM 自动提取:cargo add ai-memory-core --features llm-extraction

项目地址:github.com/peterfei/ai-memory

库地址: https://crates.io/crates/ai-memory-core

写这个库的初心很简单:AI 应该有记忆,而实现记忆不应该引入一整套基础设施。一个 Markdown 文件 + 几百行 Rust 代码,足够了。

如果你也在做 AI Agent 方向的产品,欢迎试用、提 Issue、提 PR。让 AI 不再"失忆",从这一行 Cargo.toml 开始。

标签:#AI编程 #Rust #LLM #开源工具 #AI记忆系统 #Agent开发

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

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.

相关推荐
热点推荐
卢伟权威再遭挑战!一人被换下与他争吵,拒绝听战术,刺头管不住

卢伟权威再遭挑战!一人被换下与他争吵,拒绝听战术,刺头管不住

南海浪花
2026-06-03 10:14:30
美国女子去世75年,癌细胞却活了下来,如今已分裂超过5000万吨!

美国女子去世75年,癌细胞却活了下来,如今已分裂超过5000万吨!

探谜未知世界
2026-06-03 12:58:55
差41岁的忘年恋!73岁香港卢伟国迎娶好友90后女儿,网友说的现实

差41岁的忘年恋!73岁香港卢伟国迎娶好友90后女儿,网友说的现实

寒士之言本尊
2026-06-03 13:53:46
什么叫“投名状”?意大利给全世界上了一课。

什么叫“投名状”?意大利给全世界上了一课。

阿七说史
2026-05-10 05:05:06
22岁男子入室杀害前女友,曾称“你不过是长得漂亮,你凭啥?”

22岁男子入室杀害前女友,曾称“你不过是长得漂亮,你凭啥?”

中国新闻周刊
2026-06-02 18:00:54
斯科特·福斯特执法NBA总决赛第一场,令马刺和尼克斯球迷担忧

斯科特·福斯特执法NBA总决赛第一场,令马刺和尼克斯球迷担忧

好火子
2026-06-04 01:29:04
曙光初现!中国足球小将横扫欧洲夺冠,为国足青训指明新路!

曙光初现!中国足球小将横扫欧洲夺冠,为国足青训指明新路!

田先生篮球
2026-06-02 23:13:23
四川一户人家一棵文竹,长到两层楼高,长势太狂野,网友:成精了

四川一户人家一棵文竹,长到两层楼高,长势太狂野,网友:成精了

三农老历
2026-06-04 01:34:59
钱是有灵性的,学会这4个吸财方法,钱主动来找你

钱是有灵性的,学会这4个吸财方法,钱主动来找你

金沛的国学笔记
2026-06-01 17:02:52
戴尔599美元XPS13迎战MacBook Neo

戴尔599美元XPS13迎战MacBook Neo

我是一个粉刷匠2
2026-06-03 02:16:01
毛焦尔采取强制行动!他不止骗了冯德莱恩,北约、欧盟笑不出来了

毛焦尔采取强制行动!他不止骗了冯德莱恩,北约、欧盟笑不出来了

花漾夜雨飘雪
2026-06-04 00:37:10
上海滩大佬深谋远虑,送幼子奔赴延安,晚年得以安稳善终

上海滩大佬深谋远虑,送幼子奔赴延安,晚年得以安稳善终

唠叨说历史
2026-05-25 14:18:49
世界杯在即,伊朗队仍没拿到美国签证:将驻扎墨西哥,比赛当天跨境参赛,鲁比奥发声

世界杯在即,伊朗队仍没拿到美国签证:将驻扎墨西哥,比赛当天跨境参赛,鲁比奥发声

红星新闻
2026-06-03 17:12:54
高开低走!从爆款男主到无人邀约,这四位短剧男神终是昙花一现

高开低走!从爆款男主到无人邀约,这四位短剧男神终是昙花一现

科学发掘
2026-06-03 19:01:24
我如今已68了,以亲身血泪教训告诉你:不要跟任何人,包括你的父母、子女、枕边人,分享这两件事

我如今已68了,以亲身血泪教训告诉你:不要跟任何人,包括你的父母、子女、枕边人,分享这两件事

心理观察局
2026-05-23 07:00:06
最高院:提供 “口交” “肛交”等进入式性服务,是否属卖淫行为?

最高院:提供 “口交” “肛交”等进入式性服务,是否属卖淫行为?

周军律师聊案子
2026-04-21 09:50:16
蓝色起源发射台炸了,年底还要再飞?

蓝色起源发射台炸了,年底还要再飞?

碳基打工人
2026-06-03 01:50:32
记者:出于竞技方面的考虑,斯图加特队长卡拉佐尔无缘世界杯

记者:出于竞技方面的考虑,斯图加特队长卡拉佐尔无缘世界杯

懂球帝
2026-06-03 12:55:06
结束访华不到20天,特朗普突然下禁令,要掐断中国"经济命脉"?

结束访华不到20天,特朗普突然下禁令,要掐断中国"经济命脉"?

离离言几许
2026-06-03 23:05:47
回国后我才敢说:俄罗斯小孩看中国人的眼神,和老人态度截然不同

回国后我才敢说:俄罗斯小孩看中国人的眼神,和老人态度截然不同

复转这些年
2026-05-24 16:38:35
2026-06-04 04:36:49
侃故事的阿庆
侃故事的阿庆
几分钟看完一部影视剧,诙谐幽默的娓娓道来
1203文章数 8617关注度
往期回顾 全部

科技要闻

传DeepSeek融资意向500亿:腾讯投100亿

头条要闻

男子不想上班辞职后上武当山当道士 8个月后选择下山

头条要闻

男子不想上班辞职后上武当山当道士 8个月后选择下山

体育要闻

选择中国品牌的库里,和他们的巨大野心

娱乐要闻

官方痛批乱象 刘涛郑恺等艺人遭点名

财经要闻

AI,开始偷懒了?

汽车要闻

专访蒋平:安全不做高低配 长安要让安全技术普惠

态度原创

房产
游戏
亲子
旅游
教育

房产要闻

突发!254亩调规,海口江东的超级学校真的快来了!

曝《塞尔达》传奇重制版将亮相!任天堂发布会下周见

亲子要闻

9岁孩童竟患上冠心病!医生说明原因,父母后悔不已……

旅游要闻

“一票跨两省”还游客完整壶口 | 新京报社论

教育要闻

大龄在美国学护理 RN CNA NP不同类型如何选??

无障碍浏览 进入关怀版