topoteretes/cognee | Readme-i18n

cognee - 5行代码为AI智能体构建记忆系统

Demo . Learn more · Join Discord · Join r/AIMemory . Docs . cognee community repo

🚀 我们即将推出Cognee SaaS服务：点击这里注册托管版beta测试！

通过可扩展的模块化ECL（提取、认知化、加载）管道，为智能体构建动态记忆系统，替代传统RAG方案。

更多关于使用案例和评估的信息

功能特性

互联并检索您过往的对话记录、文档、图片及音频转录内容
替代RAG系统，降低开发者工作量与成本
仅需使用Pydantic即可将数据加载至图数据库和向量数据库
从30+数据源摄取数据时进行灵活处理

快速开始

通过Google Colab notebook、Deepnote notebook或入门代码库立即体验

参与贡献

您的贡献是推动本项目成为真正开源项目的核心力量。我们高度赞赏任何形式的贡献。详情请参阅CONTRIBUTING.md

📦 安装指南

您可通过pip、poetry、uv或其他Python包管理器安装Cognee。支持Python 3.8至3.12版本

使用pip安装

pip install cognee

本地安装Cognee

可通过pip、poetry和uv安装本地Cognee仓库。使用pip本地安装时请确保pip版本高于21.3

使用UV安装（含所有可选依赖项）

uv sync --all-extras

💻 基础用法

环境配置

import os
os.environ["LLM_API_KEY"] = "YOUR OPENAI_API_KEY"

您也可以通过创建.env文件设置变量，参考我们的模板文件。如需使用不同LLM供应商，更多信息请查阅官方文档

简单示例

以下脚本将运行默认处理流程：

import cognee
import asyncio


async def main():
    # Add text to cognee
    await cognee.add("Natural language processing (NLP) is an interdisciplinary subfield of computer science and information retrieval.")

    # Generate the knowledge graph
    await cognee.cognify()

    # Query the knowledge graph
    results = await cognee.search("Tell me about NLP")

    # Display the results
    for result in results:
        print(result)


if __name__ == '__main__':
    asyncio.run(main())

示例输出：

  Natural Language Processing (NLP) is a cross-disciplinary and interdisciplinary field that involves computer science and information retrieval. It focuses on the interaction between computers and human language, enabling machines to understand and process natural language.