当前位置：首页 > news >正文

利用 LangChain 构建对话式 AI 应用

news 2025/7/8 16:01:06

随着人工智能技术的快速发展，对话式 AI 已成为现代应用的核心部分。在构建智能客服、虚拟助手以及交互式学习平台时，一个强大且灵活的框架显得尤为重要。本文将深度解析 LangChain 这一框架的功能及实际使用，帮助开发者快速上手。

什么是 LangChain？

LangChain 是一个开源的 Python 和 JavaScript 库，专注于构建由大型语言模型 (LLM) 驱动的应用程序。它提供了强大的工具来管理复杂的提示链条、持久化用户上下文以及与外部数据源交互。

LangChain 的核心功能包括：

Prompt 模板：灵活地管理多级提示。
Chains：串联任务以实现复杂功能。
Memory：支持对话上下文的记忆。
连接器：与数据库、API 和文件系统无缝集成。

接下来，我们将通过一个具体示例演示如何使用 LangChain 构建一个对话式 AI 应用。

案例构建：知识问答机器人

我们将创建一个知识问答机器人，能够根据用户的问题，实时检索相关文档并生成回答。

环境准备

安装必要库：

pip install langchain openai faiss-cpu tiktoken

获取 OpenAI 的 API 密钥：OpenAI API
准备一些示例数据，例如一个 PDF 文件，作为机器人回答问题的知识来源。

步骤 1：数据预处理

首先，我们需要将知识源（PDF 文件）转换为 LangChain 可处理的文档格式：

from langchain.document_loaders import PyPDFLoader
from langchain.text_splitter import RecursiveCharacterTextSplitter# 加载 PDF 文件
loader = PyPDFLoader("sample_document.pdf")
documents = loader.load()# 将文本拆分成小块
txt_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
split_docs = txt_splitter.split_documents(documents)

步骤 2：构建知识索引

为了快速检索答案，我们可以使用 FAISS 创建向量化搜索索引：

from langchain.vectorstores import FAISS
from langchain.embeddings.openai import OpenAIEmbeddings# 将文档向量化
embeddings = OpenAIEmbeddings()
vectorstore = FAISS.from_documents(split_docs, embeddings)# 保存索引以便后续使用
vectorstore.save_local("faiss_index")

步骤 3：定义对话逻辑

接下来，使用 LangChain 的 RetrievalQA 模块定义机器人如何从索引中检索并生成答案：

from langchain.chains import RetrievalQA
from langchain.llms import OpenAI# 加载已保存的索引
vectorstore = FAISS.load_local("faiss_index", embeddings)# 创建 LLM 和 QA Chain
llm = OpenAI(model="gpt-4", temperature=0.5)
qa_chain = RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever())# 测试问答逻辑
query = "什么是 LangChain？"
response = qa_chain.run(query)
print(response)

步骤 4：添加记忆功能

为了让机器人 "记住" 用户的上下文，可以结合 Memory 模块实现：

from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory# 初始化记忆模块
memory = ConversationBufferMemory()
conversation = ConversationChain(llm=llm, memory=memory)# 模拟多轮对话
print(conversation.run("告诉我关于LangChain的用途。"))
print(conversation.run("它支持哪些集成功能？"))

步骤 5：部署 API 服务

最后，我们可以通过 FastAPI 将这个知识问答机器人部署为一个在线服务：

from fastapi import FastAPI, Requestapp = FastAPI()@app.post("/chat")
async def chat(request: Request):data = await request.json()user_input = data["input"]response = conversation.run(user_input)return {"response": response}if __name__ == "__main__":import uvicornuvicorn.run(app, host="0.0.0.0", port=8000)