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
47
48
49
50
51
52
53
54
55
|
# frozen_string_literal: true
module Mutations
module Ml
module ModelVersions
class Create < BaseMutation
graphql_name 'MlModelVersionCreate'
authorize :write_model_registry
argument :project_path, GraphQL::Types::ID,
required: true,
description: "Project the model to mutate is in."
include FindsProject
argument :model_id, ::Types::GlobalIDType[::Ml::Model],
required: true,
description: 'Global ID of the model the version belongs to.'
argument :version, GraphQL::Types::String,
required: false,
description: 'Model version.'
argument :description, GraphQL::Types::String,
required: false,
description: 'Description of the model version.'
field :model_version,
Types::Ml::ModelVersionType,
null: true,
description: 'Model after mutation.'
def resolve(**args)
project = authorized_find!(args[:project_path])
model = ::Ml::Model.by_project_id_and_id(project.id, args[:model_id].model_id)
return { errors: ['Model not found'] } unless model
result = ::Ml::CreateModelVersionService.new(model,
{
version: args[:version],
description: args[:description],
user: current_user
}
).execute
{
model_version: result.payload[:model_version],
errors: result.message
}
end
end
end
end
end
|
