Class: Llmclt::Config

Inherits:
Object
  • Object
show all
Defined in:
lib/llmclt/config.rb,
lib/llmclt/config/safety_config.rb,
lib/llmclt/config/generation_config.rb

Defined Under Namespace

Classes: GenerationConfig, SafetyConfig

Constant Summary collapse

DEFAULT_CONFIGS = 
{
  project_id: nil,
  location_id: 'asia-northeast1',
  model: 'gemini-1.5-pro-001',
  service_account_json: nil,
  system_instruction_prompt: nil,
  open_timeout: 60,
  read_timeout: 60,
  keep_alive_timeout: 15
}.freeze

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(**kwargs) ⇒ Config 



20
21
22
23
24
25
26
27
 
# File 'lib/llmclt/config.rb', line 20

def initialize(**kwargs)
  DEFAULT_CONFIGS.each do |key, value|
    send(:"#{key}=", kwargs[key] || value)
  end

  self.safety_config = Llmclt::Config::SafetyConfig.new(kwargs[:safety_settings])
  self.generation_config = Llmclt::Config::GenerationConfig.new(kwargs[:generation_config])
end

Instance Attribute Details

#generation_config ⇒ Object

Returns the value of attribute generation_config.



18
19
20
 
# File 'lib/llmclt/config.rb', line 18

def generation_config
  @generation_config
end

#safety_config ⇒ Object

Returns the value of attribute safety_config.



17
18
19
 
# File 'lib/llmclt/config.rb', line 17

def safety_config
  @safety_config
end