1 files changed, 7 insertions, 11 deletions
diff --git a/app/services/ml/find_or_create_model_service.rb b/app/services/ml/find_or_create_model_service.rb
index 66dec7a6234..9199730e84b 100644
--- a/app/services/ml/find_or_create_model_service.rb
+++ b/app/services/ml/find_or_create_model_service.rb
@@ -2,21 +2,17 @@
 
 module Ml
   class FindOrCreateModelService
-    def initialize(project, model_name)
+    def initialize(project, name, user = nil, description = nil, metadata = [])
       @project = project
-      @name = model_name
+      @name = name
+      @description = description
+      @metadata = metadata
+      @user = user
     end
 
     def execute
-      Ml::Model.find_or_create(
-        project,
-        name,
-        Ml::FindOrCreateExperimentService.new(project, name).execute
-      )
+      FindModelService.new(@project, @name).execute ||
+        CreateModelService.new(@project, @name, @user, @description, @metadata).execute
     end
-
-    private
-
-    attr_reader :name, :project
   end
 end