Knowledge Engine for AI Agent Memory in 6 lines of code
💡 Use Cases
最适合需要为AI系统添加持久化、结构化记忆能力的场景,特别是涉及多轮对话、复杂文档关系和多模态数据的应用。
AI客服记忆持久化
Problem: AI客服在多轮对话中经常忘记之前的对话内容,导致用户体验割裂,需要重复信息。
Solution: 使用Cognee将历史对话记录转化为知识图谱,让AI客服能记住用户偏好和历史问题,实现个性化服务。
Example: 电商客服AI能记住用户上次咨询的商品型号、尺寸偏好,下次对话时直接提供相关推荐。
文档智能问答系统
Problem: 传统RAG系统只能基于向量相似度检索,无法理解文档间的复杂关系,回答不够精准。
Solution: 用Cognee的ECL管道将文档库转化为图数据库+向量搜索的混合记忆层,实现基于关系的精准检索。
Example: 企业内部知识库问答,不仅能找到相关文档,还能理解“A项目是B项目的前置条件”这类关系。
多模态AI助手记忆
Problem: AI助手处理图片、音频、文档等多种数据时,记忆分散在不同系统中,无法形成统一认知。
Solution: 通过Cognee统一处理30+数据源,将多模态数据互联成知识图谱,让AI拥有跨模态的记忆能力。
Example: 设计助手能记住用户上传的参考图片风格、会议录音中的需求描述,综合给出设计方案。
LangGraph代理记忆管理
Problem: 在LangGraph等AI代理框架中,代理状态和记忆管理复杂,需要大量自定义代码。
Solution: 集成Cognee作为持久化记忆层,仅需6行代码即可为代理添加可查询、可扩展的记忆功能。
Example: 任务规划代理能记住之前执行过的任务步骤和结果,避免重复尝试失败的方案。
📊 Project Info
- Language
- Python
- Stars
- ⭐ 13,883
- Forks
- 1,387
- Today
- +270
- Ranking
- #10
- Collection
- Overall
- Trending Date
- March 15, 2026
- Last Push
- 3/15/2026
🏷️ Topics
5分钟上手Cognee:为AI代理构建持久记忆
通过6行代码,为你的AI代理添加基于向量和图数据库的持久记忆功能
🖥️ OS
⚙️ Runtime
🔧 Tools
📝 Steps
安装Cognee
使用pip安装Cognee包
安装Cognee核心包
$ pip install cognee配置API密钥
设置OpenAI API密钥(需要先注册OpenAI账号获取API密钥)
Linux/macOS设置环境变量
$ export OPENAI_API_KEY='你的API密钥'Windows设置环境变量
$ set OPENAI_API_KEY=你的API密钥运行示例代码
创建并运行一个简单的Python脚本,体验Cognee的记忆功能
单行命令运行完整示例
$ python -c "
from cognee import Cognee
cognee = Cognee()
cognee.add('AI代理记忆让对话更连贯')
print(cognee.search('对话连贯性'))
"✅ 验证成功
成功运行后,你应该能看到Cognee的记忆搜索功能正常工作
- ✓看到包含你添加内容的搜索结果
- ✓没有出现错误提示
- ✓程序正常退出
⚡ Quick Tips
🔍 Troubleshooting
❓ ModuleNotFoundError: No module named 'cognee'
→ 重新运行 pip install cognee,确保安装成功
❓ API key not provided
→ 检查OPENAI_API_KEY环境变量是否正确设置,或创建.env文件
❓ 网络连接超时
→ 检查网络连接,或尝试使用代理
🎯 Next Steps
查看官方文档
了解更高级的功能和配置选项
尝试添加自己的文档
使用cognee.add()添加更多文本,构建个性化记忆库
探索本地UI界面
运行 cognee ui 命令启动本地Web界面
