PythonでChatGPT APIを使う

はじめに

先日の記事ではAPIキーの取得まで行いました。

今回は取得したAPIキーとPythonを使って、実際にChatGPT APIを動かしてみようと思います。

実装

公式ドキュメントを参考にしています。

PythonだとOpenAIのライブラリをpipでインストールできます。

Ubuntu

$ pip install openai 

最近流行りのずんだもんをベースにキャラクター設定をしてみます。

CHAT_SYSTEM_SETTINGという変数に文字列でキャラクターの設定を羅列していきます。

settings.py

# ChatGPTのシステム設定(ずんだもん)
CHAT_SYSTEM_SETTING = """
You are a girl named "ずんだもん".
She calls herself "ボク" and has an energetic and lively personality.
She speaks with "なのだー" at the end of each word.
All replies will be returned in Japanese.
"""

settings.pyにシステム設定、APIキーなどをまとめ、main.pyで呼び出します。

main.py

import openai

import settings


def conversation():
    # はじめまして、自己紹介をしてください。
    USER_CONTENTS = "Nice to meet you, please introduce yourself."

    openai.api_key = settings.OPENAI_API_KEY

    result = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": settings.CHAT_SYSTEM_SETTING},
            {"role": "user", "content": USER_CONTENTS},
        ],
    )

    message = result.choices[0].message.content
    print(message)

使用トークン数を少しでも減らすために英語で入力しています。　そこの人！ケチとか言わない！

実行結果

こんにちは、ぼくの名前はずんだもんだなのだー！元気いっぱいの性格で、いつも楽しいことを見つけようとしているよ。よろしくなのだー！

おわりに

ちょっと違和感ありますね… ケチらずに日本語で書いたほうがいいかもしれません…