zenml-nightly 0.68.0.dev20241027__py3-none-any.whl → 0.68.1.dev20241101__py3-none-any.whl

zenml/cli/service_accounts.py

@@ -86,8 +86,8 @@ def _create_api_key(
             f"The API key value is: '{api_key.key}'\nPlease store it safely as "
             "it will not be shown again.\nTo configure a ZenML client to use "
             "this API key, run:\n\n"
-            f"zenml connect --url {zen_store.config.url} --api-key \\\n"
-            f"    '{api_key.key}'\n"
+            f"zenml login {zen_store.config.url} --api-key \n\n"
+            f"and enter the following API key when prompted: {api_key.key}\n"
         )
 
 
@@ -543,8 +543,8 @@ def rotate_api_key(
             f"The new API key value is: '{api_key.key}'\nPlease store it "
             "safely as it will not be shown again.\nTo configure a ZenML "
             "client to use this API key, run:\n\n"
-            f"zenml connect --url {zen_store.config.url} --api-key \\\n"
-            f"    '{api_key.key}'\n"
+            f"zenml login {zen_store.config.url} --api-key \n\n"
+            f"and enter the following API key when prompted: {api_key.key}\n"
         )
 
 

zenml/cli/stack.py

@@ -75,6 +75,7 @@ from zenml.models.v2.core.service_connector import (
 from zenml.service_connectors.service_connector_utils import (
     get_resources_options_from_resource_model_for_full_stack,
 )
+from zenml.utils import requirements_utils
 from zenml.utils.dashboard_utils import get_component_url, get_stack_url
 from zenml.utils.yaml_utils import read_yaml, write_yaml
 
@@ -428,7 +429,7 @@ def register_stack(
                             object_type=component_type.value.replace("_", " "),
                             choices=[
                                 [
-                                    component.flavor,
+                                    component.flavor_name,
                                     component.name,
                                     component.configuration or "",
                                     component.connector_resource_id,
@@ -1624,7 +1625,7 @@ Stack [{deployed_stack.stack.name}]({get_stack_url(deployed_stack.stack)}):\n"""
         if components:
             component = components[0]
             stack_desc += (
-                f" * `{component.flavor}` {component_type.value}: "
+                f" * `{component.flavor_name}` {component_type.value}: "
                 f"[{component.name}]({get_component_url(component)})\n"
             )
 
@@ -1944,3 +1945,77 @@ def _get_stack_component_info(
         service_connector_index=service_connector_index,
         service_connector_resource_id=service_connector_resource_id,
     )
+
+
+@stack.command(
+    name="export-requirements", help="Export the stack requirements."
+)
+@click.argument(
+    "stack_name_or_id",
+    type=click.STRING,
+    required=False,
+)
+@click.option(
+    "--output-file",
+    "-o",
+    "output_file",
+    type=str,
+    required=False,
+    help="File to which to export the stack requirements. If not "
+    "provided, the requirements will be printed to stdout instead.",
+)
+@click.option(
+    "--overwrite",
+    "-ov",
+    "overwrite",
+    type=bool,
+    required=False,
+    is_flag=True,
+    help="Overwrite the output file if it already exists. This option is "
+    "only valid if the output file is provided.",
+)
+def export_requirements(
+    stack_name_or_id: Optional[str] = None,
+    output_file: Optional[str] = None,
+    overwrite: bool = False,
+) -> None:
+    """Exports stack requirements so they can be installed using pip.
+
+    Args:
+        stack_name_or_id: Stack name or ID. If not given, the active stack will
+            be used.
+        output_file: Optional path to the requirements output file.
+        overwrite: Overwrite the output file if it already exists. This option
+            is only valid if the output file is provided.
+    """
+    try:
+        stack_model: "StackResponse" = Client().get_stack(
+            name_id_or_prefix=stack_name_or_id
+        )
+    except KeyError as err:
+        cli_utils.error(str(err))
+
+    requirements, _ = requirements_utils.get_requirements_for_stack(
+        stack_model
+    )
+
+    if not requirements:
+        cli_utils.declare(f"Stack `{stack_model.name}` has no requirements.")
+        return
+
+    if output_file:
+        try:
+            with open(output_file, "x") as f:
+                f.write("\n".join(requirements))
+        except FileExistsError:
+            if overwrite or cli_utils.confirmation(
+                "A file already exists at the specified path. "
+                "Would you like to overwrite it?"
+            ):
+                with open(output_file, "w") as f:
+                    f.write("\n".join(requirements))
+        cli_utils.declare(
+            f"Requirements for stack `{stack_model.name}` exported to {output_file}."
+        )
+    else:
+        click.echo(" ".join(requirements), nl=False)

zenml/cli/stack_components.py

@@ -123,10 +123,9 @@ def generate_stack_component_describe_command(
                 active_component_id = active_components[0].id
 
             if component_.connector:
-                # We also need the flavor to get the connector requirements
-                connector_requirements = client.get_flavor_by_name_and_type(
-                    name=component_.flavor, component_type=component_type
-                ).connector_requirements
+                connector_requirements = (
+                    component_.flavor.connector_requirements
+                )
             else:
                 connector_requirements = None
 
@@ -570,7 +569,7 @@ def generate_stack_component_copy_command(
 
             copied_component = client.create_stack_component(
                 name=target_component,
-                flavor=component_to_copy.flavor,
+                flavor=component_to_copy.flavor_name,
                 component_type=component_to_copy.type,
                 configuration=component_to_copy.configuration,
                 labels=component_to_copy.labels,
@@ -1366,17 +1365,7 @@ def connect_stack_component_with_service_connector(
     except KeyError as err:
         cli_utils.error(str(err))
 
-    try:
-        flavor_model = client.get_flavor_by_name_and_type(
-            name=component_model.flavor, component_type=component_type
-        )
-    except KeyError as err:
-        cli_utils.error(
-            f"Could not find flavor '{component_model.flavor}' for "
-            f"{display_name} '{name_id_or_prefix}': {str(err)}"
-        )
-
-    requirements = flavor_model.connector_requirements
+    requirements = component_model.flavor.connector_requirements
 
     if not requirements:
         cli_utils.error(

zenml/cli/user_management.py

@@ -367,18 +367,6 @@ def change_user_password(
         f"Successfully updated password for active user '{active_user.name}'."
     )
 
-    store = GlobalConfiguration().store_configuration
-    if store.type == StoreType.REST:
-        from zenml.zen_stores.rest_zen_store import RestZenStoreConfiguration
-
-        assert isinstance(store, RestZenStoreConfiguration)
-
-        if store.password is not None:
-            cli_utils.declare(
-                "You may need to log in again with your new password by "
-                "running `zenml connect`."
-            )
-
 
 @user.command(
     "deactivate",

zenml/cli/utils.py

@@ -79,7 +79,6 @@ 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
@@ -223,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(
@@ -241,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:
@@ -258,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.
 
@@ -270,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()
@@ -326,6 +338,8 @@ 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
@@ -350,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
@@ -613,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}'."
     )
 
@@ -1391,74 +1406,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.
 
@@ -1632,7 +1579,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)
@@ -2719,8 +2666,8 @@ 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."
         )
 
 

zenml/client.py

@@ -346,6 +346,8 @@ class Client(metaclass=ClientMetaClass):
     """
 
     _active_user: Optional["UserResponse"] = None
+    _active_workspace: Optional["WorkspaceResponse"] = None
+    _active_stack: Optional["StackResponse"] = None
 
     def __init__(
         self,
@@ -1113,9 +1115,13 @@ class Client(metaclass=ClientMetaClass):
         Raises:
             RuntimeError: If the active workspace is not set.
         """
-        if ENV_ZENML_ACTIVE_WORKSPACE_ID in os.environ:
-            workspace_id = os.environ[ENV_ZENML_ACTIVE_WORKSPACE_ID]
-            return self.get_workspace(workspace_id)
+        if workspace_id := os.environ.get(ENV_ZENML_ACTIVE_WORKSPACE_ID):
+            if not self._active_workspace or self._active_workspace.id != UUID(
+                workspace_id
+            ):
+                self._active_workspace = self.get_workspace(workspace_id)
+
+            return self._active_workspace
 
         from zenml.constants import DEFAULT_WORKSPACE_NAME
 
@@ -1465,8 +1471,13 @@ class Client(metaclass=ClientMetaClass):
         Raises:
             RuntimeError: If the active stack is not set.
         """
-        if ENV_ZENML_ACTIVE_STACK_ID in os.environ:
-            return self.get_stack(os.environ[ENV_ZENML_ACTIVE_STACK_ID])
+        if env_stack_id := os.environ.get(ENV_ZENML_ACTIVE_STACK_ID):
+            if not self._active_stack or self._active_stack.id != UUID(
+                env_stack_id
+            ):
+                self._active_stack = self.get_stack(env_stack_id)
+
+            return self._active_stack
 
         stack_id: Optional[UUID] = None
 
@@ -1534,6 +1545,8 @@ class Client(metaclass=ClientMetaClass):
         remote_components: List[str] = []
         assert stack.components is not None
         for component_type, components in stack.components.items():
+            component_flavor: Union[FlavorResponse, str]
+
             for component in components:
                 if isinstance(component, UUID):
                     component_response = self.get_stack_component(
@@ -1554,7 +1567,7 @@ class Client(metaclass=ClientMetaClass):
 
                 configuration = validate_stack_component_config(
                     configuration_dict=component_config,
-                    flavor_name=component_flavor,
+                    flavor=component_flavor,
                     component_type=component_type,
                     # Always enforce validation of custom flavors
                     validate_custom_flavors=True,
@@ -1563,13 +1576,18 @@ class Client(metaclass=ClientMetaClass):
                 # `validate_custom_flavors=True` above
                 assert configuration is not None
                 warn_if_config_server_mismatch(configuration)
+                flavor_name = (
+                    component_flavor.name
+                    if isinstance(component_flavor, FlavorResponse)
+                    else component_flavor
+                )
                 if configuration.is_local:
                     local_components.append(
-                        f"{component_type.value}: {component_flavor}"
+                        f"{component_type.value}: {flavor_name}"
                     )
                 elif configuration.is_remote:
                     remote_components.append(
-                        f"{component_type.value}: {component_flavor}"
+                        f"{component_type.value}: {flavor_name}"
                     )
 
         if local_components and remote_components:
@@ -1906,6 +1924,7 @@ class Client(metaclass=ClientMetaClass):
         user_id: Optional[Union[str, UUID]] = None,
         connector_id: Optional[Union[str, UUID]] = None,
         stack_id: Optional[Union[str, UUID]] = None,
+        user: Optional[Union[UUID, str]] = None,
         hydrate: bool = False,
     ) -> Page[ComponentResponse]:
         """Lists all registered stack components.
@@ -1925,6 +1944,7 @@ class Client(metaclass=ClientMetaClass):
             connector_id: The id of the connector to filter by.
             stack_id: The id of the stack to filter by.
             name: The name of the component to filter by.
+            user: The ID of name of the user to filter by.
             hydrate: Flag deciding whether to hydrate the output model(s)
                 by including metadata fields in the response.
 
@@ -1946,6 +1966,7 @@ class Client(metaclass=ClientMetaClass):
             id=id,
             created=created,
             updated=updated,
+            user=user,
         )
         component_filter_model.set_scope_workspace(self.active_workspace.id)
 
@@ -1982,7 +2003,7 @@ class Client(metaclass=ClientMetaClass):
 
         validated_config = validate_stack_component_config(
             configuration_dict=configuration,
-            flavor_name=flavor,
+            flavor=flavor,
             component_type=component_type,
             # Always enforce validation of custom flavors
             validate_custom_flavors=True,
@@ -2083,7 +2104,7 @@ class Client(metaclass=ClientMetaClass):
 
             validated_config = validate_stack_component_config(
                 configuration_dict=existing_configuration,
-                flavor_name=component.flavor,
+                flavor=component.flavor,
                 component_type=component.type,
                 # Always enforce validation of custom flavors
                 validate_custom_flavors=True,
@@ -3778,6 +3799,7 @@ class Client(metaclass=ClientMetaClass):
         code_repository: Optional[Union[UUID, str]] = None,
         model: Optional[Union[UUID, str]] = None,
         stack: Optional[Union[UUID, str]] = None,
+        stack_component: Optional[Union[UUID, str]] = None,
         hydrate: bool = False,
     ) -> Page[PipelineRunResponse]:
         """List all pipeline runs.
@@ -3816,6 +3838,7 @@ class Client(metaclass=ClientMetaClass):
             code_repository: Filter by code repository name/ID.
             model: Filter by model name/ID.
             stack: Filter by stack name/ID.
+            stack_component: Filter by stack component name/ID.
             hydrate: Flag deciding whether to hydrate the output model(s)
                 by including metadata fields in the response.
 
@@ -3854,6 +3877,7 @@ class Client(metaclass=ClientMetaClass):
             code_repository=code_repository,
             stack=stack,
             model=model,
+            stack_component=stack_component,
             templatable=templatable,
         )
         runs_filter_model.set_scope_workspace(self.active_workspace.id)
@@ -3919,7 +3943,7 @@ class Client(metaclass=ClientMetaClass):
         workspace_id: Optional[Union[str, UUID]] = None,
         user_id: Optional[Union[str, UUID]] = None,
         model_version_id: Optional[Union[str, UUID]] = None,
-        num_outputs: Optional[Union[int, str]] = None,
+        model: Optional[Union[UUID, str]] = None,
         hydrate: bool = False,
     ) -> Page[StepRunResponse]:
         """List all pipelines.
@@ -3940,11 +3964,11 @@ class Client(metaclass=ClientMetaClass):
             deployment_id: The id of the deployment to filter by.
             original_step_run_id: The id of the original step run to filter by.
             model_version_id: The ID of the model version to filter by.
+            model: Filter by model name/ID.
             name: The name of the step run to filter by.
             cache_key: The cache key of the step run to filter by.
             code_hash: The code hash of the step run to filter by.
             status: The name of the run to filter by.
-            num_outputs: The number of outputs for the step run
             hydrate: Flag deciding whether to hydrate the output model(s)
                 by including metadata fields in the response.
 
@@ -3971,7 +3995,7 @@ class Client(metaclass=ClientMetaClass):
             workspace_id=workspace_id,
             user_id=user_id,
             model_version_id=model_version_id,
-            num_outputs=num_outputs,
+            model=model,
         )
         step_run_filter_model.set_scope_workspace(self.active_workspace.id)
         return self.zen_store.list_run_steps(
@@ -7243,6 +7267,7 @@ class Client(metaclass=ClientMetaClass):
             NotImplementedError: If the client is not connected to a ZenML
                 server.
         """
+        from zenml.login.credentials_store import get_credentials_store
         from zenml.zen_stores.rest_zen_store import RestZenStore
 
         zen_store = self.zen_store
@@ -7251,8 +7276,15 @@ class Client(metaclass=ClientMetaClass):
                 "API key configuration is only supported if connected to a "
                 "ZenML server."
             )
+
+        credentials_store = get_credentials_store()
         assert isinstance(zen_store, RestZenStore)
-        zen_store.set_api_key(api_key=key)
+
+        credentials_store.set_api_key(server_url=zen_store.url, api_key=key)
+
+        # Force a re-authentication to start using the new API key
+        # right away.
+        zen_store.authenticate(force=True)
 
     def list_api_keys(
         self,

zenml/config/compiler.py

@@ -208,6 +208,7 @@ class Compiler:
                 enable_artifact_visualization=config.enable_artifact_visualization,
                 enable_step_logs=config.enable_step_logs,
                 settings=config.settings,
+                tags=config.tags,
                 extra=config.extra,
                 model=config.model,
                 parameters=config.parameters,

zenml/config/global_config.py

@@ -689,6 +689,15 @@ class GlobalConfiguration(BaseModel, metaclass=GlobalConfigMetaClass):
         self._configure_store(config, skip_default_registrations, **kwargs)
         logger.info("Updated the global store configuration.")
 
+    @property
+    def is_initialized(self) -> bool:
+        """Check if the global configuration is initialized.
+
+        Returns:
+            `True` if the global configuration is initialized.
+        """
+        return self._zen_store is not None
+
     @property
     def zen_store(self) -> "BaseZenStore":
         """Initialize and/or return the global zen store.

zenml/config/pipeline_configurations.py

@@ -13,7 +13,7 @@
 #  permissions and limitations under the License.
 """Pipeline configuration classes."""
 
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
 from pydantic import SerializeAsAny, field_validator
 
@@ -39,6 +39,7 @@ class PipelineConfigurationUpdate(StrictBaseModel):
     enable_artifact_visualization: Optional[bool] = None
     enable_step_logs: Optional[bool] = None
     settings: Dict[str, SerializeAsAny[BaseSettings]] = {}
+    tags: Optional[List[str]] = None
     extra: Dict[str, Any] = {}
     failure_hook_source: Optional[SourceWithValidator] = None
     success_hook_source: Optional[SourceWithValidator] = None

zenml/config/pipeline_run_configuration.py

@@ -13,7 +13,7 @@
 #  permissions and limitations under the License.
 """Pipeline run configuration class."""
 
-from typing import Any, Dict, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 from uuid import UUID
 
 from pydantic import Field, SerializeAsAny
@@ -45,6 +45,7 @@ class PipelineRunConfiguration(
     )
     steps: Dict[str, StepConfigurationUpdate] = {}
     settings: Dict[str, SerializeAsAny[BaseSettings]] = {}
+    tags: Optional[List[str]] = None
     extra: Dict[str, Any] = {}
     model: Optional[Model] = None
     parameters: Optional[Dict[str, Any]] = None

zenml/constants.py

@@ -134,6 +134,7 @@ APP_NAME = "zenml"
 # Environment variables
 ENV_ZENML_LOGGING_COLORS_DISABLED = "ZENML_LOGGING_COLORS_DISABLED"
 ENV_ZENML_ANALYTICS_OPT_IN = "ZENML_ANALYTICS_OPT_IN"
+ENV_ZENML_USER_ID = "ZENML_USER_ID"
 ENV_ZENML_CONFIG_PATH = "ZENML_CONFIG_PATH"
 ENV_ZENML_DEBUG = "ZENML_DEBUG"
 ENV_ZENML_LOGGING_VERBOSITY = "ZENML_LOGGING_VERBOSITY"
@@ -163,7 +164,6 @@ ENV_ZENML_DISABLE_CLIENT_SERVER_MISMATCH_WARNING = (
 ENV_ZENML_DISABLE_WORKSPACE_WARNINGS = "ZENML_DISABLE_WORKSPACE_WARNINGS"
 ENV_ZENML_SKIP_IMAGE_BUILDER_DEFAULT = "ZENML_SKIP_IMAGE_BUILDER_DEFAULT"
 ENV_ZENML_SERVER = "ZENML_SERVER"
-ENV_ZENML_LOCAL_SERVER = "ZENML_LOCAL_SERVER"
 ENV_ZENML_ENFORCE_TYPE_ANNOTATIONS = "ZENML_ENFORCE_TYPE_ANNOTATIONS"
 ENV_ZENML_ENABLE_IMPLICIT_AUTH_METHODS = "ZENML_ENABLE_IMPLICIT_AUTH_METHODS"
 ENV_ZENML_DISABLE_STEP_LOGS_STORAGE = "ZENML_DISABLE_STEP_LOGS_STORAGE"
@@ -186,6 +186,7 @@ ENV_ZENML_RUN_SINGLE_STEPS_WITHOUT_STACK = (
     "ZENML_RUN_SINGLE_STEPS_WITHOUT_STACK"
 )
 ENV_ZENML_PREVENT_CLIENT_SIDE_CACHING = "ZENML_PREVENT_CLIENT_SIDE_CACHING"
+ENV_ZENML_DISABLE_CREDENTIALS_DISK_CACHING = "DISABLE_CREDENTIALS_DISK_CACHING"
 
 # Logging variables
 IS_DEBUG_ENV: bool = handle_bool_env_var(ENV_ZENML_DEBUG, default=False)
@@ -336,6 +337,7 @@ API_TOKEN = "/api_token"
 ARTIFACTS = "/artifacts"
 ARTIFACT_VERSIONS = "/artifact_versions"
 ARTIFACT_VISUALIZATIONS = "/artifact_visualizations"
+AUTH = "/auth"
 CODE_REFERENCES = "/code_references"
 CODE_REPOSITORIES = "/code_repositories"
 COMPONENT_TYPES = "/component-types"
@@ -468,11 +470,3 @@ BANNED_NAME_CHARACTERS = "\t\n\r\v\f"
 
 
 STACK_DEPLOYMENT_API_TOKEN_EXPIRATION = 60 * 6  # 6 hours
-
-# ZenML Pro
-ZENML_PRO_CONNECTION_ISSUES_SUSPENDED_PAUSED_TENANT_HINT = (
-    "\nHINT: Since you are trying to communicate with the ZenML Pro Tenant, "
-    "please make sure that your tenant is in RUNNING state on your "
-    "Organization page. If the tenant is PAUSED you can `Resume` it via UI "
-    "and try again."
-)

zenml/enums.py

@@ -198,7 +198,7 @@ class SecretValidationLevel(StrEnum):
 class ServerProviderType(StrEnum):
     """ZenML server providers."""
 
-    LOCAL = "local"
+    DAEMON = "daemon"
     DOCKER = "docker"