开源!6行代码让AI Agent拥有永久记忆
AI圈有个老笑话:AI Agent就像金鱼,每次对话结束,之前聊的全部忘光。
这真不是调侃,是所有做Agent开发的人的痛。你辛辛苦苦调教好的Agent,关掉再打开,它就不认识你了。之前学的知识、记住的偏好,全部清零。
但现在,这个难题被一个开源项目解决了。
Cognee是个啥?
这个项目叫Cognee,读音大概是"考尼",24K+ Star,已经在GitHub火了一波。
简单说,它就是一个给AI Agent用的"记忆系统"。你喂给它文档、聊天记录、数据库,它就能自动构建知识图谱,让Agent跨会话记住所有内容。
更夸张的是,实现这一切只需要6行代码:
import cognee
import asyncio
async def main():
await cognee.remember("Cognee turns documents into AI memory.")
results = await cognee.recall("What does Cognee do?")
for result in results:
print(result)
asyncio.run(main())
就这?没错,就这。
它是怎么工作的?
Cognee用了三招来解决Agent记忆问题:
第一招:啥都能吃。 不管是PDF、Word、网页、数据库,还是API返回的JSON,Cognee都能吞进去。它会自动解析、提取、转换成统一的格式。 第二招:知识图谱+向量双索引。 传统方案要么用向量数据库(按语义搜),要么用知识图谱(按关系搜)。Cognee两个都做,取长补短。你既可以搜"跟这个相似的",也可以搜"跟这个有关系的"。 第三招:会话记忆+永久记忆分开。 短期对话存在session里,快;长期知识存到graph里,持久。两者可以独立使用,也可以一起配合。实际用起来啥效果?
我试了一下,确实香。
比如你让Agent帮你处理一个项目,它读了一堆需求文档、代码库、之前的聊天记录。正常情况下,你关掉再打开,这些全忘了。Agent还得从头开始理解上下文。
用Cognee的话,它会把这些全部存到知识图谱里。下次打开,你问它"上次那个项目进度怎么样了",它能直接答上来。
而且它支持语义搜索。你不用记得 exact 的关键词,只要意思对,它就能找到相关记忆。
还能怎么玩?
Cognee已经出了官方插件,支持OpenClaw和Claude Code。装上之后,你的AI工具自动拥有记忆能力。
对企业来说,它可以用来做"公司大脑"——把各种内部文档、客服记录、业务数据全部灌进去,然后让Agent基于这些知识来回答问题。
对个人来说,你可以给自己做一个"第二大脑"。读书笔记、学习心得、工作思考,全部存进去,AI帮你整理和回忆。
怎么上手?
门槛很低,会Python就能玩:
pip install cognee
然后设置一下API Key:
import os
os.environ["LLM_API_KEY"] = "你的key"
官方文档很详细,Colab也有现成的教程可以跑。
GitHub链接:https://github.com/topoteretes/cognee
意味着什么?
说白了,Cognee解决的是一个"让AI真正变成助手"的问题。
以前的AI工具,用完就忘,每次都是陌生人。现在它能记住你、记住你教它的东西、记住它学到的知识。这才是真正的"AI助手"该有的样子。
而且它是开源的、自己能托管的。数据不用跑别人服务器,安全自己掌控。
这波,我愿称之为AI Agent开发者的必备工具。
评论区