Class: LLMBench::ConfigurationManager

Inherits:
Object
  • Object
show all
Defined in:
lib/llm_bench/configuration_manager.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(config_path: nil) ⇒ ConfigurationManager

Returns a new instance of ConfigurationManager.



10
11
12
13
 
# File 'lib/llm_bench/configuration_manager.rb', line 10

def initialize(config_path: nil)
  @config_path = config_path || File.join(__dir__, "..", "..", "models.yaml")
  @config = load_config_from_file
end

Instance Attribute Details

#config ⇒ Object (readonly)

Returns the value of attribute config.



8
9
10
 
# File 'lib/llm_bench/configuration_manager.rb', line 8

def config
  @config
end

Instance Method Details

#load_config_from_file ⇒ Object 



15
16
17
18
19
20
21
22
 
# File 'lib/llm_bench/configuration_manager.rb', line 15

def load_config_from_file
  unless File.exist?(config_path)
    warn Colors.error("Error: Configuration file not found at #{config_path}")
    exit 1
  end

  YAML.load_file(config_path)
end

#validate_provider_and_model!(provider_name:, model_nickname:) ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/llm_bench/configuration_manager.rb', line 24

def validate_provider_and_model!(provider_name:, model_nickname:)
  provider_config = find_provider(provider_name:)
  model_config = find_model(provider_config:, model_nickname:)

  validate_api_format!(model_config:)

  [provider_config, model_config]
rescue StandardError => e
  warn Colors.error("Error: #{e.message}")
  exit 1
end