Fix ZenML Pro project URLs for pipeline runs and model versions

Author: stefannica

src/zenml/constants.py

@@ -368,6 +368,7 @@ def handle_int_env_var(var: str, default: int = 0) -> int:
 PIPELINES = "/pipelines"
 PIPELINE_SPEC = "/pipeline-spec"
 PLUGIN_FLAVORS = "/plugin-flavors"
+PROJECTS = "/projects"
 REFRESH = "/refresh"
 RUNS = "/runs"
 RUN_TEMPLATES = "/run_templates"

src/zenml/orchestrators/step_run_utils.py

@@ -352,7 +352,7 @@ def log_model_version_dashboard_url(
     """
     from zenml.utils.dashboard_utils import get_model_version_url
 
-    if model_version_url := get_model_version_url(model_version.id):
+    if model_version_url := get_model_version_url(model_version):
         logger.info(
             "Dashboard URL for Model Version `%s (%s)`:\n%s",
             model_version.model.name,

src/zenml/utils/dashboard_utils.py

@@ -15,7 +15,6 @@
 
 import os
 from typing import Optional
-from uuid import UUID
 
 from zenml import constants
 from zenml.client import Client
@@ -24,6 +23,7 @@
 from zenml.logger import get_logger
 from zenml.models import (
     ComponentResponse,
+    ModelVersionResponse,
     PipelineRunResponse,
     ServerDeploymentType,
     StackResponse,
@@ -115,7 +115,7 @@ def get_run_url(run: PipelineRunResponse) -> Optional[str]:
     """
     cloud_url = get_cloud_dashboard_url()
     if cloud_url:
-        return f"{cloud_url}{constants.RUNS}/{run.id}"
+        return f"{cloud_url}{constants.PROJECTS}/{run.project.id}{constants.RUNS}/{run.id}"
 
     dashboard_url = get_server_dashboard_url()
     if dashboard_url:
@@ -124,18 +124,20 @@ def get_run_url(run: PipelineRunResponse) -> Optional[str]:
     return None
 
 
-def get_model_version_url(model_version_id: UUID) -> Optional[str]:
+def get_model_version_url(
+    model_version: ModelVersionResponse,
+) -> Optional[str]:
     """Function to get the dashboard URL of a given model version.
 
     Args:
-        model_version_id: the id of the model version.
+        model_version: the response model of the given model version.
 
     Returns:
         the URL to the model version if the dashboard is available, else None.
     """
     cloud_url = get_cloud_dashboard_url()
     if cloud_url:
-        return f"{cloud_url}/model-versions/{str(model_version_id)}"
+        return f"{cloud_url}{constants.PROJECTS}/{model_version.project.id}/model-versions/{str(model_version.id)}"
 
     return None