module RubyLLM::Providers::Mistral::Models

# File lib/ruby_llm/providers/mistral/models.rb, line 16
def headers(config)
  {
    'Authorization' => "Bearer #{config.mistral_api_key}"
  }
end

# File lib/ruby_llm/providers/mistral/models.rb, line 12
def models_url
  'models'
end

# File lib/ruby_llm/providers/mistral/models.rb, line 22
def parse_list_models_response(response, slug, capabilities)
  Array(response.body['data']).map do |model_data|
    model_id = model_data['id']

    release_date = capabilities.release_date_for(model_id)
    created_at = release_date ? Time.parse(release_date) : nil

    Model::Info.new(
      id: model_id,
      name: capabilities.format_display_name(model_id),
      provider: slug,
      family: capabilities.model_family(model_id),
      created_at: created_at,
      context_window: capabilities.context_window_for(model_id),
      max_output_tokens: capabilities.max_tokens_for(model_id),
      modalities: capabilities.modalities_for(model_id),
      capabilities: capabilities.capabilities_for(model_id),
      pricing: capabilities.pricing_for(model_id),
      metadata: {
        object: model_data['object'],
        owned_by: model_data['owned_by']
      }
    )
  end
end