zenml-nightly 0.55.0.dev20240124__py3-none-any.whl → 0.72.0.dev20250116__py3-none-any.whl

zenml/cli/utils.py

@@ -12,10 +12,12 @@
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
 """Utility functions for the CLI."""
+
 import contextlib
 import datetime
 import json
 import os
+import platform
 import re
 import subprocess
 import sys
@@ -38,22 +40,23 @@ from typing import (
 )
 
 import click
+import pkg_resources
 import yaml
 from pydantic import BaseModel, SecretStr
 from rich import box, table
+from rich.console import Console
 from rich.emoji import Emoji, NoEmoji
 from rich.markdown import Markdown
 from rich.markup import escape
-from rich.prompt import Confirm
+from rich.prompt import Confirm, Prompt
 from rich.style import Style
+from rich.table import Table
 
 from zenml.client import Client
 from zenml.console import console, zenml_style_defaults
 from zenml.constants import (
     FILTERING_DATETIME_FORMAT,
     IS_DEBUG_ENV,
-    NOT_INSTALLED_MESSAGE,
-    TERRAFORM_NOT_INSTALLED_MESSAGE,
 )
 from zenml.enums import GenericFilterOps, StackComponentType
 from zenml.logger import get_logger
@@ -63,18 +66,19 @@ from zenml.model_registries.base_model_registry import (
 )
 from zenml.models import (
     BaseFilter,
-    BaseResponse,
+    BaseIdentifiedResponse,
     BoolFilter,
     NumericFilter,
     Page,
+    ServiceConnectorRequirements,
     StrFilter,
     UUIDFilter,
 )
+from zenml.models.v2.base.filter import FilterGenerator
 from zenml.services import BaseService, ServiceState
 from zenml.stack import StackComponent
 from zenml.stack.stack_component import StackComponentConfig
 from zenml.utils import secret_utils
-from zenml.zen_server.deploy import ServerDeployment
 
 if TYPE_CHECKING:
     from uuid import UUID
@@ -103,7 +107,7 @@ logger = get_logger(__name__)
 MAX_ARGUMENT_VALUE_SIZE = 10240
 
 
-T = TypeVar("T", bound=BaseResponse)  # type: ignore[type-arg]
+T = TypeVar("T", bound=BaseIdentifiedResponse)  # type: ignore[type-arg]
 
 
 def title(text: str) -> None:
@@ -218,6 +222,8 @@ def print_table(
         caption: Caption of the table.
         columns: Optional column configurations to be used in the table.
     """
+    from rich.text import Text
+
     column_keys = {key: None for dict_ in obj for key in dict_}
     column_names = [columns.get(key, key.upper()) for key in column_keys]
     rich_table = table.Table(
@@ -236,10 +242,19 @@ def print_table(
             if key is None:
                 values.append(None)
             else:
-                value = str(dict_.get(key) or " ")
-                # escape text when square brackets are used
-                if "[" in value:
-                    value = escape(value)
+                v = dict_.get(key) or " "
+                if isinstance(v, str) and (
+                    v.startswith("http://") or v.startswith("https://")
+                ):
+                    # Display the URL as a hyperlink in a way that doesn't break
+                    # the URL when it needs to be wrapped over multiple lines
+                    value: Union[str, Text] = Text(v, style=f"link {v}")
+                else:
+                    value = str(v)
+                    # Escape text when square brackets are used, but allow
+                    # links to be decorated as rich style links
+                    if "[" in value and "[link=" not in value:
+                        value = escape(value)
                 values.append(value)
         rich_table.add_row(*values)
     if len(rich_table.columns) > 1:
@@ -253,6 +268,7 @@ def print_pydantic_models(
     exclude_columns: Optional[List[str]] = None,
     active_models: Optional[List[T]] = None,
     show_active: bool = False,
+    rename_columns: Dict[str, str] = {},
 ) -> None:
     """Prints the list of Pydantic models in a table.
 
@@ -265,6 +281,7 @@ def print_pydantic_models(
         active_models: Optional list of active models of the given type T.
         show_active: Flag to decide whether to append the active model on the
             top of the list.
+        rename_columns: Optional dictionary to rename columns.
     """
     if exclude_columns is None:
         exclude_columns = list()
@@ -285,32 +302,34 @@ def print_pydantic_models(
         """
         # Explicitly defined columns take precedence over exclude columns
         if not columns:
-            if isinstance(model, BaseResponse):
+            if isinstance(model, BaseIdentifiedResponse):
                 include_columns = ["id"]
 
-                if "name" in model.__fields__:
+                if "name" in model.model_fields:
                     include_columns.append("name")
 
                 include_columns.extend(
                     [
                         k
-                        for k in model.__fields__[
-                            "body"
-                        ].type_.__fields__.keys()
-                        if k not in exclude_columns
-                    ]
-                    + [
-                        k
-                        for k in model.__fields__[
-                            "metadata"
-                        ].type_.__fields__.keys()
+                        for k in model.get_body().model_fields.keys()
                         if k not in exclude_columns
                     ]
                 )
 
+                if model.metadata is not None:
+                    include_columns.extend(
+                        [
+                            k
+                            for k in model.get_metadata().model_fields.keys()
+                            if k not in exclude_columns
+                        ]
+                    )
+
             else:
                 include_columns = [
-                    k for k in model.dict().keys() if k not in exclude_columns
+                    k
+                    for k in model.model_dump().keys()
+                    if k not in exclude_columns
                 ]
         else:
             include_columns = columns
@@ -319,11 +338,13 @@ def print_pydantic_models(
 
         for k in include_columns:
             value = getattr(model, k)
+            if k in rename_columns:
+                k = rename_columns[k]
             # In case the response model contains nested `BaseResponse`s
             #  we want to attempt to represent them by name, if they contain
             #  such a field, else the id is used
-            if isinstance(value, BaseResponse):
-                if "name" in value.__fields__:
+            if isinstance(value, BaseIdentifiedResponse):
+                if "name" in value.model_fields:
                     items[k] = str(getattr(value, "name"))
                 else:
                     items[k] = str(value.id)
@@ -332,8 +353,8 @@ def print_pydantic_models(
             #  the list and extract either name or id
             elif isinstance(value, list):
                 for v in value:
-                    if isinstance(v, BaseResponse):
-                        if "name" in v.__fields__:
+                    if isinstance(v, BaseIdentifiedResponse):
+                        if "name" in v.model_fields:
                             items.setdefault(k, []).append(
                                 str(getattr(v, "name"))
                             )
@@ -343,6 +364,7 @@ def print_pydantic_models(
                 items[k] = [str(v) for v in value]
             else:
                 items[k] = str(value)
+
         # prepend an active marker if a function to mark active was passed
         if not active_models and not show_active:
             return items
@@ -352,9 +374,11 @@ def print_pydantic_models(
             marker = "current"
         if active_models is not None and show_active_column:
             return {
-                marker: ":point_right:"
-                if any(model.id == a.id for a in active_models)
-                else "",
+                marker: (
+                    ":point_right:"
+                    if any(model.id == a.id for a in active_models)
+                    else ""
+                ),
                 **items,
             }
 
@@ -418,30 +442,34 @@ def print_pydantic_model(
         exclude_columns = set()
 
     if not columns:
-        if isinstance(model, BaseResponse):
+        if isinstance(model, BaseIdentifiedResponse):
             include_columns = ["id"]
 
-            if "name" in model.__fields__:
+            if "name" in model.model_fields:
                 include_columns.append("name")
 
             include_columns.extend(
                 [
                     k
-                    for k in model.__fields__["body"].type_.__fields__.keys()
-                    if k not in exclude_columns
-                ]
-                + [
-                    k
-                    for k in model.__fields__[
-                        "metadata"
-                    ].type_.__fields__.keys()
+                    for k in model.get_body().model_fields.keys()
                     if k not in exclude_columns
                 ]
             )
 
+            if model.metadata is not None:
+                include_columns.extend(
+                    [
+                        k
+                        for k in model.get_metadata().model_fields.keys()
+                        if k not in exclude_columns
+                    ]
+                )
+
         else:
             include_columns = [
-                k for k in model.dict().keys() if k not in exclude_columns
+                k
+                for k in model.model_dump().keys()
+                if k not in exclude_columns
             ]
     else:
         include_columns = list(columns)
@@ -450,8 +478,8 @@ def print_pydantic_model(
 
     for k in include_columns:
         value = getattr(model, k)
-        if isinstance(value, BaseResponse):
-            if "name" in value.__fields__:
+        if isinstance(value, BaseIdentifiedResponse):
+            if "name" in value.model_fields:
                 items[k] = str(getattr(value, "name"))
             else:
                 items[k] = str(value.id)
@@ -460,8 +488,8 @@ def print_pydantic_model(
         #  the list and extract either name or id
         elif isinstance(value, list):
             for v in value:
-                if isinstance(v, BaseResponse):
-                    if "name" in v.__fields__:
+                if isinstance(v, BaseIdentifiedResponse):
+                    if "name" in v.model_fields:
                         items.setdefault(k, []).append(str(getattr(v, "name")))
                     else:
                         items.setdefault(k, []).append(str(v.id))
@@ -508,43 +536,6 @@ def format_integration_list(
     return list_of_dicts
 
 
-def print_stack_outputs(stack: "StackResponse") -> None:
-    """Prints outputs for stacks deployed with mlstacks.
-
-    Args:
-        stack: Instance of a stack model.
-    """
-    verify_mlstacks_prerequisites_installation()
-
-    if not stack.stack_spec_path:
-        declare("No stack spec path is set for this stack.")
-        return
-    stack_caption = f"'{stack.name}' stack"
-    rich_table = table.Table(
-        box=box.HEAVY_EDGE,
-        title="MLStacks Outputs",
-        caption=stack_caption,
-        show_lines=True,
-    )
-    rich_table.add_column("OUTPUT_KEY", overflow="fold")
-    rich_table.add_column("OUTPUT_VALUE", overflow="fold")
-
-    from mlstacks.utils.terraform_utils import get_stack_outputs
-
-    stack_spec_file = stack.stack_spec_path
-    stack_outputs = get_stack_outputs(stack_path=stack_spec_file)
-
-    for output_key, output_value in stack_outputs.items():
-        rich_table.add_row(output_key, output_value)
-
-    # capitalize entries in first column
-    rich_table.columns[0]._cells = [
-        component.upper()  # type: ignore[union-attr]
-        for component in rich_table.columns[0]._cells
-    ]
-    console.print(rich_table)
-
-
 def print_stack_configuration(stack: "StackResponse", active: bool) -> None:
     """Prints the configuration options of a stack.
 
@@ -572,14 +563,28 @@ def print_stack_configuration(stack: "StackResponse", active: bool) -> None:
         for component in rich_table.columns[0]._cells
     ]
     console.print(rich_table)
+
+    if not stack.labels:
+        declare("No labels are set for this stack.")
+    else:
+        rich_table = table.Table(
+            box=box.HEAVY_EDGE,
+            title="Labels",
+            show_lines=True,
+        )
+        rich_table.add_column("LABEL")
+        rich_table.add_column("VALUE", overflow="fold")
+
+        for label, value in stack.labels.items():
+            rich_table.add_row(label, str(value))
+
+        console.print(rich_table)
+
     declare(
         f"Stack '{stack.name}' with id '{stack.id}' is "
         f"{f'owned by user {stack.user.name}.' if stack.user else 'unowned.'}"
     )
 
-    if stack.stack_spec_path:
-        declare(f"Stack spec path for `mlstacks`: '{stack.stack_spec_path}'")
-
 
 def print_flavor_list(flavors: Page["FlavorResponse"]) -> None:
     """Prints the list of flavors.
@@ -603,13 +608,18 @@ def print_flavor_list(flavors: Page["FlavorResponse"]) -> None:
 
 
 def print_stack_component_configuration(
-    component: "ComponentResponse", active_status: bool
+    component: "ComponentResponse",
+    active_status: bool,
+    connector_requirements: Optional[ServiceConnectorRequirements] = None,
 ) -> None:
     """Prints the configuration options of a stack component.
 
     Args:
         component: The stack component to print.
         active_status: Whether the stack component is active.
+        connector_requirements: Connector requirements for the component, taken
+            from the component flavor. Only needed if the component has a
+            connector.
     """
     if component.user:
         user_name = component.user.name
@@ -618,7 +628,7 @@ def print_stack_component_configuration(
 
     declare(
         f"{component.type.value.title()} '{component.name}' of flavor "
-        f"'{component.flavor}' with id '{component.id}' is owned by "
+        f"'{component.flavor_name}' with id '{component.id}' is owned by "
         f"user '{user_name}'."
     )
 
@@ -665,7 +675,7 @@ def print_stack_component_configuration(
         rich_table.add_column("VALUE", overflow="fold")
 
         for label, value in component.labels.items():
-            rich_table.add_row(label, value)
+            rich_table.add_row(label, str(value))
 
         console.print(rich_table)
 
@@ -680,11 +690,17 @@ def print_stack_component_configuration(
         rich_table.add_column("PROPERTY")
         rich_table.add_column("VALUE", overflow="fold")
 
+        resource_type = (
+            connector_requirements.resource_type
+            if connector_requirements
+            else component.connector.resource_types[0]
+        )
+
         connector_dict = {
             "ID": str(component.connector.id),
             "NAME": component.connector.name,
             "TYPE": component.connector.type,
-            "RESOURCE TYPE": component.connector.resource_types[0],
+            "RESOURCE TYPE": resource_type,
             "RESOURCE NAME": component.connector_resource_id
             or component.connector.resource_id
             or "N/A",
@@ -695,11 +711,6 @@ def print_stack_component_configuration(
 
         console.print(rich_table)
 
-    if component.component_spec_path:
-        declare(
-            f"Component spec path for `mlstacks`: {component.component_spec_path}"
-        )
-
 
 def expand_argument_value_from_file(name: str, value: str) -> str:
     """Expands the value of an argument pointing to a file into the contents of that file.
@@ -915,12 +926,14 @@ def prompt_configuration(
     config_dict = {}
     for attr_name, attr_schema in config_schema.get("properties", {}).items():
         title = attr_schema.get("title", attr_name)
-        attr_type = attr_schema.get("type", "string")
+        attr_type_name = attr_type = attr_schema.get("type", "string")
+        if attr_type == "array":
+            attr_type_name = "list (CSV or JSON)"
         title = f"[{attr_name}] {title}"
         required = attr_name in config_schema.get("required", [])
         hidden = attr_schema.get("format", "") == "password"
         subtitles: List[str] = []
-        subtitles.append(attr_type)
+        subtitles.append(attr_type_name)
         if hidden:
             subtitles.append("secret")
         if required:
@@ -938,6 +951,8 @@ def prompt_configuration(
                 if hidden and not show_secrets:
                     title += " is currently set to: [HIDDEN]"
                 else:
+                    if attr_type == "array":
+                        existing_value = json.dumps(existing_value)
                     title += f" is currently set to: '{existing_value}'"
             else:
                 title += " is not currently set"
@@ -999,12 +1014,19 @@ def prompt_configuration(
 def install_packages(
     packages: List[str],
     upgrade: bool = False,
+    use_uv: bool = False,
 ) -> None:
-    """Installs pypi packages into the current environment with pip.
+    """Installs pypi packages into the current environment with pip or uv.
+
+    When using with `uv`, a virtual environment is required.
 
     Args:
         packages: List of packages to install.
         upgrade: Whether to upgrade the packages if they are already installed.
+        use_uv: Whether to use uv for package installation.
+
+    Raises:
+        e: If the package installation fails.
     """
     if "neptune" in packages:
         declare(
@@ -1013,53 +1035,120 @@ def install_packages(
         )
         uninstall_package("neptune-client")
 
+    if "prodigy" in packages:
+        packages.remove("prodigy")
+        declare(
+            "The `prodigy` package should be installed manually using your "
+            "license key. Please visit https://prodi.gy/docs/install for more "
+            "information."
+        )
+    if not packages:
+        # if user only tried to install prodigy, we can
+        # just return without doing anything
+        return
+
+    pip_command = ["uv", "pip"] if use_uv else ["pip"]
     if upgrade:
-        command = [
-            sys.executable,
-            "-m",
-            "pip",
-            "install",
-            "--upgrade",
-        ] + packages
+        command = (
+            [
+                sys.executable,
+                "-m",
+            ]
+            + pip_command
+            + [
+                "install",
+                "--upgrade",
+            ]
+            + packages
+        )
     else:
-        command = [sys.executable, "-m", "pip", "install"] + packages
+        command = [sys.executable, "-m"] + pip_command + ["install"] + packages
 
     if not IS_DEBUG_ENV:
-        command += [
-            "-qqq",
-            "--no-warn-conflicts",
-        ]
-
-    subprocess.check_call(command)
+        quiet_flag = "-q" if use_uv else "-qqq"
+        command.append(quiet_flag)
+        if not use_uv:
+            command.append("--no-warn-conflicts")
 
-    if "label-studio" in packages:
-        warning(
-            "There is a known issue with Label Studio installations "
-            "via zenml. You might find that the Label Studio "
-            "installation breaks the ZenML CLI. In this case, please "
-            "run `pip install 'pydantic<1.11,>=1.9.0'` to fix the "
-            "issue or message us on Slack if you need help with this. "
-            "We are working on a more definitive fix."
-        )
+    try:
+        subprocess.check_call(command)
+    except subprocess.CalledProcessError as e:
+        if (
+            use_uv
+            and "Failed to locate a virtualenv or Conda environment" in str(e)
+        ):
+            error(
+                "Failed to locate a virtualenv or Conda environment. "
+                "When using uv, a virtual environment is required. "
+                "Run `uv venv` to create a virtualenv and retry."
+            )
+        else:
+            raise e
 
 
-def uninstall_package(package: str) -> None:
-    """Uninstalls pypi package from the current environment with pip.
+def uninstall_package(package: str, use_uv: bool = False) -> None:
+    """Uninstalls pypi package from the current environment with pip or uv.
 
     Args:
         package: The package to uninstall.
+        use_uv: Whether to use uv for package uninstallation.
     """
-    subprocess.check_call(
-        [
-            sys.executable,
-            "-m",
-            "pip",
-            "uninstall",
-            "-qqq",
-            "-y",
-            package,
-        ]
-    )
+    pip_command = ["uv", "pip"] if use_uv else ["pip"]
+    quiet_flag = "-q" if use_uv else "-qqq"
+
+    if use_uv:
+        subprocess.check_call(
+            [
+                sys.executable,
+                "-m",
+            ]
+            + pip_command
+            + [
+                "uninstall",
+                quiet_flag,
+                package,
+            ]
+        )
+    else:
+        subprocess.check_call(
+            [
+                sys.executable,
+                "-m",
+            ]
+            + pip_command
+            + [
+                "uninstall",
+                quiet_flag,
+                "-y",
+                package,
+            ]
+        )
+
+
+def is_uv_installed() -> bool:
+    """Check if uv is installed in the current environment.
+
+    Returns:
+        True if uv is installed, False otherwise.
+    """
+    try:
+        pkg_resources.get_distribution("uv")
+        return True
+    except pkg_resources.DistributionNotFound:
+        return False
+
+
+def is_pip_installed() -> bool:
+    """Check if pip is installed in the current environment.
+
+    Returns:
+        True if pip is installed, False otherwise.
+    """
+    try:
+        pkg_resources.get_distribution("pip")
+        return True
+    except pkg_resources.DistributionNotFound:
+        return False
 
 
 def pretty_print_secret(
@@ -1127,6 +1216,10 @@ def get_service_state_emoji(state: "ServiceState") -> str:
         return ":pause_button:"
     if state == ServiceState.ERROR:
         return ":heavy_exclamation_mark:"
+    if state == ServiceState.PENDING_STARTUP:
+        return ":hourglass:"
+    if state == ServiceState.SCALED_TO_ZERO:
+        return ":chart_decreasing:"
     return ":hourglass_not_done:"
 
 
@@ -1141,15 +1234,18 @@ def pretty_print_model_deployer(
     """
     model_service_dicts = []
     for model_service in model_services:
-        served_model_info = model_deployer.get_model_server_info(model_service)
         dict_uuid = str(model_service.uuid)
         dict_pl_name = model_service.config.pipeline_name
         dict_pl_stp_name = model_service.config.pipeline_step_name
-        dict_model_name = served_model_info.get("MODEL_NAME", "")
+        dict_model_name = model_service.config.model_name
+        type = model_service.SERVICE_TYPE.type
+        flavor = model_service.SERVICE_TYPE.flavor
         model_service_dicts.append(
             {
                 "STATUS": get_service_state_emoji(model_service.status.state),
                 "UUID": dict_uuid,
+                "TYPE": type,
+                "FLAVOR": flavor,
                 "PIPELINE_NAME": dict_pl_name,
                 "PIPELINE_STEP_NAME": dict_pl_stp_name,
                 "MODEL_NAME": dict_model_name,
@@ -1194,9 +1290,11 @@ def pretty_print_model_version_table(
             "NAME": model_version.registered_model.name,
             "MODEL_VERSION": model_version.version,
             "VERSION_DESCRIPTION": model_version.description,
-            "METADATA": model_version.metadata.dict()
-            if model_version.metadata
-            else {},
+            "METADATA": (
+                model_version.metadata.model_dump()
+                if model_version.metadata
+                else {}
+            ),
         }
         for model_version in model_versions
     ]
@@ -1226,15 +1324,21 @@ def pretty_print_model_version_details(
         "REGISTERED_MODEL_NAME": model_version.registered_model.name,
         "VERSION": model_version.version,
         "VERSION_DESCRIPTION": model_version.description,
-        "CREATED_AT": str(model_version.created_at)
-        if model_version.created_at
-        else "N/A",
-        "UPDATED_AT": str(model_version.last_updated_at)
-        if model_version.last_updated_at
-        else "N/A",
-        "METADATA": model_version.metadata.dict()
-        if model_version.metadata
-        else {},
+        "CREATED_AT": (
+            str(model_version.created_at)
+            if model_version.created_at
+            else "N/A"
+        ),
+        "UPDATED_AT": (
+            str(model_version.last_updated_at)
+            if model_version.last_updated_at
+            else "N/A"
+        ),
+        "METADATA": (
+            model_version.metadata.model_dump()
+            if model_version.metadata
+            else {}
+        ),
         "MODEL_SOURCE_URI": model_version.model_source_uri,
         "STAGE": model_version.stage.value,
     }
@@ -1276,9 +1380,10 @@ def print_served_model_configuration(
         **served_model_info,
         "UUID": str(model_service.uuid),
         "STATUS": get_service_state_emoji(model_service.status.state),
+        "TYPE": model_service.SERVICE_TYPE.type,
+        "FLAVOR": model_service.SERVICE_TYPE.flavor,
         "STATUS_MESSAGE": model_service.status.last_error,
         "PIPELINE_NAME": model_service.config.pipeline_name,
-        "RUN_NAME": model_service.config.run_name,
         "PIPELINE_STEP_NAME": model_service.config.pipeline_step_name,
     }
 
@@ -1296,74 +1401,6 @@ def print_served_model_configuration(
     console.print(rich_table)
 
 
-def print_server_deployment_list(servers: List["ServerDeployment"]) -> None:
-    """Print a table with a list of ZenML server deployments.
-
-    Args:
-        servers: list of ZenML server deployments
-    """
-    server_dicts = []
-    for server in servers:
-        status = ""
-        url = ""
-        connected = ""
-        if server.status:
-            status = get_service_state_emoji(server.status.status)
-            if server.status.url:
-                url = server.status.url
-            if server.status.connected:
-                connected = ":point_left:"
-        server_dicts.append(
-            {
-                "STATUS": status,
-                "NAME": server.config.name,
-                "PROVIDER": server.config.provider.value,
-                "URL": url,
-                "CONNECTED": connected,
-            }
-        )
-    print_table(server_dicts)
-
-
-def print_server_deployment(server: "ServerDeployment") -> None:
-    """Prints the configuration and status of a ZenML server deployment.
-
-    Args:
-        server: Server deployment to print
-    """
-    server_name = server.config.name
-    title_ = f"ZenML server '{server_name}'"
-
-    rich_table = table.Table(
-        box=box.HEAVY_EDGE,
-        title=title_,
-        show_header=False,
-        show_lines=True,
-    )
-    rich_table.add_column("", overflow="fold")
-    rich_table.add_column("", overflow="fold")
-
-    server_info = []
-
-    if server.status:
-        server_info.extend(
-            [
-                ("URL", server.status.url or ""),
-                ("STATUS", get_service_state_emoji(server.status.status)),
-                ("STATUS_MESSAGE", server.status.status_message or ""),
-                (
-                    "CONNECTED",
-                    ":white_check_mark:" if server.status.connected else "",
-                ),
-            ]
-        )
-
-    for item in server_info:
-        rich_table.add_row(*item)
-
-    console.print(rich_table)
-
-
 def describe_pydantic_object(schema_json: Dict[str, Any]) -> None:
     """Describes a Pydantic object based on the dict-representation of its schema.
 
@@ -1537,7 +1574,7 @@ def print_components_table(
             "ACTIVE": ":point_right:" if is_active else "",
             "NAME": component.name,
             "COMPONENT ID": component.id,
-            "FLAVOR": component.flavor,
+            "FLAVOR": component.flavor_name,
             "OWNER": f"{component.user.name if component.user else '-'}",
         }
         configurations.append(component_config)
@@ -1593,7 +1630,7 @@ def expires_in(
         expires_at -= datetime.timedelta(seconds=skew_tolerance)
     if expires_at < now:
         return expired_str
-    return seconds_to_human_readable((expires_at - now).seconds)
+    return seconds_to_human_readable(int((expires_at - now).total_seconds()))
 
 
 def print_service_connectors_table(
@@ -1656,13 +1693,15 @@ def print_service_connectors_table(
             "RESOURCE TYPES": "\n".join(connector.emojified_resource_types),
             "RESOURCE NAME": resource_name,
             "OWNER": f"{connector.user.name if connector.user else '-'}",
-            "EXPIRES IN": expires_in(
-                connector.expires_at,
-                ":name_badge: Expired!",
-                connector.expires_skew_tolerance,
-            )
-            if connector.expires_at
-            else "",
+            "EXPIRES IN": (
+                expires_in(
+                    connector.expires_at,
+                    ":name_badge: Expired!",
+                    connector.expires_skew_tolerance,
+                )
+                if connector.expires_at
+                else ""
+            ),
             "LABELS": "\n".join(labels),
         }
         configurations.append(connector_config)
@@ -1718,15 +1757,17 @@ def print_service_connector_resource_table(
             resource_row = {}
             if not show_resources_only:
                 resource_row = {
-                    "CONNECTOR ID": str(resource_model.id)
-                    if not printed_connector
-                    else "",
-                    "CONNECTOR NAME": resource_model.name
-                    if not printed_connector
-                    else "",
-                    "CONNECTOR TYPE": resource_model.emojified_connector_type
-                    if not printed_connector
-                    else "",
+                    "CONNECTOR ID": (
+                        str(resource_model.id) if not printed_connector else ""
+                    ),
+                    "CONNECTOR NAME": (
+                        resource_model.name if not printed_connector else ""
+                    ),
+                    "CONNECTOR TYPE": (
+                        resource_model.emojified_connector_type
+                        if not printed_connector
+                        else ""
+                    ),
                 }
             resource_row.update(
                 {
@@ -1771,8 +1812,7 @@ def print_service_connector_configuration(
         )
     else:
         declare(
-            f"Service connector '{connector.name}' of type "
-            f"'{connector.type}'."
+            f"Service connector '{connector.name}' of type '{connector.type}'."
         )
 
     title_ = f"'{connector.name}' {connector.type} Service Connector Details"
@@ -1802,16 +1842,20 @@ def print_service_connector_configuration(
             "RESOURCE NAME": connector.resource_id or "<multiple>",
             "SECRET ID": connector.secret_id or "",
             "SESSION DURATION": expiration,
-            "EXPIRES IN": expires_in(
-                connector.expires_at,
-                ":name_badge: Expired!",
-                connector.expires_skew_tolerance,
-            )
-            if connector.expires_at
-            else "N/A",
-            "EXPIRES_SKEW_TOLERANCE": connector.expires_skew_tolerance
-            if connector.expires_skew_tolerance
-            else "N/A",
+            "EXPIRES IN": (
+                expires_in(
+                    connector.expires_at,
+                    ":name_badge: Expired!",
+                    connector.expires_skew_tolerance,
+                )
+                if connector.expires_at
+                else "N/A"
+            ),
+            "EXPIRES_SKEW_TOLERANCE": (
+                connector.expires_skew_tolerance
+                if connector.expires_skew_tolerance
+                else "N/A"
+            ),
             "OWNER": user_name,
             "WORKSPACE": connector.workspace.name,
             "CREATED_AT": connector.created,
@@ -1825,16 +1869,20 @@ def print_service_connector_configuration(
             "RESOURCE TYPES": ", ".join(connector.emojified_resource_types),
             "RESOURCE NAME": connector.resource_id or "<multiple>",
             "SESSION DURATION": expiration,
-            "EXPIRES IN": expires_in(
-                connector.expires_at,
-                ":name_badge: Expired!",
-                connector.expires_skew_tolerance,
-            )
-            if connector.expires_at
-            else "N/A",
-            "EXPIRES_SKEW_TOLERANCE": connector.expires_skew_tolerance
-            if connector.expires_skew_tolerance
-            else "N/A",
+            "EXPIRES IN": (
+                expires_in(
+                    connector.expires_at,
+                    ":name_badge: Expired!",
+                    connector.expires_skew_tolerance,
+                )
+                if connector.expires_at
+                else "N/A"
+            ),
+            "EXPIRES_SKEW_TOLERANCE": (
+                connector.expires_skew_tolerance
+                if connector.expires_skew_tolerance
+                else "N/A"
+            ),
         }
 
     for item in properties.items():
@@ -1944,7 +1992,7 @@ def print_service_connector_resource_type(
     message = f"{title}\n" if title else ""
     emoji = replace_emojis(resource_type.emoji) if resource_type.emoji else ""
     supported_auth_methods = [
-        f'{Emoji("lock")} {a}' for a in resource_type.auth_methods
+        f"{Emoji('lock')} {a}" for a in resource_type.auth_methods
     ]
     message += (
         f"{heading} {emoji} {resource_type.name} "
@@ -2068,7 +2116,7 @@ def print_service_connector_type(
     """
     message = f"{title}\n" if title else ""
     supported_auth_methods = [
-        f'{Emoji("lock")} {a.auth_method}' for a in connector_type.auth_methods
+        f"{Emoji('lock')} {a.auth_method}" for a in connector_type.auth_methods
     ]
     supported_resource_types = [
         f"{replace_emojis(r.emoji)} {r.resource_type}"
@@ -2153,7 +2201,7 @@ def _scrub_secret(config: StackComponentConfig) -> Dict[str, Any]:
         A configuration with secret values removed.
     """
     config_dict = {}
-    config_fields = dict(config.__class__.__fields__)
+    config_fields = config.__class__.model_fields
     for key, value in config_fields.items():
         if secret_utils.is_secret_field(value):
             config_dict[key] = "********"
@@ -2329,12 +2377,13 @@ def create_filter_help_text(filter_model: Type[BaseFilter], field: str) -> str:
     Returns:
         The help text.
     """
-    if filter_model.is_sort_by_field(field):
+    filter_generator = FilterGenerator(filter_model)
+    if filter_generator.is_sort_by_field(field):
         return (
             "[STRING] Example: --sort_by='desc:name' to sort by name in "
             "descending order. "
         )
-    if filter_model.is_datetime_field(field):
+    if filter_generator.is_datetime_field(field):
         return (
             f"[DATETIME] The following datetime format is supported: "
             f"'{FILTERING_DATETIME_FORMAT}'. Make sure to keep it in "
@@ -2343,23 +2392,23 @@ def create_filter_help_text(filter_model: Type[BaseFilter], field: str) -> str:
             f"'{GenericFilterOps.GTE}:{FILTERING_DATETIME_FORMAT}' to "
             f"filter for everything created on or after the given date."
         )
-    elif filter_model.is_uuid_field(field):
+    elif filter_generator.is_uuid_field(field):
         return (
             f"[UUID] Example: --{field}='{GenericFilterOps.STARTSWITH}:ab53ca' "
             f"to filter for all UUIDs starting with that prefix."
         )
-    elif filter_model.is_int_field(field):
+    elif filter_generator.is_int_field(field):
         return (
             f"[INTEGER] Example: --{field}='{GenericFilterOps.GTE}:25' to "
             f"filter for all entities where this field has a value greater than "
             f"or equal to the value."
         )
-    elif filter_model.is_bool_field(field):
+    elif filter_generator.is_bool_field(field):
         return (
             f"[BOOL] Example: --{field}='True' to "
             f"filter for all instances where this field is true."
         )
-    elif filter_model.is_str_field(field):
+    elif filter_generator.is_str_field(field):
         return (
             f"[STRING] Example: --{field}='{GenericFilterOps.CONTAINS}:example' "
             f"to filter everything that contains the query string somewhere in "
@@ -2381,27 +2430,28 @@ def create_data_type_help_text(
     Returns:
         The help text.
     """
-    if filter_model.is_datetime_field(field):
+    filter_generator = FilterGenerator(filter_model)
+    if filter_generator.is_datetime_field(field):
         return (
             f"[DATETIME] supported filter operators: "
             f"{[str(op) for op in NumericFilter.ALLOWED_OPS]}"
         )
-    elif filter_model.is_uuid_field(field):
+    elif filter_generator.is_uuid_field(field):
         return (
             f"[UUID] supported filter operators: "
             f"{[str(op) for op in UUIDFilter.ALLOWED_OPS]}"
         )
-    elif filter_model.is_int_field(field):
+    elif filter_generator.is_int_field(field):
         return (
             f"[INTEGER] supported filter operators: "
             f"{[str(op) for op in NumericFilter.ALLOWED_OPS]}"
         )
-    elif filter_model.is_bool_field(field):
+    elif filter_generator.is_bool_field(field):
         return (
             f"[BOOL] supported filter operators: "
             f"{[str(op) for op in BoolFilter.ALLOWED_OPS]}"
         )
-    elif filter_model.is_str_field(field):
+    elif filter_generator.is_str_field(field):
         return (
             f"[STRING] supported filter operators: "
             f"{[str(op) for op in StrFilter.ALLOWED_OPS]}"
@@ -2430,7 +2480,7 @@ def list_options(filter_model: Type[BaseFilter]) -> Callable[[F], F]:
     def inner_decorator(func: F) -> F:
         options = []
         data_type_descriptors = set()
-        for k, v in filter_model.__fields__.items():
+        for k, v in filter_model.model_fields.items():
             if k not in filter_model.CLI_EXCLUDE_FIELDS:
                 options.append(
                     click.option(
@@ -2610,29 +2660,100 @@ def print_model_url(url: Optional[str]) -> None:
         warning(
             "You can display various ZenML entities including pipelines, "
             "runs, stacks and much more on the ZenML Dashboard. "
-            "You can try it locally, by running `zenml up`, or remotely, "
-            "by deploying ZenML on the infrastructure of your choice."
+            "You can try it locally, by running `zenml local --local`, or "
+            "remotely, by deploying ZenML on the infrastructure of your choice."
         )
 
 
-def warn_deprecated_example_subcommand() -> None:
-    """Warning for deprecating example subcommand."""
-    warning(
-        "The `example` CLI subcommand has been deprecated and will be removed "
-        "in a future release."
-    )
-
+def is_jupyter_installed() -> bool:
+    """Checks if Jupyter notebook is installed.
 
-def verify_mlstacks_prerequisites_installation() -> None:
-    """Checks if the `mlstacks` package is installed."""
+    Returns:
+        bool: True if Jupyter notebook is installed, False otherwise.
+    """
     try:
-        import mlstacks  # noqa: F401
-        import python_terraform  # noqa: F401
+        pkg_resources.get_distribution("notebook")
+        return True
+    except pkg_resources.DistributionNotFound:
+        return False
+
+
+def multi_choice_prompt(
+    object_type: str,
+    choices: List[List[Any]],
+    headers: List[str],
+    prompt_text: str,
+    allow_zero_be_a_new_object: bool = False,
+    default_choice: Optional[str] = None,
+) -> Optional[int]:
+    """Prompts the user to select a choice from a list of choices.
+
+    Args:
+        object_type: The type of the object
+        choices: The list of choices
+        prompt_text: The prompt text
+        headers: The list of headers.
+        allow_zero_be_a_new_object: Whether to allow zero as a new object
+        default_choice: The default choice
 
-        subprocess.check_output(
-            ["terraform", "--version"], universal_newlines=True
+    Returns:
+        The selected choice index or None for new object
+
+    Raises:
+        RuntimeError: If no choice is made.
+    """
+    table = Table(
+        title=f"Available {object_type}",
+        show_header=True,
+        border_style=None,
+        expand=True,
+        show_lines=True,
+    )
+    table.add_column("Choice", justify="left", width=1)
+    for h in headers:
+        table.add_column(
+            h.replace("_", " ").capitalize(), justify="left", width=10
         )
-    except ImportError:
-        error(NOT_INSTALLED_MESSAGE)
-    except subprocess.CalledProcessError:
-        error(TERRAFORM_NOT_INSTALLED_MESSAGE)
+
+    i_shift = 0
+    if allow_zero_be_a_new_object:
+        i_shift = 1
+        table.add_row(
+            "[0]",
+            *([f"Create a new {object_type}"] * len(headers)),
+        )
+    for i, one_choice in enumerate(choices):
+        table.add_row(f"[{i + i_shift}]", *[str(x) for x in one_choice])
+    Console().print(table)
+
+    selected = Prompt.ask(
+        prompt_text,
+        choices=[str(i) for i in range(0, len(choices) + 1)],
+        default=default_choice,
+        show_choices=False,
+    )
+    if selected is None:
+        raise RuntimeError(f"No {object_type} was selected")
+
+    if selected == "0" and allow_zero_be_a_new_object:
+        return None
+    else:
+        return int(selected) - i_shift
+
+
+def requires_mac_env_var_warning() -> bool:
+    """Checks if a warning needs to be shown for a local Mac server.
+
+    This is for the case where a user is on a macOS system, trying to run a
+    local server but is missing the `OBJC_DISABLE_INITIALIZE_FORK_SAFETY`
+    environment variable.
+
+    Returns:
+        bool: True if a warning needs to be shown, False otherwise.
+    """
+    if sys.platform == "darwin":
+        mac_version_tuple = tuple(map(int, platform.release().split(".")[:2]))
+        return not os.getenv(
+            "OBJC_DISABLE_INITIALIZE_FORK_SAFETY"
+        ) and mac_version_tuple >= (10, 13)
+    return False