Disable default project behaviour for pro workspaces (#3605)

* Disable default project behaviour for pro workspaces

* Docstring

Author: schustmi

src/zenml/zen_stores/base_zen_store.py

@@ -450,20 +450,6 @@ def _default_project_name(self) -> str:
         """
         return os.getenv(ENV_ZENML_DEFAULT_PROJECT_NAME, DEFAULT_PROJECT_NAME)
 
-    def _get_default_project(self) -> ProjectResponse:
-        """Get the default project.
-
-        Raises:
-            KeyError: If the default project doesn't exist.
-
-        Returns:
-            The default project.
-        """
-        try:
-            return self.get_project(self._default_project_name)
-        except KeyError:
-            raise KeyError("Unable to find default project.")
-
     def _get_default_stack(
         self,
     ) -> StackResponse:

src/zenml/zen_stores/sql_zen_store.py

@@ -1155,19 +1155,7 @@ def _initialize(self) -> None:
 
     def _initialize_database(self) -> None:
         """Initialize the database if not already initialized."""
-        # When running in a Pro ZenML server, the default project is not
-        # created on database initialization but on server onboarding.
-        create_default_project = True
-        if ENV_ZENML_SERVER in os.environ:
-            from zenml.config.server_config import ServerConfiguration
-
-            if (
-                ServerConfiguration.get_server_config().deployment_type
-                == ServerDeploymentType.CLOUD
-            ):
-                create_default_project = False
-
-        if create_default_project:
+        if self._default_project_enabled:
             # Make sure the default project exists
             self._get_or_create_default_project()
         # Make sure the default stack exists
@@ -9440,7 +9428,8 @@ def update_project(
                 session=session,
             )
             if (
-                existing_project.name == self._default_project_name
+                self._default_project_enabled
+                and existing_project.name == self._default_project_name
                 and "name" in project_update.model_fields_set
                 and project_update.name != existing_project.name
             ):
@@ -9481,7 +9470,10 @@ def delete_project(self, project_name_or_id: Union[str, UUID]) -> None:
                 schema_class=ProjectSchema,
                 session=session,
             )
-            if project.name == self._default_project_name:
+            if (
+                self._default_project_enabled
+                and project.name == self._default_project_name
+            ):
                 raise IllegalOperationError(
                     "The default project cannot be deleted."
                 )
@@ -9542,6 +9534,27 @@ def _get_or_create_default_project(self) -> ProjectResponse:
                 ProjectRequest(name=default_project_name)
             )
 
+    @property
+    def _default_project_enabled(self) -> bool:
+        """Check if the default project is enabled.
+
+        When running in a Pro ZenML server, the default project is not enabled.
+
+        Returns:
+            True if the default project is enabled, False otherwise.
+        """
+        default_project_enabled = True
+        if ENV_ZENML_SERVER in os.environ:
+            from zenml.config.server_config import ServerConfiguration
+
+            if (
+                ServerConfiguration.get_server_config().deployment_type
+                == ServerDeploymentType.CLOUD
+            ):
+                default_project_enabled = False
+
+        return default_project_enabled
+
     # =======================
     # Internal helper methods
     # =======================
@@ -9915,7 +9928,7 @@ def _set_filter_project_id(
                 session=session,
             )
             project_id = project.id
-        else:
+        elif self._default_project_enabled:
             # Use the default project as a last resort.
             try:
                 project = self._get_schema_by_name_or_id(
@@ -9926,6 +9939,8 @@ def _set_filter_project_id(
                 project_id = project.id
             except KeyError:
                 raise ValueError("Project scope missing from the filter")
+        else:
+            raise ValueError("Project scope missing from the filter")
 
         filter_model.project = project_id