Class: CompletionKit::ModelDiscoveryService

Inherits:
Object
  • Object
show all
Defined in:
app/services/completion_kit/model_discovery_service.rb

Instance Method Summary collapse

Constructor Details

#initialize(config:) ⇒ ModelDiscoveryService

Returns a new instance of ModelDiscoveryService.



7
8
9
10
11
 
# File 'app/services/completion_kit/model_discovery_service.rb', line 7

def initialize(config:)
  @provider = config[:provider]
  @api_key = config[:api_key]
  @api_endpoint = config[:api_endpoint]
end

Instance Method Details

#refresh!(&on_progress) ⇒ Object 



13
14
15
16
17
18
 
# File 'app/services/completion_kit/model_discovery_service.rb', line 13

def refresh!(&on_progress)
  models_with_names = fetch_models
  reconcile(models_with_names)
  return if %w[openrouter ollama].include?(@provider)
  probe_new_models(&on_progress)
end