Dividindo dados externos para recuperação

Desenvolvimento de aplicativos de LLM com LangChain

Jonathan Bennion

AI Engineer & LangChain Contributor

Etapas de desenvolvimento de RAG

O fluxo geral de RAG: um carregador de documentos, um divisor de documentos e o processo de armazenamento e recuperação.

Divisão de documentos: dividir em chunks
Quebrar documentos para caber na janela de contexto da LLM

Pensando na divisão...

O primeiro parágrafo da introdução do artigo Attention is All You Need.

Linha 1:

Recurrent neural networks, long short-term memory [13] and gated recurrent [7] neural networks

Linha 2:

in particular, have been firmly established as state of the art approaches in sequence modeling and

¹ https://arxiv.org/abs/1706.03762

Sobreposição de chunks

O primeiro parágrafo da introdução do artigo Attention is All You Need dividido em dois chunks com sobreposição.

Qual é a melhor estratégia de divisão?

A palavra "context" dividida em letras individuais.

CharacterTextSplitter
RecursiveCharacterTextSplitter
Muitos outros

¹ Wikipedia Commons

quote = '''One machine can do the work of fifty ordinary humans.\nNo machine can do
the work of one extraordinary human.'''

len(quote)

chunk_size = 24
chunk_overlap = 3

¹ Elbert Hubbard

from langchain_text_splitters import CharacterTextSplitter


ct_splitter = CharacterTextSplitter(
    separator='.',
    chunk_size=chunk_size,
    chunk_overlap=chunk_overlap)


docs = ct_splitter.split_text(quote) 
print(docs)

print([len(doc) for doc in docs])

['One machine can do the work of fifty ordinary humans',
 'No machine can do the work of one extraordinary human']

[52, 53]

Divide no separador para ficar < chunk_size, mas pode falhar!

from langchain_text_splitters import RecursiveCharacterTextSplitter


rc_splitter = RecursiveCharacterTextSplitter(
    separators=["\n\n", "\n", " ", ""],
    chunk_size=chunk_size,
    chunk_overlap=chunk_overlap)


docs = rc_splitter.split_text(quote) 
print(docs)

RecursiveCharacterTextSplitter

separators=["\n\n", "\n", " ", ""]

['One machine can do the',
 'work of fifty ordinary',
 'humans.',
 'No machine can do the',
 'work of one',
 'extraordinary human.']

Tenta dividir por parágrafo: "\n\n"
Tenta dividir por frase: "\n"
Tenta dividir por palavras: " "

RecursiveCharacterTextSplitter com HTML

from langchain_community.document_loaders import UnstructuredHTMLLoader 
from langchain_text_splitters import RecursiveCharacterTextSplitter


loader = UnstructuredHTMLLoader("white_house_executive_order_nov_2023.html")
data = loader.load()


rc_splitter = RecursiveCharacterTextSplitter(
    chunk_size=chunk_size,
    chunk_overlap=chunk_overlap,
    separators=['.'])


docs = rc_splitter.split_documents(data) 
print(docs[0])

Document(page_content="To search this site, enter a search term [...]

Vamos praticar!

Desenvolvimento de aplicativos de LLM com LangChain