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 Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(config:) ⇒ ModelDiscoveryService

Returns a new instance of ModelDiscoveryService.



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

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

Instance Attribute Details

#catalog_model_count ⇒ Object (readonly)

Returns the value of attribute catalog_model_count.



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

def catalog_model_count
  @catalog_model_count
end

Instance Method Details

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



21
22
23
24
25
26
27
28
 
# File 'app/services/completion_kit/model_discovery_service.rb', line 21

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