class AiModelCatalog::Catalog

# File lib/ai/model_catalog.rb, line 149
def initialize(entries)
  @entries = entries
end

# File lib/ai/model_catalog.rb, line 161
def base?(base)
  bases.include?(base.to_s.downcase)
end

# File lib/ai/model_catalog.rb, line 157
def bases
  entries.map(&:base).uniq.sort
end

# File lib/ai/model_catalog.rb, line 153
def empty?
  entries.empty?
end

# File lib/ai/model_catalog.rb, line 178
def variants_for(base, version)
  base = base.to_s.downcase
  version_key = version&.to_s
  entries.select do |entry|
    entry.base == base && entry.version.to_s == version_key.to_s
  end
end

# File lib/ai/model_catalog.rb, line 174
def version?(base, version)
  versions_for(base).include?(version.to_s)
end

# File lib/ai/model_catalog.rb, line 165
def versions_for(base)
  base = base.to_s.downcase
  entries
    .select { |entry| entry.base == base && entry.version }
    .map(&:version)
    .uniq
    .sort_by { |version| version_sort_key(version) }
end