44.7k联系我们登录免费试用

使用 LangChain、LangChain vector store、Google Vertex AI Gemini 2.0 Flash 和 Ollama mxbai-embed-large 构建 RAG 聊天机器人

什么是 RAG

检索增强生成（Retrieval-Augmented Generation，简称 RAG）正引领生成式 AI，尤其是对话式 AI 的新潮流。它将预训练的大语言模型（LLM，如 OpenAI 的 GPT）与存储于向量数据库（如 Milvus、Zilliz Cloud）中的外部知识源相结合，从而让模型输出更准确、更具上下文相关性，并且能够及时融合最新信息。 一个完整的 RAG 系统通常包含以下四大核心组件：

  • 向量数据库：用于存储与检索向量化后的知识；
  • 嵌入模型：将文本转为向量表示，为后续的相似度搜索提供支持；
  • 大语言模型（LLM）：根据检索到的上下文和用户提问生成回答；
  • 框架：负责将上述组件串联成可用的应用。

核心组件说明

本教程将带你在 Python 环境下，借助以下组件一步步搭建一个初级的 RAG 聊天机器人：

  • LangChain: 一个开源框架，帮助你协调大语言模型、向量数据库、嵌入模型等之间的交互，使集成检索增强生成（RAG）管道变得更容易。
  • LangChain in-memory vector store: 一个内存型， 临时性 的向量存储，将嵌入数据存储在内存中，并通过精确的线性搜索找到最相似的嵌入。默认的相似度度量是余弦相似度，但可以更改为 ml-distance 支持的任何相似度度量。目前该存储仅适用于演示，不支持 ID 或删除操作。 (如果您需要为应用程序或企业项目提供更具扩展性的解决方案，我们推荐使用 Zilliz Cloud，这是一个基于开源项目 Milvus构建的全托管向量数据库服务，并提供支持最多 100 万个向量的免费套餐。)
  • Google Vertex AI Gemini 2.0 Flash: 该模型旨在快速部署和高性能机器学习任务。凭借增强的自然语言处理和图像识别能力，它在实时分析和自动决策方面表现出色。非常适合寻求快速洞察和高效模型训练的企业，应用领域包括聊天机器人和数据分析。
  • Ollama mxbai-embed-large: 这个先进的人工智能模型专注于为自然语言处理任务生成高质量的嵌入。它的优势在于捕捉文本数据中的细微含义和关系，使其非常适合进行语义搜索、推荐系统和内容聚类应用。

完成本教程后，你将拥有一个能够基于自定义知识库回答问题的完整聊天机器人。

注意事项: 使用专有模型前请确保已获取有效 API 密钥。

实战：搭建 RAG 聊天机器人

第 1 步：安装并配置 LangChain

%pip install --quiet --upgrade langchain-text-splitters langchain-community langgraph

第 2 步：安装并配置 Google Vertex AI Gemini 2.0 Flash

pip install -qU "langchain[google-vertexai]"

# Ensure your VertexAI credentials are configured

from langchain.chat_models import init_chat_model

llm = init_chat_model("gemini-2.0-flash-001", model_provider="google_vertexai")

第 3 步：安装并配置 Ollama mxbai-embed-large

pip install -qU langchain-ollama

from langchain_ollama import OllamaEmbeddings

embeddings = OllamaEmbeddings(model="mxbai-embed-large")

第 4 步：安装并配置 LangChain vector store

pip install -qU langchain-core

from langchain_core.vectorstores import InMemoryVectorStore

vector_store = InMemoryVectorStore(embeddings)

第 5 步：正式构建 RAG 聊天机器人

在设置好所有组件之后，我们来搭建一个简单的聊天机器人。我们将使用 Milvus介绍文档 作为私有知识库。你可以用你自己的数据集替换它，来定制你自己的 RAG 聊天机器人。

import bs4
from langchain import hub
from langchain_community.document_loaders import WebBaseLoader
from langchain_core.documents import Document
from langchain_text_splitters import RecursiveCharacterTextSplitter
from langgraph.graph import START, StateGraph
from typing_extensions import List, TypedDict

# 加载并拆分博客内容
loader = WebBaseLoader(
    web_paths=("https://milvus.io/docs/overview.md",),
    bs_kwargs=dict(
        parse_only=bs4.SoupStrainer(
            class_=("doc-style doc-post-content")
        )
    ),
)

docs = loader.load()

text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
all_splits = text_splitter.split_documents(docs)

# 索引分块
_ = vector_store.add_documents(documents=all_splits)

# Define prompt for question-answering
prompt = hub.pull("rlm/rag-prompt")


# 定义应用状态
class State(TypedDict):
    question: str
    context: List[Document]
    answer: str


# 定义应用步骤
def retrieve(state: State):
    retrieved_docs = vector_store.similarity_search(state["question"])
    return {"context": retrieved_docs}


def generate(state: State):
    docs_content = "\n\n".join(doc.page_content for doc in state["context"])
    messages = prompt.invoke({"question": state["question"], "context": docs_content})
    response = llm.invoke(messages)
    return {"answer": response.content}


# 编译应用并测试
graph_builder = StateGraph(State).add_sequence([retrieve, generate])
graph_builder.add_edge(START, "retrieve")
graph = graph_builder.compile()

测试聊天机器人

Yeah! You've built your own chatbot. Let's ask the chatbot a question.

response = graph.invoke({"question": "What data types does Milvus support?"})
print(response["answer"])

示例输出

Milvus 支持多种数据类型，包括稀疏向量、二进制向量、JSON 和数组。此外，它还支持常见的数值类型和字符类型，使其能够满足不同的数据建模需求。这使得用户能够高效地管理非结构化或多模态数据。

优化小贴士

在搭建 RAG 系统时，合理调优能显著提升性能与效率。下面为各组件提供一些实用建议：

LangChain 优化建议

为了优化 LangChain，需要通过高效地构建链路和代理来减少工作流程中的冗余操作。使用缓存避免重复计算，从而加快系统速度，并尝试采用模块化设计，确保模型或数据库等组件能够轻松替换。这将提供灵活性和效率，使您能够快速扩展系统，而无需不必要的延迟或复杂性。

LangChain in-memory vector store 优化建议

LangChain 内存型向量存储只是一个临时性的向量存储，它将嵌入数据存储在内存中，并进行精确的线性搜索以找到最相似的嵌入。它的功能非常有限，仅适用于演示。如果您计划构建一个功能完整甚至生产级的解决方案，我们推荐使用 Zilliz Cloud，这是一个基于开源项目 Milvus构建的全托管向量数据库服务，并提供支持最多 100 万个向量的免费套餐。)

Google Vertex AI Gemini 2.0 Flash 优化建议

Gemini 2.0 Flash 专为超快速的 RAG 应用程序而构建，优化效率和速度至关重要。通过最小化检索文档的数量来提高检索效率，从而避免不必要的令牌消耗。简洁地构建提示，使用项目符号或编号列表来提高清晰度。将温度设定为 0.1–0.2 以获取事实性回应，微调 top-p 和 top-k 来控制变异性。对经常查询的主题实现响应缓存，以减少延迟。在负载下，利用 Google Cloud 的自动扩展和 GPU 加速来保持平稳性能。如果使用多个模型，利用 Flash 进行快速摘要和初步分析，然后再将查询升级到更大的模型。

Ollama mxbai-embed-large 优化建议

为了优化Ollama mxbai-embed-large组件在检索增强生成（RAG）设置中的表现，可以考虑在领域特定数据上对嵌入模型进行微调，以提升检索任务的相关性。利用批处理输入查询来提高吞吐量和效率，并实施缓存机制，对频繁访问的嵌入数据进行缓存以减少延迟。监控和分析性能指标，以识别瓶颈，并迭代调整超参数，如学习率和嵌入大小，同时利用混合精度训练来平衡准确性和资源利用。最后，定期用新数据更新嵌入数据库，以保持检索结果的准确性，确保您的RAG系统始终有效和响应迅速。

通过系统性实施这些优化方案，RAG 系统将在响应速度、结果准确率、资源利用率等维度获得全面提升。 AI 技术迭代迅速，建议定期进行压力测试与架构调优，持续跟踪最新优化方案，确保系统在技术发展中始终保持竞争优势。

RAG 成本计算器

估算 RAG 成本时，需要分析向量存储、计算资源和 API 使用等方面的开销。主要成本驱动因素包括向量数据库查询、嵌入生成和 LLM 推理。RAG 成本计算器是一款免费的在线工具，可快速估算构建 RAG 的费用，涵盖切块（chunking）、嵌入、向量存储/搜索和 LLM 生成。能帮助你发现节省费用的机会，最高可通过无服务器方案在向量存储成本上实现 10 倍降本。

立即使用 RAG 成本计算器

Calculate your RAG cost Calculate your RAG cost

收获与总结

通过深入这个教程，你已经解锁了从零开始构建 RAG 系统的魔力，使用前沿工具！你了解到 LangChain 如何作为粘合剂，将所有组件无缝连接在一起，协调数据流动。而 LangChain 向量存储则成为你快速文档存储和检索的首选，确保你的系统即使在大规模数据集中也能迅速找到相关信息。借助 Google Vertex AI Gemini 2.0 Flash 的强大功能，你看到如何生成精准、上下文敏感的响应，几乎感觉像是人类—同时保持高效和可扩展性。我们还不能忘记 Ollama 的 mxbai-embed-large 模型，它将原始文本转化为丰富的嵌入，使你的系统能够理解语义关系，从非结构化数据中提取有意义的模式。所有这些组件共同形成了一个动态管道：获取文档、嵌入文档、存储以便检索，并生成既准确又自然的答案。

但这还不是全部！你还学到了优化 RAG 系统的专业技巧——比如调整块大小以更好地保留上下文，或者在检索中平衡速度和准确性——甚至发现一个免费的 RAG 成本计算器可以帮助你在扩展时更智能地预算。现在你已经看到这些工具如何协作将数据转化为可操作的洞察，是时候掌控局面了。想象一下你可以构建的应用程序：更智能的聊天机器人、研究助理或定制知识中心。工具在你手中，可能性是无穷无尽的。快去尝试、迭代，突破 RAG 的界限。你的下一个突破距离只需几行代码。让我们一起构建一些了不起的东西！ 🚀

欢迎反馈！

我们很期待听到你的使用心得与建议！ 🌟 你可以：

  • 在下方留言；
  • 加入 Milvus Discord 社区，与全球 AI 爱好者一起交流。 如果你觉得本教程对你有帮助，别忘了给 Milvus GitHub 仓库点个 ⭐，这将激励我们不断创作！💖

大规模向量数据库

Zilliz Cloud 是一个专为大规模应用构建的全托管向量数据库，完美适配您的 RAG 应用。

免费试用 Zilliz Cloud
快速免费计算您的 RAG 成本

继续阅读

AI Assistant