add CollectorType

Granulate · Jul 3, 2023 · f97b36d · f97b36d
1 parent fcb4919
commit f97b36d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/granulate_utils/config_feeder/core/models/cluster.py b/granulate_utils/config_feeder/core/models/cluster.py
@@ -18,6 +18,12 @@ class CloudProvider(str, Enum):
     GCP = "gcp"
 
 
+class CollectorType(str, Enum):
+    UNKOWN = "unknown"
+    SAGENT = "sagent"
+    GPROFILER = "gprofiler"
+
+
 class ClusterBase(BaseModel):
     provider: CloudProvider
     external_id: str

diff --git a/granulate_utils/config_feeder/core/models/collection.py b/granulate_utils/config_feeder/core/models/collection.py
@@ -0,0 +1,7 @@
+from enum import Enum
+
+
+class CollectorType(str, Enum):
+    UNKOWN = "unknown"
+    SAGENT = "sagent"
+    GPROFILER = "gprofiler"