Welcome to mirror list, hosted at ThFree Co, Russian Federation.

gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'app/graphql/resolvers/ml/model_detail_resolver.rb')
-rw-r--r--app/graphql/resolvers/ml/model_detail_resolver.rb27
1 files changed, 27 insertions, 0 deletions
diff --git a/app/graphql/resolvers/ml/model_detail_resolver.rb b/app/graphql/resolvers/ml/model_detail_resolver.rb
new file mode 100644
index 00000000000..01c025c1d8a
--- /dev/null
+++ b/app/graphql/resolvers/ml/model_detail_resolver.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Resolvers
+ module Ml
+ class ModelDetailResolver < Resolvers::BaseResolver
+ extension ::Gitlab::Graphql::Limit::FieldCallCount, limit: 1
+
+ type ::Types::Ml::ModelType, null: true
+
+ argument :id, ::Types::GlobalIDType[::Ml::Model],
+ required: true,
+ description: 'ID of the model.'
+
+ def resolve(id:)
+ Gitlab::Graphql::Lazy.with_value(find_object(id: id)) do |ml_model|
+ ml_model if current_user.can?(:read_model_registry, ml_model&.project)
+ end
+ end
+
+ private
+
+ def find_object(id:)
+ GitlabSchema.find_by_gid(id)
+ end
+ end
+ end
+end