1 files changed, 22 insertions, 0 deletions
diff --git a/app/services/ml/find_or_create_model_version_service.rb b/app/services/ml/find_or_create_model_version_service.rb
new file mode 100644
index 00000000000..1316b2546b9
--- /dev/null
+++ b/app/services/ml/find_or_create_model_version_service.rb
@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module Ml
+  class FindOrCreateModelVersionService
+    def initialize(project, params = {})
+      @project = project
+      @name = params[:model_name]
+      @version = params[:version]
+      @package = params[:package]
+    end
+
+    def execute
+      model = Ml::FindOrCreateModelService.new(project, name).execute
+
+      Ml::ModelVersion.find_or_create!(model, version, package)
+    end
+
+    private
+
+    attr_reader :version, :name, :project, :package
+  end
+end