GPT APIを試す方法

新しいAPIへのアクセスが解放されるたびに、ささっと動作を確認したくなるので、メモ

import openai
import os
from datetime import datetime

start_time = datetime.now()

openai.api_key = os.getenv("OPENAI_API_KEY")
model_name = "gpt-4"
# model_name = "gpt-3.5-turbo"
# model_name = "gpt-4-32k"

question = "日本の初代の総理大臣の名前を教えて"

response = openai.ChatCompletion.create(
    model=model_name,
    messages=[
        {"role": "user", "content": question},
    ],
)
print(response)

参考

アクセスできないAPIを選ぶと下記のようになります。