Class: Legion::Extensions::Llm::Gemini::Provider

Inherits:
Provider
  • Object
show all
Includes:
Logging::Helper
Defined in:
lib/legion/extensions/llm/gemini/provider.rb

Overview

Gemini provider implementation for the Legion::Extensions::Llm base provider contract.

Defined Under Namespace

Modules: Capabilities

Class Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Class Attribute Details

.registry_publisher ⇒ Object 



21
22
23
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 21

def registry_publisher
  @registry_publisher ||= Legion::Extensions::Llm::RegistryPublisher.new(provider_family: :gemini)
end

Class Method Details

.capabilities ⇒ Object 



19
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 19

def capabilities = Capabilities

.configuration_options ⇒ Object 



17
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 17

def configuration_options = %i[gemini_api_key gemini_api_base]

.configuration_requirements ⇒ Object 



18
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 18

def configuration_requirements = %i[gemini_api_key]

.slug ⇒ Object 



16
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 16

def slug = 'gemini'

Instance Method Details

#api_base ⇒ Object 



80
81
82
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 80

def api_base
  config.gemini_api_base || settings[:endpoint] || 'https://generativelanguage.googleapis.com/v1beta'
end

#completion_url ⇒ Object 



88
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 88

def completion_url = generate_content_url(model: @model)

#embed_content_url(model:) ⇒ Object 



101
102
103
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 101

def embed_content_url(model:)
  "#{model_path(model)}:embedContent"
end

#embedding_url(model:) ⇒ Object 



91
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 91

def embedding_url(model:) = embed_content_url(model:)

#generate_content_url(model:) ⇒ Object 



93
94
95
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 93

def generate_content_url(model:)
  "#{model_path(model)}:generateContent"
end

#headers ⇒ Object 



84
85
86
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 84

def headers
  identity_headers.merge('x-goog-api-key' => config.gemini_api_key)
end

#list_models ⇒ Object 



105
106
107
108
109
110
111
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 105

def list_models(**)
  log.info { 'listing available Gemini models' }
  super.tap do |models|
    log.info { "discovered #{models.size} Gemini model(s); publishing to registry" }
    self.class.registry_publisher.publish_models_async(models, readiness: readiness(live: false))
  end
end

#models_url ⇒ Object 



90
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 90

def models_url = 'models'

#settings ⇒ Object 



76
77
78
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 76

def settings
  Gemini.default_settings
end

#stream_generate_content_url(model:) ⇒ Object 



97
98
99
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 97

def stream_generate_content_url(model:)
  "#{model_path(model)}:streamGenerateContent?alt=sse"
end

#stream_url ⇒ Object 



89
 
# File 'lib/legion/extensions/llm/gemini/provider.rb', line 89

def stream_url = stream_generate_content_url(model: @model)