zenml-nightly 0.82.1.dev20250527__py3-none-any.whl → 0.83.0.dev20250529__py3-none-any.whl

zenml/integrations/skypilot/utils.py

@@ -0,0 +1,273 @@
+"""Utility functions for Skypilot orchestrators."""
+
+import re
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple
+
+import sky
+
+from zenml.integrations.skypilot.flavors.skypilot_orchestrator_base_vm_config import (
+    SkypilotBaseOrchestratorSettings,
+)
+from zenml.logger import get_logger
+
+logger = get_logger(__name__)
+
+if TYPE_CHECKING:
+    from sky.clouds.cloud import Cloud
+
+
+def sanitize_cluster_name(name: str) -> str:
+    """Sanitize the value to be used in a cluster name.
+
+    Args:
+        name: Arbitrary input cluster name.
+
+    Returns:
+        Sanitized cluster name.
+    """
+    name = re.sub(
+        r"[^a-z0-9-]", "-", name.lower()
+    )  # replaces any character that is not a lowercase letter, digit, or hyphen with a hyphen
+    name = re.sub(r"^[-]+", "", name)  # trim leading hyphens
+    name = re.sub(r"[-]+$", "", name)  # trim trailing hyphens
+    return name
+
+
+def prepare_docker_setup(
+    container_registry_uri: str,
+    credentials: Optional[Tuple[str, str]] = None,
+    use_sudo: bool = True,
+) -> Tuple[Optional[str], Dict[str, str]]:
+    """Prepare Docker login setup command and environment variables.
+
+    Args:
+        container_registry_uri: URI of the container registry.
+        credentials: Optional credentials (username, password) tuple.
+        use_sudo: Whether to use sudo prefix in docker commands.
+
+    Returns:
+        Tuple of (setup command, environment variables)
+    """
+    if credentials:
+        docker_username, docker_password = credentials
+        sudo_prefix = "sudo " if use_sudo else ""
+        setup = (
+            f"{sudo_prefix}docker login --username $DOCKER_USERNAME --password "
+            f"$DOCKER_PASSWORD {container_registry_uri}"
+        )
+        task_envs = {
+            "DOCKER_USERNAME": docker_username,
+            "DOCKER_PASSWORD": docker_password,
+        }
+    else:
+        setup = None
+        task_envs = {}
+
+    return setup, task_envs
+
+
+def create_docker_run_command(
+    image: str,
+    entrypoint_str: str,
+    arguments_str: str,
+    environment: Dict[str, str],
+    docker_run_args: List[str],
+    use_sudo: bool = True,
+) -> str:
+    """Create a Docker run command string.
+
+    Args:
+        image: Docker image to run.
+        entrypoint_str: Entrypoint command.
+        arguments_str: Command arguments.
+        environment: Environment variables.
+        docker_run_args: Additional Docker run arguments.
+        use_sudo: Whether to use sudo prefix in docker commands.
+
+    Returns:
+        Docker run command as string.
+    """
+    docker_environment_str = " ".join(
+        f"-e {k}={v}" for k, v in environment.items()
+    )
+    custom_run_args = " ".join(docker_run_args)
+    if custom_run_args:
+        custom_run_args += " "
+
+    sudo_prefix = "sudo " if use_sudo else ""
+    return f"{sudo_prefix}docker run --rm {custom_run_args}{docker_environment_str} {image} {entrypoint_str} {arguments_str}"
+
+
+def prepare_task_kwargs(
+    settings: SkypilotBaseOrchestratorSettings,
+    run_command: str,
+    setup: Optional[str],
+    task_envs: Dict[str, str],
+    task_name: str,
+) -> Dict[str, Any]:
+    """Prepare task keyword arguments for sky.Task.
+
+    Args:
+        settings: Skypilot orchestrator settings.
+        run_command: Command to run.
+        setup: Setup command.
+        task_envs: Task environment variables.
+        task_name: Task name.
+
+    Returns:
+        Task keyword arguments dictionary.
+    """
+    # Merge envs from settings with existing task_envs
+    merged_envs = {}
+
+    # First add user-provided envs
+    if settings.envs:
+        merged_envs.update(settings.envs)
+
+    # Then add task_envs which take precedence
+    if task_envs:
+        merged_envs.update(task_envs)
+
+    task_kwargs = {
+        "run": run_command,
+        "setup": setup,
+        "envs": merged_envs,
+        "name": settings.task_name or task_name,
+        "workdir": settings.workdir,
+        "file_mounts_mapping": settings.file_mounts,
+        **settings.task_settings,  # Add any arbitrary task settings
+    }
+
+    # Remove None values to avoid overriding SkyPilot defaults
+    return {k: v for k, v in task_kwargs.items() if v is not None}
+
+
+def prepare_resources_kwargs(
+    cloud: "Cloud",
+    settings: SkypilotBaseOrchestratorSettings,
+    default_instance_type: Optional[str] = None,
+    kubernetes_image: Optional[str] = None,
+) -> Dict[str, Any]:
+    """Prepare resources keyword arguments for sky.Resources.
+
+    Args:
+        cloud: Skypilot cloud.
+        settings: Skypilot orchestrator settings.
+        default_instance_type: Default instance type.
+        kubernetes_image: Image to use for Kubernetes (if applicable).
+
+    Returns:
+        Resources keyword arguments dictionary.
+    """
+    resources_kwargs = {
+        "cloud": cloud,
+        "instance_type": settings.instance_type or default_instance_type,
+        "cpus": settings.cpus,
+        "memory": settings.memory,
+        "accelerators": settings.accelerators,
+        "accelerator_args": settings.accelerator_args,
+        "use_spot": settings.use_spot,
+        "job_recovery": settings.job_recovery,
+        "region": settings.region,
+        "zone": settings.zone,
+        "image_id": kubernetes_image
+        if kubernetes_image
+        else settings.image_id,
+        "disk_size": settings.disk_size,
+        "disk_tier": settings.disk_tier,
+        "ports": settings.ports,
+        "labels": settings.labels,
+        "any_of": settings.any_of,
+        "ordered": settings.ordered,
+        **settings.resources_settings,  # Add any arbitrary resource settings
+    }
+
+    # Remove None values to avoid overriding SkyPilot defaults
+    return {k: v for k, v in resources_kwargs.items() if v is not None}
+
+
+def prepare_launch_kwargs(
+    settings: SkypilotBaseOrchestratorSettings,
+    down: Optional[bool] = None,
+    idle_minutes_to_autostop: Optional[int] = None,
+) -> Dict[str, Any]:
+    """Prepare launch keyword arguments for sky.launch.
+
+    Args:
+        settings: Skypilot orchestrator settings.
+        down: Whether to tear down the cluster after job completion.
+        idle_minutes_to_autostop: Minutes to autostop after idleness.
+
+    Returns:
+        Launch keyword arguments dictionary.
+    """
+    # Determine values falling back to settings where applicable
+    down_value = down if down is not None else settings.down
+    idle_value = (
+        idle_minutes_to_autostop
+        if idle_minutes_to_autostop is not None
+        else settings.idle_minutes_to_autostop
+    )
+
+    # The following parameters were removed from sky.launch in versions > 0.8.
+    # We therefore no longer include them in the kwargs passed to the call.
+    # • stream_logs – handled by explicitly calling sky.stream_and_get
+    # • detach_setup / detach_run – setup/run are now detached by default
+
+    launch_kwargs = {
+        "retry_until_up": settings.retry_until_up,
+        "idle_minutes_to_autostop": idle_value,
+        "down": down_value,
+        "backend": None,
+        **settings.launch_settings,  # Keep user-provided extras
+    }
+
+    # Remove keys that are no longer supported by sky.launch.
+    for _deprecated in (
+        "stream_logs",
+        "detach_setup",
+        "detach_run",
+        "num_nodes",
+    ):
+        launch_kwargs.pop(_deprecated, None)
+
+    # Remove None values to avoid overriding SkyPilot defaults
+    return {k: v for k, v in launch_kwargs.items() if v is not None}
+
+
+def sky_job_get(request_id: str, stream_logs: bool, cluster_name: str) -> Any:
+    """Handle SkyPilot request results based on stream_logs setting.
+
+    SkyPilot API exec and launch methods are asynchronous and return a request ID.
+    This method waits for the operation to complete and returns the result.
+    If stream_logs is True, it will also stream the logs and wait for the
+    job to complete.
+
+    Args:
+        request_id: The request ID returned from a SkyPilot operation.
+        stream_logs: Whether to stream logs while waiting for completion.
+        cluster_name: The name of the cluster to tail logs for.
+
+    Returns:
+        The result of the SkyPilot operation.
+
+    Raises:
+        Exception: If the SkyPilot job fails.
+    """
+    if stream_logs:
+        # Stream logs and wait for completion
+        job_id, _ = sky.stream_and_get(request_id)
+    else:
+        # Just wait for completion without streaming logs
+        job_id, _ = sky.get(request_id)
+
+    status = 0  # 0=Successful, 100=Failed
+    if stream_logs:
+        status = sky.tail_logs(
+            cluster_name=cluster_name, job_id=job_id, follow=True
+        )
+
+    if status != 0:
+        raise Exception(f"SkyPilot job {job_id} failed with status {status}")
+
+    return job_id

zenml/integrations/skypilot_aws/__init__.py

@@ -32,7 +32,7 @@ class SkypilotAWSIntegration(Integration):
 
     NAME = SKYPILOT_AWS
     # all 0.6.x versions of skypilot[aws] are compatible
-    REQUIREMENTS = ["skypilot[aws]~=0.8.0"]
+    REQUIREMENTS = ["skypilot[aws]==0.9.3"]
     APT_PACKAGES = ["openssh-client", "rsync"]
 
     @classmethod
@@ -47,4 +47,3 @@ class SkypilotAWSIntegration(Integration):
         )
 
         return [SkypilotAWSOrchestratorFlavor]
-

zenml/integrations/skypilot_azure/__init__.py

@@ -31,7 +31,7 @@ class SkypilotAzureIntegration(Integration):
     """Definition of Skypilot (Azure) Integration for ZenML."""
 
     NAME = SKYPILOT_AZURE
-    REQUIREMENTS = ["skypilot[azure]~=0.8.0"]
+    REQUIREMENTS = ["skypilot[azure]==0.9.3"]
     APT_PACKAGES = ["openssh-client", "rsync"]
 
     @classmethod
@@ -46,4 +46,3 @@ class SkypilotAzureIntegration(Integration):
         )
 
         return [SkypilotAzureOrchestratorFlavor]
-

zenml/integrations/skypilot_gcp/__init__.py

@@ -31,7 +31,15 @@ class SkypilotGCPIntegration(Integration):
     """Definition of Skypilot (GCP) Integration for ZenML."""
 
     NAME = SKYPILOT_GCP
-    REQUIREMENTS = ["skypilot[gcp]~=0.8.0"]
+    REQUIREMENTS = [
+        "skypilot[gcp]==0.9.3",
+        # TODO: Remove this once the issue is fixed:
+        # Adding the dependencies of the GCP integration on top of the
+        # requirements of the skypilot integration results in a
+        # very long resolution time for pip. This is a workaround to
+        # speed up the resolution.
+        "protobuf>=4.25.0,<5.0.0",
+    ]
     APT_PACKAGES = ["openssh-client", "rsync"]
 
     @classmethod

zenml/integrations/skypilot_kubernetes/__init__.py

@@ -31,8 +31,8 @@ class SkypilotKubernetesIntegration(Integration):
     """Definition of Skypilot Kubernetes Integration for ZenML."""
 
     NAME = SKYPILOT_KUBERNETES
-    # all 0.6.x versions of skypilot[kubernetes] are compatible
-    REQUIREMENTS = ["skypilot[kubernetes]~=0.8.0"]
+
+    REQUIREMENTS = ["skypilot[kubernetes]==0.9.3"]
     APT_PACKAGES = ["openssh-client", "rsync"]
 
     @classmethod
@@ -47,4 +47,3 @@ class SkypilotKubernetesIntegration(Integration):
         )
 
         return [SkypilotKubernetesOrchestratorFlavor]
-

zenml/integrations/skypilot_lambda/__init__.py

@@ -31,7 +31,7 @@ class SkypilotLambdaIntegration(Integration):
     """Definition of Skypilot Lambda Integration for ZenML."""
 
     NAME = SKYPILOT_LAMBDA
-    REQUIREMENTS = ["skypilot[lambda]~=0.8.0"]
+    REQUIREMENTS = ["skypilot[lambda]==0.9.3"]
 
     @classmethod
     def flavors(cls) -> List[Type[Flavor]]:
@@ -45,4 +45,3 @@ class SkypilotLambdaIntegration(Integration):
         )
 
         return [SkypilotLambdaOrchestratorFlavor]
-

zenml/login/web_login.py

@@ -239,7 +239,7 @@ def web_login(
         else:
             # There was another error with the request
             raise AuthorizationException(
-                f"Error: {response.status_code} {response.json()['error']}"
+                f"Error: {response.status_code} {response.text}"
             )
 
     # Save the token in the credentials store

zenml/model/model.py

@@ -29,6 +29,7 @@ from zenml.enums import MetadataResourceTypes, ModelStages
 from zenml.exceptions import EntityExistsError
 from zenml.logger import get_logger
 from zenml.pipelines.pipeline_context import get_pipeline_context
+from zenml.utils import pagination_utils
 from zenml.utils.pydantic_utils import before_validator_handler
 from zenml.utils.string_utils import format_name_template
 
@@ -38,7 +39,6 @@ if TYPE_CHECKING:
         ArtifactVersionResponse,
         ModelResponse,
         ModelVersionResponse,
-        PipelineRunResponse,
     )
 
 logger = get_logger(__name__)
@@ -303,17 +303,6 @@ class Model(BaseModel):
             version=version,
         )
 
-    def get_pipeline_run(self, name: str) -> "PipelineRunResponse":
-        """Get pipeline run linked to this version.
-
-        Args:
-            name: The name of the pipeline run to retrieve.
-
-        Returns:
-            PipelineRun as PipelineRunResponse
-        """
-        return self._get_or_create_model_version().get_pipeline_run(name=name)
-
     def set_stage(
         self, stage: Union[str, ModelStages], force: bool = False
     ) -> None:
@@ -431,19 +420,20 @@ class Model(BaseModel):
 
         client = Client()
 
+        mv = self._get_or_create_model_version()
+
         if not only_link and delete_from_artifact_store:
-            mv = self._get_model_version()
-            artifact_responses = mv.data_artifacts
-            artifact_responses.update(mv.model_artifacts)
-            artifact_responses.update(mv.deployment_artifacts)
-
-            for artifact_ in artifact_responses.values():
-                for artifact_response_ in artifact_.values():
-                    client._delete_artifact_from_artifact_store(
-                        artifact_version=artifact_response_
-                    )
+            artifact_versions = pagination_utils.depaginate(
+                client.list_artifact_versions,
+                model_version_id=mv.id,
+                project=mv.project_id,
+            )
+            for artifact_version in artifact_versions:
+                client._delete_artifact_from_artifact_store(
+                    artifact_version=artifact_version
+                )
 
-        client.delete_all_model_version_artifact_links(self.id, only_link)
+        client.delete_all_model_version_artifact_links(mv.id, only_link)
 
     def _lazy_artifact_get(
         self,

zenml/models/__init__.py

@@ -86,6 +86,7 @@ from zenml.models.v2.core.artifact import (
     ArtifactResponse,
     ArtifactResponseBody,
     ArtifactResponseMetadata,
+    ArtifactResponseResources,
     ArtifactUpdate,
 )
 from zenml.models.v2.core.artifact_version import (
@@ -94,6 +95,7 @@ from zenml.models.v2.core.artifact_version import (
     ArtifactVersionResponse,
     ArtifactVersionResponseBody,
     ArtifactVersionResponseMetadata,
+    ArtifactVersionResponseResources,
     ArtifactVersionUpdate,
     LazyArtifactVersionResponse
 )
@@ -125,6 +127,7 @@ from zenml.models.v2.core.code_repository import (
     CodeRepositoryResponse,
     CodeRepositoryResponseBody,
     CodeRepositoryResponseMetadata,
+    CodeRepositoryResponseResources,
 )
 from zenml.models.v2.core.component import (
     ComponentBase,
@@ -149,6 +152,7 @@ from zenml.models.v2.core.device import (
     OAuthDeviceResponse,
     OAuthDeviceResponseBody,
     OAuthDeviceResponseMetadata,
+    OAuthDeviceResponseResources,
     OAuthDeviceInternalRequest,
     OAuthDeviceInternalUpdate,
     OAuthDeviceInternalResponse,
@@ -160,6 +164,7 @@ from zenml.models.v2.core.flavor import (
     FlavorResponse,
     FlavorResponseBody,
     FlavorResponseMetadata,
+    FlavorResponseResources,
 )
 from zenml.models.v2.core.logs import (
     LogsRequest,
@@ -172,6 +177,7 @@ from zenml.models.v2.core.model import (
     ModelResponse,
     ModelResponseBody,
     ModelResponseMetadata,
+    ModelResponseResources,
     ModelRequest,
     ModelUpdate,
 )
@@ -212,6 +218,7 @@ from zenml.models.v2.core.pipeline_build import (
     PipelineBuildResponse,
     PipelineBuildResponseBody,
     PipelineBuildResponseMetadata,
+    PipelineBuildResponseResources,
 )
 from zenml.models.v2.core.pipeline_deployment import (
     PipelineDeploymentBase,
@@ -251,6 +258,7 @@ from zenml.models.v2.core.schedule import (
     ScheduleResponse,
     ScheduleResponseBody,
     ScheduleResponseMetadata,
+    ScheduleResponseResources,
 )
 from zenml.models.v2.core.secret import (
     SecretFilter,
@@ -258,6 +266,7 @@ from zenml.models.v2.core.secret import (
     SecretResponse,
     SecretResponseBody,
     SecretResponseMetadata,
+    SecretResponseResources,
     SecretUpdate,
 )
 from zenml.models.v2.core.service_account import (
@@ -275,6 +284,7 @@ from zenml.models.v2.core.service_connector import (
     ServiceConnectorResponse,
     ServiceConnectorResponseBody,
     ServiceConnectorResponseMetadata,
+    ServiceConnectorResponseResources,
 )
 from zenml.models.v2.core.stack import (
     DefaultStackRequest,
@@ -284,6 +294,7 @@ from zenml.models.v2.core.stack import (
     StackResponse,
     StackResponseBody,
     StackResponseMetadata,
+    StackResponseResources
 )
 from zenml.models.v2.misc.statistics import (
     ProjectStatistics,
@@ -303,6 +314,7 @@ from zenml.models.v2.core.tag import (
     TagResponse,
     TagResponseBody,
     TagResponseMetadata,
+    TagResponseResources,
     TagRequest,
     TagUpdate,
 )
@@ -381,6 +393,7 @@ from zenml.models.v2.misc.auth_models import (
     OAuthRedirectResponse,
     OAuthTokenResponse,
 )
+from zenml.models.v2.misc.pipeline_run_dag import PipelineRunDAG
 from zenml.models.v2.misc.run_metadata import (
     RunMetadataEntry,
     RunMetadataResource,
@@ -423,12 +436,15 @@ APIKeyResponseBody.model_rebuild()
 ArtifactResponse.model_rebuild()
 ArtifactResponseBody.model_rebuild()
 ArtifactResponseMetadata.model_rebuild()
+ArtifactResponseResources.model_rebuild()
 ArtifactVersionRequest.model_rebuild()
 ArtifactVersionResponseBody.model_rebuild()
 ArtifactVersionResponseMetadata.model_rebuild()
+ArtifactVersionResponseResources.model_rebuild()
 CodeReferenceResponseBody.model_rebuild()
 CodeRepositoryResponseBody.model_rebuild()
 CodeRepositoryResponseMetadata.model_rebuild()
+CodeRepositoryResponseResources.model_rebuild()
 ComponentResponseBody.model_rebuild()
 ComponentResponseMetadata.model_rebuild()
 ComponentResponseResources.model_rebuild()
@@ -437,21 +453,24 @@ EventSourceResponseMetadata.model_rebuild()
 EventSourceResponseResources.model_rebuild()
 FlavorResponseBody.model_rebuild()
 FlavorResponseMetadata.model_rebuild()
+FlavorResponseResources.model_rebuild()
 LazyArtifactVersionResponse.model_rebuild()
 ModelResponseBody.model_rebuild()
 ModelResponseMetadata.model_rebuild()
+ModelResponseResources.model_rebuild()
 ModelVersionResponseBody.model_rebuild()
 ModelVersionResponseMetadata.model_rebuild()
 ModelVersionResponseResources.model_rebuild()
 ModelVersionArtifactResponseBody.model_rebuild()
 ModelVersionPipelineRunResponseBody.model_rebuild()
-OAuthDeviceResponseBody.model_rebuild()
+OAuthDeviceResponseResources.model_rebuild()
 PipelineResponseBody.model_rebuild()
 PipelineResponseMetadata.model_rebuild()
 PipelineResponseResources.model_rebuild()
 PipelineBuildBase.model_rebuild()
 PipelineBuildResponseBody.model_rebuild()
 PipelineBuildResponseMetadata.model_rebuild()
+PipelineBuildResponseResources.model_rebuild()
 PipelineDeploymentRequest.model_rebuild()
 PipelineDeploymentResponseBody.model_rebuild()
 PipelineDeploymentResponseMetadata.model_rebuild()
@@ -466,22 +485,26 @@ RunTemplateResponseResources.model_rebuild()
 RunTemplateResponseBody.model_rebuild()
 ScheduleResponseBody.model_rebuild()
 ScheduleResponseMetadata.model_rebuild()
+ScheduleResponseResources.model_rebuild()
 SecretResponseBody.model_rebuild()
 SecretResponseMetadata.model_rebuild()
+SecretResponseResources.model_rebuild()
 ServiceResponseBody.model_rebuild()
 ServiceResponseMetadata.model_rebuild()
 ServiceResponseResources.model_rebuild()
 ServiceConnectorResponseBody.model_rebuild()
 ServiceConnectorResponseMetadata.model_rebuild()
+ServiceConnectorResponseResources.model_rebuild()
 StackRequest.model_rebuild()
 StackResponseBody.model_rebuild()
 StackResponseMetadata.model_rebuild()
-StackResponse.model_rebuild()
+StackResponseResources.model_rebuild()
 StepRunRequest.model_rebuild()
 StepRunResponseBody.model_rebuild()
 StepRunResponseMetadata.model_rebuild()
 StepRunResponseResources.model_rebuild()
 TagResponseBody.model_rebuild()
+TagResponseResources.model_rebuild()
 TriggerExecutionResponseResources.model_rebuild()
 TriggerResponseBody.model_rebuild()
 TriggerResponseMetadata.model_rebuild()
@@ -552,12 +575,14 @@ __all__ = [
     "ArtifactResponse",
     "ArtifactResponseBody",
     "ArtifactResponseMetadata",
+    "ArtifactResponseResources",
     "ArtifactUpdate",
     "ArtifactVersionRequest",
     "ArtifactVersionFilter",
     "ArtifactVersionResponse",
     "ArtifactVersionResponseBody",
     "ArtifactVersionResponseMetadata",
+    "ArtifactVersionResponseResources",
     "ArtifactVersionUpdate",
     "ArtifactVisualizationRequest",
     "ArtifactVisualizationResponse",
@@ -573,6 +598,7 @@ __all__ = [
     "CodeRepositoryResponse",
     "CodeRepositoryResponseBody",
     "CodeRepositoryResponseMetadata",
+    "CodeRepositoryResponseResources",
     "ComponentBase",
     "ComponentRequest",
     "ComponentUpdate",
@@ -593,6 +619,7 @@ __all__ = [
     "FlavorResponse",
     "FlavorResponseBody",
     "FlavorResponseMetadata",
+    "FlavorResponseResources",
     "LogsRequest",
     "LogsResponse",
     "LogsResponseBody",
@@ -602,6 +629,7 @@ __all__ = [
     "ModelResponse",
     "ModelResponseBody",
     "ModelResponseMetadata",
+    "ModelResponseResources",
     "ModelUpdate",
     "ModelVersionFilter",
     "ModelVersionRequest",
@@ -623,6 +651,7 @@ __all__ = [
     "OAuthDeviceResponse",
     "OAuthDeviceResponseBody",
     "OAuthDeviceResponseMetadata",
+    "OAuthDeviceResponseResources",
     "OAuthDeviceInternalRequest",
     "OAuthDeviceInternalUpdate",
     "OAuthDeviceInternalResponse",
@@ -639,12 +668,14 @@ __all__ = [
     "PipelineBuildResponse",
     "PipelineBuildResponseBody",
     "PipelineBuildResponseMetadata",
+    "PipelineBuildResponseResources",
     "PipelineDeploymentBase",
     "PipelineDeploymentRequest",
     "PipelineDeploymentFilter",
     "PipelineDeploymentResponse",
     "PipelineDeploymentResponseBody",
     "PipelineDeploymentResponseMetadata",
+    "PipelineDeploymentResponseResources",
     "PipelineRunRequest",
     "PipelineRunUpdate",
     "PipelineRunFilter",
@@ -666,11 +697,13 @@ __all__ = [
     "ScheduleResponse",
     "ScheduleResponseBody",
     "ScheduleResponseMetadata",
+    "ScheduleResponseResources",
     "SecretFilter",
     "SecretRequest",
     "SecretResponse",
     "SecretResponseBody",
     "SecretResponseMetadata",
+    "SecretResponseResources",
     "SecretUpdate",
     "ServiceResponse",
     "ServiceResponseBody",
@@ -697,12 +730,14 @@ __all__ = [
     "ServiceConnectorResponse",
     "ServiceConnectorResponseBody",
     "ServiceConnectorResponseMetadata",
+    "ServiceConnectorResponseResources",
     "StackRequest",
     "StackUpdate",
     "StackFilter",
     "StackResponse",
     "StackResponseBody",
     "StackResponseMetadata",
+    "StackResponseResources",
     "StepRunRequest",
     "StepRunUpdate",
     "StepRunFilter",
@@ -717,6 +752,7 @@ __all__ = [
     "TagResponse",
     "TagResponseBody",
     "TagResponseMetadata",
+    "TagResponseResources",
     "TagRequest",
     "TagUpdate",
     "TriggerResponse",
@@ -786,4 +822,5 @@ __all__ = [
     "RunMetadataEntry",
     "RunMetadataResource",
     "ProjectStatistics",
+    "PipelineRunDAG",
 ]