LoginSignup
5
7

PythonでOpenAIのGPT-3モデルを使ってチャットをする

Last updated at Posted at 2023-02-05

今回は、PythonでOpenAIのGPT-3モデルを使ってチャットをする一例を紹介します。PythonからOpenAI APIが提供するライブラリopenaiライブラリをつかって、GPT-3を呼び出してテキスト生成や質問応答などのタスクを行うことができます。

  • 1. 2023年2月現在、OpenAI GPT-3をPythonから使うには、OpenAIが提供するopenaiパッケージを使用します。
  • 2. openapiを使うには別途OpenAIのAPIキーを取得する必要があります。APIキーを正しく設定することで、このコードを実行することができます。
  • 3. OpenAI APIキーを取得するには、OpenAIのWebサイトにアクセスしてアカウントを作成してください。アカウント作成後、APIキーを取得するためのページにアクセスできます。URLは次の通りです。
    https://platform.openai.com/account/api-keys
  • 4. openaiのライブラリーをインストールしておきます。
pip install openai

それでは実際に、OpenAIのGPT-3モデルを使ってチャットができるコードを紹介します。このコードは、ユーザーが入力したテキストをGPT-3に送り、GPT-3からのテキスト応答を表示するものです。"quit"と入力すると、チャットは終了します。コードは次の通りです。

chatGPT.py
import openai

# APIキーの設定
openai.api_key = "取得したAPI KEY"


# GPT-3モデルの指定
model_engine = "text-davinci-002"

print("Chat with GPT-3: type 'quit' to exit")
while True:
    # ユーザーからの入力を取得
    user_input = input("You: ")
    if user_input.strip().lower() == "quit":
        break

    completions = openai.Completion.create(
        engine=model_engine,
        prompt=user_input,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )

    # GPT-3からの応答
    message = completions.choices[0].text
    print("GPT-3:", message)

2. 実行する

> python chatGPT.py

3. 結果

chatGPT.png

結果、コマンドラインベースでチャットができています。
チャットを終了するときは"quit"を入力してEnterを実行します。

この情報が皆様のなんらかのお役に立てることを望んでおります。
皆さんも、お試しください。


5
7
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
5
7