StreamlitでChatGPTのUIを作る—LangChain ReActエージェントとGoogle検索の組み合わせ

質問: 「艦これが流行っていた頃のアニメは？」

Thought: 艦これのサービス開始時期を調べる必要がある
Action: Search Google("艦これ サービス開始 2013")
Observation: 「艦隊これくしょんは2013年4月にサービス開始」

Thought: 2013年〜2015年頃のアニメを調べる
Action: Search Wikipedia("2013年アニメ一覧")
Observation: ...

Final Answer: 艦これが流行っていた2013〜2015年頃のアニメは...（関西弁で回答）

from langchain.llms import OpenAIChat
from langchain import PromptTemplate, LLMChain
from langchain.agents import ZeroShotAgent, Tool, AgentExecutor
from langchain.utilities import GoogleSearchAPIWrapper
from langchain.agents.react.base import DocstoreExplorer
from langchain import Wikipedia

# LLM の設定
llm = OpenAIChat(temperature=0.5)

# ツールの定義
search = GoogleSearchAPIWrapper()
docstore = DocstoreExplorer(Wikipedia())

tools = [
    Tool(
        name="Search Google",
        func=search.run,
        description="最新情報や現在の出来事について調べるときに使う",
    ),
    Tool(
        name="Search Wikipedia",
        func=docstore.search,
        description="歴史的な出来事や詳細な事典的情報を調べるときに使う",
    ),
]

prefix = """Answer the following questions as best you can.
You think step by step in English.
Finally answer in Japanese.
You have access to the following tools:"""

suffix = """Begin!
最終的な答えは日本語で答えてください。
質問に答える時には根拠も述べてください。
不足する情報がある場合は質問してください。
関西弁で答えてください。

Question: {input}
{agent_scratchpad}"""

prompt = ZeroShotAgent.create_prompt(
    tools,
    prefix=prefix,
    suffix=suffix,
    input_variables=["input", "agent_scratchpad"],
)

import streamlit as st
from streamlit_chat import message

st.markdown("## chatGPT")
st.markdown("1. 色々なことを質問してみてください")

# チャット履歴の表示
if "generated" not in st.session_state:
    st.session_state["generated"] = []
if "past" not in st.session_state:
    st.session_state["past"] = []

# 入力フォーム
user_input = st.text_input("メッセージを入力してください")

if user_input:
    output = agent_executor.run(user_input)

    st.session_state.past.append(user_input)
    st.session_state.generated.append(output)

# メッセージ表示（新しいものが上に来るよう逆順で表示）
if st.session_state["generated"]:
    for i in range(len(st.session_state["generated"]) - 1, -1, -1):
        message(st.session_state["generated"][i], key=str(i))
        message(st.session_state["past"][i], is_user=True, key=f"{i}_user")

from langchain.chat_models import ChatOpenAI
from langchain.schema import HumanMessage, SystemMessage

chat = ChatOpenAI(temperature=0)

messages = [
    SystemMessage(content="あなたは役立つアシスタントです。日本語で答えてください。"),
    HumanMessage(content=user_input),
]

response = chat(messages)
print(response.content)

export OPENAI_API_KEY="sk-..."
export GOOGLE_API_KEY="your-google-api-key"
export GOOGLE_CSE_ID="your-custom-search-engine-id"  # カスタム検索エンジンID

# エージェント版（Google検索 + Wikipedia）
poetry run streamlit run chatgpt_streamlit/streamlit_run.py

# シンプルチャット版
poetry run streamlit run chatgpt_streamlit/streamlit_run_chat.py