PythonでChatGPTを使う

Mac Ventura上のPythonでChatGPTを使うためのメモ。

まずはライブラリをインストールする必要がありますが、pip3コマンドで簡単にインストールできます。

% pip3 install openai

インストール後は、下記のような感じのプログラムを書くだけ。API Keyが必要ですので、
https://platform.openai.com/account/api-keysで取得しておきます。

## -- gpt_test.py --
import openai
import json

openai.api_key = "ここにAPI Keyを入れる"

def ask(question):
    prompt = f"I am a large language model trained by OpenAI. Ask me anything!\nQ: {question}\nA:"
    response = openai.Completion.create(
        engine="text-davinci-002",
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.7,
    )

    answer = response.choices[0].text.strip()
    return answer


x = "中川"
ans = ask("「" + x + "」の読み方をカタカナで")
print(ans)

ちなみにこのプログラム、何度か実行すると読み方がころころ変わるので、ChatGPTの精度はまだまだかなぁ。という気もしてきます。有料アカウントにして、GPT-4を使えばまた事情は変わるかもしれませんが。

% python3 gpt_test.py
ナカガワ
% python3 gpt_test.py
チュウガワ
% python3 gpt_test.py
チュウカン