Add valid service types for different cloud providers

zenml-io · Oct 27, 2024 · 1f779fd · 1f779fd
1 parent cb78c1a
commit 1f779fd
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/internal/provider/validation.go b/internal/provider/validation.go
@@ -15,6 +15,30 @@ var (
 		"mysql", "postgres", "snowflake", "databricks",
 	}
 
+	validServiceTypes = map[string][]string{
+		"aws": {
+			"artifact-store",
+			"container-registry",
+			"secrets-manager",
+			"step-operator",
+		},
+		"gcp": {
+			"artifact-store", 
+			"container-registry",
+			"secrets-manager",
+			"step-operator",
+		},
+		"azure": {
+			"artifact-store",
+			"container-registry",
+			"step-operator",
+		},
+		"kubernetes": {
+			"orchestrator",
+			"step-operator",
+		},
+	}
+
 	validAuthMethods = map[string][]string{
 		"aws": {
 			"iam-role",