zenml-nightly 0.80.0.dev20250325__py3-none-any.whl → 0.80.0.dev20250327__py3-none-any.whl

zenml/VERSION

@@ -1 +1 @@
-0.80.0.dev20250325
+0.80.0.dev20250327

zenml/cli/project.py

@@ -75,11 +75,18 @@ def list_projects(ctx: click.Context, /, **kwargs: Any) -> None:
     required=False,
     help="The display name of the project.",
 )
+@click.option(
+    "--set-default",
+    "set_default",
+    is_flag=True,
+    help="Set this project as the default project.",
+)
 @click.argument("project_name", type=str, required=True)
 def register_project(
     project_name: str,
     set_project: bool = False,
     display_name: Optional[str] = None,
+    set_default: bool = False,
 ) -> None:
     """Register a new project.
 
@@ -87,12 +94,13 @@ def register_project(
         project_name: The name of the project to register.
         set_project: Whether to set the project as active.
         display_name: The display name of the project.
+        set_default: Whether to set the project as the default project.
     """
     check_zenml_pro_project_availability()
     client = Client()
     with console.status("Creating project...\n"):
         try:
-            client.create_project(
+            project = client.create_project(
                 project_name,
                 description="",
                 display_name=display_name,
@@ -105,26 +113,51 @@ def register_project(
         client.set_active_project(project_name)
         cli_utils.declare(f"The active project has been set to {project_name}")
 
+    if set_default:
+        client.update_user(
+            name_id_or_prefix=client.active_user.id,
+            updated_default_project_id=project.id,
+        )
+        cli_utils.declare(
+            f"The default project has been set to {project.name}"
+        )
+
 
 @project.command("set")
 @click.argument("project_name_or_id", type=str, required=True)
-def set_project(project_name_or_id: str) -> None:
+@click.option(
+    "--default",
+    "default",
+    is_flag=True,
+    help="Set this project as the default project.",
+)
+def set_project(project_name_or_id: str, default: bool = False) -> None:
     """Set the active project.
 
     Args:
         project_name_or_id: The name or ID of the project to set as active.
+        default: Whether to set the project as the default project.
     """
     check_zenml_pro_project_availability()
     client = Client()
     with console.status("Setting project...\n"):
         try:
-            client.set_active_project(project_name_or_id)
+            project = client.set_active_project(project_name_or_id)
             cli_utils.declare(
                 f"The active project has been set to {project_name_or_id}"
             )
         except Exception as e:
             cli_utils.error(str(e))
 
+    if default:
+        client.update_user(
+            name_id_or_prefix=client.active_user.id,
+            updated_default_project_id=project.id,
+        )
+        cli_utils.declare(
+            f"The default project has been set to {project.name}"
+        )
+
 
 @project.command("describe")
 @click.argument("project_name_or_id", type=str, required=False)

zenml/cli/server.py

@@ -193,6 +193,7 @@ def status() -> None:
 
     gc = GlobalConfiguration()
     client = Client()
+    _ = client.zen_store
 
     store_cfg = gc.store_configuration
 

zenml/client.py

@@ -235,7 +235,7 @@ class ClientConfiguration(FileSyncModel):
         else:
             raise RuntimeError(
                 "No active project is configured. Run "
-                "`zenml project set PROJECT_NAME` to set the active "
+                "`zenml project set <NAME>` to set the active "
                 "project."
             )
 
@@ -876,6 +876,7 @@ class Client(metaclass=ClientMetaClass):
         old_password: Optional[str] = None,
         updated_is_admin: Optional[bool] = None,
         updated_metadata: Optional[Dict[str, Any]] = None,
+        updated_default_project_id: Optional[UUID] = None,
         active: Optional[bool] = None,
     ) -> UserResponse:
         """Update a user.
@@ -891,6 +892,7 @@ class Client(metaclass=ClientMetaClass):
                 update.
             updated_is_admin: Whether the user should be an admin.
             updated_metadata: The new metadata for the user.
+            updated_default_project_id: The new default project ID for the user.
             active: Use to activate or deactivate the user.
 
         Returns:
@@ -928,6 +930,9 @@ class Client(metaclass=ClientMetaClass):
         if updated_metadata is not None:
             user_update.user_metadata = updated_metadata
 
+        if updated_default_project_id is not None:
+            user_update.default_project_id = updated_default_project_id
+
         return self.zen_store.update_user(
             user_id=user.id, user_update=user_update
         )
@@ -1154,7 +1159,7 @@ class Client(metaclass=ClientMetaClass):
         if not project:
             raise RuntimeError(
                 "No active project is configured. Run "
-                "`zenml project set PROJECT_NAME` to set the active "
+                "`zenml project set <NAME>` to set the active "
                 "project."
             )
 

zenml/config/global_config.py

@@ -113,7 +113,7 @@ class GlobalConfiguration(BaseModel, metaclass=GlobalConfigMetaClass):
             global config.
         store: Store configuration.
         active_stack_id: The ID of the active stack.
-        active_project_name: The name of the active project.
+        active_project_id: The ID of the active project.
     """
 
     user_id: uuid.UUID = Field(default_factory=uuid.uuid4)
@@ -123,7 +123,7 @@ class GlobalConfiguration(BaseModel, metaclass=GlobalConfigMetaClass):
     version: Optional[str] = None
     store: Optional[SerializeAsAny[StoreConfiguration]] = None
     active_stack_id: Optional[uuid.UUID] = None
-    active_project_name: Optional[str] = None
+    active_project_id: Optional[uuid.UUID] = None
 
     _zen_store: Optional["BaseZenStore"] = None
     _active_project: Optional["ProjectResponse"] = None
@@ -393,15 +393,15 @@ class GlobalConfiguration(BaseModel, metaclass=GlobalConfigMetaClass):
         if ENV_ZENML_SERVER in os.environ:
             return
         active_project, active_stack = self.zen_store.validate_active_config(
-            self.active_project_name,
+            self.active_project_id,
             self.active_stack_id,
             config_name="global",
         )
         if active_project:
-            self.active_project_name = active_project.name
+            self.active_project_id = active_project.id
             self._active_project = active_project
         else:
-            self.active_project_name = None
+            self.active_project_id = None
             self._active_project = None
 
         self.set_active_stack(active_stack)
@@ -730,7 +730,7 @@ class GlobalConfiguration(BaseModel, metaclass=GlobalConfigMetaClass):
         Returns:
             The project that was set active.
         """
-        self.active_project_name = project.name
+        self.active_project_id = project.id
         self._active_project = project
         # Sanitize the global configuration to reflect the new project
         self._sanitize_config()
@@ -751,35 +751,35 @@ class GlobalConfiguration(BaseModel, metaclass=GlobalConfigMetaClass):
         Returns:
             The model of the active project.
         """
-        project_name = self.get_active_project_name()
+        project_id = self.get_active_project_id()
 
         if self._active_project is not None:
             return self._active_project
 
         project = self.zen_store.get_project(
-            project_name_or_id=project_name,
+            project_name_or_id=project_id,
         )
         return self.set_active_project(project)
 
-    def get_active_project_name(self) -> str:
-        """Get the name of the active project.
+    def get_active_project_id(self) -> UUID:
+        """Get the ID of the active project.
 
         Returns:
-            The name of the active project.
+            The ID of the active project.
 
         Raises:
             RuntimeError: If the active project is not set.
         """
-        if self.active_project_name is None:
+        if self.active_project_id is None:
             _ = self.zen_store
-            if self.active_project_name is None:
+            if self.active_project_id is None:
                 raise RuntimeError(
                     "No project is currently set as active. Please set the "
-                    "active project using the `zenml project set` CLI "
+                    "active project using the `zenml project set <NAME>` CLI "
                     "command."
                 )
 
-        return self.active_project_name
+        return self.active_project_id
 
     def get_active_stack_id(self) -> UUID:
         """Get the ID of the active stack.

zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py

@@ -238,7 +238,7 @@ class KubernetesOrchestrator(ContainerizedOrchestrator):
                         )
                     if kubernetes_context != active_context:
                         logger.warning(
-                            f"{msg}the Kubernetes context "
+                            f"{msg}the Kubernetes context "  # nosec
                             f"'{kubernetes_context}' configured for the "
                             f"Kubernetes orchestrator is not the same as the "
                             f"active context in the local Kubernetes "

zenml/models/v2/core/user.py

@@ -203,6 +203,10 @@ class UserUpdate(UserBase, BaseUpdate):
         "accounts. Required when updating the password.",
         max_length=STR_FIELD_MAX_LENGTH,
     )
+    default_project_id: Optional[UUID] = Field(
+        default=None,
+        title="The default project ID for the user.",
+    )
 
     @model_validator(mode="after")
     def user_email_updates(self) -> "UserUpdate":
@@ -279,6 +283,10 @@ class UserResponseBody(BaseDatedResponseBody):
     is_admin: bool = Field(
         title="Whether the account is an administrator.",
     )
+    default_project_id: Optional[UUID] = Field(
+        default=None,
+        title="The default project ID for the user.",
+    )
 
 
 class UserResponseMetadata(BaseResponseMetadata):
@@ -422,6 +430,15 @@ class UserResponse(
         """
         return self.get_metadata().user_metadata
 
+    @property
+    def default_project_id(self) -> Optional[UUID]:
+        """The `default_project_id` property.
+
+        Returns:
+            the value of the property.
+        """
+        return self.get_body().default_project_id
+
     # Helper methods
     @classmethod
     def _get_crypt_context(cls) -> "CryptContext":

zenml/service_connectors/service_connector_registry.py

@@ -216,30 +216,28 @@ class ServiceConnectorRegistry:
                     AWSServiceConnector,
                 )
             except ImportError as e:
-                logger.warning(f"Could not import AWS service connector: {e}.")
+                logger.debug(f"Could not import AWS service connector: {e}.")
 
             try:
                 from zenml.integrations.gcp.service_connectors.gcp_service_connector import (  # noqa
                     GCPServiceConnector,
                 )
             except ImportError as e:
-                logger.warning(f"Could not import GCP service connector: {e}.")
+                logger.debug(f"Could not import GCP service connector: {e}.")
 
             try:
                 from zenml.integrations.azure.service_connectors.azure_service_connector import (  # noqa
                     AzureServiceConnector,
                 )
             except ImportError as e:
-                logger.warning(
-                    f"Could not import Azure service connector: {e}."
-                )
+                logger.debug(f"Could not import Azure service connector: {e}.")
 
             try:
                 from zenml.integrations.kubernetes.service_connectors.kubernetes_service_connector import (  # noqa
                     KubernetesServiceConnector,
                 )
             except ImportError as e:
-                logger.warning(
+                logger.debug(
                     f"Could not import Kubernetes service connector: {e}."
                 )
 
@@ -248,7 +246,7 @@ class ServiceConnectorRegistry:
                     DockerServiceConnector,
                 )
             except ImportError as e:
-                logger.warning(
+                logger.debug(
                     f"Could not import Docker service connector: {e}."
                 )
 
@@ -257,7 +255,7 @@ class ServiceConnectorRegistry:
                     HyperAIServiceConnector,
                 )
             except ImportError as e:
-                logger.warning(
+                logger.debug(
                     f"Could not import HyperAI service connector: {e}."
                 )
 

zenml/zen_server/routers/users_endpoints.py

@@ -65,6 +65,11 @@ from zenml.zen_server.utils import (
 
 logger = get_logger(__name__)
 
+pass_change_limiter = RequestLimiter(
+    day_limit=server_config().login_rate_limit_day,
+    minute_limit=server_config().login_rate_limit_minute,
+)
+
 router = APIRouter(
     prefix=API + VERSION_1 + USERS,
     tags=["users"],
@@ -223,179 +228,6 @@ def get_user(
 # When the auth scheme is set to EXTERNAL, users cannot be updated via the
 # API.
 if server_config().auth_scheme != AuthScheme.EXTERNAL:
-    pass_change_limiter = RequestLimiter(
-        day_limit=server_config().login_rate_limit_day,
-        minute_limit=server_config().login_rate_limit_minute,
-    )
-
-    @router.put(
-        "/{user_name_or_id}",
-        responses={
-            401: error_response,
-            404: error_response,
-            422: error_response,
-        },
-    )
-    @handle_exceptions
-    def update_user(
-        user_name_or_id: Union[str, UUID],
-        user_update: UserUpdate,
-        request: Request,
-        auth_context: AuthContext = Security(authorize),
-    ) -> UserResponse:
-        """Updates a specific user.
-
-        Args:
-            user_name_or_id: Name or ID of the user.
-            user_update: the user to use for the update.
-            request: The request object.
-            auth_context: Authentication context.
-
-        Returns:
-            The updated user.
-
-        Raises:
-            IllegalOperationError: if the user tries change admin status,
-                while not an admin, if the user tries to change the password
-                of another user, or if the user tries to change their own
-                password without providing the old password or providing
-                an incorrect old password.
-        """
-        user = zen_store().get_user(user_name_or_id)
-
-        # Use a separate object to compute the update that will be applied to
-        # the user to avoid giving the API requester direct control over the
-        # user attributes that are updated.
-        #
-        # Exclude attributes that cannot be updated through this endpoint:
-        #
-        # - activation_token
-        # - external_user_id
-        # - old_password
-        #
-        # Exclude things that are not always safe to update and need to be
-        # validated first:
-        #
-        # - admin
-        # - active
-        # - password
-        # - email_opted_in + email
-        #
-        safe_user_update = user_update.create_copy(
-            exclude={
-                "activation_token",
-                "external_user_id",
-                "is_admin",
-                "active",
-                "password",
-                "old_password",
-                "email_opted_in",
-                "email",
-            },
-        )
-
-        if user.id != auth_context.user.id:
-            verify_admin_status_if_no_rbac(
-                auth_context.user.is_admin, "update other user account"
-            )
-            # verify_permission_for_model(
-            #     user,
-            #     action=Action.UPDATE,
-            # )
-
-        # Validate a password change
-        if user_update.password is not None:
-            if user.id != auth_context.user.id:
-                raise IllegalOperationError(
-                    "Users cannot change the password of other users. Use the "
-                    "account deactivation and activation flow instead."
-                )
-
-            # If the user is updating their own password, we need to verify
-            # the old password
-            if user_update.old_password is None:
-                raise IllegalOperationError(
-                    "The current password must be supplied when changing the "
-                    "password."
-                )
-
-            with pass_change_limiter.limit_failed_requests(request):
-                auth_user = zen_store().get_auth_user(user_name_or_id)
-                if not UserAuthModel.verify_password(
-                    user_update.old_password, auth_user
-                ):
-                    raise IllegalOperationError(
-                        "The current password is incorrect."
-                    )
-
-            # Accept the password update
-            safe_user_update.password = user_update.password
-
-        # Validate an admin status change
-        if (
-            user_update.is_admin is not None
-            and user.is_admin != user_update.is_admin
-        ):
-            if user.id == auth_context.user.id:
-                raise IllegalOperationError(
-                    "Cannot change the admin status of your own user account."
-                )
-
-            if (
-                user.id != auth_context.user.id
-                and not auth_context.user.is_admin
-            ):
-                raise IllegalOperationError(
-                    "Only admins are allowed to change the admin status of "
-                    "other user accounts."
-                )
-
-            # Accept the admin status update
-            safe_user_update.is_admin = user_update.is_admin
-
-        # Validate an active status change
-        if (
-            user_update.active is not None
-            and user.active != user_update.active
-        ):
-            if user.id == auth_context.user.id:
-                raise IllegalOperationError(
-                    "Cannot change the active status of your own user account."
-                )
-
-            if (
-                user.id != auth_context.user.id
-                and not auth_context.user.is_admin
-            ):
-                raise IllegalOperationError(
-                    "Only admins are allowed to change the active status of "
-                    "other user accounts."
-                )
-
-            # Accept the admin status update
-            safe_user_update.is_admin = user_update.is_admin
-
-        # Validate changes to private user account information
-        if (
-            user_update.email_opted_in is not None
-            or user_update.email is not None
-        ):
-            if user.id != auth_context.user.id:
-                raise IllegalOperationError(
-                    "Cannot change the private user account information for "
-                    "another user account."
-                )
-
-            # Accept the private user account information update
-            if safe_user_update.email_opted_in is not None:
-                safe_user_update.email_opted_in = user_update.email_opted_in
-                safe_user_update.email = user_update.email
-
-        updated_user = zen_store().update_user(
-            user_id=user.id,
-            user_update=safe_user_update,
-        )
-        return dehydrate_response_model(updated_user)
 
     @activation_router.put(
         "/{user_name_or_id}" + ACTIVATE,
@@ -599,6 +431,170 @@ if server_config().auth_scheme != AuthScheme.EXTERNAL:
             )
 
 
+@router.put(
+    "/{user_name_or_id}",
+    responses={
+        401: error_response,
+        404: error_response,
+        422: error_response,
+    },
+)
+@handle_exceptions
+def update_user(
+    user_name_or_id: Union[str, UUID],
+    user_update: UserUpdate,
+    request: Request,
+    auth_context: AuthContext = Security(authorize),
+) -> UserResponse:
+    """Updates a specific user.
+
+    Args:
+        user_name_or_id: Name or ID of the user.
+        user_update: the user to use for the update.
+        request: The request object.
+        auth_context: Authentication context.
+
+    Returns:
+        The updated user.
+
+    Raises:
+        IllegalOperationError: if the user tries change admin status,
+            while not an admin, if the user tries to change the password
+            of another user, or if the user tries to change their own
+            password without providing the old password or providing
+            an incorrect old password.
+    """
+    if server_config().auth_scheme == AuthScheme.EXTERNAL:
+        # For external auth, we only allow updating the default project
+        user_update = UserUpdate(
+            default_project_id=user_update.default_project_id,
+        )
+
+    user = zen_store().get_user(user_name_or_id)
+
+    # Use a separate object to compute the update that will be applied to
+    # the user to avoid giving the API requester direct control over the
+    # user attributes that are updated.
+    #
+    # Exclude attributes that cannot be updated through this endpoint:
+    #
+    # - activation_token
+    # - external_user_id
+    # - old_password
+    #
+    # Exclude things that are not always safe to update and need to be
+    # validated first:
+    #
+    # - admin
+    # - active
+    # - password
+    # - email_opted_in + email
+    #
+    safe_user_update = user_update.create_copy(
+        exclude={
+            "activation_token",
+            "external_user_id",
+            "is_admin",
+            "active",
+            "password",
+            "old_password",
+            "email_opted_in",
+            "email",
+        },
+    )
+
+    if user.id != auth_context.user.id:
+        verify_admin_status_if_no_rbac(
+            auth_context.user.is_admin, "update other user account"
+        )
+        # verify_permission_for_model(
+        #     user,
+        #     action=Action.UPDATE,
+        # )
+
+    # Validate a password change
+    if user_update.password is not None:
+        if user.id != auth_context.user.id:
+            raise IllegalOperationError(
+                "Users cannot change the password of other users. Use the "
+                "account deactivation and activation flow instead."
+            )
+
+        # If the user is updating their own password, we need to verify
+        # the old password
+        if user_update.old_password is None:
+            raise IllegalOperationError(
+                "The current password must be supplied when changing the "
+                "password."
+            )
+
+        with pass_change_limiter.limit_failed_requests(request):
+            auth_user = zen_store().get_auth_user(user_name_or_id)
+            if not UserAuthModel.verify_password(
+                user_update.old_password, auth_user
+            ):
+                raise IllegalOperationError(
+                    "The current password is incorrect."
+                )
+
+        # Accept the password update
+        safe_user_update.password = user_update.password
+
+    # Validate an admin status change
+    if (
+        user_update.is_admin is not None
+        and user.is_admin != user_update.is_admin
+    ):
+        if user.id == auth_context.user.id:
+            raise IllegalOperationError(
+                "Cannot change the admin status of your own user account."
+            )
+
+        if user.id != auth_context.user.id and not auth_context.user.is_admin:
+            raise IllegalOperationError(
+                "Only admins are allowed to change the admin status of "
+                "other user accounts."
+            )
+
+        # Accept the admin status update
+        safe_user_update.is_admin = user_update.is_admin
+
+    # Validate an active status change
+    if user_update.active is not None and user.active != user_update.active:
+        if user.id == auth_context.user.id:
+            raise IllegalOperationError(
+                "Cannot change the active status of your own user account."
+            )
+
+        if user.id != auth_context.user.id and not auth_context.user.is_admin:
+            raise IllegalOperationError(
+                "Only admins are allowed to change the active status of "
+                "other user accounts."
+            )
+
+        # Accept the admin status update
+        safe_user_update.is_admin = user_update.is_admin
+
+    # Validate changes to private user account information
+    if user_update.email_opted_in is not None or user_update.email is not None:
+        if user.id != auth_context.user.id:
+            raise IllegalOperationError(
+                "Cannot change the private user account information for "
+                "another user account."
+            )
+
+        # Accept the private user account information update
+        if safe_user_update.email_opted_in is not None:
+            safe_user_update.email_opted_in = user_update.email_opted_in
+            safe_user_update.email = user_update.email
+
+    updated_user = zen_store().update_user(
+        user_id=user.id,
+        user_update=safe_user_update,
+    )
+    return dehydrate_response_model(updated_user)
+
+
 @current_user_router.get(
     "/current-user",
     responses={401: error_response, 404: error_response, 422: error_response},

zenml/zen_stores/base_zen_store.py

@@ -22,7 +22,6 @@ from typing import (
     Optional,
     Tuple,
     Type,
-    Union,
 )
 from uuid import UUID
 
@@ -46,6 +45,7 @@ from zenml.enums import (
 from zenml.exceptions import IllegalOperationError
 from zenml.logger import get_logger
 from zenml.models import (
+    ProjectFilter,
     ProjectResponse,
     ServerDatabaseType,
     ServerDeploymentType,
@@ -232,7 +232,6 @@ class BaseZenStore(
         Returns:
             The initialized store.
         """
-        logger.debug(f"Creating store with config '{config}'...")
         store_class = BaseZenStore.get_store_class(config.type)
         store = store_class(
             config=config,
@@ -292,7 +291,7 @@ class BaseZenStore(
 
     def validate_active_config(
         self,
-        active_project_name_or_id: Optional[Union[str, UUID]] = None,
+        active_project_id: Optional[UUID] = None,
         active_stack_id: Optional[UUID] = None,
         config_name: str = "",
     ) -> Tuple[Optional[ProjectResponse], StackResponse]:
@@ -307,7 +306,7 @@ class BaseZenStore(
         stack will be returned in their stead.
 
         Args:
-            active_project_name_or_id: The name or ID of the active project.
+            active_project_id: The ID of the active project.
             active_stack_id: The ID of the active stack.
             config_name: The name of the configuration to validate (used in the
                 displayed logs/messages).
@@ -317,30 +316,62 @@ class BaseZenStore(
         """
         active_project: Optional[ProjectResponse] = None
 
-        if active_project_name_or_id:
+        if active_project_id:
             try:
-                active_project = self.get_project(active_project_name_or_id)
+                active_project = self.get_project(active_project_id)
             except (KeyError, IllegalOperationError):
-                active_project_name_or_id = None
+                active_project_id = None
                 logger.warning(
                     f"The current {config_name} active project is no longer "
                     f"available."
                 )
 
+        if active_project is None:
+            user = self.get_user()
+            if user.default_project_id:
+                try:
+                    active_project = self.get_project(user.default_project_id)
+                except (KeyError, IllegalOperationError):
+                    logger.warning(
+                        "The default project %s for the active user is no longer "
+                        "available.",
+                        user.default_project_id,
+                    )
+                else:
+                    logger.info(
+                        f"Setting the {config_name} active project "
+                        f"to '{active_project.name}'."
+                    )
+
         if active_project is None:
             try:
-                active_project = self._get_default_project()
-            except (KeyError, IllegalOperationError):
+                projects = self.list_projects(
+                    project_filter_model=ProjectFilter()
+                )
+            except Exception:
+                # There was some failure, we force the user to set the active
+                # project manually
                 logger.warning(
                     "An active project is not set. Please set the active "
-                    "project by running `zenml project set "
-                    "<project-name>`."
+                    "project by running `zenml project set <NAME>`."
                 )
             else:
-                logger.info(
-                    f"Setting the {config_name} active project "
-                    f"to '{active_project.name}'."
-                )
+                if len(projects) == 0:
+                    logger.warning(
+                        "No available projects. Please create a project by "
+                        "running `zenml project register <NAME> --set`."
+                    )
+                elif len(projects) == 1:
+                    active_project = projects.items[0]
+                    logger.info(
+                        f"Setting the {config_name} active project "
+                        f"to '{active_project.name}'."
+                    )
+                else:
+                    logger.warning(
+                        "Multiple projects are available. Please set the "
+                        "active project by running `zenml project set <NAME>`."
+                    )
 
         active_stack: StackResponse
 

zenml/zen_stores/migrations/versions/1f1105204aaf_add_user_default_project.py

@@ -0,0 +1,51 @@
+"""Add user default project [1f1105204aaf].
+
+Revision ID: 1f1105204aaf
+Revises: 0.80.0
+Create Date: 2025-03-25 11:11:13.480745
+
+"""
+
+import sqlalchemy as sa
+import sqlmodel
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "1f1105204aaf"
+down_revision = "0.80.0"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    """Upgrade database schema and/or data, creating a new revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.add_column(
+            sa.Column(
+                "default_project_id",
+                sqlmodel.sql.sqltypes.GUID(),
+                nullable=True,
+            )
+        )
+        batch_op.create_foreign_key(
+            "fk_user_default_project_id_project",
+            "project",
+            ["default_project_id"],
+            ["id"],
+            ondelete="SET NULL",
+        )
+
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade database schema and/or data back to the previous revision."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.drop_constraint(
+            "fk_user_default_project_id_project", type_="foreignkey"
+        )
+        batch_op.drop_column("default_project_id")
+
+    # ### end Alembic commands ###

zenml/zen_stores/schemas/user_schemas.py

@@ -34,6 +34,7 @@ from zenml.models import (
 )
 from zenml.utils.time_utils import utc_now
 from zenml.zen_stores.schemas.base_schemas import NamedSchema
+from zenml.zen_stores.schemas.schema_utils import build_foreign_key_field
 
 if TYPE_CHECKING:
     from zenml.zen_stores.schemas import (
@@ -83,6 +84,15 @@ class UserSchema(NamedSchema, table=True):
     is_admin: bool = Field(default=False)
     user_metadata: Optional[str] = Field(nullable=True)
 
+    default_project_id: Optional[UUID] = build_foreign_key_field(
+        source=__tablename__,
+        target="project",
+        source_column="default_project_id",
+        target_column="id",
+        ondelete="SET NULL",
+        nullable=True,
+    )
+
     stacks: List["StackSchema"] = Relationship(back_populates="user")
     components: List["StackComponentSchema"] = Relationship(
         back_populates="user",
@@ -299,6 +309,7 @@ class UserSchema(NamedSchema, table=True):
                 created=self.created,
                 updated=self.updated,
                 is_admin=self.is_admin,
+                default_project_id=self.default_project_id,
             ),
             metadata=metadata,
         )

{zenml_nightly-0.80.0.dev20250325.dist-info → zenml_nightly-0.80.0.dev20250327.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.80.0.dev20250325
+Version: 0.80.0.dev20250327
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.80.0.dev20250325.dist-info → zenml_nightly-0.80.0.dev20250327.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=q0fYek6a4p9kBXsu1CnsztzozW1boCTn9m_2yfxmbqY,19
+zenml/VERSION,sha256=aOovWdLqRIV4qVkeYqC1inlhULjhqv_F5RCcFDdaXWc,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -42,10 +42,10 @@ zenml/cli/login.py,sha256=-tPuTaWctRDkWZOcY4Af2uens6_oEWFkehl8ITeKFHk,38836
 zenml/cli/model.py,sha256=7lmeMLwD3BIWaw_P3WBFRe42Twfvb11K67p_xmvRWH0,22817
 zenml/cli/model_registry.py,sha256=zzxWXXFhKu2B1Wp0u7prKVnN1ftM-JdGdQwlD-5G-QM,20786
 zenml/cli/pipeline.py,sha256=Vlz1OgGb1Ep-4Ekgd-Wz5SmieWigfx56i8wA5BGl228,19222
-zenml/cli/project.py,sha256=7kqc-MaNr7gFbkl2NlAPwQ4jMipRX-TW_Eml5fQJqao,5417
+zenml/cli/project.py,sha256=MD0HNHc5HSOYvG-mMNJGvDXv0DlGCZsnSDBzwEhQo9s,6407
 zenml/cli/secret.py,sha256=zwt07v0xoIf_dLf-qY5CFdbKepBEuwmXD2HIMcLe_xU,20164
 zenml/cli/served_model.py,sha256=3w1UcAbg6Geu37fr7ej1_81GBCt3fF7j3Ge799YE4Mc,14974
-zenml/cli/server.py,sha256=30DYWQI9cNQgRTXeX2ff23RAdQcb9kYmDAtUNfV0SoQ,26105
+zenml/cli/server.py,sha256=EmElo8MlPujj15waddSN7wb7M7vu0HCfkIFgILGLL6Q,26130
 zenml/cli/service_accounts.py,sha256=DkSrBroYvQ3zNDHVYbY9IxqmbN7f8wGsAmt0Wyjq1FE,17766
 zenml/cli/service_connectors.py,sha256=ZGo1X3UwoHZ2BEFVev216Vfaf9wBEb0PKduwtX3mnk4,74332
 zenml/cli/stack.py,sha256=J5FzZWOrn9RLoJ6WjsxDl75pcTrBMiPOJeQj_lNix6U,67236
@@ -55,7 +55,7 @@ zenml/cli/text_utils.py,sha256=bY1GIjoULt1cW2FyrPlMoAXNS2R7cSOjDFEZQqrpVQ8,3553
 zenml/cli/user_management.py,sha256=sNnhaUxH-cHecbZBR1L0mEU0TnLNZHzI6ZBCUSQa7OY,13078
 zenml/cli/utils.py,sha256=6Bv2-gqkHC8DdUT4b_HL6pgGMErY8lwjuJb-eHgPTJA,86383
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
-zenml/client.py,sha256=y_5faIEjAj0WeTG3NkNX4ayjFAWH2g1cNTFPnsji3PY,292150
+zenml/client.py,sha256=l8nHsajF_Ctn3ne4unrn7OHW-VG4OKOKObt-POYFZzE,292402
 zenml/client_lazy_loader.py,sha256=MOBgS1ITYqGvPUnWQ6edn9s8Hr_72YfWbwEIfHKUr9g,7104
 zenml/code_repositories/__init__.py,sha256=W5bDfzAG8OXIKZSV1L-VHuzMcSCYa9qzTdPb3jqfyYw,920
 zenml/code_repositories/base_code_repository.py,sha256=Id6VjbUu8N3ZpNvBGhOgbahtoMiCAtYXed3G7YQ_iAc,5225
@@ -68,7 +68,7 @@ zenml/config/build_configuration.py,sha256=Jsng7ebpeaRbzXlbszU-uYkaVihgQ4OrD839y
 zenml/config/compiler.py,sha256=pBRJRAzjHloYShJiknDugbH5FGiit5G1o2IJ2dnE9-E,23318
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
 zenml/config/docker_settings.py,sha256=KTUR9FaRn3ivcagzFgpLe4eev8fvd5Wq1oV8wGNMB-g,13322
-zenml/config/global_config.py,sha256=pmw99M1gW5BE9ShHXz0y0UIiDlGShxgAe38jazLQyJM,29578
+zenml/config/global_config.py,sha256=3w2PLnbfmu6qUdJ1LIn1cdmylSy7VhB2pG1_sxbOPjQ,29556
 zenml/config/pipeline_configurations.py,sha256=f6mxkTFCCKhGXSPgNQ2JUdBzfL2m_VYuRxWG-OKx54M,3751
 zenml/config/pipeline_run_configuration.py,sha256=uWTD4778HcCjDpPSfGmBlVqddxTq4cq1_du5kQIIGBA,2263
 zenml/config/pipeline_spec.py,sha256=uWpiIfznJvXAiKs1yMIUDT1h1tYEFNO-RDVTYcIv9CE,2821
@@ -338,7 +338,7 @@ zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=0
 zenml/integrations/kubernetes/flavors/kubernetes_step_operator_flavor.py,sha256=ILN-H4cl7z3i4ltb4UBs55wbtIo871b4ib28pYkQoyQ,5605
 zenml/integrations/kubernetes/orchestrators/__init__.py,sha256=TJID3OTieZBox36WpQpzD0jdVRA_aZVcs_bNtfXS8ik,811
 zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=roAYEGTvjnjCERCbjnN2o4v6kL_2xKI-eFnhiRBs1oY,14759
-zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=sahuh1is5zj642F0QEHta3-WZQWBc3EEMCusnV7RK2U,24801
+zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=GsOyxmDW6BUNSowK-vWCVXxsO-P7xROtIB9vmP3l1Kg,24810
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=T8r6k9sRBl_0PvsiG-k56gT9uG9VB4YxYrIoiGQZ4IM,7721
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=Y7uGU8eksMluGyXYsf688CwpiXwI_W6WYLscYwRZXRY,2494
 zenml/integrations/kubernetes/orchestrators/manifest_utils.py,sha256=Owyuz5Iix-QvCnKObC6xhcuQtNG_ik-8Vbdmk13eWfc,12557
@@ -662,7 +662,7 @@ zenml/models/v2/core/tag.py,sha256=eS9Xzuay91OURfKfNa_fE1CtGmlmPkFDIr6T_QWFZ34,7
 zenml/models/v2/core/tag_resource.py,sha256=H7CPj9oaUpym8vSkwuYSJ6_rWfHMEKVRedPw2kyxmII,2640
 zenml/models/v2/core/trigger.py,sha256=L6-pzWPM2HFsfA6-6rM1XeHx4UChZrIY58G6ZG1PA10,12782
 zenml/models/v2/core/trigger_execution.py,sha256=WMYifA3nFKjVXLhx_cg4yMEjGDGo5a8PkV32CzRFhlc,3377
-zenml/models/v2/core/user.py,sha256=evyC5xqduJuLmq_SoVQ19rb94NQS1BRuX37P6ORSR6E,13943
+zenml/models/v2/core/user.py,sha256=Y7Qbe3JSeeovp5x3JOekWu9itX406GB2VHjOvBSNouc,14435
 zenml/models/v2/misc/__init__.py,sha256=knhroJ2h0uHBCGzAiBBGJEiuhEA3cwI6XYBRIyXdbkQ,613
 zenml/models/v2/misc/auth_models.py,sha256=j31CQt9qfH20tyCjY0depqbKP0YZEewcgGKrixqBUbg,3688
 zenml/models/v2/misc/build_item.py,sha256=66Fywatv8bDY86pf0k530fsHTk_L4IkHL6uJL7-6dAo,1938
@@ -715,7 +715,7 @@ zenml/secret/schemas/gcp_secret_schema.py,sha256=tDSxPXlfh4BOSICZjfJjUukfzEyBEgx
 zenml/service_connectors/__init__.py,sha256=gIBAVUPBZtdO6JtEBCixy9YG4wZRA1mnaxaGAxi3T1Q,645
 zenml/service_connectors/docker_service_connector.py,sha256=_6WPqYCcSUf8JPakipbkJRvaN2ghhY7zCr38WCHG0SI,13218
 zenml/service_connectors/service_connector.py,sha256=rVYXZDVp2XJEYIbhkQW3zEjlgsvxqGN9ZOmmVE_pum4,54977
-zenml/service_connectors/service_connector_registry.py,sha256=mCabyKAr7Y6bcAbIFXbir9YrHczDe1ZJ8Bks5bOQouk,9658
+zenml/service_connectors/service_connector_registry.py,sha256=U5MdC9M_RfkHK56AoERB9HtDP4vxk2hDFoZRlWzBA9g,9608
 zenml/service_connectors/service_connector_utils.py,sha256=25eD-2BDWHmRFwpdV5xCAtM3osgGwBzZcZecYRox6Tk,17958
 zenml/services/__init__.py,sha256=BhqBztNXLrqN-4FZr24WHP9ujP_M0plt6axQvakh1jQ,2784
 zenml/services/container/__init__.py,sha256=dFHcmlXgNXjUaCR18oGQJ19-I_6f3UeAUURHjqldjTs,668
@@ -1056,7 +1056,7 @@ zenml/zen_server/routers/steps_endpoints.py,sha256=By-9_VKrJeSo6UUqrggDhGKtrnhxi
 zenml/zen_server/routers/tag_resource_endpoints.py,sha256=AdmbujrwKK6arnSx2VHpJWbZXnlviGlay7LCov4x9Fc,3203
 zenml/zen_server/routers/tags_endpoints.py,sha256=lRjnMaJOdTnooMHoQYIViKfgu32cMq7rI3Lyeaxq4QI,4467
 zenml/zen_server/routers/triggers_endpoints.py,sha256=5fb0EITa3bs--AtKQRQZ5QswHu4pb37OrG5y0OlnGXE,9983
-zenml/zen_server/routers/users_endpoints.py,sha256=4fn-QbdhctQl_KtqW4WCmUpU_YThyAMKy2aLtdz_1Yg,24839
+zenml/zen_server/routers/users_endpoints.py,sha256=b1zZRbVJF8MT3XjwAwj7FJ0scplnQuNT4AzmbCU8PcU,24344
 zenml/zen_server/routers/webhook_endpoints.py,sha256=KOJsuykv_TMjL3oEItpC4OWWP75p-OEvVjRSUBd5Mro,3983
 zenml/zen_server/secure_headers.py,sha256=glh6QujnjyeoH1_FK-tAS-105G-qKS_34AqSzqJ6TRc,4182
 zenml/zen_server/template_execution/__init__.py,sha256=79knXLKfegsvVSVSWecpqrepq6iAavTUA4hKuiDk-WE,613
@@ -1066,7 +1066,7 @@ zenml/zen_server/template_execution/workload_manager_interface.py,sha256=CL9c7z8
 zenml/zen_server/utils.py,sha256=Jc2Q4UBaYG2ruHdsN9JmbOWfWU_eWD9wTBBEgcGAbqg,17439
 zenml/zen_server/zen_server_api.py,sha256=ALyv5096frXXRNySegEtsmkDbHLLqHd402bbQI7RnII,17941
 zenml/zen_stores/__init__.py,sha256=6LTgH6XwDeDxKqVJ1JTfGhmS8II1NLopPloINGmdyI0,691
-zenml/zen_stores/base_zen_store.py,sha256=EQsuB5DrS0ivgSjB2wmr7qL8EN1_Xc7-GAhLyOrXMhM,15997
+zenml/zen_stores/base_zen_store.py,sha256=UtclHihv9-YA0vgbznh8Hp7Rpuivw8ZF7J7oPUM-IuY,17290
 zenml/zen_stores/migrations/README.md,sha256=x04jsb6EOP6PBEGMQlDELiqKEham2O-iztAs9AylMFc,4898
 zenml/zen_stores/migrations/__init__.py,sha256=N9CHfdz0AZ6KniQ450VCIV3H0CuWtx83AloYy82woho,657
 zenml/zen_stores/migrations/alembic.py,sha256=JDqx7Md6DxnHtP3xrZG1I0cNv6NyTR0oO3tPRUPaS2I,7455
@@ -1173,6 +1173,7 @@ zenml/zen_stores/migrations/versions/1ac1b9c04da1_make_secrets_values_optional.p
 zenml/zen_stores/migrations/versions/1cb6477f72d6_move_artifact_save_type.py,sha256=sv2-qjqwVifEiJr1IRERmqBVPaOB1BEkJLyOfJXOyHk,3151
 zenml/zen_stores/migrations/versions/1d74e596abb8_add_run_once_start_time_to_schedule.py,sha256=GIGHE1qkwjske0vtc7XsVBRlIr7A2XKf1-tsfxrov94,1032
 zenml/zen_stores/migrations/versions/1d8f30c54477_migrate_to_new_.py,sha256=ICZLr5MsjGqOnMWwG8W6ixKq3OX2rfs3XkXRzHhSUnA,4200
+zenml/zen_stores/migrations/versions/1f1105204aaf_add_user_default_project.py,sha256=vN-MvyhO-W9D7NVcwaU6_p0JA3MVdl2l-yrntDcnahM,1427
 zenml/zen_stores/migrations/versions/1f9d1cd00b90_add_unique_name_constraints.py,sha256=NSjB4G94c7akAAKT3PpBVKvJHuGMWKob60lPxF65LoA,8044
 zenml/zen_stores/migrations/versions/248dfd320b68_update_size_of_flavor_config_schema.py,sha256=rdHrVpqtRtVPzOUawWXx7SrBfv8tog-uXJQr_k4r4_0,1229
 zenml/zen_stores/migrations/versions/25155145c545_separate_actions_and_triggers.py,sha256=T1A3e0YAtSIdGFmJUjUSfp6f8v82T91XHT-kziui4N4,7588
@@ -1294,7 +1295,7 @@ zenml/zen_stores/schemas/stack_schemas.py,sha256=SkMspxsnR420KU4mZlulPbEOR4DMSRy
 zenml/zen_stores/schemas/step_run_schemas.py,sha256=NuJtCXZ_Zacz4_X7GZ5PQ-fLpzVBhnhScGx2fBqAlDg,16333
 zenml/zen_stores/schemas/tag_schemas.py,sha256=GMKdyT5pZb977vscOW41N2beVt-IHcELSiSSIBHqlUI,6377
 zenml/zen_stores/schemas/trigger_schemas.py,sha256=IOfd6enpAr9mwN5p4KbA2jvQEHlMEUXDXAMqIULS4lc,10711
-zenml/zen_stores/schemas/user_schemas.py,sha256=2cGZQlxr550JXXNzRMwhd6XgnRwYvWcCS9iEb5qdr5I,11110
+zenml/zen_stores/schemas/user_schemas.py,sha256=wLSHtfjjrYBWnX_PXv5KPz_DRdheTxZVjcgBe-cP__c,11497
 zenml/zen_stores/schemas/utils.py,sha256=82-LQj1EV4jp69kKE2GolQw_zTDBvAkyA4cHDaSIj6U,4160
 zenml/zen_stores/secrets_stores/__init__.py,sha256=VGMVerFv3xUWxnp5X_bHyQJwGCeN7BTovUDwho_w-4w,651
 zenml/zen_stores/secrets_stores/aws_secrets_store.py,sha256=o2kftfG0nr6oTodEDHWYvSK6zKbs1WGfqgMd5bJ4k0k,13799
@@ -1308,8 +1309,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=nEO0bAPlULBLxLVk-UTR
 zenml/zen_stores/sql_zen_store.py,sha256=RmcpUHnSlPOMLNZ9StPxi_9FPqB1HgfiK_Humst9sns,440431
 zenml/zen_stores/template_utils.py,sha256=GWBP5QEOyvhzndS_MLPmvh28sQaOPpPoZFXCIX9CRL4,9065
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.80.0.dev20250325.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.80.0.dev20250325.dist-info/METADATA,sha256=kiJ-oLS8naAVjQWFIvU_832XJ3asEuL3EuKvwRYRk64,24227
-zenml_nightly-0.80.0.dev20250325.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-zenml_nightly-0.80.0.dev20250325.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.80.0.dev20250325.dist-info/RECORD,,
+zenml_nightly-0.80.0.dev20250327.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.80.0.dev20250327.dist-info/METADATA,sha256=riu8q-gWcij4HNT2KkKdnB_-vTAiJyuuN66-l53C3-4,24227
+zenml_nightly-0.80.0.dev20250327.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+zenml_nightly-0.80.0.dev20250327.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.80.0.dev20250327.dist-info/RECORD,,