ChatGPT | 修改RetrievalQA推荐答案的数量

知识库经常遇到一个问题会在一个文件的多处或者多个文件出现,这时候如果只回答一个结果就欠佳,最理想的做法是模仿推荐功能,把合适的多个答案及其出处汇总给用户。

如图,一个接口文档里面提到多处“http请求URL”:

ChatGPT | 修改RetrievalQA推荐答案的数量_第1张图片

使用上面接口文档embedding之后,如果不对RetrievalQA做修改,默认是返回4个结果

ChatGPT | 修改RetrievalQA推荐答案的数量_第2张图片

 对RetrievalQA做修改的话,可以返回更多结果

ChatGPT | 修改RetrievalQA推荐答案的数量_第3张图片

 代码也好简单,重点是search_kwargs={"k":10} 这部分

qa_RTCS = RetrievalQA.from_chain_type(llm=openAiLLm,chain_type="stuff",
                                      retriever=db_RTCS.as_retriever(search_kwargs={"k":10}),#k表示 推荐结果条数,越大推荐越全面
                                      return_source_documents = True) #如果需要打印上下文才使用

你可能感兴趣的:(ChatGPT,ChatGPT,LangChain,RetrievalQA,as_retriever)