LLM連携アプリの開発を支援するライブラリ LangChain の使い方 (3) - エージェント

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

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

# パッケージのインポート
from langchain.agents import load_tools
from langchain.agents import initialize_agent
from langchain.llms import OpenAI

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

# ツールの準備
tools = load_tools(["serpapi", "llm-math"], llm=llm)

# エージェントの準備
agent = initialize_agent(
    tools, 
    llm, 
    agent="zero-shot-react-description", 
    verbose=True
)

# エージェントの実行
agent.run("富士山の高さに2を掛けると？")

> Entering new AgentExecutor chain...
 I need to calculate the result of multiplying the height of Mt. Fuji by 2.
Action: Calculator
Action Input: 3776 * 2
Observation: Answer: 7552
Thought: I now know the final answer
Final Answer: 富士山の高さを2倍すると7552mになります。
> Finished AgentExecutor chain.
'富士山の高さを2倍すると7552mになります。

# エージェントの実行
agent.run("ぼっち・ざ・ろっくのぼっちちゃんの本名は？")

> Entering new AgentExecutor chain...
 I need to find out the real name of the character.
Action: Search
Action Input: "Bocchi the Rock's real name"
Observation: (TV) Plot Summary: Hitori "Bocchi-chan" ("Loner") Gotō is a lonely high school girl whose heart lies in her guitar. She does nothing every day except strum her guitar by herself at home.
Thought: This doesn't answer my question.
Action: Search
Action Input: "Bocchi the Rock real name"
Observation: Bocchi the Rock! is written and illustrated by Aki Hamaji. It was initially serialized on Houbunsha's Manga Time Kirara Max magazine on December 19, 2017, as a guest work. A full serialization started in the same magazine on March 19, 2018. It has been collected into five tankōbon volumes as of November 26, 2022.
Thought: This still doesn't answer my question.
Action: Search
Action Input: "Bocchi the Rock character real name"
Observation: Gotou, also known as "Bocchi-chan," is a lonely girl who loves the guitar.
Thought: This is the answer I'm looking for.
Final Answer: Gotou
> Finished AgentExecutor chain.
'Gotou'