LCEL検索チェーンの構築

LangChainで学ぶ検索拡張生成（RAG）

Meri Nova

Machine Learning Engineer

検索用データの準備

ドキュメントチャンクが保存される。

LangChainで学ぶ検索拡張生成（RAG）

RAGのためのLCELとは

ユーザー入力。

LangChainで学ぶ検索拡張生成（RAG）

RAGのためのLCELとは

ユーザー入力がRunnablePassthroughを通じてチェーンに入り、「question」に割り当てられる様子。

LangChainで学ぶ検索拡張生成（RAG）

RAGのためのLCELとは

入力が検索システムにクエリとして使用され、「context」として割り当てられる様子。

LangChainで学ぶ検索拡張生成（RAG）

RAGのためのLCELとは

「context」と「question」がプロンプトテンプレートに統合されている様子。

LangChainで学ぶ検索拡張生成（RAG）

RAGのためのLCELとは

プロンプトがLLM（モデル）に統合され、出力が生成される様子。

LangChainで学ぶ検索拡張生成（RAG）

RAGのためのLCELとは

モデル出力がパーサーに渡される様子。

LangChainで学ぶ検索拡張生成（RAG）

リトリーバーのインスタンス化

vector_store = Chroma.from_documents(
    documents=chunks, 
    embedding=embedding_model
)


retriever = vector_store.as_retriever(
    search_type="similarity",
    search_kwargs={"k": 2}
)
LangChainで学ぶ検索拡張生成（RAG）

プロンプトテンプレートの作成

from langchain_core.prompts import ChatPromptTemplate


prompt = ChatPromptTemplate.from_template("""
Use the following pieces of context to answer the question at the end.
If you don't know the answer, say that you don't know.
Context: {context}
Question: {question}
""")

llm = ChatOpenAI(model="gpt-4o-mini", api_key="...", temperature=0)
LangChainで学ぶ検索拡張生成（RAG）

LCEL検索チェーンの構築

from langchain_core.runnables import RunnablePassthrough
from langchain_core.output_parsers import StrOutputParser


chain = (

    {"context": retriever, "question": RunnablePassthrough()}

    | prompt

    | llm

    | StrOutputParser()

)
LangChainで学ぶ検索拡張生成（RAG）

検索チェーンの実行

result = chain.invoke({"question": "What are the key findings or results presented in the paper?"})
print(result)

- Top Performance: RAG models set new records on open-domain question answering tasks...
- Better Generation: RAG models produce more specific, diverse, and factual language...
- Dynamic Knowledge Use: The non-parametric memory allows RAG models to access and ...
LangChainで学ぶ検索拡張生成（RAG）

お疲れさまでした！

LangChainで学ぶ検索拡張生成（RAG）

Preparing Video For Download...