跳到内容

GraphRAG 索引 🤖

GraphRAG 索引包是一个数据管道和转换套件,旨在利用大型语言模型(LLM)从非结构化文本中提取有意义的结构化数据。

索引管道是可配置的。它们由工作流、标准和自定义步骤、提示模板以及输入/输出适配器组成。我们的标准管道旨在

  • 从原始文本中提取实体、关系和声明
  • 对实体进行社区检测
  • 生成多粒度级别的社区摘要和报告
  • 将实体嵌入到图向量空间中
  • 将文本块嵌入到文本向量空间中

管道的输出默认以 Parquet 表的形式存储,嵌入数据则写入您配置的向量存储中。

入门

要求

有关设置开发环境的详细信息,请参阅入门中的要求部分。

要配置 GraphRAG,请参阅配置文档。拥有配置文件后,您可以使用 CLI 或 Python API 运行管道。

用法

CLI

uv run poe index --root <data_root> # default config mode

Python API

有关直接从 Python 代码调用的推荐方法,请参阅索引 API 的Python 文件

延伸阅读

  • 要在 GraphRAG 项目中开始开发,请参阅入门
  • 要了解索引库的基本概念和执行模型,请参阅架构文档
  • 要阅读更多关于配置索引引擎的信息,请参阅配置文档