This is the multi-page printable view of this section. Click here to print.

Return to the regular view of this page.

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