module RubyLLM::Models

def all
  @all ||= begin
    data = JSON.parse(File.read(File.expand_path('models.json', __dir__)))
    data['models'].map { |model| ModelInfo.new(model.transform_keys(&:to_sym)) }
  end
rescue Errno::ENOENT
  [] # Return empty array if file doesn't exist yet
end

def audio_models
  all.select { |m| m.type == 'audio' }
end

def by_family(family)
  all.select { |m| m.family == family }
end

def chat_models
  all.select { |m| m.type == 'chat' }
end

def default_model
  'gpt-4o-mini'
end

def embedding_models
  all.select { |m| m.type == 'embedding' }
end

def find(model_id)
  all.find { |m| m.id == model_id } or raise Error, "Unknown model: #{model_id}"
end

def image_models
  all.select { |m| m.type == 'image' }
end

def provider_for(model)
  Provider.for(model)
end

def refresh!
  @all = nil
end