PythonでChatGPTのAPIを叩く【GPT3.5, GPT4】

メッセージを格納したい場合などにお使いください

<API_KEY>の部分を書き換え
exec_gpt, exec_gpt4の引数にpromptを入れると使えます

ソースコード

GPT3.5

import openai

def exec_gpt(prompt):
    openai.api_key = '<API_KEY>'
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "user", "content": prompt},
        ],
    )
    return response.choices[0].message.content.strip()

GPT4

import openai

def exec_gpt4(prompt):
    openai.api_key = '<API_KEY>'
    response = openai.ChatCompletion.create(
        model='gpt-4',
        messages=[
            {"role": "user", "content": prompt},
        ],
    )
    return response.choices[0].message.content.strip()

使用例

prompt = '' # 任意のプロンプト
message1 = exec_gpt(prompt)
message2 = exec_gpt4(prompt)

---

購読者限定情報をお届け『DeepMagazine』