RAG入门秘籍：构建检索增强生成应用程序的终极指南！

AI老猴子

已于 2024-09-12 11:57:06 修改

阅读量1.4k

点赞数 38

文章标签：人工智能大模型 AI大模型 ai 检索增强 RAG

于 2024-08-31 09:00:00 首次发布

本文链接：https://blog.csdn.net/star_nwe/article/details/141727242

版权

概述：RAG（检索增强生成）是一种结合信息检索和生成模型的技术，允许用户将自己的数据与大型语言模型（LLM）结合，生成更精确和上下文相关的输出。这篇文章将简要介绍RAG的基本概念，并提供一个简化的教程，帮助初学者从零开始构建RAG应用程序。

一、学习RAG的挑战

在快速变化的AI领域中，特别是关于RAG，存在大量噪音和复杂性。供应商往往将其过度复杂化，试图将他们的工具、生态系统和愿景注入其中。本教程旨在帮助初学者消除这些干扰，专注于从头构建一个简单的RAG系统。

二、什么是检索增强生成（RAG）

RAG的核心思想是通过检索工具将用户的自有数据添加到传递给大型语言模型的提示中，以此生成输出。相比单纯依赖预训练模型，这种方法带来了多项优势：

避免幻觉：通过在提示中包含事实信息，减少LLM产生幻觉的风险。
参考事实：允许用户在回应查询时参考真实数据，验证潜在问题。
利用未训练数据：可以使用LLM未曾训练过的数据。

三、RAG系统的基本组件

一个RAG系统由以下几个组件构成：

文档集合（语料库）
用户输入
文档集合与用户输入之间的相似性度量

四、RAG系统的操作步骤

接收用户输入
执行相似性测量
对用户输入和检索到的文档进行后处理

初学者可以通过以下步骤从头构建一个RAG系统，并逐步学习复杂的变体。

五、示例：构建最简单的RAG系统

1. 获取文档集合

首先，我们定义一个简单的文档集合：

corpus_of_documents = [
    "Take a leisurely walk in the park and enjoy the fresh air.",
    "Visit a local museum and discover something new.",
    "Attend a live music concert and feel the rhythm.",
    "Go for a hike and admire the natural scenery.",
    "Have a picnic with friends and share some laughs.",
    "Explore a new cuisine by dining at an ethnic restaurant.",
    "Take a yoga class and stretch your body and mind.",
    "Join a local sports league and enjoy some friendly competition.",
    "Attend a workshop or lecture on a topic you're interested in.",
    "Visit an amusement park and ride the roller coasters."
]

2. 定义与执行相似性度量

为了比较用户输入和文档集合的相似性，我们可以使用Jaccard相似度。Jaccard相似度是指两个集合的交集除以并集的大小。以下是一个简单的实现：

def jaccard_similarity(query, document):
    query = query.lower().split(" ")
    document = document.lower().split(" ")
    intersection = set(query).intersection(set(document))
    union = set(query).union(set(document))
    return len(intersection)/len(union)

接下来，我们定义一个函数来返回与用户输入最相似的文档：

def return_response(query, corpus):
    similarities = []
    for doc in corpus:
        similarity = jaccard_similarity(query, doc)
        similarities.append(similarity)
    return corpus_of_documents[similarities.index(max(similarities))]

3. 运行示例

我们可以通过以下代码运行一个简单的示例：

user_prompt = "What is a leisure activity that you like?"
user_input = "I like to hike"
response = return_response(user_input, corpus_of_documents)
print(response)

输出将是：

'Go for a hike and admire the natural scenery.'

恭喜，您已经构建了一个基本的RAG应用程序！

六、改进相似性度量与添加LLM

虽然我们使用了简单的Jaccard相似度来学习，但它在处理语义上存在局限。为了进一步提升生成效果，我们可以引入大型语言模型（LLM）进行后处理。以下是一个集成开源LLM的简单示例：

import requests
import json

def query_llama(user_input, document):
    # 假设你有一个运行的LLM服务，可以发送HTTP请求获取生成的文本
    url = "http://localhost:8000/query"  # LLM服务的URL
    payload = {
        "prompt": f"User input: {user_input}\nDocument: {document}\nResponse:",
        "max_tokens": 50
    }
    headers = {"Content-Type": "application/json"}
    response = requests.post(url, data=json.dumps(payload), headers=headers)
    return response.json().get("text")

response_with_llm = query_llama(user_input, response)
print(response_with_llm)