Contents

# frozen_string_literal: true

module Luo
  class OpenAI
    include Configurable

    setting :access_token, default: ENV.fetch('OPENAI_ACCESS_TOKEN')
    setting :retries, default: ENV.fetch('OPENAI_REQUEST_RETRIES', 3).to_i
    setting :host, default: ENV.fetch('OPENAI_HOST', 'https://api.openai.com')
    setting :temperature, default: ENV.fetch('OPENAI_TEMPERATURE', 1).to_i

    include HttpClient.init_client

    PARAMS = Dry::Schema.Params do
      required(:model).filled(:string)
      required(:temperature).filled(:float)
      required(:messages).filled(:array)
      optional(:top_p).maybe(:float)
      optional(:n).maybe(:integer)
      optional(:stream).maybe(:bool)
      optional(:stop).maybe(:array)
      optional(:max_tokens).maybe(:integer)
      optional(:presence_penalty).maybe(:float)
      optional(:frequency_penalty).maybe(:float)
      optional(:logit_bias).maybe(:hash)
      optional(:user).maybe(:string)
    end

    def chat_completions(params)
      client.post('/v1/chat/completions', params.to_h)
    end

    def chat(messages)
      if messages.is_a?(Messages)
        messages = messages.to_a
      end
      params = PARAMS.call(
        model: 'gpt-3.5-turbo',
        temperature: config.temperature,
        messages: messages
      )
      return params.errors unless params.success?
      chat_completions(params).body.dig("choices", 0, "message", "content")
    end

  end

end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
luo-0.1.4	lib/luo/open_ai.rb
luo-0.1.3	lib/luo/open_ai.rb
luo-0.1.2	lib/luo/open_ai.rb
luo-0.1.1	lib/luo/open_ai.rb