NodeRAG 索引构建
python -m NodeRAG.build -f path/to/main_foulder
当您首次使用此命令
时,它将在main_folder
目录中创建Node_config.yaml
文件。
根据以下说明修改配置文件(添加 API 和服务提供商),以确保 NodeRAG 可以访问正确的 API。
要快速使用 NodeRAG 演示,请设置您的 OpenAI 账户的 API 密钥。如果您没有 API 密钥,请参考 OpenAI 认证。确保您在model_config
和embedding_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