OpenAI Responses API 入門 (5) - 会話状態

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

import os
from google.colab import userdata

# 環境変数の準備 (左端の鍵アイコンでOPENAI_API_KEYを設定)
os.environ["OPENAI_API_KEY"] = userdata.get("OPENAI_API_KEY")

from openai import OpenAI

# クライアントの準備
client = OpenAI()

# 推論の実行
response = client.responses.create(
    model="gpt-4o",
    input=[
        {"role": "user", "content": "日本一高い山は？"},
        {"role": "assistant", "content": "日本一高い山は、富士山です。"},
        {"role": "user", "content": "その山は何県にある？"},
    ],
)
print(response.output_text)

# メッセージリストの準備
messages = [
    {"role": "user", "content": "日本一高い山は？"}
]

# 推論の実行
response = client.responses.create(
    model="gpt-4o",
    input=messages,
    store=False
)
print(response.output_text)

# メッセージリストに応答を追加
messages += [{"role": el.role, "content": el.content} for el in response.output]

# 新しいメッセージの追加
messages.append({ "role": "user", "content": "その山は何県にある？" })

# 推論の実行
second_response = client.responses.create(
    model="gpt-4o-mini",
    input=messages,
    store=False
)
print(second_response.output_text)

# 推論の実行
response = client.responses.create(
    model="gpt-4o-mini",
    input=[{"role": "user", "content": "日本一高い山は？"}],
)
print(response.output_text)

# 推論の実行
second_response = client.responses.create(
    model="gpt-4o",
    previous_response_id=response.id,
    input=[{"role": "user", "content": "その山は何県にある？"}],
)
print(second_response.output_text)