# 安装，在使用 RAPTOR 之前，请确保已安装 Python 3.8+。克隆 RAPTOR 存储库并安装必要的依赖项：git clone https://github.com/parthsarthi03/raptor.gitcd raptorpip install -r requirements.txt
# 开始使用 RAPTOR，请按照以下步骤操作：# 首先，设置您的 OpenAI API 密钥并初始化 RAPTOR 配置：import osos.environ["OPENAI_API_KEY"] = "your-openai-api-key"
from raptor import RetrievalAugmentation
# Initialize with default configuration. For advanced configurations, check the documentation. [WIP]RA = RetrievalAugmentation()
# 将您的文本文档添加到 RAPTOR 进行索引：with open('sample.txt', 'r') as file:text = file.read()RA.add_documents(text)
# 现在可以使用 RAPTOR 根据索引文档回答问题：question = "How did Cinderella reach her happy ending?"answer = RA.answer_question(question=question)print("Answer: ", answer)
# 将构造好的树保存到指定路径：SAVE_PATH = "demo/cinderella"RA.save(SAVE_PATH)
# 将保存的树重新加载到 RAPTOR 中：RA = RetrievalAugmentation(tree=SAVE_PATH)answer = RA.answer_question(question=question)