RubyからOpenAIのAPIを叩く

はじめに

RubyからChatGPTで有名なOpenAIのAPIをたたく記事を探したが、意外となかったので忘れるまえに書いておく。

これは少し古い記事なので、ChatGPTのAPIを叩きたい人はこちらの記事を参照してください。

利用するライブラリ

これを使う。

gem install "ruby-openai"

ChatGPTの API Key を取得しておく。

あとは次のようにすれば、利用することができる。（自分の場合は、以前に登録したときの無料枠が失効しており、クレジットカードを登録する必要があった。）

require "ruby/openai"

client = OpenAI::Client.new(access_token: "ACCESS_TOKEN_GOES_HERE")

res = client.completions(
        parameters: {
            model: "text-davinci-003",
            prompt: "Are you a cat?",
            max_tokens: 10
        })

puts res["choices"].map{|i| i["text"]}.join
# No, I am not a cat.

ChatGPTのAPIを叩くまでは知らなかったが、文章校正や、プログラミングなどに特化したモデルもあり、そういったものを選ぶこともできる。

この記事は以上です。