Split SAML deployment process

polyaxon · Nov 15, 2023 · b4c21ee · b4c21ee
1 parent ce910ce
commit b4c21ee
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/haupt/haupt/common/events/event_actions.py b/haupt/haupt/common/events/event_actions.py
@@ -51,6 +51,7 @@
 GITLAB = "gitlab"
 BITBUCKET = "bitbucket"
 GOOGLE = "google"
+SAML = "saml"
 
 
 WRITE_ACTIONS = [

diff --git a/haupt/haupt/schemas/platform_config.py b/haupt/haupt/schemas/platform_config.py
@@ -278,11 +278,14 @@ class PlatformConfig(BaseSchemaModel):
     auth_google_options: Optional[Dict] = Field(
         alias="POLYAXON_AUTH_GOOGLE_OPTIONS", default={}
     )
-    auth_azure_options: Optional[Dict] = Field(
-        alias="POLYAXON_AUTH_AZURE_OPTIONS", default={}
+    auth_okta_options: Optional[Dict] = Field(
+        alias="POLYAXON_AUTH_OKTA_OPTIONS", default={}
     )
-    auth_saml_options: Optional[Dict] = Field(
-        alias="POLYAXON_AUTH_SAML_OPTIONS", default={}
+    auth_onelogin_options: Optional[Dict] = Field(
+        alias="POLYAXON_AUTH_ONELOGIN_OPTIONS", default={}
+    )
+    auth_azuread_options: Optional[Dict] = Field(
+        alias="POLYAXON_AUTH_AZUREAD_OPTIONS", default={}
     )
     ui_in_sandbox: Optional[bool] = Field(alias=ENV_KEYS_UI_IN_SANDBOX, default=False)
     ui_admin_enabled: Optional[bool] = Field(
@@ -327,8 +330,9 @@ def validate_str_list(cls, v, field: "ModelField"):
         "auth_gitlab_options",
         "auth_bitbucket_options",
         "auth_google_options",
-        "auth_azure_options",
-        "auth_saml_options",
+        "auth_okta_options",
+        "auth_onelogin_options",
+        "auth_azuread_options",
         pre=True,
     )
     def validate_json_fields(cls, v, field: "ModelField"):