LangChain数据连接器教程:如何优化大模型数据加载与文档转换

释放双眼,带上耳机,听听看~!
本教程将详细介绍LangChain的数据连接器,包括文档加载器和文档转换器的优化方法,帮助您更好地处理大模型数据并进行AI应用开发。

LangChain学习来到第三篇,如果您是刚入门LangChain,请先查阅前两篇。

LangChain 一 hello LLM – 掘金 (juejin.cn)

LangChain 二 模型 – 掘金 (juejin.cn)

前言

   上篇我们学习了LangChain的模型接口。主要分为两类,聊天场景的chat_models和接入各家大模型的langchain.llms,我们可以使用predict方法与模型沟通。

  另外, 上篇文章,我们还了解到聊天业务中的langchain.schema.messages中的三个消息类,后期我们在实战中,将与memory模块一起,完成RAG或AI聊天应用开发。现在,让我们进入Data Connections,看看LangChain相比LlamaIndex,是如何做数据连接的。我之前写过LlamaIndex 四 数据连接器 – 掘金 (juejin.cn),总结部分我们会做个对比。

Data Connections

  大模型有时并不包含我们需要的知识或数据,这是它有时会“瞎编”的原因。比如某些企业或单位,闭源的数据,或模型数据训练后新的数据,特别是当我们在做RAG应用时,我们可以为大模型连接一些数据,让它基于连接的数据,继续完成Completion或Chat工作。

相应模块

  将数据加载到大模型的工作流程,每个阶段LangChain都提供了相应的组件类。

  • 文档加载器

  负责加载各种格式的数据。LangChain提供了各种类型的加载器,本地、网络皆可加载。具体可见Document loaders | 🦜️🔗 Langchain

from langchain.document_loaders import TextLoader

loader = TextLoader("./sample_data/README.md")
docs = loader.load()

LangChain数据连接器教程:如何优化大模型数据加载与文档转换

  langchain通过document_loaders模块加载文档,这里引入了TextLoader, 加载本地系统的README.md文件,并返回一个loader对象。loader的load方法,加载文档,并返回一个document对象, 和LlamaIndex挺像的。

  • 文档转换器

  拆分文档,将文档转换为问答格式,去除冗余文档。接下来,我们做文档拆分:

from langchain.text_splitter import CharacterTextSplitter
text_splitter = CharacterTextSplitter(        
    separator = "nn",
    chunk_size = 1000,
    chunk_overlap  = 200,
    length_function = len,
)

split_docs = text_splitter.split_documents(docs)

  CharacterTextSplitter是最简单的拆分器,它将文档拆分为固定长度的文本块,方便等下做嵌入。这里将上面的Document进行分割,分割符为两个换行符,每1000个字符一份,chunk_overlap 的意思是前后重叠为100个,这样可以更好的弥补因为在分割的周围出错断片影响语音的问题。

  RecursiveCharacterTextSplitter还可以根据编程语言的特性,将代码分为合适的文本块。

from langchain.text_splitter import Language,RecursiveCharacterTextSplitter

PYTHON_CODE = """
def hello_langchain():
    print("Hello, Langchain!")

# Call the function
hello_langchain()
"""
python_splitter = RecursiveCharacterTextSplitter.from_language(
    language=Language.PYTHON, chunk_size=50, chunk_overlap=0
)
python_docs = python_splitter.create_documents([PYTHON_CODE])
python_docs

LangChain数据连接器教程:如何优化大模型数据加载与文档转换

   从上图可以看出在切割的时候做了语言分析,代码分割的很正确,这里使用的是RecursiveCharacterTextSplitter方法

from langchain.text_splitter import MarkdownHeaderTextSplitter

markdown_document = "# Chapter 1nn    ## Section 1nnHi this is the 1st sectionnnWelcomenn ### Module 1 nn Hi this is the first module nn ## Section 2nn Hi this is the 2nd section"

headers_to_split_on = [
    ("#", "Header 1"),
    ("##", "Header 2"),
    ("###", "Header 3"),
]

splitter = MarkdownHeaderTextSplitter(headers_to_split_on=headers_to_split_on)
splits = splitter.split_text(markdown_document)

   MarkdownHeaderTextSplitter 可以将Markdown文档按照段落结构,基于Markdown语法进行文档分块。

  • 文本嵌入模型

  将非结构化文本转换为向量方式,这是适合大模型计算的方式。比如,我们调用OpenAI embedding接口。

LangChain数据连接器教程:如何优化大模型数据加载与文档转换

LangChain数据连接器教程:如何优化大模型数据加载与文档转换

  • 向量存储

  将上一步转化的向量,存入chromaDB或啥,并支持向量检索

pip install -q chromadb

from langchain.document_loaders import TextLoader
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.text_splitter import CharacterTextSplitter
from langchain.vectorstores import Chroma

text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
documents = text_splitter.split_documents(docs)
db = Chroma.from_documents(documents, OpenAIEmbeddings())

  • 检索器

  提供数据查询的通用接口。

向量数据库提供的重要接口就是相似性查询。如上述内容提到,文本相似性的衡量,由文本的向量表示的欧几里得距离来衡量。向量数据库提供了该接口,用于查询与给定文本最相似的文本。

query = "什么是WTF Langchain?"
docs = db.similarity_search(query)
print(docs[0].page_content)

实战

  • 了解了加载器、分割器
  • 了解了基于OpenAI的embedding 向量数据库存储和检索的方法。

参考资料

本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
AI教程

实时人工智能应用:ClipDrop、Fal.ai和Leonardo AI的革命性技术

2023-12-19 16:58:00

AI教程

大语言模型角色扮演:深度学习科学家在Nature发表论文

2023-12-19 17:00:00

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索