Module: Legion::Extensions::Apollo::Helpers::Capability

Extended by:
Logging::Helper, Settings::Helper
Defined in:
lib/legion/extensions/apollo/helpers/capability.rb

Constant Summary collapse

EMBEDDING_MODELS = 
%w[mxbai-embed-large bge-large snowflake-arctic-embed].freeze
PRIVILEGE_MUTEX = 
Mutex.new

Class Method Summary collapse

Class Method Details

.apollo_write_enabled? ⇒ Boolean

Returns:

  • (Boolean) 


35
36
37
38
39
40
 
# File 'lib/legion/extensions/apollo/helpers/capability.rb', line 35

def apollo_write_enabled?
  settings[:data][:apollo_write] == true
rescue StandardError => e
  handle_exception(e, level: :warn, operation: 'apollo.capability.apollo_write_enabled')
  false
end

.can_embed? ⇒ Boolean

Returns:

  • (Boolean) 


16
17
18
19
20
21
22
23
 
# File 'lib/legion/extensions/apollo/helpers/capability.rb', line 16

def can_embed?
  return false unless defined?(Legion::LLM) && Legion::LLM.started?

  ollama_embedding_available? || cloud_embedding_configured?
rescue StandardError => e
  handle_exception(e, level: :warn, operation: 'apollo.capability.can_embed')
  false
end

.can_write? ⇒ Boolean

Returns:

  • (Boolean) 


25
26
27
28
29
30
31
32
33
 
# File 'lib/legion/extensions/apollo/helpers/capability.rb', line 25

def can_write?
  return false unless apollo_write_enabled?
  return false unless defined?(Legion::Data) && Legion::Data.connected?

  check_db_write_privilege
rescue StandardError => e
  handle_exception(e, level: :warn, operation: 'apollo.capability.can_write')
  false
end

.check_db_write_privilege ⇒ Object 



60
61
62
63
64
65
66
67
68
69
70
71
 
# File 'lib/legion/extensions/apollo/helpers/capability.rb', line 60

def check_db_write_privilege
  PRIVILEGE_MUTEX.synchronize do
    return @apollo_write_privilege unless @apollo_write_privilege.nil?

    @apollo_write_privilege = Legion::Data.connection
                                          .fetch("SELECT has_table_privilege(current_user, 'apollo_entries', 'INSERT') AS can_insert")
                                          .first[:can_insert] == true
  end
rescue StandardError => e
  handle_exception(e, level: :warn, operation: 'apollo.capability.check_db_write_privilege')
  @apollo_write_privilege = false
end

.cloud_embedding_configured? ⇒ Boolean

Returns:

  • (Boolean) 


51
52
53
54
55
56
57
58
 
# File 'lib/legion/extensions/apollo/helpers/capability.rb', line 51

def cloud_embedding_configured?
  provider = settings[:embedding][:provider]
  model = settings[:embedding][:model]
  !provider.nil? && !model.nil?
rescue StandardError => e
  handle_exception(e, level: :warn, operation: 'apollo.capability.cloud_embedding_configured')
  false
end

.ollama_embedding_available? ⇒ Boolean

Returns:

  • (Boolean) 


42
43
44
45
46
47
48
49
 
# File 'lib/legion/extensions/apollo/helpers/capability.rb', line 42

def ollama_embedding_available?
  return false unless defined?(Legion::LLM::Discovery::Ollama)

  EMBEDDING_MODELS.any? { |m| Legion::LLM::Discovery::Ollama.model_available?(m) }
rescue StandardError => e
  handle_exception(e, level: :warn, operation: 'apollo.capability.ollama_embedding_available')
  false
end

.reset! ⇒ Object 



73
74
75
 
# File 'lib/legion/extensions/apollo/helpers/capability.rb', line 73

def reset!
  @apollo_write_privilege = nil
end