news 2026/7/10 3:25:21

RAG 落地实战——从文档解析到向量检索到重排序的代码示例

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
RAG 落地实战——从文档解析到向量检索到重排序的代码示例

要构建一个工业级标准的 RAG(检索增强生成)系统,核心链路通常遵循:文档加载 → 文本分块 → 向量检索(粗排) → 重排序(精排) → LLM 生成的流程。下面我将为你提供一个基于 Python 和 LangChain 框架的完整实战代码示例。

核心环境准备与基础配置

首先,我们需要加载必要的库,并配置文本分块和向量检索的参数。合理的分块策略是保证检索质量的第一步。

importosimporttorchfromdotenvimportload_dotenvfromlangchain.text_splitterimportRecursiveCharacterTextSplitterfromlangchain_community.document_loadersimportTextLoaderfromlangchain_community.vectorstoresimportFAISSfromlangchain_community.embeddingsimportHuggingFaceBgeEmbeddings### 加载环境变量load_dotenv()### 全局配置DOC_PATH="./long_document.txt"FAISS_PATH="./faiss_rag_rerank_db"分块配置(滑动窗口),保证上下文连贯 CHUNK_SIZE=800CHUNK_OVERLAP=300检索配置 TOP_K_COARSE=20# 粗检索召回20条TOP_K_FINE=5# 重排序后保留5条重排序模型 RERANK_MODEL="BAAI/bge-reranker-v2-m3"

文档解析与向量库构建

将原始文档进行分块,并使用 Embedding 模型将其转化为向量存入 FAISS 向量数据库中。

defbuild_vector_db():loader=TextLoader(DOC_PATH,encoding="utf-8")docs=loader.load()# 文本分块splitter=RecursiveCharacterTextSplitter(chunk_size=CHUNK_SIZE,chunk_overlap=CHUNK_OVERLAP)chunks=splitter.split_documents(docs)# 加载 BGE 嵌入模型并构建向量库embedding=HuggingFaceBgeEmbeddings(model_name="BAAI/bge-small-zh-v1.5",model_kwargs={"device":"cuda"iftorch.cuda.is_available()else"cpu"},encode_kwargs={"normalize_embeddings":True})db=FAISS.from_documents(chunks,embedding)db.save_local(FAISS_PATH)print("✅ 向量库构建完成")returndb

多路召回与重排序(核心优化)

为了提升检索精度,生产环境中通常会采用“混合检索”(向量检索 + BM25 关键词检索),随后引入重排序模型(如 BGE-Reranker)对候选结果进行精细打分。

fromlangchain.retrieversimportBM25Retriever,EnsembleRetrieverfromlangchain_community.cross_encodersimportHuggingFaceCrossEncoderfromlangchain.retrieversimportContextualCompressionRetrieverfromlangchain.retrievers.document_compressorsimportCrossEncoderRerankdefget_reranked_retriever(split_docs):# 1. 初始化 BM25 稀疏召回bm25_retriever=BM25Retriever.from_documents(split_docs)bm25_retriever.k=10# 2. 初始化稠密向量召回vector_retriever=vector_db.as_retriever(search_kwargs={"k":10})# 3. 融合两路召回结果(混合检索)ensemble_retriever=EnsembleRetriever(retrievers=[bm25_retriever,vector_retriever],weights=[0.4,0.6])# 4. 加载 BGE 重排序模型,筛选最相关的 Top-Nrerank_model=HuggingFaceCrossEncoder(model_name=RERANK_MODEL)compressor=CrossEncoderRerank(model=rerank_model,top_n=TOP_K_FINE)# 5. 组合成带重排序的检索器compression_retriever=ContextualCompressionRetriever(base_retriever=ensemble_retriever,base_compressor=compressor)returncompression_retriever

LLM 生成与防幻觉控制

最后,将重排序后的高质量上下文与用户问题结合,通过精心设计的 Prompt 让大模型生成答案,并严格限制其“幻觉”。

fromlangchain.promptsimportPromptTemplatefromlangchain_openaiimportChatOpenAIdefgenerate_answer(query,context_docs):# 私有知识库专属 Prompt 模板template="""你是一个专业的企业内部助手,你需要根据提供的上下文信息回答用户的问题。 如果你不知道答案,就直接说你不知道,绝对不要编造答案。 上下文信息如下: {context} 用户问题: {question} 回答:"""prompt=PromptTemplate.from_template(template)# 拼接上下文context="nn".join([doc.page_contentfordocincontext_docs])formatted_prompt=prompt.format(context=context,question=query)# 调用大模型生成(设置低 temperature 降低随机性)llm=ChatOpenAI(model="gpt-3.5-turbo",temperature=0.1)response=llm.invoke(formatted_prompt)returnresponse.content

💡 生产环境避坑与优化建议防幻觉与溯源:在 Prompt 中必须强调“只根据资料回答”,并在返回答案时附带来源文档信息(如来源:《员工报销制度》第 2 条),这能大幅提升用户信任度并便于排查错误。。
缓存策略:对于“营业时间”、“报销流程”等高频问题,建议使用 Redis 等对检索结果或最终答案进行缓存,以降低响应延迟和 API 调用成本。
模型选择:重排序模型(Reranker)是提升准确率的关键。实测中,BGE-Reranker-v2-m3 的效果通常优于通用的 Cross-Encoder,生产环境建议优先采用。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/10 3:22:52

你的微博记忆正在悄悄消失?这款免费工具帮你永久保存珍贵瞬间

你的微博记忆正在悄悄消失?这款免费工具帮你永久保存珍贵瞬间 【免费下载链接】Speechless 把新浪微博的内容,导出成 PDF 文件进行备份的 Chrome Extension。 项目地址: https://gitcode.com/gh_mirrors/sp/Speechless 你是否曾有过这样的经历&am…

作者头像 李华
网站建设 2026/7/10 3:22:41

中兴光猫工厂模式深度解析与专业解锁技术指南

中兴光猫工厂模式深度解析与专业解锁技术指南 【免费下载链接】zteOnu A tool that can open ZTE onu device factory mode 项目地址: https://gitcode.com/gh_mirrors/zt/zteOnu 你是否曾因中兴光猫的功能限制而无法进行深度网络调试?是否渴望获得设备的完整…

作者头像 李华
网站建设 2026/7/10 3:21:12

STM32与TS2007FC构建高性能嵌入式音频系统

1. 项目背景与硬件选型解析在嵌入式音频系统开发领域,如何选择合适的功放芯片与主控平台往往决定了最终产品的音质表现和功能上限。TS2007FC作为意法半导体推出的3W无滤波D类音频功率放大器,与STM32F437ZG这款高性能ARM Cortex-M4微控制器的组合&#xf…

作者头像 李华
网站建设 2026/7/10 3:21:07

macOS安装机制深度解析:签名、公证与权限模型

1. 项目概述:这不是一句简单的标题,而是一道高频、高痛、高隐蔽性的 macOS 系统入场关卡“Installation (macOS)”——短短两个英文单词,出现在 GitHub 仓库 README 里、开源工具文档首页、开发者论坛的求助帖中,甚至是你双击下载…

作者头像 李华
网站建设 2026/7/10 3:21:03

怎么选中式庭院火锅店不踩坑?内行5条选购标准+实测推荐

一、行业痛点:普通人吃庭院火锅,极易踩的4个坑近两年中式庭院火锅爆火,区别于密闭商圈火锅店,凭借古风场景、露天烟火气收割亲子、商务、年轻食客,但大量网红门店重装修、轻内核,普通消费者就餐极易翻车&am…

作者头像 李华