5分钟上手Cognee:为AI代理构建持久记忆

通过6行代码,为你的AI代理添加基于向量和图数据库的持久记忆功能

环境要求

支持的操作系统

macOS Windows Linux

运行环境

Python 3.10 - 3.13 必需

所需工具

pip

Python包管理器,用于安装Cognee

必需

操作步骤

1

安装Cognee

使用pip安装Cognee包

安装Cognee核心包
pip install cognee

预期结果:看到类似 'Successfully installed cognee-x.x.x' 的提示

如果遇到权限问题,可以尝试添加 --user 参数

2

配置API密钥

设置OpenAI API密钥(需要先注册OpenAI账号获取API密钥)

Linux/macOS设置环境变量
export OPENAI_API_KEY='你的API密钥'
Windows设置环境变量
set OPENAI_API_KEY=你的API密钥

预期结果:无报错,环境变量设置成功

也可以创建.env文件:复制项目中的.env.template并填入你的API密钥

3

运行示例代码

创建并运行一个简单的Python脚本,体验Cognee的记忆功能

单行命令运行完整示例
python -c "
from cognee import Cognee
cognee = Cognee()
cognee.add('AI代理记忆让对话更连贯')
print(cognee.search('对话连贯性'))
"

预期结果:看到包含 'AI代理记忆让对话更连贯' 的搜索结果

第一次运行会下载必要的模型,可能需要等待1-2分钟

验证成功

成功运行后,你应该能看到Cognee的记忆搜索功能正常工作

看到包含你添加内容的搜索结果
没有出现错误提示
程序正常退出

快速提示

性能

第一次运行较慢是正常的,因为需要下载和初始化模型

配置

确保OPENAI_API_KEY环境变量已正确设置,否则会报错

其他

可以尝试添加更多文本,然后搜索相关概念,体验语义搜索

常见问题

1

ModuleNotFoundError: No module named 'cognee'

重新运行 pip install cognee,确保安装成功

2

API key not provided

检查OPENAI_API_KEY环境变量是否正确设置,或创建.env文件

3

网络连接超时

检查网络连接,或尝试使用代理

下一步

查看官方文档

了解更高级的功能和配置选项

尝试添加自己的文档

使用cognee.add()添加更多文本,构建个性化记忆库

探索本地UI界面

运行 cognee ui 命令启动本地Web界面

助手