【Streamlit】LangChainとGPT-4.5-preview を使ったチャットアプリ構築

import streamlit as st
from langchain_community.chat_models import ChatOpenAI
from langchain.schema import HumanMessage, AIMessage, SystemMessage
import os

# OpenAI クライアントを生成
chat = ChatOpenAI(
    model="gpt-4.5-preview",  # GPT-4.5-preview モデルを指定
    temperature=0.7,           # 応答のランダム性を調整
    streaming=True,            # ストリーミング応答を有効化
    openai_api_key=os.getenv("OPENAI_API_KEY")  # 環境変数から API キーを取得
)

# セッションステートを初期化
if "messages" not in st.session_state:
    st.session_state.messages = [SystemMessage(content="You are a helpful assistant.")]

# タイトルを表示
st.title("GPT-4.5-preview チャット")

# メッセージ履歴を表示
for message in st.session_state.messages:
    if isinstance(message, HumanMessage):
        with st.chat_message("user"):
            st.markdown(message.content)
    elif isinstance(message, AIMessage):
        with st.chat_message("assistant"):
            st.markdown(message.content)

# ユーザー入力を処理
if prompt := st.chat_input("What is up?"):
    user_message = HumanMessage(content=prompt)
    st.session_state.messages.append(user_message)

    with st.chat_message("user"):
        st.markdown(prompt)

    # AI 応答を生成
    with st.chat_message("assistant"):
        full_response = ""
        placeholder = st.empty()

        for chunk in chat.stream(st.session_state.messages):
            if chunk.content:
                full_response += chunk.content
                placeholder.markdown(full_response + "▌")

        placeholder.markdown(full_response)

    # AI のメッセージを履歴に追加
    ai_message = AIMessage(content=full_response)
    st.session_state.messages.append(ai_message)

with st.chat_message("assistant"):
    full_response = ""
    placeholder = st.empty()

    for chunk in chat.stream(st.session_state.messages):
        if chunk.content:
            full_response += chunk.content
            placeholder.markdown(full_response + "▌")

    placeholder.markdown(full_response)