Module: RubyLLM::Protocols::Gemini::Models

Defined in:
lib/ruby_llm/protocols/gemini/models.rb

Overview

Models methods for the Gemini API integration

Instance Method Summary collapse

Instance Method Details

#list_models ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
 
# File 'lib/ruby_llm/protocols/gemini/models.rb', line 8

def list_models
  models = []
  page_token = nil

  loop do
    response = @connection.get(models_url) do |req|
      req.params = { pageSize: 1000 }
      req.params[:pageToken] = page_token if page_token
    end

    models.concat(parse_list_models_response(response, @provider.slug))
    page_token = response.body['nextPageToken']
    break unless page_token
  end

  models
end