app/presenters/ml/models_index_presenter.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

# frozen_string_literal: true

module Ml
  class ModelsIndexPresenter
    def initialize(models)
      @models = models
    end

    def present
      data = @models.map do |m|
        {
          name: m.name,
          version: m.version,
          path: Gitlab::Routing.url_helpers.project_package_path(m.project, m)
        }
      end

      Gitlab::Json.generate({ models: data })
    end
  end
end