Module: RubyLLM::Providers::VertexAI::Capabilities

Defined in:
lib/ruby_llm/providers/vertexai/capabilities.rb

Overview

Feature capability gaps not represented in upstream model catalogs.

Class Method Summary collapse

Class Method Details

.augment(capabilities, model_id:, modalities:) ⇒ Object



8
9
10
11
12
13
14
15
# File 'lib/ruby_llm/providers/vertexai/capabilities.rb', line 8

def self.augment(capabilities, model_id:, modalities:)
  return capabilities if model_id.include?('embedding')

  additions = []
  additions << 'tool_choice' if model_id == 'gemini-2.5-flash'
  additions << 'transcription' if modalities[:input].include?('audio') && modalities[:output].include?('text')
  capabilities | additions
end