首页 > 其他分享 >基于RAG的私人知识库聊天机器人

基于RAG的私人知识库聊天机器人

时间:2024-06-20 18:01:35浏览次数:21  
标签:RAG 机器人 AI MongoDB 模型 知识库 langchain import

随着人工智能的快速发展,特别是大语言模型的惊艳表现,让我们见证了信息检索和知识管理系统的重大进步。在这篇文章中,将探讨如何利用 Retrieval-Augmented Generation(RAG)技术,结合 MongoDB 数据库和 LangChain 框架,来构建一个先进的私人知识聊天机器人。

大模型聊天机器人的不足

在过去的这段时间,相信大家都领会到了生成式人工智能聊天机器人的魅力,它们展现出具有令人难以置信的能力,可以即时创建回复。然而,直接使用像 ChatGPT、ChatGLM、文心一言这样的模型生成的答案也存在问题。

幻觉

大语言模型有时也会生成所谓的 “幻觉” 文本,这些文本可能描述了现实中不存在的事件、人物或地点,导致以下问题:

  1. 现实与虚构混淆:模型无法准确区分历史事实与虚构内容,从而生成关于虚构事件或人物的响应,这可能误导那些寻求确凿信息的用户。
  2. 错误信息传播:当模型提供不真实或误导性的信息时,特别是在科学或医学领域,可能对公众造成误导,影响决策和信念体系。

无法访问私人信息

虽然大语言模型能够生成广泛的答案,但它们无法访问私人或定制化的信息,这限制了它们的应用范围:

  1. 缺乏个性化解答:这些模型通常无法访问用户的私人数据,如超出范围的历史记录或个人偏好,因此无法提供高度个性化的答案。
  2. 标准化回答的局限性:由于缺乏对用户特定需求的了解,这些模型生成的答案可能无法满足特定用户群体或行业的独特需求。

答案太笼统

大语言模型通常在提供答案时过于泛化,缺乏针对特定领域或行业的深度和专业性:

  1. 行业特定知识的缺乏:模型在处理涉及专业领域(如金融、法律或医疗)的查询时,可能无法提供足够专业的见解或详细的技术数据。
  2. 泛化与专业性的平衡:在努力适应广泛的查询类型时,模型可能难以在泛化能力和特定领域的深度知识之间找到平衡点,导致答案可能不足以满足专业人士或具有特定需求的用户的期望。

为了应对这些挑战,我们转向 Retrieval-Augmented Generation(RAG),即检索增强生成。RAG是一个人工智能框架,旨在从外部知识库中获取准确和最新的事实,它在大语言模型的生成过程中提供上下文和理解至关重要。

MongoDB Atlas 和 LangChain

MongoDB Atlas 向量搜索

MongoDB Atlas 作为向量数据库的优势:

  • MongoDB Atlas 与现有的MongoDB数据库无缝集成,使其成为已经在数据管理中使用 MongoDB 的组织的自然扩展。这确保了在不需要进行大规模系统改造的情况下,顺利过渡到向量存储。
  • MongoDB Atlas 专为处理大规模、运营关键的应用程序而设计,展示了其稳健性和可靠性。这在数据准确性和可用性至关重要的应用程序中变得特别关键,它的横向扩展能力确保了效率。
  • MongoDB Atlas 在处理各种数据类型和结构方面的灵活性,使其成为容纳向量嵌入复杂性的理想选择。
  • MongoDB Atlas 定位在企业级,拥有高安全标准、跨多个云的可用性和全面管理。这确保了组织可以信任它来实现安全、可靠和高效的操作。

LangChain 框架

LangChain 是一个强大的语言模型驱动应用程序框架,它改变了构建智能聊天机器人的规则。LangChain 的突出优势:

  • 上下文感知:LangChain 使应用程序,尤其是聊天机器人,能够根据来自各种来源的上下文信息(如提示指令和历史交互)进行理解和响应。
  • 推理能力:LangChain 赋予应用程序有效推理的能力,使它们能够根据提供的上下文做出明智的决策。

由 RAG 驱动的生成式 AI 聊天机器人实现

原理图解

image.png

建立知识库

加载数据

需要准备好获取问题回答所需的数据,这可以通过从网络资源、PDF、Word 文档以及其他可用的文本资料中提取信息来实现。这一步骤至关重要,因为数据的质量直接影响到机器人回答问题的能力。

生成嵌入

获取数据后,对其进行处理,将其分割成可管理的数据块,这有助于信息的高效处理。每个数据块都经过嵌入过程,这一转换过程将原始文本转换为数字表示形式,便于计算机分析。采用的嵌入方法还需确保在 MongoDB Atlas 中存储数据时的数据安全性与高效检索能力。

MongoDB 向量存储

为实现高效检索,生成的嵌入向量将存储在 MongoDB 中。MongoDB 提供了高效的存储方案,可以快速地对这些嵌入向量进行检索,这对于问答系统的快速响应至关重要。

RAG 问答

问题嵌入

当用户提出问题时,会生成该特定问题的嵌入。这一步骤准备好将问题与 MongoDB Atlas 中存储的数据进行比较。

检索相似的数据块

利用 MongoDB 的向量搜索功能,系统检索与提出的问题最匹配的相关数据块,这些数据块是在知识库中准备好的。

基于上下文和问题的提示词

检索到的数据块和用户问题合并后,创建一个包含全面上下文的查询。这个上下文为大语言模型(LLM)提供了必要的信息,使其能够基于全面的理解生成回答。

通过RAG进行定制答案生成

配备了上下文的大语言模型(如 GPT 或类似模型)利用 RAG(Retrieval-Augmented Generation)机制生成定制化的答案。这样的答案既符合用户的问题,也充分考虑了数据集中的信息,确保了回答的个性化和准确性。这一过程是生成式 AI 聊天机器人的核心,它结合了检索式和生成式方法,以提供更为准确和灵活的回答。

代码实现

python复制代码import os  
import re  
from openai import OpenAI  
import time  
from dotenv import load_dotenv  
from pymongo import MongoClient  
from langchain.chat_models import ChatOpenAI  
from langchain.schema import AIMessage, HumanMessage  
from langchain.llms import OpenAI  
from langchain.document_loaders import PyPDFLoader  
from langchain.text_splitter import RecursiveCharacterTextSplitter  
from langchain.embeddings import OpenAIEmbeddings  
from langchain.vectorstores import MongoDBAtlasVectorSearch  
from langchain.chains import RetrievalQA  
from langchain.schema.language_model import BaseLanguageModel  
import gradio as gr

load_dotenv(override=True)  
  
# 连接 MongoDB 和 OpenAI
OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]  
MONGO_URI = os.environ["MONGO_URI"]  
DB_NAME = "pdfchatbot"  
COLLECTION_NAME = "pdfText"  
ATLAS_VECTOR_SEARCH_INDEX_NAME = "vector_index"  
  
embeddings = OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)  
   
EMBEDDING_FIELD_NAME = "embedding"  
TEXT_FIELD_NAME = "text"  
  
client = MongoClient(MONGO_URI)  
db = client[DB_NAME]  
collection = db[COLLECTION_NAME]

# PDF处理
def process_pdf(file,progress=gr.Progress()):  
    progress(0, desc="Starting")  
    time.sleep(1)  
    progress(0.05)  
    new_string = ""  
    for letter in progress.tqdm(file.name, desc="Uploading Your PDF into MongoDB Atlas"):  
    time.sleep(0.25)  

    loader = PyPDFLoader(file.name)  
    pages = loader.load_and_split()  
    print(pages)  
 
    text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50)  
    docs = text_splitter.split_documents(pages)  
    docs_as_strings = [str(doc) for doc in docs]  

    vectorStore = MongoDBAtlasVectorSearch(  
    collection, embeddings, index_name=ATLAS_VECTOR_SEARCH_INDEX_NAME  
    )  

    docsearch = vectorStore.from_documents(  
    docs,  
    embeddings,  
    collection=collection,  
    index_name=ATLAS_VECTOR_SEARCH_INDEX_NAME,  
    )  

    return docsearch

# 查询和显示    
def query_and_display(query,history):  
    history_langchain_format = []  
    for human, ai in history:  
        history_langchain_format.append(HumanMessage(content=human))  
        history_langchain_format.append(AIMessage(content=ai))  
    history_langchain_format.append(HumanMessage(content=query))  
    # Set up MongoDBAtlasVectorSearch with embeddings  
    vectorStore = MongoDBAtlasVectorSearch(collection,OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY),index_name=ATLAS_VECTOR_SEARCH_INDEX_NAME,)  
    print(query)    
    print("---------------")  
    docs = vectorStore.max_marginal_relevance_search(query, K=5)  
    
    llm = OpenAI(openai_api_key=OPENAI_API_KEY, temperature=0)  
    retriever = vectorStore.as_retriever(search_type="similarity",search_kwargs={"k": 5},)  

    for document in retriever:  
        print(str(document) + "\n")  

    qa = RetrievalQA.from_chain_type(llm, chain_type="stuff",retriever=retriever)  
    retriever_output = qa.run(query)  

    print(retriever_output)  
    return retriever_output
    
# 用户界面
with gr.Blocks(css=".gradio-container {background-color: AliceBlue}") as demo:  
    gr.Markdown("YBCarry Chatbot")  

    with gr.Tab("Upload PDF"):  
        with gr.Row():  
            pdf_input = gr.File()  
            pdf_output = gr.Textbox()  
        pdf_button = gr.Button("Upload PDF")  

    with gr.Tab("Ask question"):  
        gr.ChatInterface(query_and_display)  

    pdf_button.click(process_pdf, inputs=pdf_input, outputs=pdf_output)  

demo.launch()

image.png

image.png

结语

通过RAG,我们开始着手解决与大语言模型相关的挑战,确保了精准答复和上下文相关的响应。跟随这一方向,相信大模型可以更好地服务于我们的日常和工作。

如何学习大模型 AI ?

由于新岗位的生产效率,要优于被取代岗位的生产效率,所以实际上整个社会的生产效率是提升的。

但是具体到个人,只能说是:

“最先掌握AI的人,将会比较晚掌握AI的人有竞争优势”。

这句话,放在计算机、互联网、移动互联网的开局时期,都是一样的道理。

我在一线互联网企业工作十余年里,指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家,也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑,所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限,很多互联网行业朋友无法获得正确的资料得到学习提升,故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

在这里插入图片描述

第一阶段(10天):初阶应用

该阶段让大家对大模型 AI有一个最前沿的认识,对大模型 AI 的理解超过 95% 的人,可以在相关讨论时发表高级、不跟风、又接地气的见解,别人只会和 AI 聊天,而你能调教 AI,并能用代码将大模型和业务衔接。

  • 大模型 AI 能干什么?
  • 大模型是怎样获得「智能」的?
  • 用好 AI 的核心心法
  • 大模型应用业务架构
  • 大模型应用技术架构
  • 代码示例:向 GPT-3.5 灌入新知识
  • 提示工程的意义和核心思想
  • Prompt 典型构成
  • 指令调优方法论
  • 思维链和思维树
  • Prompt 攻击和防范

第二阶段(30天):高阶应用

该阶段我们正式进入大模型 AI 进阶实战学习,学会构造私有知识库,扩展 AI 的能力。快速开发一个完整的基于 agent 对话机器人。掌握功能最强的大模型开发框架,抓住最新的技术进展,适合 Python 和 JavaScript 程序员。

  • 为什么要做 RAG
  • 搭建一个简单的 ChatPDF
  • 检索的基础概念
  • 什么是向量表示(Embeddings)
  • 向量数据库与向量检索
  • 基于向量检索的 RAG
  • 搭建 RAG 系统的扩展知识
  • 混合检索与 RAG-Fusion 简介
  • 向量模型本地部署

第三阶段(30天):模型训练

恭喜你,如果学到这里,你基本可以找到一份大模型 AI相关的工作,自己也能训练 GPT 了!通过微调,训练自己的垂直大模型,能独立训练开源多模态大模型,掌握更多技术方案。

到此为止,大概2个月的时间。你已经成为了一名“AI小子”。那么你还想往下探索吗?

  • 为什么要做 RAG
  • 什么是模型
  • 什么是模型训练
  • 求解器 & 损失函数简介
  • 小实验2:手写一个简单的神经网络并训练它
  • 什么是训练/预训练/微调/轻量化微调
  • Transformer结构简介
  • 轻量化微调
  • 实验数据集的构建

第四阶段(20天):商业闭环

对全球大模型从性能、吞吐量、成本等方面有一定的认知,可以在云端和本地等多种环境下部署大模型,找到适合自己的项目/创业方向,做一名被 AI 武装的产品经理。

  • 硬件选型
  • 带你了解全球大模型
  • 使用国产大模型服务
  • 搭建 OpenAI 代理
  • 热身:基于阿里云 PAI 部署 Stable Diffusion
  • 在本地计算机运行大模型
  • 大模型的私有化部署
  • 基于 vLLM 部署大模型
  • 案例:如何优雅地在阿里云私有部署开源大模型
  • 部署一套开源 LLM 项目
  • 内容安全
  • 互联网信息服务算法备案

学习是一个过程,只要学习就会有挑战。天道酬勤,你越努力,就会成为越优秀的自己。

如果你能在15天内完成所有的任务,那你堪称天才。然而,如果你能完成 60-70% 的内容,你就已经开始具备成为一名大模型 AI 的正确特征了。

这份完整版的大模型 AI 学习资料已经上传CSDN,朋友们如果需要可以微信扫描下方CSDN官方认证二维码免费领取【保证100%免费

标签:RAG,机器人,AI,MongoDB,模型,知识库,langchain,import
From: https://blog.csdn.net/python1234567_/article/details/139828148

相关文章

  • RAG 如何消除大模型幻觉
    什么是大模型幻觉假设我们有一个基于大型生成模型(如GPT-3)的问答系统,该系统用于回答药企内部知识库中的问题。我们向其提出一个问题:“阿司匹林的主要药理作用是什么?”正确的答案应该是:“阿司匹林主要通过抑制环氧酶(COX)酶的活性,降低前列腺素和血栓素的生成,从而起到抗血小板......
  • 打开 RAG 对接大模型的黑盒 —— 9 大隐藏问题
    前一段时间,各个大模型在争斗:谁能携带更长、更大的上下文Prompt,比如Kimi说200万字,阿里通义千问又说自己能达1000万字;大家都知道Prompt很重要,但是RAG和长的上下文文本携带是两个不同的技术方向。RAG先来简单介绍一下什么是RAG(增强搜索生成),很简单:当我们问Ch......
  • Dragon Boat Festival Revised
    DragonBoatFestivalWiththeapproachingoftheDragonBoatFestival,thedesireofgoingbackhomebecamestrongerandstronger.AssoonasIarrivedhome,Ifeltrelaxedbecausethesmellofstickyriceandreedleavesfilledwiththeair.Itisfantast......
  • ROS机器人虚拟仿真挑战赛持续学习笔记-20240619
    cartographer需要全手工编译……比较麻烦。如果使用新版ceres-solver,版本2.x,需要修改源码,部分“接口代码”有改动。稳妥使用ceres-solver-1.13.0,且需要安装abseil-cpp。验证是否成功,使用roscd或roslaunch,查看一下是否有对于功能包:map只有room_mini和tianracer_racetr......
  • cbrutekrag:一款针对SSH服务器的渗透测试工具
    关于cbrutekragcbrutekrag是一款针对SSH服务器的渗透测试工具,该工具基于C语言开发,可以帮助广大研究人员使用字典攻击来对目标SSH服务器执行渗透测试。该工具仅出于教育目的设计和开发,请在获得测试授权后再使用该工具。 工具要求libssh 工具下载广大研究人员可以使......
  • 12.1.k8s中的pod数据持久化-pv与pvc资源及动态存储StorageClass
    目录一、pc与pvc的概念二、pvc与pv初体验1,准别nfs环境1.1.所有节点安装nfs工具1.2.harbor节点编辑nfs配置文件 2,创建3个pv资源3,harbor节点创建pv对应的nfs存储路径 4,创建pvc关联pv5,创建pod引入pvc6,编辑index访问文件到harbor存储目录下7,浏览器访问测试三、Storag......
  • 大模型微调和RAG的应用场景
      大家好,我是herosunly。985院校硕士毕业,现担任算法研究员一职,热衷于机器学习算法研究与应用。曾获得阿里云天池比赛第一名,CCF比赛第二名,科大讯飞比赛第三名。拥有多项发明专利。对机器学习和深度学习拥有自己独到的见解。曾经辅导过若干个非计算机专业的学生进入到算法行......
  • fragment2
    packagecom.example.dataapplication.fragment.frag2;importandroid.os.Bundle;importandroidx.annotation.NonNull;importandroidx.annotation.Nullable;importandroidx.fragment.app.Fragment;importandroidx.recyclerview.widget.LinearLayoutManager;import......
  • Fragment
    Fragment可以做到把屏幕划分成几块,然后进行分组,进行模块化的管理。并且可以更加方便的在运行过程中动态地更新Activity的用户界面Fragment并不能单独使用,他需要嵌套在Activity中使用,尽管他拥有自己的生命周期,但是还是会受到宿主Activity的生命周期的影响,比如Activity被destor......
  • Fragment的动态创建
    Fragment的动态创建动态创建不同于静态创建,不需要写固定的xml文件,但是依然要有一个xml文件来当容器。1.我们需要使用<androidx.fragment.app.FragmentContainerView/><?xmlversion="1.0"encoding="utf-8"?><LinearLayoutxmlns:android="http://schemas.android.com/apk/re......