zenml-nightly 0.80.2.dev20250414__py3-none-any.whl → 0.80.2.dev20250416__py3-none-any.whl

zenml/logging/step_logging.py

@@ -13,6 +13,7 @@
 #  permissions and limitations under the License.
 """ZenML logging handler."""
 
+import logging
 import os
 import re
 import sys
@@ -48,6 +49,7 @@ logger = get_logger(__name__)
 redirected: ContextVar[bool] = ContextVar("redirected", default=False)
 
 LOGS_EXTENSION = ".log"
+PIPELINE_RUN_LOGS_FOLDER = "pipeline_runs"
 
 
 def remove_ansi_escape_codes(text: str) -> str:
@@ -65,14 +67,14 @@ def remove_ansi_escape_codes(text: str) -> str:
 
 def prepare_logs_uri(
     artifact_store: "BaseArtifactStore",
-    step_name: str,
+    step_name: Optional[str] = None,
     log_key: Optional[str] = None,
 ) -> str:
     """Generates and prepares a URI for the log file or folder for a step.
 
     Args:
         artifact_store: The artifact store on which the artifact will be stored.
-        step_name: Name of the step.
+        step_name: Name of the step. Skipped for global pipeline run logs.
         log_key: The unique identification key of the log file.
 
     Returns:
@@ -81,11 +83,8 @@ def prepare_logs_uri(
     if log_key is None:
         log_key = str(uuid4())
 
-    logs_base_uri = os.path.join(
-        artifact_store.path,
-        step_name,
-        "logs",
-    )
+    subfolder = step_name or PIPELINE_RUN_LOGS_FOLDER
+    logs_base_uri = os.path.join(artifact_store.path, subfolder, "logs")
 
     # Create the dir
     if not artifact_store.exists(logs_base_uri):
@@ -210,7 +209,7 @@ def fetch_logs(
         artifact_store.cleanup()
 
 
-class StepLogsStorage:
+class PipelineLogsStorage:
     """Helper class which buffers and stores logs to a given URI."""
 
     def __init__(
@@ -324,6 +323,18 @@ class StepLogsStorage:
             self.disabled = True
 
             try:
+                # The configured logging handler uses a lock to ensure that
+                # logs generated by different threads are not interleaved.
+                # Given that most artifact stores are based on fsspec, which
+                # use a separate thread for async operations, it may happen that
+                # the fsspec library itself will log something, which will end
+                # up in a deadlock.
+                # To avoid this, we temporarily disable the lock in the logging
+                # handler while writing to the file.
+                logging_handler = logging.getLogger().handlers[0]
+                logging_lock = logging_handler.lock
+                logging_handler.lock = None
+
                 if self.buffer:
                     if self.artifact_store.config.IS_IMMUTABLE_FILESYSTEM:
                         _logs_uri = self._get_timestamped_filename()
@@ -353,6 +364,9 @@ class StepLogsStorage:
                 # I/O errors.
                 logger.error(f"Error while trying to write logs: {e}")
             finally:
+                # Restore the original logging handler lock
+                logging_handler.lock = logging_lock
+
                 self.buffer = []
                 self.last_save_time = time.time()
 
@@ -418,27 +432,32 @@ class StepLogsStorage:
                         )
 
 
-class StepLogsStorageContext:
-    """Context manager which patches stdout and stderr during step execution."""
+class PipelineLogsStorageContext:
+    """Context manager which patches stdout and stderr during pipeline run execution."""
 
     def __init__(
-        self, logs_uri: str, artifact_store: "BaseArtifactStore"
+        self,
+        logs_uri: str,
+        artifact_store: "BaseArtifactStore",
+        prepend_step_name: bool = True,
     ) -> None:
         """Initializes and prepares a storage object.
 
         Args:
             logs_uri: the URI of the logs file.
-            artifact_store: Artifact Store from the current step context.
+            artifact_store: Artifact Store from the current pipeline run context.
+            prepend_step_name: Whether to prepend the step name to the logs.
         """
-        self.storage = StepLogsStorage(
+        self.storage = PipelineLogsStorage(
             logs_uri=logs_uri, artifact_store=artifact_store
         )
+        self.prepend_step_name = prepend_step_name
 
-    def __enter__(self) -> "StepLogsStorageContext":
+    def __enter__(self) -> "PipelineLogsStorageContext":
         """Enter condition of the context manager.
 
         Wraps the `write` method of both stderr and stdout, so each incoming
-        message gets stored in the step logs storage.
+        message gets stored in the pipeline logs storage.
 
         Returns:
             self
@@ -499,14 +518,17 @@ class StepLogsStorageContext:
         """
 
         def wrapped_write(*args: Any, **kwargs: Any) -> Any:
-            # Check if step names in logs are disabled via env var
-            step_names_disabled = handle_bool_env_var(
-                ENV_ZENML_DISABLE_STEP_NAMES_IN_LOGS, default=False
+            step_names_disabled = (
+                handle_bool_env_var(
+                    ENV_ZENML_DISABLE_STEP_NAMES_IN_LOGS, default=False
+                )
+                or not self.prepend_step_name
             )
 
             if step_names_disabled:
                 output = method(*args, **kwargs)
             else:
+                message = args[0]
                 # Try to get step context if not available yet
                 step_context = None
                 try:
@@ -515,9 +537,7 @@ class StepLogsStorageContext:
                     pass
 
                 if step_context and args[0] != "\n":
-                    message = f"[{step_context.step_name}] " + args[0]
-                else:
-                    message = args[0]
+                    message = f"[{step_context.step_name}] " + message
 
                 output = method(message, *args[1:], **kwargs)
 

zenml/login/credentials_store.py

@@ -25,6 +25,7 @@ from zenml.constants import (
 from zenml.io import fileio
 from zenml.logger import get_logger
 from zenml.login.credentials import APIToken, ServerCredentials, ServerType
+from zenml.login.pro.constants import ZENML_PRO_API_URL
 from zenml.login.pro.workspace.models import WorkspaceRead
 from zenml.models import OAuthTokenResponse, ServerModel
 from zenml.utils import yaml_utils
@@ -396,6 +397,36 @@ class CredentialsStore(metaclass=SingletonMetaClass):
         """
         return self.get_pro_token(pro_api_url) is not None
 
+    def can_login(self, server_url: str) -> bool:
+        """Check if credentials to login to the given server exist.
+
+        Args:
+            server_url: The server URL for which to check the authentication.
+
+        Returns:
+            True if the credentials store contains credentials that can be used
+            to login to the given server URL, False otherwise.
+        """
+        self.check_and_reload_from_file()
+        credentials = self.get_credentials(server_url)
+        if not credentials:
+            return False
+
+        if credentials.api_key is not None:
+            return True
+        elif (
+            credentials.username is not None
+            and credentials.password is not None
+        ):
+            return True
+        elif credentials.type == ServerType.PRO:
+            pro_api_url = credentials.pro_api_url or ZENML_PRO_API_URL
+            pro_token = self.get_pro_token(pro_api_url, allow_expired=False)
+            if pro_token:
+                return True
+
+        return False
+
     def set_api_key(
         self,
         server_url: str,

zenml/materializers/path_materializer.py

@@ -26,6 +26,7 @@ from zenml.constants import (
 from zenml.enums import ArtifactType
 from zenml.io import fileio
 from zenml.materializers.base_materializer import BaseMaterializer
+from zenml.utils.io_utils import is_path_within_directory
 
 
 class PathMaterializer(BaseMaterializer):
@@ -71,7 +72,15 @@ class PathMaterializer(BaseMaterializer):
 
                 # Extract the archive to the temporary directory
                 with tarfile.open(archive_path_local, "r:gz") as tar:
-                    tar.extractall(path=directory)
+                    # Validate archive members to prevent path traversal attacks
+                    # Filter members to only those with safe paths
+                    safe_members = []
+                    for member in tar.getmembers():
+                        if is_path_within_directory(member.name, directory):
+                            safe_members.append(member)
+
+                    # Extract only safe members
+                    tar.extractall(path=directory, members=safe_members)  # nosec B202 - members are filtered through is_path_within_directory
 
                 # Clean up the archive file
                 os.remove(archive_path_local)
@@ -93,8 +102,14 @@ class PathMaterializer(BaseMaterializer):
 
         Args:
             data: Path to a local directory or file to store. Must be a Path object.
+
+        Raises:
+            TypeError: If data is not a Path object.
         """
-        assert isinstance(data, Path)
+        if not isinstance(data, Path):
+            raise TypeError(
+                f"Expected a Path object, got {type(data).__name__}"
+            )
 
         if data.is_dir():
             # Handle directory artifact

zenml/models/v2/base/base.py

@@ -134,7 +134,7 @@ class BaseResponse(BaseZenModel, Generic[AnyBody, AnyMetadata, AnyResources]):
             )
 
         # Check if the name has changed
-        if "name" in self.model_fields:
+        if "name" in type(self).model_fields:
             original_name = getattr(self, "name")
             hydrated_name = getattr(hydrated_model, "name")
 
@@ -172,7 +172,7 @@ class BaseResponse(BaseZenModel, Generic[AnyBody, AnyMetadata, AnyResources]):
                     )
 
         # Check all the fields in the body
-        for field in self.get_body().model_fields:
+        for field in type(self.get_body()).model_fields:
             original_value = getattr(self.get_body(), field)
             hydrated_value = getattr(hydrated_model.get_body(), field)
 
@@ -255,7 +255,9 @@ class BaseResponse(BaseZenModel, Generic[AnyBody, AnyMetadata, AnyResources]):
         """
         if self.metadata is None:
             # If the metadata is not there, check the class first.
-            metadata_annotation = self.model_fields["metadata"].annotation
+            metadata_annotation = (
+                type(self).model_fields["metadata"].annotation
+            )
             assert metadata_annotation is not None, (
                 "For each response model, an annotated metadata"
                 "field should exist."
@@ -293,7 +295,9 @@ class BaseResponse(BaseZenModel, Generic[AnyBody, AnyMetadata, AnyResources]):
         """
         if self.resources is None:
             # If the resources are not there, check the class first.
-            resources_annotation = self.model_fields["resources"].annotation
+            resources_annotation = (
+                type(self).model_fields["resources"].annotation
+            )
             assert resources_annotation is not None, (
                 "For each response model, an annotated resources"
                 "field should exist."

zenml/models/v2/base/filter.py

@@ -665,7 +665,7 @@ class BaseFilter(BaseModel):
             A list of Filter models.
         """
         return self._generate_filter_list(
-            {key: getattr(self, key) for key in self.model_fields}
+            {key: getattr(self, key) for key in type(self).model_fields}
         )
 
     @property

zenml/models/v2/core/pipeline_run.py

@@ -45,6 +45,7 @@ from zenml.models.v2.base.scoped import (
     RunMetadataFilterMixin,
     TaggableFilter,
 )
+from zenml.models.v2.core.logs import LogsRequest
 from zenml.models.v2.core.model_version import ModelVersionResponse
 from zenml.models.v2.core.tag import TagResponse
 from zenml.utils.tag_utils import Tag
@@ -55,6 +56,7 @@ if TYPE_CHECKING:
     from zenml.models import TriggerExecutionResponse
     from zenml.models.v2.core.artifact_version import ArtifactVersionResponse
     from zenml.models.v2.core.code_reference import CodeReferenceResponse
+    from zenml.models.v2.core.logs import LogsResponse
     from zenml.models.v2.core.pipeline import PipelineResponse
     from zenml.models.v2.core.pipeline_build import (
         PipelineBuildResponse,
@@ -124,6 +126,10 @@ class PipelineRunRequest(ProjectScopedRequest):
         default=None,
         title="Tags of the pipeline run.",
     )
+    logs: Optional[LogsRequest] = Field(
+        default=None,
+        title="Logs of the pipeline run.",
+    )
 
     model_config = ConfigDict(protected_namespaces=())
 
@@ -252,6 +258,10 @@ class PipelineRunResponseResources(ProjectScopedResponseResources):
     tags: List[TagResponse] = Field(
         title="Tags associated with the pipeline run.",
     )
+    logs: Optional["LogsResponse"] = Field(
+        title="Logs associated with this pipeline run.",
+        default=None,
+    )
 
     # TODO: In Pydantic v2, the `model_` is a protected namespaces for all
     #  fields defined under base models. If not handled, this raises a warning.
@@ -579,6 +589,15 @@ class PipelineRunResponse(
         """
         return self.get_resources().tags
 
+    @property
+    def logs(self) -> Optional["LogsResponse"]:
+        """The `logs` property.
+
+        Returns:
+            the value of the property.
+        """
+        return self.get_resources().logs
+
 
 # ------------------ Filter Model ------------------
 

zenml/orchestrators/step_launcher.py

@@ -158,7 +158,7 @@ class StepLauncher:
                 step_name=self._step_name,
             )
 
-            logs_context = step_logging.StepLogsStorageContext(
+            logs_context = step_logging.PipelineLogsStorageContext(
                 logs_uri=logs_uri, artifact_store=self._stack.artifact_store
             )  # type: ignore[assignment]
 
@@ -240,7 +240,7 @@ class StepLauncher:
                             # the external jobs in step operators
                             if isinstance(
                                 logs_context,
-                                step_logging.StepLogsStorageContext,
+                                step_logging.PipelineLogsStorageContext,
                             ):
                                 force_write_logs = partial(
                                     logs_context.storage.save_to_file,
@@ -421,7 +421,6 @@ class StepLauncher:
         )
         environment = orchestrator_utils.get_config_environment_vars(
             pipeline_run_id=step_run_info.run_id,
-            step_run_id=step_run_info.step_run_id,
         )
         if last_retry:
             environment[ENV_ZENML_IGNORE_FAILURE_HOOK] = str(False)

zenml/orchestrators/step_runner.py

@@ -40,7 +40,7 @@ from zenml.constants import (
 from zenml.enums import ArtifactSaveType
 from zenml.exceptions import StepInterfaceError
 from zenml.logger import get_logger
-from zenml.logging.step_logging import StepLogsStorageContext, redirected
+from zenml.logging.step_logging import PipelineLogsStorageContext, redirected
 from zenml.materializers.base_materializer import BaseMaterializer
 from zenml.models.v2.core.step_run import StepRunInputResponse
 from zenml.orchestrators.publish_utils import (
@@ -136,7 +136,7 @@ class StepRunner:
         logs_context = nullcontext()
         if step_logging_enabled and not redirected.get():
             if step_run.logs:
-                logs_context = StepLogsStorageContext(  # type: ignore[assignment]
+                logs_context = PipelineLogsStorageContext(  # type: ignore[assignment]
                     logs_uri=step_run.logs.uri,
                     artifact_store=self._stack.artifact_store,
                 )

zenml/orchestrators/utils.py

@@ -105,7 +105,6 @@ def is_setting_enabled(
 def get_config_environment_vars(
     schedule_id: Optional[UUID] = None,
     pipeline_run_id: Optional[UUID] = None,
-    step_run_id: Optional[UUID] = None,
 ) -> Dict[str, str]:
     """Gets environment variables to set for mirroring the active config.
 
@@ -119,7 +118,6 @@ def get_config_environment_vars(
         schedule_id: Optional schedule ID to use to generate a new API token.
         pipeline_run_id: Optional pipeline run ID to use to generate a new API
             token.
-        step_run_id: Optional step run ID to use to generate a new API token.
 
     Returns:
         Environment variable dict.
@@ -138,7 +136,7 @@ def get_config_environment_vars(
         credentials_store = get_credentials_store()
         url = global_config.store_configuration.url
         api_token = credentials_store.get_token(url, allow_expired=False)
-        if schedule_id or pipeline_run_id or step_run_id:
+        if schedule_id or pipeline_run_id:
             assert isinstance(global_config.zen_store, RestZenStore)
 
             # The user has the option to manually set an expiration for the API
@@ -173,7 +171,7 @@ def get_config_environment_vars(
                 # If only a schedule is given, the pipeline run credentials will
                 # be valid for the entire duration of the schedule.
                 api_key = credentials_store.get_api_key(url)
-                if not api_key and not pipeline_run_id and not step_run_id:
+                if not api_key and not pipeline_run_id:
                     logger.warning(
                         "An API token without an expiration time will be generated "
                         "and used to run this pipeline on a schedule. This is very "
@@ -194,7 +192,6 @@ def get_config_environment_vars(
                     token_type=APITokenType.WORKLOAD,
                     schedule_id=schedule_id,
                     pipeline_run_id=pipeline_run_id,
-                    step_run_id=step_run_id,
                 )
 
             environment_vars[ENV_ZENML_STORE_PREFIX + "API_TOKEN"] = (

zenml/pipelines/pipeline_context.py

@@ -109,6 +109,7 @@ class PipelineContext:
             pipeline_configuration.enable_artifact_visualization
         )
         self.enable_step_logs = pipeline_configuration.enable_step_logs
+        self.enable_pipeline_logs = pipeline_configuration.enable_pipeline_logs
         self.settings = pipeline_configuration.settings
         self.extra = pipeline_configuration.extra
         self.model = pipeline_configuration.model

zenml/pipelines/pipeline_decorator.py

@@ -50,6 +50,7 @@ def pipeline(
     enable_cache: Optional[bool] = None,
     enable_artifact_metadata: Optional[bool] = None,
     enable_step_logs: Optional[bool] = None,
+    enable_pipeline_logs: Optional[bool] = None,
     settings: Optional[Dict[str, "SettingsOrDict"]] = None,
     tags: Optional[List[Union[str, "Tag"]]] = None,
     extra: Optional[Dict[str, Any]] = None,
@@ -67,6 +68,7 @@ def pipeline(
     enable_cache: Optional[bool] = None,
     enable_artifact_metadata: Optional[bool] = None,
     enable_step_logs: Optional[bool] = None,
+    enable_pipeline_logs: Optional[bool] = None,
     settings: Optional[Dict[str, "SettingsOrDict"]] = None,
     tags: Optional[List[Union[str, "Tag"]]] = None,
     extra: Optional[Dict[str, Any]] = None,
@@ -84,6 +86,7 @@ def pipeline(
         enable_cache: Whether to use caching or not.
         enable_artifact_metadata: Whether to enable artifact metadata or not.
         enable_step_logs: If step logs should be enabled for this pipeline.
+        enable_pipeline_logs: If pipeline logs should be enabled for this pipeline.
         settings: Settings for this pipeline.
         tags: Tags to apply to runs of the pipeline.
         extra: Extra configurations for this pipeline.
@@ -108,6 +111,7 @@ def pipeline(
             enable_cache=enable_cache,
             enable_artifact_metadata=enable_artifact_metadata,
             enable_step_logs=enable_step_logs,
+            enable_pipeline_logs=enable_pipeline_logs,
             settings=settings,
             tags=tags,
             extra=extra,

zenml/pipelines/pipeline_definition.py

@@ -16,7 +16,7 @@
 import copy
 import hashlib
 import inspect
-from contextlib import contextmanager
+from contextlib import contextmanager, nullcontext
 from pathlib import Path
 from typing import (
     TYPE_CHECKING,
@@ -56,8 +56,13 @@ from zenml.enums import StackComponentType
 from zenml.exceptions import EntityExistsError
 from zenml.hooks.hook_validators import resolve_and_validate_hook
 from zenml.logger import get_logger
+from zenml.logging.step_logging import (
+    PipelineLogsStorageContext,
+    prepare_logs_uri,
+)
 from zenml.models import (
     CodeReferenceRequest,
+    LogsRequest,
     PipelineBuildBase,
     PipelineBuildResponse,
     PipelineDeploymentBase,
@@ -130,6 +135,7 @@ class Pipeline:
         enable_artifact_metadata: Optional[bool] = None,
         enable_artifact_visualization: Optional[bool] = None,
         enable_step_logs: Optional[bool] = None,
+        enable_pipeline_logs: Optional[bool] = None,
         settings: Optional[Mapping[str, "SettingsOrDict"]] = None,
         tags: Optional[List[Union[str, "Tag"]]] = None,
         extra: Optional[Dict[str, Any]] = None,
@@ -149,6 +155,7 @@ class Pipeline:
             enable_artifact_visualization: If artifact visualization should be
                 enabled for this pipeline.
             enable_step_logs: If step logs should be enabled for this pipeline.
+            enable_pipeline_logs: If pipeline logs should be enabled for this pipeline.
             settings: Settings for this pipeline.
             tags: Tags to apply to runs of this pipeline.
             extra: Extra configurations for this pipeline.
@@ -174,6 +181,7 @@ class Pipeline:
                 enable_artifact_metadata=enable_artifact_metadata,
                 enable_artifact_visualization=enable_artifact_visualization,
                 enable_step_logs=enable_step_logs,
+                enable_pipeline_logs=enable_pipeline_logs,
                 settings=settings,
                 tags=tags,
                 extra=extra,
@@ -293,6 +301,7 @@ class Pipeline:
         enable_artifact_metadata: Optional[bool] = None,
         enable_artifact_visualization: Optional[bool] = None,
         enable_step_logs: Optional[bool] = None,
+        enable_pipeline_logs: Optional[bool] = None,
         settings: Optional[Mapping[str, "SettingsOrDict"]] = None,
         tags: Optional[List[Union[str, "Tag"]]] = None,
         extra: Optional[Dict[str, Any]] = None,
@@ -322,6 +331,7 @@ class Pipeline:
             enable_artifact_visualization: If artifact visualization should be
                 enabled for this pipeline.
             enable_step_logs: If step logs should be enabled for this pipeline.
+            enable_pipeline_logs: If pipeline logs should be enabled for this pipeline.
             settings: settings for this pipeline.
             tags: Tags to apply to runs of this pipeline.
             extra: Extra configurations for this pipeline.
@@ -364,6 +374,7 @@ class Pipeline:
                 "enable_artifact_metadata": enable_artifact_metadata,
                 "enable_artifact_visualization": enable_artifact_visualization,
                 "enable_step_logs": enable_step_logs,
+                "enable_pipeline_logs": enable_pipeline_logs,
                 "settings": settings,
                 "tags": tags,
                 "extra": extra,
@@ -588,6 +599,7 @@ To avoid this consider setting pipeline parameters only in one place (config or
         enable_artifact_metadata: Optional[bool] = None,
         enable_artifact_visualization: Optional[bool] = None,
         enable_step_logs: Optional[bool] = None,
+        enable_pipeline_logs: Optional[bool] = None,
         schedule: Optional[Schedule] = None,
         build: Union[str, "UUID", "PipelineBuildBase", None] = None,
         settings: Optional[Mapping[str, "SettingsOrDict"]] = None,
@@ -610,6 +622,8 @@ To avoid this consider setting pipeline parameters only in one place (config or
             enable_artifact_visualization: If artifact visualization should be
                 enabled for this pipeline run.
             enable_step_logs: If step logs should be enabled for this pipeline.
+            enable_pipeline_logs: If pipeline logs should be enabled for this
+                pipeline run.
             schedule: Optional schedule to use for the run.
             build: Optional build to use for the run.
             settings: Settings for this pipeline run.
@@ -641,6 +655,7 @@ To avoid this consider setting pipeline parameters only in one place (config or
             enable_artifact_metadata=enable_artifact_metadata,
             enable_artifact_visualization=enable_artifact_visualization,
             enable_step_logs=enable_step_logs,
+            enable_pipeline_logs=enable_pipeline_logs,
             steps=step_configurations,
             settings=settings,
             schedule=schedule,
@@ -723,7 +738,7 @@ To avoid this consider setting pipeline parameters only in one place (config or
         if prevent_build_reuse:
             logger.warning(
                 "Passing `prevent_build_reuse=True` to "
-                "`pipeline.with_opitions(...)` is deprecated. Use "
+                "`pipeline.with_options(...)` is deprecated. Use "
                 "`DockerSettings.prevent_build_reuse` instead."
             )
 
@@ -806,31 +821,77 @@ To avoid this consider setting pipeline parameters only in one place (config or
 
         with track_handler(AnalyticsEvent.RUN_PIPELINE) as analytics_handler:
             stack = Client().active_stack
-            deployment = self._create_deployment(**self._run_args)
 
-            self.log_pipeline_deployment_metadata(deployment)
-            run = create_placeholder_run(deployment=deployment)
+            # Enable or disable pipeline run logs storage
+            if self._run_args.get("schedule"):
+                # Pipeline runs scheduled to run in the future are not logged
+                # via the client.
+                logging_enabled = False
+            elif constants.handle_bool_env_var(
+                constants.ENV_ZENML_DISABLE_PIPELINE_LOGS_STORAGE, False
+            ):
+                logging_enabled = False
+            else:
+                logging_enabled = self._run_args.get(
+                    "enable_pipeline_logs",
+                    self.configuration.enable_pipeline_logs
+                    if self.configuration.enable_pipeline_logs is not None
+                    else True,
+                )
 
-            analytics_handler.metadata = self._get_pipeline_analytics_metadata(
-                deployment=deployment,
-                stack=stack,
-                run_id=run.id if run else None,
-            )
+            logs_context = nullcontext()
+            logs_model = None
 
-            if run:
-                run_url = dashboard_utils.get_run_url(run)
-                if run_url:
-                    logger.info(f"Dashboard URL for Pipeline Run: {run_url}")
-                else:
-                    logger.info(
-                        "You can visualize your pipeline runs in the `ZenML "
-                        "Dashboard`. In order to try it locally, please run "
-                        "`zenml login --local`."
+            if logging_enabled:
+                # Configure the logs
+                logs_uri = prepare_logs_uri(
+                    stack.artifact_store,
+                )
+
+                logs_context = PipelineLogsStorageContext(
+                    logs_uri=logs_uri,
+                    artifact_store=stack.artifact_store,
+                    prepend_step_name=False,
+                )  # type: ignore[assignment]
+
+                logs_model = LogsRequest(
+                    uri=logs_uri,
+                    artifact_store_id=stack.artifact_store.id,
+                )
+
+            with logs_context:
+                deployment = self._create_deployment(**self._run_args)
+
+                self.log_pipeline_deployment_metadata(deployment)
+                run = create_placeholder_run(
+                    deployment=deployment, logs=logs_model
+                )
+
+                analytics_handler.metadata = (
+                    self._get_pipeline_analytics_metadata(
+                        deployment=deployment,
+                        stack=stack,
+                        run_id=run.id if run else None,
                     )
+                )
+
+                if run:
+                    run_url = dashboard_utils.get_run_url(run)
+                    if run_url:
+                        logger.info(
+                            f"Dashboard URL for Pipeline Run: {run_url}"
+                        )
+                    else:
+                        logger.info(
+                            "You can visualize your pipeline runs in the `ZenML "
+                            "Dashboard`. In order to try it locally, please run "
+                            "`zenml login --local`."
+                        )
+
+                deploy_pipeline(
+                    deployment=deployment, stack=stack, placeholder_run=run
+                )
 
-            deploy_pipeline(
-                deployment=deployment, stack=stack, placeholder_run=run
-            )
             if run:
                 return Client().get_pipeline_run(run.id)
             return None