Class: Legion::Extensions::Llm::Bedrock::Actor::DiscoveryRefresh

Inherits:
Actors::Every
  • Object
show all
Includes:
Inventory::ScopedRefresher, Logging::Helper
Defined in:
lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb

Overview

rubocop:disable Style/Documentation,Metrics/ClassLength

Constant Summary collapse

EMBED_TYPES = 
%i[embed embedding].freeze

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.every_seconds ⇒ Object 



33
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 33

def self.every_seconds = 3600

Instance Method Details

#check_subtask? ⇒ Boolean

Returns:

  • (Boolean) 


39
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 39

def check_subtask?  = false

#compute_lanes_for_scope ⇒ Object 



52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 52

def compute_lanes_for_scope(**)
  return [] unless defined?(Legion::LLM::Call::Registry)

  settings = Legion::Settings.dig(:extensions, :llm, :bedrock) || {}
  fleet_enabled = settings.dig(:fleet, :dispatch, :enabled)

  instances = Legion::LLM::Call::Registry.all_instances.select do |e|
    (e[:provider] || '').to_sym == :bedrock
  end

  instances.flat_map { |inst| lanes_for_instance(inst, fleet_enabled: fleet_enabled) }
rescue StandardError => e
  handle_exception(e, level: :warn, handled: true,
                      operation: 'bedrock.actor.discovery_refresh.compute_lanes')
  []
end

#credential_hash ⇒ Object 



69
70
71
72
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 69

def credential_hash(**)
  raw = Legion::Settings.dig(:extensions, :llm, :bedrock) || {}
  Digest::SHA256.hexdigest(raw[:api_key].to_s + raw[:instances].to_s)[0, 16]
end

#generate_task? ⇒ Boolean

Returns:

  • (Boolean) 


40
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 40

def generate_task?  = false

#manual ⇒ Object 



74
75
76
77
78
79
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 74

def manual(**)
  tick if defined?(Legion::Extensions::Llm::Inventory::ScopedRefresher) &&
          self.class.ancestors.include?(Legion::Extensions::Llm::Inventory::ScopedRefresher)
rescue StandardError => e
  handle_exception(e, level: :warn, handled: true, operation: 'bedrock.actor.discovery_refresh')
end

#run_now? ⇒ Boolean

Returns:

  • (Boolean) 


37
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 37

def run_now?        = true

#runner_class ⇒ Object 



35
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 35

def runner_class    = self.class

#runner_function ⇒ Object 



36
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 36

def runner_function = 'manual'

#scope_key ⇒ Object 



48
49
50
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 48

def scope_key(**)
  { provider: :bedrock }
end

#time ⇒ Object 



42
43
44
45
46
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 42

def time
  return self.class.every_seconds unless defined?(Legion::Settings)

  Legion::Settings.dig(:extensions, :llm, :bedrock, :discovery_interval) || self.class.every_seconds
end

#use_runner? ⇒ Boolean

Returns:

  • (Boolean) 


38
 
# File 'lib/legion/extensions/llm/bedrock/actors/discovery_refresh.rb', line 38

def use_runner?     = false