LLM連携アプリの開発を支援するライブラリ LangChain の使い方 (1) - LLMとプロンプト・チェーン

# パッケージのインストール
!pip install langchain
!pip install openai

# 環境変数の準備
import os
os.environ["OPENAI_API_KEY"] = "<OpenAI_APIのトークン>"

from langchain.llms import OpenAI

# LLMの準備
llm = OpenAI(temperature=0.9)

# LLMの呼び出し
print(llm("コンピュータゲームを作る新会社名の案を1つ出してください"))

ツクルゲームス

from langchain.prompts import PromptTemplate
from langchain.chains import LLMChain

# プロンプトテンプレートの準備
prompt = PromptTemplate(
    input_variables=["product"],
    template="{product}を作る新会社名の案を1つ出してください",
)

# LLMChainの準備
chain = LLMChain(llm=llm, prompt=prompt)

# LLMChainの実行
print(chain.run("家庭用ロボット"))

ポケットロボットカンパニー

# パッケージのインポート
from langchain.llms import OpenAI
from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate

# 1つ目のチェーンの準備 : 劇のタイトルからあらすじを生成

# LLMの準備
llm = OpenAI(temperature=.7)

# テンプレートの準備
template = """あなたは劇作家です。劇のタイトルが与えられた場合、そのタイトルのあらすじを書くのがあなたの仕事です。

タイトル:{title}
あらすじ:"""

# プロンプトテンプレートの準備
prompt_template = PromptTemplate(
    input_variables=["title"], 
    template=template
)

# LLMChainの準備
synopsis_chain = LLMChain(llm=llm, prompt=prompt_template)

# 2つ目のチェーン : 劇のあらすじからレビューを生成

# LLMの準備準備
llm = OpenAI(temperature=.7)

# テンプレートの準備
template = """あなたは演劇評論家です。 劇のあらすじが与えられた場合、そのあらすじのレビューを書くのがあなたの仕事です。

あらすじ:
{synopsis}
レビュー:"""

# プロンプトテンプレートの準備
prompt_template = PromptTemplate(
    input_variables=["synopsis"], 
    template=template
)

# LLMChainの準備
review_chain = LLMChain(llm=llm, prompt=prompt_template)

from langchain.chains import SimpleSequentialChain

# SimpleSequentialChainで2つのチェーンを繋ぐ
overall_chain = SimpleSequentialChain(
    chains=[synopsis_chain, review_chain], 
    verbose=True
)

# SimpleSequentialChainの実行
review = overall_chain.run("浜辺の夕暮れ時の悲劇")

> Entering new SimpleSequentialChain chain...

浜辺の街で暮らす家族がいます。末っ子のジョエルは、詩を書いたり音楽を作ったりするような夢を持っていましたが、家族からは実家を継ぐことを強く望まれていました。しかし、ジョエルはそんな夢を捨てきれずにいました。

ある夕暮れ時、ジョエルは家族の猛反対を押し切り、自身の夢に向かって大きな一歩を踏み出そうとしましたが、未だ夢を叶えることができず、最後には彼は家族の悲劇を


「浜辺の街」は、一家の悲劇を描いた心温まるドラマです。ジョエルは、詩や音楽に関する夢を抱いていますが、家族は実家を継ぐよう望んでいます。ジョエルは夢を捨てきれずにいますが、家族の猛反対を押し切り、自身の夢に向かうところまでは来ました。しかし、最後に彼は夢を叶えることができず、家族の悲劇が終わるまでずっと心痛む状態にあります。 本作は、家族の愛情を描いた

> Finished SimpleSequentialChain chain.

# レビューの表示
print(review)

「浜辺の街」は、一家の悲劇を描いた心温まるドラマです。ジョエルは、詩や音楽に関する夢を抱いていますが、家族は実家を継ぐよう望んでいます。ジョエルは夢を捨てきれずにいますが、家族の猛反対を押し切り、自身の夢に向かうところまでは来ました。しかし、最後に彼は夢を叶えることができず、家族の悲劇が終わるまでずっと心痛む状態にあります。 本作は、家族の愛情を描いた

# 1つ目のチェーン : 劇のタイトルと時代からあらすじを生成

# LLMの準備
llm = OpenAI(temperature=.7)

# テンプレートの準備
# テンプレートの準備
template = """あなたは劇作家です。劇のタイトルと時代が与えられた場合、そのタイトルのあらすじを書くのがあなたの仕事です。

タイトル:{title}
時代:{era}
あらすじ:"""

# プロンプトテンプレートの生成
prompt_template = PromptTemplate(
    input_variables=["title", 'era'], 
    template=template
)

# LLMChainの準備
synopsis_chain = LLMChain(
    llm=llm, 
    prompt=prompt_template, 
    output_key="synopsis"
)

# 2つ目のチェーン : 劇のあらすじからレビューを生成

# LLMの準備準備
llm = OpenAI(temperature=.7)

# テンプレートの準備
template = """あなたは演劇評論家です。 劇のあらすじが与えられた場合、そのあらすじのレビューを書くのがあなたの仕事です。

あらすじ:
{synopsis}
レビュー:"""

# プロンプトテンプレートの準備
prompt_template = PromptTemplate(
    input_variables=["synopsis"], 
    template=template
)

# LLMChainの準備
review_chain = LLMChain(
    llm=llm, prompt=prompt_template, 
    output_key="review"

from langchain.chains import SequentialChain

# SequentialChainで2つのチェーンを繋ぐ
overall_chain = SequentialChain(
    chains=[synopsis_chain, review_chain],
    input_variables=["era", "title"],
    output_variables=["synopsis", "review"],
    verbose=True
)

# SequentialChainの実行
review = overall_chain({"title":"浜辺の夕暮れ時の悲劇", "era": "戦国時代"})

> Entering new SequentialChain chain...
Chain 0:
{'synopsis': '\n\nこの物語は戦国時代の日本を舞台にしています。ある小さな集落が存在し、そこには若い男女が住んでいました。\n\n主人公の一人は、少し変わった性格である少年のジョンです。ジョンは毎日のように浜辺を散歩していました。ある日、ジョンは浜辺で若い女性と出会いました。その女性は、武士であり、彼女の気高い態度や誠実な気持ちに心を奪われました。\n\nしかし、ジョンは彼女の名前を知り'}

Chain 1:
{'review': '\n\nこの物語は戦国時代の日本を舞台にした、素晴らしいストーリーです。主人公のジョンが、浜辺で若い女性と出会うことで、時代の背景を見事に表現しています。ジョンは、彼女の気高い態度や誠実な気持ちに心を奪われます。これは、女性の持つ強さと柔らかさを象徴しています。物語を通して、多くの人々が、自分の内なる力を信じることを学べるでしょう。戦国時代を舞台にしたこの物語'}


> Finished SequentialChain chain.

# レビューの表示
print(review)

{'review': '\n\nこの物語は戦国時代の日本を舞台にした、素晴らしいストーリーです。主人公のジョンが、浜辺で若い女性と出会うことで、時代の背景を見事に表現しています。ジョンは、彼女の気高い態度や誠実な気持ちに心を奪われます。これは、女性の持つ強さと柔らかさを象徴しています。物語を通して、多くの人々が、自分の内なる力を信じることを学べるでしょう。戦国時代を舞台にしたこの物語'}