【LangChain v0.2入門】第3回:チャットボットの作り方

# 必要なライブラリをインストール
!pip install langchain
!pip install langchain-openai
!pip install langchain_community

import os
from google.colab import userdata

# 環境変数の準備
os.environ["OPENAI_API_KEY"] = userdata.get("OPENAI_API_KEY")

from langchain_openai import ChatOpenAI

# LLMの準備
llm = ChatOpenAI(model="gpt-4o")

from langchain_core.messages import HumanMessage

# 出力パーサーの準備
parser = StrOutputParser()

# 会話
result = llm.invoke([HumanMessage(content="こんにちは!私の名前はつくもちです。")])
parser.invoke(result)

こんにちは、つくもちさん！お会いできて嬉しいです。今日はどんなお手伝いができますか？

# 過去の会話内容についての質問
result = llm.invoke([HumanMessage(content="私の名前は？")])
parser.invoke(result)

ごめんなさい、あなたの名前はわかりません。お手数ですが、教えていただけますか？

from langchain_core.messages import AIMessage

# 会話履歴付きで過去の会話内容についての質問
result = llm.invoke(
    [
        HumanMessage(content="こんにちは! 私の名前はつくもちです。"),
        AIMessage(content="こんにちは、つくもちさん！お会いできて嬉しいです。今日はどんなお手伝いができますか？"),
        HumanMessage(content="私の名前は？"),
    ]
)

parser.invoke(result)

あなたの名前は「つくもち」さんですね。どうぞよろしくお願いします！今日は何か特別なことがありますか？

from langchain_community.chat_message_histories import ChatMessageHistory
from langchain_core.chat_history import BaseChatMessageHistory
from langchain_core.runnables.history import RunnableWithMessageHistory

# 会話履歴の保存場所の準備
store = {}

# セッション毎の会話履歴の取得
def get_session_history(session_id: str) -> BaseChatMessageHistory:
    if session_id not in store:
        store[session_id] = ChatMessageHistory()
    return store[session_id]

# 会話履歴付きLLMの準備
message_history = RunnableWithMessageHistory(llm, get_session_history)

# コンフィグの準備
config = {"configurable": {"session_id": "abc1"}}

# 過去の会話内容についての質問
response = message_history.invoke(
    [HumanMessage(content="私の名前は？")],
    config=config,
)
response.content

あなたの名前はつくもちさんですね。何か他にお手伝いできることがあれば教えてください！

# 過去の会話内容についての質問
response = llm_with_history.invoke(
    [HumanMessage(content="私の名前は？")],
    config=config,
)
response.content

あなたの名前は「つくもち」さんですね。何か他に知りたいことやお話したいことがあれば、ぜひ教えてください！

from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder

prompt = ChatPromptTemplate.from_messages(
    [
        ("system", "あなたは親切なアシスタントです。できる限りすべての質問に答えてください。"),
        MessagesPlaceholder(variable_name="messages"),
    ]
)

chain = prompt | llm

response = chain.invoke(
    {"messages": [HumanMessage(content="こんにちは！私はボブです")], "language": "日本語"}
)

response.content

こんにちは、ボブさん！お会いできて嬉しいです。今日はどんなことについてお話ししますか？

# 会話履歴付きLLMの準備
message_history = RunnableWithMessageHistory(
    chain,
    get_session_history,
    input_messages_key="messages",
)

# コンフィグの準備
config = {"configurable": {"session_id": "abc2"}}

# 会話（1回目）
response = message_history.invoke(
    {"messages": [HumanMessage(content="こんにちは！ 私の名前はつくもちです。")], "language": "日本語"},
    config=config,
)
response.content

こんにちは、つくもちさん！今日はどのようなお手伝いをしましょうか？

# 会話（2回目）
response = message_history.invoke(
    {"messages": [HumanMessage(content="私の名前は？")], "language": "日本語"},
    config=config,
)
response.content

あなたの名前は「つくもち」さんですね。どうぞよろしくお願いします！

from langchain_core.runnables import RunnablePassthrough

# メッセージフィルタ (最近のk個のメッセージ)
def filter_messages(messages, k=4):
    return messages[-k:]

# チェーンの準備
chain = (
    RunnablePassthrough.assign(messages=lambda x: filter_messages(x["messages"]))
    | prompt
    | llm
)

# 会話履歴付きLLMの準備
message_history = RunnableWithMessageHistory(
    chain,
    get_session_history,
    input_messages_key="messages",
)

# コンフィグの準備
config = {"configurable": {"session_id": "abc3"}}

# 会話
response = message_history.invoke(
    {"messages": [HumanMessage(content="こんにちは！ 私の名前はつくもちです。")], "language": "日本語"},
    config=config,
)
response.content

こんにちは、つくもちさん！お知らせいただきありがとうございます。今日はどのようなことでお手伝いできますか？

# 過去の会話内容についての質問
response = message_history.invoke(
    {"messages": [HumanMessage(content="私の名前は？")], "language": "日本語"},
    config=config,
)
response.content

あなたの名前はつくもちさんですね。ほかに何かお手伝いできることはありますか？

# 過去の会話内容についての質問
response = message_history.invoke(
    {"messages": [HumanMessage(content="私の名前は？")], "language": "日本語"},
    config=config,
)
response.content

申し訳ありませんが、以前のメッセージの履歴は保持されていません。そのため、あなたの名前はわかりません。お名前を教えていただけますか？