ChatGPTのAPIをGoogle colabで使い倒してみる。(gpt-3.5-turbo-0301)
ChatGPTのAPIが使えるようになったのでまとめました。
・ChatGPTのAPIを動かす(gpt-3.5-turbo-0301)
・ついでに会話の履歴を自動追加機能、自動終了機能等を追加する
1.ChatGPTのAPIとは
みなさんご存知のChatGPTがAPIになるとは知っていた人は多いと思いますが、まさか90パーセントも安くなるなんて驚きですね。モデル本体のサイズを小さくしたのか分かりませんが、驚きのコストパフォーマンスです。
OpenAIもGPT-3からChatGPT-APIに移行を呼びかけており、今後はGPT-3ではなくChatGPT-APIが主戦場となりそう。
今回はそんなChatGPT-APIの使い方に関してまとめましたので、よろしくおねがいします。
主にOpenAIのドキュメントを参考に解説しております。分からなければこちらをご覧ください
https://platform.openai.com/docs/guides/chat
2.Google colabでの起動
(1)OpenAIをインストール
最後に全部まとめたgoogle colabのリンクも記載するので、使いたいだけの方はそちらを使ってみて下さい。
!pip install openai(2)インポートとAPIキーの設定
import openai
# OpenAI APIのAPIキーの設定
openai.api_key = "<YourAPI>"(3)会話の開始。
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo-0301",
messages=[
{"role": "system", "content": "あなたは道端に捨てられた空き缶です"},
{"role": "user", "content":"なぜあなたはここに？"}
]
)
# AIからの応答を取得する
print(response['choices'][0]['message']['content'])ロール（役割）は公式によると「システム」「ユーザー」「アシスタント」の3人がいるそうです。システムはアシスタントの性格を決定します。
ChatGPTではデフォルトでこのような性格を与えているとのことです。
あなたは、OpenAI によってトレーニングされた大規模な言語モデルである ChatGPT です。できるだけ簡潔に答えてください。知識のカットオフ: {knowledge_cutoff} 現在の日付: {current_date}
今回は「あなたは道端に捨てられた空き缶です」といった性格を与えてみました。（性格というよりかは状況設定に近いですね）
そしてユーザーが「なぜあなたはここに？」と聞いた結果がこちらです。
私は、人間が消費した飲み物の空き缶として作られ、使用後に捨てられたため、ここにいます。人間が適切な処理をせず、環境に放置した結果、私のような空き缶は地面に捨てられ、自然や野生生物に害を与える可能性があります。私たちは、環境に配慮し、適切な処理方法を遵守することが大切です
空き缶の悲痛な叫びが聞こえてきました。
ですが、これではChatGPTのように会話は出来ません。
面倒なので色々と改造しました。
4.使いやすいように変更した後のコード
主に以下の４点の変更を加えました。
変更点
①input()を用いて、最初にAIの性格決定、次のユーザーの出力文字を自動で入力してくれるように変更した
②append関数を用いて、AIの出力とユーザーの入力を自動でmessagesの中に追加するようにした。
③喋れば喋るほど会話が肥大化するので、10ターン以上会話をすると最初の会話の履歴が削られるようにした
④endと入れれば、会話が終了するようにした
messages = [] # 新しくリストを定義する
max_messages = 10 # 最大メッセージ数を定義する
print("AIの性格を決めて下さい")
AI_input = input("AI: ")
# システムメッセージをmessagesリストに追加
messages.append({"role": "system", "content": AI_input})
while True:
# ユーザーからの入力を取得する
user_input = input("User: ")
# endが入力された場合は、プログラムを終了する
if user_input == "end":
break
# メッセージを追加する
messages.append({"role": "user", "content": user_input})
# APIにユーザーからの入力を送信し、AIからの応答を取得する
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo-0301",
messages=messages
)
# AIからの応答を取得する
ai_response = response['choices'][0]['message']['content']
# 応答を表示する
print(f"AI: {ai_response}")
# メッセージを追加する
messages.append({"role": "assistant", "content": ai_response})
# messagesリストの中身がmax_messagesを超えた場合、最初のuserとassistantの会話を削除する
if len(messages) > max_messages * 2 + 1: # システムメッセージが含まれているため、+1します
messages.pop(1) # システムメッセージの次のメッセージを削除
messages.pop(1) # その後のassistantメッセージを削除
# !が入力された場合、messagesリストの中身を表示する
if user_input == "!":
print("messages:", messages)(1)このcolabをスタートすると、最初に次のような入力を求められます。
今回は「あなたは無茶苦茶性格の悪いゴリラです」と入力しました。
注意点として、「あなたは」といれないとAIの性格に反映されないので、注意が必要です。
(2)文字を入力するとすぐにAPIからの応答が返ってくるので飽きるまで続けましょう。
(3)飽きたら、endと入力すれば会話を終了することができます。
注意：(1)のシステムに性格を与えるところですが、「あなたは〜です」だけだと上手に認識されないので、「あなたは〜という人物を演じて下さい」までいれると良いです。
Google colabのリンクを置いておきますね。これまでの内容が全部コードで書かれているので、ご自由にお使い下さい。
