zenml-nightly 0.82.1.dev20250526__py3-none-any.whl → 0.82.1.dev20250528__py3-none-any.whl

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",
 ]

zenml/models/v2/base/scoped.py

@@ -106,9 +106,7 @@ class ProjectScopedRequest(UserScopedRequest):
 class UserScopedResponseBody(BaseDatedResponseBody):
     """Base user-owned body."""
 
-    user: Optional["UserResponse"] = Field(
-        title="The user who created this resource.", default=None
-    )
+    user_id: Optional[UUID] = Field(title="The user id.", default=None)
 
 
 class UserScopedResponseMetadata(BaseResponseMetadata):
@@ -118,6 +116,10 @@ class UserScopedResponseMetadata(BaseResponseMetadata):
 class UserScopedResponseResources(BaseResponseResources):
     """Base class for all resource models associated with the user."""
 
+    user: Optional["UserResponse"] = Field(
+        title="The user who created this resource.", default=None
+    )
+
 
 UserBody = TypeVar("UserBody", bound=UserScopedResponseBody)
 UserMetadata = TypeVar("UserMetadata", bound=UserScopedResponseMetadata)
@@ -141,19 +143,28 @@ class UserScopedResponse(
             The analytics metadata.
         """
         metadata = super().get_analytics_metadata()
-        if self.user is not None:
-            metadata["user_id"] = self.user.id
+        if user_id := self.user_id:
+            metadata["user_id"] = user_id
         return metadata
 
     # Body and metadata properties
+    @property
+    def user_id(self) -> Optional[UUID]:
+        """The user ID property.
+
+        Returns:
+            the value of the property.
+        """
+        return self.get_body().user_id
+
     @property
     def user(self) -> Optional["UserResponse"]:
-        """The `user` property.
+        """The user property.
 
         Returns:
             the value of the property.
         """
-        return self.get_body().user
+        return self.get_resources().user
 
 
 class UserScopedFilter(BaseFilter):
@@ -289,12 +300,12 @@ class UserScopedFilter(BaseFilter):
 class ProjectScopedResponseBody(UserScopedResponseBody):
     """Base project-scoped body."""
 
+    project_id: UUID = Field(title="The project id.")
+
 
 class ProjectScopedResponseMetadata(UserScopedResponseMetadata):
     """Base project-scoped metadata."""
 
-    project: "ProjectResponse" = Field(title="The project of this resource.")
-
 
 class ProjectScopedResponseResources(UserScopedResponseResources):
     """Base project-scoped resources."""
@@ -318,8 +329,6 @@ class ProjectScopedResponse(
     Used as a base class for all domain models that are project-scoped.
     """
 
-    project_id: UUID = Field(title="The project id.")
-
     # Analytics
     def get_analytics_metadata(self) -> Dict[str, Any]:
         """Fetches the analytics metadata for project scoped models.
@@ -328,11 +337,19 @@ class ProjectScopedResponse(
             The analytics metadata.
         """
         metadata = super().get_analytics_metadata()
-        if self.project is not None:
-            metadata["project_id"] = self.project.id
+        metadata["project_id"] = self.project_id
         return metadata
 
-    # Body and metadata properties
+    @property
+    def project_id(self) -> UUID:
+        """The project ID property.
+
+        Returns:
+            the value of the property.
+        """
+        return self.get_body().project_id
+
+    # Helper
     @property
     def project(self) -> "ProjectResponse":
         """The project property.
@@ -340,7 +357,9 @@ class ProjectScopedResponse(
         Returns:
             the value of the property.
         """
-        return self.get_metadata().project
+        from zenml.client import Client
+
+        return Client().get_project(self.project_id)
 
 
 # ---------------------- Filter Models ----------------------

zenml/models/v2/core/artifact.py

@@ -87,12 +87,6 @@ class ArtifactUpdate(BaseUpdate):
 class ArtifactResponseBody(ProjectScopedResponseBody):
     """Response body for artifacts."""
 
-    tags: List[TagResponse] = Field(
-        title="Tags associated with the model",
-    )
-    latest_version_name: Optional[str] = None
-    latest_version_id: Optional[UUID] = None
-
 
 class ArtifactResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for artifacts."""
@@ -106,6 +100,13 @@ class ArtifactResponseMetadata(ProjectScopedResponseMetadata):
 class ArtifactResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the Artifact Entity."""
 
+    tags: List[TagResponse] = Field(
+        title="Tags associated with the artifact.",
+    )
+    # TODO: maybe move these back to body or figure out a better solution
+    latest_version_name: Optional[str] = None
+    latest_version_id: Optional[UUID] = None
+
 
 class ArtifactResponse(
     ProjectScopedResponse[
@@ -139,7 +140,7 @@ class ArtifactResponse(
         Returns:
             the value of the property.
         """
-        return self.get_body().tags
+        return self.get_resources().tags
 
     @property
     def latest_version_name(self) -> Optional[str]:
@@ -148,7 +149,7 @@ class ArtifactResponse(
         Returns:
             the value of the property.
         """
-        return self.get_body().latest_version_name
+        return self.get_resources().latest_version_name
 
     @property
     def latest_version_id(self) -> Optional[UUID]:
@@ -157,7 +158,7 @@ class ArtifactResponse(
         Returns:
             the value of the property.
         """
-        return self.get_body().latest_version_id
+        return self.get_resources().latest_version_id
 
     @property
     def has_custom_name(self) -> bool:

zenml/models/v2/core/artifact_version.py

@@ -194,13 +194,6 @@ class ArtifactVersionResponseBody(ProjectScopedResponseBody):
     data_type: SourceWithValidator = Field(
         title="Data type of the artifact.",
     )
-    tags: List[TagResponse] = Field(
-        title="Tags associated with the model",
-    )
-    producer_pipeline_run_id: Optional[UUID] = Field(
-        title="The ID of the pipeline run that generated this artifact version.",
-        default=None,
-    )
     save_type: ArtifactSaveType = Field(
         title="The save type of the artifact version.",
     )
@@ -234,10 +227,6 @@ class ArtifactVersionResponseBody(ProjectScopedResponseBody):
 class ArtifactVersionResponseMetadata(ProjectScopedResponseMetadata):
     """Response metadata for artifact versions."""
 
-    producer_step_run_id: Optional[UUID] = Field(
-        title="ID of the step run that produced this artifact.",
-        default=None,
-    )
     visualizations: Optional[List["ArtifactVisualizationResponse"]] = Field(
         default=None, title="Visualizations of the artifact."
     )
@@ -249,6 +238,18 @@ class ArtifactVersionResponseMetadata(ProjectScopedResponseMetadata):
 class ArtifactVersionResponseResources(ProjectScopedResponseResources):
     """Class for all resource models associated with the artifact version entity."""
 
+    tags: List[TagResponse] = Field(
+        title="Tags associated with the artifact version.",
+    )
+    producer_step_run_id: Optional[UUID] = Field(
+        title="ID of the step run that produced this artifact.",
+        default=None,
+    )
+    producer_pipeline_run_id: Optional[UUID] = Field(
+        title="The ID of the pipeline run that generated this artifact version.",
+        default=None,
+    )
+
 
 class ArtifactVersionResponse(
     ProjectScopedResponse[
@@ -280,7 +281,7 @@ class ArtifactVersionResponse(
         return self.get_body().artifact
 
     @property
-    def version(self) -> Union[str, int]:
+    def version(self) -> str:
         """The `version` property.
 
         Returns:
@@ -313,7 +314,7 @@ class ArtifactVersionResponse(
         Returns:
             the value of the property.
         """
-        return self.get_body().tags
+        return self.get_resources().tags
 
     @property
     def producer_pipeline_run_id(self) -> Optional[UUID]:
@@ -322,7 +323,7 @@ class ArtifactVersionResponse(
         Returns:
             the value of the property.
         """
-        return self.get_body().producer_pipeline_run_id
+        return self.get_resources().producer_pipeline_run_id
 
     @property
     def save_type(self) -> ArtifactSaveType:
@@ -349,7 +350,7 @@ class ArtifactVersionResponse(
         Returns:
             the value of the property.
         """
-        return self.get_metadata().producer_step_run_id
+        return self.get_resources().producer_step_run_id
 
     @property
     def visualizations(
@@ -710,7 +711,6 @@ class LazyArtifactVersionResponse(ArtifactVersionResponse):
     """
 
     id: Optional[UUID] = None  # type: ignore[assignment]
-    project_id: Optional[UUID] = None  # type: ignore[assignment]
     lazy_load_name: Optional[str] = None
     lazy_load_version: Optional[str] = None
     lazy_load_model_name: str

zenml/models/v2/core/logs.py

@@ -13,12 +13,12 @@
 #  permissions and limitations under the License.
 """Models representing logs."""
 
-from typing import Any, Optional, Union
+from typing import Any, Optional
 from uuid import UUID
 
 from pydantic import Field, field_validator
 
-from zenml.constants import STR_FIELD_MAX_LENGTH, TEXT_FIELD_MAX_LENGTH
+from zenml.constants import TEXT_FIELD_MAX_LENGTH
 from zenml.models.v2.base.base import (
     BaseDatedResponseBody,
     BaseIdentifiedResponse,
@@ -60,27 +60,6 @@ class LogsRequest(BaseRequest):
         )
         return value
 
-    @field_validator("artifact_store_id")
-    @classmethod
-    def str_field_max_length_check(cls, value: Any) -> Any:
-        """Checks if the length of the value exceeds the maximum text length.
-
-        Args:
-            value: the value set in the field
-
-        Returns:
-            the value itself.
-
-        Raises:
-            AssertionError: if the length of the field is longer than the
-                maximum threshold.
-        """
-        assert len(str(value)) < STR_FIELD_MAX_LENGTH, (
-            "The length of the value for this field can not "
-            f"exceed {STR_FIELD_MAX_LENGTH}"
-        )
-        return value
-
 
 # ------------------ Update Model ------------------
 
@@ -101,41 +80,20 @@ class LogsResponseBody(BaseDatedResponseBody):
 class LogsResponseMetadata(BaseResponseMetadata):
     """Response metadata for logs."""
 
-    step_run_id: Optional[Union[str, UUID]] = Field(
+    step_run_id: Optional[UUID] = Field(
         title="Step ID to associate the logs with.",
         default=None,
         description="When this is set, pipeline_run_id should be set to None.",
-        union_mode="left_to_right",
     )
-    pipeline_run_id: Optional[Union[str, UUID]] = Field(
+    pipeline_run_id: Optional[UUID] = Field(
         title="Pipeline run ID to associate the logs with.",
         default=None,
         description="When this is set, step_run_id should be set to None.",
-        union_mode="left_to_right",
     )
-    artifact_store_id: Union[str, UUID] = Field(
+    artifact_store_id: UUID = Field(
         title="The artifact store ID to associate the logs with.",
-        union_mode="left_to_right",
     )
 
-    @field_validator("artifact_store_id")
-    @classmethod
-    def str_field_max_length_check(cls, value: Any) -> Any:
-        """Checks if the length of the value exceeds the maximum text length.
-
-        Args:
-            value: the value set in the field
-
-        Returns:
-            the value itself.
-
-        Raises:
-            AssertionError: if the length of the field is longer than the
-                maximum threshold.
-        """
-        assert len(str(value)) < STR_FIELD_MAX_LENGTH
-        return value
-
 
 class LogsResponseResources(BaseResponseResources):
     """Class for all resource models associated with the Logs entity."""
@@ -169,7 +127,7 @@ class LogsResponse(
         return self.get_body().uri
 
     @property
-    def step_run_id(self) -> Optional[Union[str, UUID]]:
+    def step_run_id(self) -> Optional[UUID]:
         """The `step_run_id` property.
 
         Returns:
@@ -178,7 +136,7 @@ class LogsResponse(
         return self.get_metadata().step_run_id
 
     @property
-    def pipeline_run_id(self) -> Optional[Union[str, UUID]]:
+    def pipeline_run_id(self) -> Optional[UUID]:
         """The `pipeline_run_id` property.
 
         Returns:
@@ -187,7 +145,7 @@ class LogsResponse(
         return self.get_metadata().pipeline_run_id
 
     @property
-    def artifact_store_id(self) -> Union[str, UUID]:
+    def artifact_store_id(self) -> UUID:
         """The `artifact_store_id` property.
 
         Returns: