Open AI API Access in Ruby

client.rb

# gem 'wrest' in Gemfile
# optionally, `Wrest.logger = Rails.logger` in `config/initializers/wrest.rb`


module OpenAI
  def self.base_uri
    'https://api.openai.com/v1'.to_uri(
      default_headers: {
        'Content-Type' => 'application/json',
        'Authorization' => "Bearer #{ENV.fetch('OPENAI_KEY', nil)}"
      }
    )
  end

  class Client
    PATH = '/chat/completions'.freeze
    def initialize(model = 'gpt-3.5-turbo')
      @uri = OpenAI.base_uri[PATH]
      @model = model
    end

    # messages = [{ role: 'user', content: 'Say this is a test!' }]
    # https://platform.openai.com/docs/api-reference/chat/create
    def chat_completions_create(messages, temperature = 0.8)
      @uri.post(
        {
          model: @model,
          messages: messages,
          temperature: temperature
        }.to_json
      ).deserialise
    end
  end
end