Class: OllamaAgent::RuntimeCommandSystem::Completers::ModelCompleter

Inherits:
BaseCompleter
  • Object
show all
Defined in:
lib/ollama_agent/runtime_command_system/completers.rb

Instance Method Summary collapse

Instance Method Details

#suggestions(ast:, session: {}) ⇒ Object



18
19
20
21
22
23
24
25
26
27
28
29
30
31
# File 'lib/ollama_agent/runtime_command_system/completers.rb', line 18

def suggestions(ast:, session: {}, **)
  token = current_token(ast)
  models = session[:model_registry_cache] ||= Providers::ModelRegistry.all(agent: session[:agent])
  select_matches(models, token).map do |model|
    Suggestion.new(
      text: model.name,
      type: :model,
      description: (model),
      metadata: { provider: model.provider, context_size: model.context_size, status: model.status },
      capabilities: model.capabilities - [:chat],
      replacement_start: current_argument_start(ast)
    )
  end
end