RubyでOpenAI APIを使ってみる

Rubyのライブラリを使ってOpenAIのAPIを使ってみました。

まずは環境変数にOpenAIのAPI Keyをセットします。
キーは事前にOpenAIのサイトから取得しておきます。

export OPENAI_API_KEY=sk-XXXXXXX

次はGemfileでライブラリを追加します。

gem "openai"

最後に下のような実装でAPIを呼び出します。

openai_client = OpenAI::Client.new(api_key: "sk-XXXXXXX", default_engine: "text-davinci-003")
result = openai_client.completions(prompt: "夕食のおすすめはある?", max_tokens: 100)

呼び出し結果は下の通りです。
finish_reason="length"なのでtokensが小さすぎた事が分かります。
こういう時はmax_tokensを大きくして再度APIを呼び出します。
ただ長くすると返答に時間がかかるので注意が必要です。