1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
# frozen_string_literal: true
require 'mime/types'
module API
# MLFlow integration API, replicating the Rest API https://www.mlflow.org/docs/latest/rest-api.html#rest-api
module Ml
module MlflowArtifacts
class Artifacts < ::API::Base
feature_category :mlops
helpers ::API::Helpers::PackagesHelpers
before do
check_api_read!
check_api_model_registry_read!
end
desc 'MLflow artifact API' do
detail 'MLflow artifacts mapping to GitLab artifacts'
end
route_setting :api, write: true
route_setting :model_registry, write: true
params do
optional :path, type: String,
desc: 'Path to the artifact, model version id, optionally followed by path. E.g. 15/MLmodel'
end
get 'artifacts', urgency: :low do
model_version, path = params[:path].split('/', 2)
# MLflow handles directories differently than GitLab does so when MLflow checks if a path is a directory
# we return an empty array as 404s would cause issues for MLflow
files = path.present? ? [] : list_model_artifacts(user_project, model_version).all
package_files = { files: files }
present package_files, with: Entities::Ml::MlflowArtifacts::ArtifactsList
end
get 'artifacts/:model_version/*file_path', format: false, urgency: :low do
present_package_file!(find_model_artifact(user_project, params[:model_version],
CGI.escape(params[:file_path])))
end
end
end
end
end
|
