RAG数据集自动构造探索, 附PROMPT

方法介绍

大型语言模型 (LLM) 能够有效地将简单问题转化为更复杂的问题。为了从提供的文档中生成中等难度到困难难度的样本，我们采用了以下方法：

• 推理：重写问题，以增强推理的需要，从而有效地回答问题。

• 增加条件：修改问题以引入条件元素，这增加了问题的复杂性。

• 多上下文：以需要来自多个相关部分或块的信息来形成答案的方式重新措辞问题。

• 对话式：部分问题经过演化，可以转化为对话式样本。这些问题模拟了聊天式的问答互动，模仿了聊天式问答流程。

实现Prompt展示

推理

根据提供的上下文将问题重写为多跳推理问题，从而使给定问题复杂化。回答这个问题应该要求读者利用给定上下文中的信息做出多个逻辑联系或推断。改写题目时要遵循的规则:1. 确保改写后的问题可以完全根据上下文提供的信息来回答。2. 不要提出超过15个单词的问题。尽可能使用缩写。3. 确保你的问题清晰明了。4. “基于所提供的上下文”、“根据上下文”等短语不允许出现在问题中。

多上下文

任务是重写并使给定的问题复杂化，使回答问题需要从context1和context2中获得信息。按照下面给出的规则重写问题。1. 改写后的问题不应该很长。尽可能使用缩写。2. 改写后的问题必须是合理的，必须被人类理解和回应。3. 改写后的问题必须完全符合上下文1和上下文2中的信息。4. 阅读并理解上下文中的内容，然后重写问题，这样回答问题时就需要同时了解上下文1和上下文2。5. 像“基于提供的上下文”、“根据上下文?”等不允许出现在问题中。

条件prompt

通过引入条件元素重写提供的问题以增加其复杂性。目标是通过加入影响问题上下文的场景或条件，使问题更加复杂。按照下面给出的规则重写问题。1. 改写后的问题不应超过25个单词。尽可能使用缩写。2. 改写后的问题必须是合理的，必须被人类理解和回应。3. 改写后的问题必须能从目前的信息中完全回答。

对话式

将提供的问题重新格式化为两个单独的问题，就像它是对话的一部分一样。每个问题都应该集中在与原问题相关的一个特定方面或子主题上。按照下面给出的规则重写问题。1. 改写后的问题不应超过25个单词。尽可能使用缩写。2. 改写后的问题必须是合理的，必须被人类理解和回应。3. 改写后的问题必须能从目前的信息中完全回答。4. 像“提供上下文”、“根据上下文?”等不允许出现在问题中。

详细可参考：

https://github.com/explodinggradients/ragas/blob/main/src/ragas/testset/prompts.py

代码实现

from ragas.testset.generator import TestsetGeneratorfrom ragas.testset.evolutions import simple, reasoning, multi_contextfrom langchain_openai import ChatOpenAI, OpenAIEmbeddings
# documents = load your documents
# generator with openai modelsgenerator_llm = ChatOpenAI(model="gpt-3.5-turbo-16k")critic_llm = ChatOpenAI(model="gpt-4")embeddings = OpenAIEmbeddings()
generator = TestsetGenerator.from_langchain(generator_llm,critic_llm,embeddings)
# Change resulting question type distributiondistributions = {simple: 0.5,multi_context: 0.4,reasoning: 0.1}
# use generator.generate_with_llamaindex_docs if you use llama-index as document loadertestset = generator.generate_with_langchain_docs(documents, 10, distributions) testset.to_pandas()