module RubyLLM::Providers::Perplexity::Models

# File lib/ruby_llm/providers/perplexity/models.rb, line 25
def create_model_info(id, slug, capabilities)
  Model::Info.new(
    id: id,
    name: id,
    provider: slug,
    context_window: capabilities.context_window_for(id),
    max_output_tokens: capabilities.max_tokens_for(id),
    capabilities: capabilities.critical_capabilities_for(id),
    pricing: capabilities.pricing_for(id),
    metadata: {}
  )
end

# File lib/ruby_llm/providers/perplexity/models.rb, line 16
def list_models(**)
  slug = 'perplexity'
  parse_list_models_response(nil, slug, Perplexity::Capabilities)
end

# File lib/ruby_llm/providers/perplexity/models.rb, line 21
def parse_list_models_response(_response, slug, capabilities)
  MODEL_IDS.map { |id| create_model_info(id, slug, capabilities) }
end