NodeRAG

探索 NodeRAG 实现 本节详细介绍如何复现 NodeRAG 的结果。

NodeRAG 索引构建

python -m NodeRAG.build -f path/to/main_foulder

当您首次使用此命令时,它将在main_folder目录中创建Node_config.yaml文件。

创建配置文件

根据以下说明修改配置文件(添加 API 和服务提供商),以确保 NodeRAG 可以访问正确的 API。

要快速使用 NodeRAG 演示,请设置您的 OpenAI 账户的 API 密钥。如果您没有 API 密钥,请参考 OpenAI 认证。确保您在model_configembedding_config部分都输入了 API 密钥。

有关详细配置和修改说明,请参阅配置指南

#==============================================================================
# AI 模型配置
#==============================================================================
model_config:
  model_name: gpt-4o-mini            # 用于文本生成的模型名称
  api_keys:    # 您的 API 密钥(可选)

embedding_config:
  api_keys:    # 您的 API 密钥(可选)

构建

设置配置后,重新运行以下命令:

python -m NodeRAG.build -f path/to/main_folder

终端将显示状态树:

状态树

y继续。等待工作流程完成。

处理中

完成

有关 NodeRAG 配置、索引和使用的更多详细信息,请参阅我们的文档

回答和评估

首先,根据基准测试格式准备您的测试问题。您需要创建一个包含问题及其对应答案键的测试集 parquet 文件。准备好后,您可以使用以下命令运行评估:

python -m /eval/eval_node -f path/to/main_folder -q path/to/question_parquet

Last modified April 5, 2025: update reproduce (f23a25c)