zenml-nightly 0.66.0.dev20240930__py3-none-any.whl → 0.68.0.dev20241026__py3-none-any.whl

zenml/cli/__init__.py

@@ -1870,13 +1870,7 @@ Connecting to a ZenML Server
 
 The ZenML client can be [configured to connect to a remote database or ZenML
 server](https://docs.zenml.io/how-to/connecting-to-zenml)
-with the `zenml connect` command. If no arguments are supplied, ZenML
-will attempt to connect to the last ZenML server deployed from the local host
-using the 'zenml deploy' command:
-
-```bash
-zenml connect
-```
+with the `zenml connect` command.
 
 To connect to a ZenML server, you can either pass the configuration as command
 line arguments or as a YAML file:
@@ -2517,37 +2511,6 @@ The verbosity of the ZenML client output can be configured using the
 ```bash
 zenml logging set-verbosity DEBUG
 ```
-
-Deploying ZenML to the cloud
-----------------------------
-
-The ZenML CLI provides a simple way to deploy ZenML to the cloud. Simply run
-
-```bash
-zenml deploy
-```
-
-You will be prompted to provide a name for your deployment and details like what
-cloud provider you want to deploy to — and that's it! It creates the
-database and any VPCs, permissions, and more that are needed.
-
-In order to be able to run the deploy command, you should have your cloud
-provider's CLI configured locally with permissions to create resources like
-MySQL databases and networks.
-
-Deploying Stack Components
---------------------------
-
-Stack components can be deployed directly via the CLI. You can use the `deploy`
-subcommand for this. For example, you could deploy a GCP artifact store using
-the following command:
-
-```shell
-zenml artifact-store deploy -f gcp -p gcp -r us-east1 -x project_id=zenml-core basic_gcp_artifact_store
-```
-
-For full documentation on this functionality, please refer to [the dedicated
-documentation on stack component deploy](https://docs.zenml.io/how-to/stack-deployment/deploy-a-stack-component).
 """
 
 from zenml.cli.version import *  # noqa
@@ -2570,7 +2533,6 @@ from zenml.cli.service_accounts import *  # noqa
 from zenml.cli.service_connectors import *  # noqa
 from zenml.cli.stack import *  # noqa
 from zenml.cli.stack_components import *  # noqa
-from zenml.cli.stack_recipes import *  # noqa
 from zenml.cli.user_management import *  # noqa
 from zenml.cli.workspace import *  # noqa
 from zenml.cli.tag import *  # noqa

zenml/cli/base.py

@@ -50,7 +50,7 @@ from zenml.io import fileio
 from zenml.logger import get_logger
 from zenml.utils.io_utils import copy_dir, get_global_config_directory
 from zenml.utils.yaml_utils import write_yaml
-from zenml.zen_server.utils import get_active_deployment
+from zenml.zen_server.utils import get_local_server
 
 logger = get_logger(__name__)
 # WT_SESSION is a Windows Terminal specific environment variable. If it
@@ -79,11 +79,11 @@ class ZenMLProjectTemplateLocation(BaseModel):
 ZENML_PROJECT_TEMPLATES = dict(
     e2e_batch=ZenMLProjectTemplateLocation(
         github_url="zenml-io/template-e2e-batch",
-        github_tag="2024.08.29",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
+        github_tag="2024.10.10",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
     ),
     starter=ZenMLProjectTemplateLocation(
         github_url="zenml-io/template-starter",
-        github_tag="2024.09.23",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
+        github_tag="2024.09.24",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
     ),
     nlp=ZenMLProjectTemplateLocation(
         github_url="zenml-io/template-nlp",
@@ -91,7 +91,7 @@ ZENML_PROJECT_TEMPLATES = dict(
     ),
     llm_finetuning=ZenMLProjectTemplateLocation(
         github_url="zenml-io/template-llm-finetuning",
-        github_tag="2024.08.29",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
+        github_tag="2024.09.24",  # Make sure it is aligned with .github/workflows/update-templates-to-examples.yml
     ),
 )
 
@@ -358,7 +358,7 @@ def clean(yes: bool = False, local: bool = False) -> None:
         )
 
     if yes or confirm:
-        server = get_active_deployment(local=True)
+        server = get_local_server()
 
         if server:
             from zenml.zen_server.deploy.deployer import ServerDeployer

zenml/cli/integration.py

@@ -282,15 +282,6 @@ def install(
     else:
         integration_set = set(integrations)
 
-    # TODO: remove once python 3.8 is deprecated
-    if sys.version_info.minor == 8 and "tensorflow" in integration_set:
-        warning(
-            "Python 3.8 with TensorFlow is not fully compatible with "
-            "Pydantic 2 requirements. Consider upgrading to a "
-            "higher Python version if you would like to use the "
-            "Tensorflow integration."
-        )
-
     if sys.version_info.minor == 12 and "tensorflow" in integration_set:
         warning(
             "The TensorFlow integration is not yet compatible with Python "

zenml/cli/pipeline.py

@@ -33,7 +33,7 @@ from zenml.models import (
     PipelineRunFilter,
     ScheduleFilter,
 )
-from zenml.new.pipelines.pipeline import Pipeline
+from zenml.pipelines.pipeline_definition import Pipeline
 from zenml.utils import source_utils, uuid_utils
 from zenml.utils.yaml_utils import write_yaml
 

zenml/cli/served_model.py

@@ -215,11 +215,18 @@ def register_model_deployer_subcommands() -> None:  # noqa: C901
                     )
                     or "No hostname specified for this service"
                 )
+                prediction_apis_urls = (
+                    model_deployer.get_model_server_info(served_models[0]).get(
+                        "PREDICTION_APIS_URLS"
+                    )
+                    or "No prediction APIs URLs specified for this service"
+                )
                 declare(
                     f"  Prediction URL of Served Model {served_model_uuid} "
                     f"is:\n"
                     f"  {prediction_url}\n"
-                    f"  and the hostname is: {prediction_hostname}"
+                    f"  and the hostname is: {prediction_hostname}\n"
+                    f"  and the prediction APIs URLs are: {prediction_apis_urls}\n"
                 )
             except KeyError:
                 warning("The deployed model instance has no 'prediction_url'.")

zenml/cli/server.py

@@ -23,8 +23,6 @@ import yaml
 from rich.errors import MarkupError
 
 import zenml
-from zenml.analytics.enums import AnalyticsEvent
-from zenml.analytics.utils import track_handler
 from zenml.cli import utils as cli_utils
 from zenml.cli.cli import cli
 from zenml.cli.web_login import web_login
@@ -35,8 +33,8 @@ from zenml.constants import ENV_ZENML_LOCAL_SERVER
 from zenml.enums import ServerProviderType, StoreType
 from zenml.exceptions import AuthorizationException, IllegalOperationError
 from zenml.logger import get_logger
-from zenml.utils import terraform_utils, yaml_utils
-from zenml.zen_server.utils import get_active_deployment
+from zenml.utils import yaml_utils
+from zenml.zen_server.utils import get_local_server
 
 logger = get_logger(__name__)
 
@@ -94,13 +92,6 @@ LOCAL_ZENML_SERVER_NAME = "local"
     default=None,
     help="Specify an ngrok auth token to use for exposing the ZenML server.",
 )
-@click.option(
-    "--legacy",
-    is_flag=True,
-    help="Start the legacy ZenML dashboard instead of the new ZenML dashboard.",
-    default=False,
-    type=click.BOOL,
-)
 def up(
     docker: bool = False,
     ip_address: Union[
@@ -111,7 +102,6 @@ def up(
     connect: bool = False,
     image: Optional[str] = None,
     ngrok_token: Optional[str] = None,
-    legacy: bool = False,
 ) -> None:
     """Start the ZenML dashboard locally and connect the client to it.
 
@@ -127,8 +117,6 @@ def up(
         ngrok_token: An ngrok auth token to use for exposing the ZenML dashboard
             on a public domain. Primarily used for accessing the dashboard in
             Colab.
-        legacy: Start the legacy ZenML dashboard instead of the new ZenML
-            dashboard.
     """
     from zenml.zen_server.deploy.deployer import ServerDeployer
 
@@ -181,7 +169,7 @@ def up(
 
     deployer = ServerDeployer()
 
-    server = get_active_deployment(local=True)
+    server = get_local_server()
     if server and server.config.provider != provider:
         deployer.remove_server(LOCAL_ZENML_SERVER_NAME)
 
@@ -200,7 +188,6 @@ def up(
         ServerProviderType.DOCKER,
     ]:
         config_attrs["ip_address"] = ip_address
-    config_attrs["use_legacy_dashboard"] = legacy
 
     from zenml.zen_server.deploy.deployment import ServerDeploymentConfig
 
@@ -266,7 +253,7 @@ def show(ngrok_token: Optional[str] = None) -> None:
 @cli.command("down", help="Shut down the local ZenML dashboard.")
 def down() -> None:
     """Shut down the local ZenML dashboard."""
-    server = get_active_deployment(local=True)
+    server = get_local_server()
 
     if not server:
         cli_utils.declare("The local ZenML dashboard is not running.")
@@ -284,180 +271,6 @@ def down() -> None:
         gc.set_default_store()
 
 
-@cli.command("deploy", help="Deploy ZenML in the cloud.")
-@click.option(
-    "--provider",
-    "-p",
-    type=click.Choice(
-        [
-            ServerProviderType.AWS.value,
-            ServerProviderType.GCP.value,
-            ServerProviderType.AZURE.value,
-        ],
-        case_sensitive=True,
-    ),
-    default=None,
-    help="Server deployment provider.",
-)
-@click.option(
-    "--name",
-    type=str,
-    help="A name for the ZenML server deployment. This is used as a prefix for "
-    "the names of deployed resources, such as database services and Kubernetes "
-    "resources.",
-)
-@click.option(
-    "--timeout",
-    "-t",
-    type=click.INT,
-    default=None,
-    help="Time in seconds to wait for the server to be deployed.",
-)
-@click.option(
-    "--config",
-    help="Use a YAML or JSON configuration or configuration file.",
-    required=False,
-    type=str,
-)
-@click.option(
-    "--gcp-project-id",
-    help="The project in GCP to deploy the server to. ",
-    required=False,
-    type=str,
-)
-def deploy(
-    provider: Optional[str] = None,
-    name: Optional[str] = None,
-    timeout: Optional[int] = None,
-    config: Optional[str] = None,
-    gcp_project_id: Optional[str] = None,
-) -> None:
-    """Deploy the ZenML server in a cloud provider.
-
-    Args:
-        name: Name for the ZenML server deployment.
-        provider: ZenML server provider name.
-        timeout: Time in seconds to wait for the server to start.
-        config: A YAML or JSON configuration or configuration file to use.
-        gcp_project_id: The project in GCP to deploy the server to.
-    """
-    with track_handler(
-        event=AnalyticsEvent.ZENML_SERVER_DEPLOYED
-    ) as analytics_handler:
-        try:
-            terraform_utils.verify_terraform_installation()
-        except RuntimeError as e:
-            cli_utils.error(str(e))
-
-        config_dict: Dict[str, Any] = {}
-
-        if config:
-            if os.path.isfile(config):
-                config_dict = yaml_utils.read_yaml(config)
-            else:
-                config_dict = yaml.safe_load(config)
-            if not isinstance(config_dict, dict):
-                cli_utils.error(
-                    "The configuration argument must be JSON/YAML content or "
-                    "point to a valid configuration file."
-                )
-
-            name = config_dict.get("name", name)
-            provider = config_dict.get("provider", provider)
-
-        if not name:
-            name = click.prompt(
-                "ZenML server name (used as a prefix for the names of deployed "
-                "resources)",
-                default="zenml",
-            )
-        config_dict["name"] = name
-
-        if not provider:
-            provider = click.prompt(
-                "ZenML server provider",
-                type=click.Choice(
-                    [
-                        ServerProviderType.AWS.value,
-                        ServerProviderType.GCP.value,
-                        ServerProviderType.AZURE.value,
-                    ],
-                    case_sensitive=True,
-                ),
-                default=ServerProviderType.AWS.value,
-            )
-        config_dict["provider"] = provider
-
-        if provider == ServerProviderType.GCP.value:
-            if "project_id" not in config_dict:
-                if not gcp_project_id:
-                    gcp_project_id = click.prompt(
-                        "GCP project ID",
-                    )
-                config_dict["project_id"] = gcp_project_id
-
-        from zenml.zen_server.deploy.deployment import ServerDeploymentConfig
-
-        server_config = ServerDeploymentConfig.model_validate(config_dict)
-
-        from zenml.zen_server.deploy.deployer import ServerDeployer
-
-        deployer = ServerDeployer()
-
-        server = get_active_deployment(local=False)
-        if server:
-            if server.config.provider != provider:
-                cli_utils.error(
-                    "ZenML is already deployed using a different provider "
-                    f"({server.config.provider}). Please tear down the "
-                    "existing deployment by running `zenml destroy` before "
-                    "deploying a new one."
-                )
-
-            if server.config.name != name:
-                cli_utils.error(
-                    f"An existing deployment with a different name "
-                    f"'{server.config.name}' already exists. Please tear down "
-                    f"the existing deployment by running `zenml destroy` "
-                    f"before deploying a new one."
-                )
-
-        server = deployer.deploy_server(server_config, timeout=timeout)
-
-        metadata = {
-            "server_deployment": str(server.config.provider),
-        }
-
-        analytics_handler.metadata = metadata
-
-        if server.status and server.status.url:
-            cli_utils.declare(
-                f"ZenML server '{name}' running at '{server.status.url}'. To "
-                "connect to the server, run `zenml connect --url "
-                f"{server.status.url}`."
-            )
-
-
-@cli.command(
-    "destroy", help="Tear down and clean up the cloud ZenML deployment."
-)
-def destroy() -> None:
-    """Tear down and clean up a cloud ZenML deployment."""
-    server = get_active_deployment(local=False)
-    if not server:
-        cli_utils.declare("No cloud ZenML server has been deployed.")
-        return
-
-    from zenml.zen_server.deploy.deployer import ServerDeployer
-
-    deployer = ServerDeployer()
-    deployer.remove_server(server.config.name)
-
-    cli_utils.declare(
-        "The ZenML server has been torn down and all resources removed."
-    )
-
-
 @cli.command(
     "status", help="Show information about the current configuration."
 )
@@ -497,18 +310,11 @@ def status() -> None:
         f"Local store files are located at: '{gc.local_stores_path}'"
     )
 
-    server = get_active_deployment(local=True)
+    server = get_local_server()
     if server:
         cli_utils.declare("The status of the local dashboard:")
         cli_utils.print_server_deployment(server)
 
-    server = get_active_deployment(local=False)
-    if server:
-        cli_utils.declare(
-            "The status of the cloud ZenML server deployed from this host:"
-        )
-        cli_utils.print_server_deployment(server)
-
 
 @cli.command(
     "connect",
@@ -661,7 +467,7 @@ def connect(
 
     # Raise an error if a local server is running when trying to connect to
     # another server
-    active_deployment = get_active_deployment(local=True)
+    active_deployment = get_local_server()
     if (
         active_deployment
         and active_deployment.status
@@ -701,24 +507,6 @@ def connect(
         api_key = api_key or store_dict.get("api_key")
         verify_ssl = store_dict.get("verify_ssl", verify_ssl)
 
-    elif url is None:
-        server = get_active_deployment(local=False)
-
-        if server is None or not server.status or not server.status.url:
-            cli_utils.warning(
-                "Running `zenml connect` without arguments can only be used to "
-                "connect to a ZenML server previously deployed from this host "
-                "with `zenml deploy`, but no such active deployment was found. "
-                "Please use the `--url` or `--config` command line arguments "
-                "to configure how to connect to a remote third party ZenML "
-                "server. Alternatively, call `zenml up` to start the ZenML "
-                "dashboard locally."
-            )
-            return
-        url = server.status.url
-        if server.status.ca_crt:
-            verify_ssl = server.status.ca_crt
-
     if not url:
         url = click.prompt("ZenML server URL", type=str)
     else:
@@ -792,12 +580,7 @@ def disconnect_server() -> None:
         cli_utils.declare("Restored default store configuration.")
 
 
-@cli.command("logs", help="Show the logs for the local or cloud ZenML server.")
-@click.option(
-    "--local",
-    is_flag=True,
-    help="Show the logs for the local ZenML server.",
-)
+@cli.command("logs", help="Show the logs for the local ZenML server.")
 @click.option(
     "--follow",
     "-f",
@@ -818,7 +601,6 @@ def disconnect_server() -> None:
     help="Show raw log contents (don't pretty-print logs).",
 )
 def logs(
-    local: bool = False,
     follow: bool = False,
     raw: bool = False,
     tail: Optional[int] = None,
@@ -826,17 +608,11 @@ def logs(
     """Display the logs for a ZenML server.
 
     Args:
-        local: Whether to show the logs for the local ZenML server.
         follow: Continue to output new log data as it becomes available.
         tail: Only show the last NUM lines of log output.
         raw: Show raw log contents (don't pretty-print logs).
     """
-    server = get_active_deployment(local=True)
-    if not local:
-        remote_server = get_active_deployment(local=False)
-        if remote_server is not None:
-            server = remote_server
-
+    server = get_local_server()
     if server is None:
         cli_utils.error(
             "The local ZenML dashboard is not running. Please call `zenml "