zenml-nightly 0.80.0.dev20250326__tar.gz → 0.80.0.dev20250327__tar.gz

{zenml_nightly-0.80.0.dev20250326 → zenml_nightly-0.80.0.dev20250327}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.80.0.dev20250326
+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.dev20250326 → zenml_nightly-0.80.0.dev20250327}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zenml-nightly"
-version = "0.80.0.dev20250326"
+version = "0.80.0.dev20250327"
 packages = [{ include = "zenml", from = "src" }]
 description = "ZenML: Write production-ready ML code."
 authors = ["ZenML GmbH <info@zenml.io>"]

zenml_nightly-0.80.0.dev20250327/src/zenml/VERSION

@@ -0,0 +1 @@
+0.80.0.dev20250327

{zenml_nightly-0.80.0.dev20250326 → zenml_nightly-0.80.0.dev20250327}/src/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_nightly-0.80.0.dev20250326 → zenml_nightly-0.80.0.dev20250327}/src/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_nightly-0.80.0.dev20250326 → zenml_nightly-0.80.0.dev20250327}/src/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_nightly-0.80.0.dev20250326 → zenml_nightly-0.80.0.dev20250327}/src/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_nightly-0.80.0.dev20250326 → zenml_nightly-0.80.0.dev20250327}/src/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_nightly-0.80.0.dev20250326 → zenml_nightly-0.80.0.dev20250327}/src/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},