Contents

module RubyAI
  class Configuration
    BASE_URL = "https://api.openai.com/v1/chat/completions"

    MODELS = {
      "gpt-4" => "gpt-4",
      "gpt-4-32k" => "gpt-4-32k",
      "gpt-4-turbo" => "gpt-4-turbo",
      "gpt-4o-mini" => "gpt-4o-mini",
      "o1-mini" => "o1-mini",
      "o1-preview" => "o1-preview",
      "text-davinci-003" => "text-davinci-003"
    }

    DEFAULT_MODEL = "gpt-3.5-turbo"

    attr_accessor :api_key, :model, :messages, :temperature

    def initialize(config = {})
      @api_key = config[:api_key]
      @model = config.fetch(:model, DEFAULT_MODEL)
      @messages = config.fetch(:messages, nil)
      @temperature = config.fetch(:temperature, 0.7)
    end
  end

  def self.configuration
    @configuration ||= Configuration.new
  end

  def self.configure
    yield(configuration)
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rubyai-0.5	lib/rubyai/configuration.rb
rubyai-0.4	lib/rubyai/configuration.rb