Class: Legion::Extensions::Llm::Vertex::Actor::DiscoveryRefresh

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

Overview

rubocop:disable Style/Documentation

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.every_secondsObject



31
# File 'lib/legion/extensions/llm/vertex/actors/discovery_refresh.rb', line 31

def self.every_seconds = 3600

Instance Method Details

#check_subtask?Boolean

Returns:

  • (Boolean)


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

def check_subtask?  = false

#compute_lanes_for_scopeObject



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

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

  settings      = Legion::Settings.dig(:extensions, :llm, :vertex) || {}
  fleet_enabled = settings.dig(:fleet, :dispatch, :enabled)
  instances     = Legion::LLM::Call::Registry.all_instances.select do |e|
    (e[:provider] || '').to_sym == :vertex
  end

  instances.flat_map do |entry|
    lanes_for_instance(entry, fleet_enabled: fleet_enabled)
  end
rescue StandardError => e
  handle_exception(e, level: :warn, handled: true, operation: 'vertex.actor.compute_lanes_for_scope')
  []
end

#credential_hashObject



134
135
136
137
# File 'lib/legion/extensions/llm/vertex/actors/discovery_refresh.rb', line 134

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

#generate_task?Boolean

Returns:

  • (Boolean)


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

def generate_task?  = false

#manualObject



139
140
141
142
143
# File 'lib/legion/extensions/llm/vertex/actors/discovery_refresh.rb', line 139

def manual
  tick if respond_to?(:tick)
rescue StandardError => e
  handle_exception(e, level: :warn, handled: true, operation: 'vertex.actor.discovery_refresh')
end

#run_now?Boolean

Returns:

  • (Boolean)


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

def run_now?        = true

#runner_classObject



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

def runner_class    = self.class

#runner_functionObject



34
# File 'lib/legion/extensions/llm/vertex/actors/discovery_refresh.rb', line 34

def runner_function = 'manual'

#scope_keyObject



46
47
48
# File 'lib/legion/extensions/llm/vertex/actors/discovery_refresh.rb', line 46

def scope_key
  { provider: :vertex }
end

#timeObject



40
41
42
43
44
# File 'lib/legion/extensions/llm/vertex/actors/discovery_refresh.rb', line 40

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

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

#use_runner?Boolean

Returns:

  • (Boolean)


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

def use_runner?     = false