Class: ActiveHarness::ModelConfig

Inherits:
Object
  • Object
show all
Defined in:
lib/active_harness/agent/models.rb

Overview

DSL builder — used by Agent.model { … }

Instance Method Summary collapse

Constructor Details

#initialize ⇒ ModelConfig

Returns a new instance of ModelConfig.



133
134
135
 
# File 'lib/active_harness/agent/models.rb', line 133

def initialize
  @models = []
end

Instance Method Details

#to_h ⇒ Object 



152
153
154
 
# File 'lib/active_harness/agent/models.rb', line 152

def to_h
  { models: @models }
end

#use(provider:, model:, temperature: nil, name: nil, size: nil, quality: nil, retry_attempts: nil, retry_delay: nil) ⇒ Object Also known as: fallback 



137
138
139
140
141
142
143
144
145
146
147
148
 
# File 'lib/active_harness/agent/models.rb', line 137

def use(provider:, model:, temperature: nil, name: nil, size: nil, quality: nil, retry_attempts: nil, retry_delay: nil)
  @models << {
    provider:       provider,
    model:          model,
    temperature:    temperature,
    name:           name,
    size:           size,
    quality:        quality,
    retry_attempts: retry_attempts,
    retry_delay:    retry_delay
  }.compact
end