[API] OpenAIのGPT-4o-miniとAPI-keyでやりとりするpythonコードサンプル

2024-12-02 2024-12-02

事前に、OpenAIのプラットフォームでAPI-KEYを取得しておき、exportコマンドで変数設定

export OPENAI_API_KEY=ここにapi-keyを設定する #注意! 通常の変数のように""で囲わない。APIエラーで沼になることがある。

# requestsをインストール
pip3 install requests

import os
import requests

api_key = os.environ.get("OPENAI_API_KEY")
if api_key is None:
    raise ValueError("OPENAI_API_KEY environment variable not set")

headers = {
    "Content-Type": "application/json",
    "Authorization": f"Bearer {api_key}",
}

data = {
    "model": "gpt-4o-mini",
    "messages": [{"role": "user", "content": "こんにちは、調子はいかがですか？"}],
    "temperature": 0.5,
    "max_tokens": 150,
}

response = requests.post("https://api.openai.com/v1/chat/completions", headers=headers, json=data)

if response.status_code == 200:
    try:
        content = response.json()["choices"][0]["message"]["content"]
        print(content)
    except (KeyError, IndexError) as e:
        print(f"Error extracting content: {e}.  Response: {response.json()}")
else:
    print(f"Error: {response.status_code} - {response.text}")

# 出力例

> こんにちは！私は元気です。あなたはいかがですか？何かお手伝いできることがあれば教えてください。