Class: CompletionKit::ModelDiscoveryService

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

Defined Under Namespace

Classes: DiscoveryError

Constant Summary collapse

AZURE_HOST_SUFFIXES = 
[".openai.azure.com", ".services.ai.azure.com"].freeze

Instance Method Summary collapse

Constructor Details

#initialize(config:) ⇒ ModelDiscoveryService

Returns a new instance of ModelDiscoveryService.



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

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

Instance Method Details

#refresh!(force: false, &on_progress) ⇒ Object 



18
19
20
21
22
23
24
25
 
# File 'app/services/completion_kit/model_discovery_service.rb', line 18

def refresh!(force: false, &on_progress)
  discovered = fetch_models
  reconcile(discovered)
  return if @provider == "openrouter"

  reset_failed_generation if force
  probe_new_models(&on_progress)
end