zenml-nightly 0.55.0.dev20240124__py3-none-any.whl → 0.72.0.dev20250116__py3-none-any.whl

zenml/models/__init__.py

@@ -17,11 +17,15 @@
 
 # V2 Base
 from zenml.models.v2.base.base import (
+    BaseZenModel,
     BaseRequest,
     BaseResponse,
+    BaseUpdate,
+    BaseIdentifiedResponse,
     BaseResponseBody,
     BaseResponseMetadata,
-    BaseZenModel,
+    BaseResponseResources,
+    BaseDatedResponseBody,
 )
 from zenml.models.v2.base.scoped import (
     UserScopedRequest,
@@ -34,7 +38,8 @@ from zenml.models.v2.base.scoped import (
     WorkspaceScopedResponse,
     WorkspaceScopedResponseBody,
     WorkspaceScopedResponseMetadata,
-    WorkspaceScopedTaggableFilter
+    WorkspaceScopedResponseResources,
+    WorkspaceScopedTaggableFilter,
 )
 from zenml.models.v2.base.filter import (
     BaseFilter,
@@ -46,6 +51,21 @@ from zenml.models.v2.base.filter import (
 from zenml.models.v2.base.page import Page
 
 # V2 Core
+from zenml.models.v2.core.action import (
+    ActionFilter,
+    ActionRequest,
+    ActionResponse,
+    ActionResponseBody,
+    ActionResponseMetadata,
+    ActionResponseResources,
+    ActionUpdate,
+)
+from zenml.models.v2.core.action_flavor import (
+    ActionFlavorResponse,
+    ActionFlavorResponseBody,
+    ActionFlavorResponseMetadata,
+    ActionFlavorResponseResources,
+)
 from zenml.models.v2.core.api_key import (
     APIKey,
     APIKeyRequest,
@@ -73,6 +93,7 @@ from zenml.models.v2.core.artifact_version import (
     ArtifactVersionResponseBody,
     ArtifactVersionResponseMetadata,
     ArtifactVersionUpdate,
+    LazyArtifactVersionResponse
 )
 from zenml.models.v2.core.artifact_visualization import (
     ArtifactVisualizationRequest,
@@ -80,6 +101,15 @@ from zenml.models.v2.core.artifact_visualization import (
     ArtifactVisualizationResponseBody,
     ArtifactVisualizationResponseMetadata,
 )
+from zenml.models.v2.core.service import (
+    ServiceResponse,
+    ServiceResponseBody,
+    ServiceResponseMetadata,
+    ServiceUpdate,
+    ServiceFilter,
+    ServiceRequest,
+    ServiceResponseResources,
+)
 from zenml.models.v2.core.code_reference import (
     CodeReferenceRequest,
     CodeReferenceResponse,
@@ -102,6 +132,13 @@ from zenml.models.v2.core.component import (
     ComponentResponse,
     ComponentResponseBody,
     ComponentResponseMetadata,
+    ComponentResponseResources
+)
+from zenml.models.v2.core.event_source_flavor import (
+    EventSourceFlavorResponse,
+    EventSourceFlavorResponseBody,
+    EventSourceFlavorResponseMetadata,
+    EventSourceFlavorResponseResources,
 )
 from zenml.models.v2.core.device import (
     OAuthDeviceUpdate,
@@ -142,6 +179,7 @@ from zenml.models.v2.core.model_version import (
     ModelVersionResponseMetadata,
     ModelVersionFilter,
     ModelVersionUpdate,
+    ModelVersionResponseResources,
 )
 from zenml.models.v2.core.model_version_artifact import (
     ModelVersionArtifactFilter,
@@ -162,6 +200,7 @@ from zenml.models.v2.core.pipeline import (
     PipelineResponse,
     PipelineResponseBody,
     PipelineResponseMetadata,
+    PipelineResponseResources
 )
 from zenml.models.v2.core.pipeline_build import (
     PipelineBuildBase,
@@ -178,6 +217,7 @@ from zenml.models.v2.core.pipeline_deployment import (
     PipelineDeploymentResponse,
     PipelineDeploymentResponseBody,
     PipelineDeploymentResponseMetadata,
+    PipelineDeploymentResponseResources,
 )
 from zenml.models.v2.core.pipeline_run import (
     PipelineRunRequest,
@@ -186,13 +226,20 @@ from zenml.models.v2.core.pipeline_run import (
     PipelineRunResponse,
     PipelineRunResponseBody,
     PipelineRunResponseMetadata,
-)
+    PipelineRunResponseResources
+)
+from zenml.models.v2.core.run_template import (
+    RunTemplateRequest,
+    RunTemplateUpdate,
+    RunTemplateResponse,
+    RunTemplateResponseBody,
+    RunTemplateResponseMetadata,
+    RunTemplateResponseResources,
+    RunTemplateFilter,
+)
+from zenml.models.v2.base.base_plugin_flavor import BasePluginFlavorResponse
 from zenml.models.v2.core.run_metadata import (
     RunMetadataRequest,
-    RunMetadataFilter,
-    RunMetadataResponse,
-    RunMetadataResponseBody,
-    RunMetadataResponseMetadata,
 )
 from zenml.models.v2.core.schedule import (
     ScheduleRequest,
@@ -241,6 +288,7 @@ from zenml.models.v2.core.step_run import (
     StepRunResponse,
     StepRunResponseBody,
     StepRunResponseMetadata,
+    StepRunResponseResources
 )
 from zenml.models.v2.core.tag import (
     TagFilter,
@@ -280,17 +328,40 @@ from zenml.models.v2.misc.service_connector_type import (
     ServiceConnectorTypedResourcesModel,
     ResourceTypeModel,
 )
-from zenml.models.v2.misc.server_models import ServerDatabaseType, ServerModel
+from zenml.models.v2.misc.server_models import (
+    ServerDatabaseType,
+    ServerLoadInfo,
+    ServerModel,
+)
+from zenml.models.v2.core.trigger import (
+    TriggerRequest,
+    TriggerFilter,
+    TriggerUpdate,
+    TriggerResponse,
+    TriggerResponseBody,
+    TriggerResponseMetadata,
+    TriggerResponseResources,
+)
+from zenml.models.v2.core.trigger_execution import (
+    TriggerExecutionRequest,
+    TriggerExecutionFilter,
+    TriggerExecutionResponse,
+    TriggerExecutionResponseBody,
+    TriggerExecutionResponseMetadata,
+    TriggerExecutionResponseResources,
+)
+from zenml.models.v2.core.event_source import (
+    EventSourceRequest,
+    EventSourceFilter,
+    EventSourceUpdate,
+    EventSourceResponse,
+    EventSourceResponseBody,
+    EventSourceResponseMetadata,
+    EventSourceResponseResources,
+)
 from zenml.models.v2.misc.user_auth import UserAuthModel
 from zenml.models.v2.misc.build_item import BuildItem
 from zenml.models.v2.misc.loaded_visualization import LoadedVisualization
-from zenml.models.v2.misc.hub_plugin_models import (
-    HubPluginRequestModel,
-    HubPluginResponseModel,
-    HubUserResponseModel,
-    HubPluginBaseModel,
-    PluginStatus,
-)
 from zenml.models.v2.misc.external_user import ExternalUserModel
 from zenml.models.v2.misc.auth_models import (
     OAuthDeviceAuthorizationRequest,
@@ -301,174 +372,120 @@ from zenml.models.v2.misc.auth_models import (
     OAuthRedirectResponse,
     OAuthTokenResponse,
 )
+from zenml.models.v2.misc.run_metadata import (
+    RunMetadataEntry,
+    RunMetadataResource,
+)
 from zenml.models.v2.misc.server_models import (
     ServerModel,
     ServerDatabaseType,
     ServerDeploymentType,
 )
+from zenml.models.v2.core.server_settings import (
+    ServerActivationRequest,
+    ServerSettingsResponse,
+    ServerSettingsResponseResources,
+    ServerSettingsResponseBody,
+    ServerSettingsResponseMetadata,
+    ServerSettingsUpdate,
+)
+from zenml.models.v2.misc.stack_deployment import (
+    DeployedStack,
+    StackDeploymentConfig,
+    StackDeploymentInfo,
+)
+from zenml.models.v2.misc.info_models import (
+    ComponentInfo,
+    ServiceConnectorInfo,
+    ServiceConnectorResourcesInfo,
+    ResourcesInfo,
+)
 
 # ----------------------------- Forward References -----------------------------
 
 # V2
-APIKeyResponseBody.update_forward_refs(
-    ServiceAccountResponse=ServiceAccountResponse,
-)
-ArtifactVersionRequest.update_forward_refs(
-    ArtifactVisualizationRequest=ArtifactVisualizationRequest,
-)
-ArtifactVersionResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-ArtifactVersionResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-    ArtifactVisualizationResponse=ArtifactVisualizationResponse,
-    RunMetadataResponse=RunMetadataResponse,
-)
-CodeReferenceResponseBody.update_forward_refs(
-    CodeRepositoryResponse=CodeRepositoryResponse,
-)
-CodeRepositoryResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-CodeRepositoryResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-)
-ComponentResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-ComponentResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-    ServiceConnectorResponse=ServiceConnectorResponse,
-)
-FlavorResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-FlavorResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-)
-ModelResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-    TagResponse=TagResponse,
-)
-ModelResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-)
-ModelVersionResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-    ModelResponse=ModelResponse,
-    RunMetadataResponse=RunMetadataResponse,
-)
-ModelVersionResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-    RunMetadataResponse=RunMetadataResponse,
-)
-ModelVersionArtifactResponseBody.update_forward_refs(
-    ArtifactVersionResponse=ArtifactVersionResponse,
-)
-ModelVersionPipelineRunResponseBody.update_forward_refs(
-    PipelineRunResponse=PipelineRunResponse
-)
-OAuthDeviceResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-PipelineResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-PipelineResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-)
-PipelineBuildBase.update_forward_refs(
-    BuildItem=BuildItem,
-)
-PipelineBuildResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-PipelineBuildResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-    PipelineResponse=PipelineResponse,
-    StackResponse=StackResponse,
-    BuildItem=BuildItem,
-)
-PipelineDeploymentRequest.update_forward_refs(
-    CodeReferenceRequest=CodeReferenceRequest,
-)
-PipelineDeploymentResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-PipelineDeploymentResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-    PipelineResponse=PipelineResponse,
-    StackResponse=StackResponse,
-    PipelineBuildResponse=PipelineBuildResponse,
-    ScheduleResponse=ScheduleResponse,
-    CodeReferenceResponse=CodeReferenceResponse,
-)
-PipelineRunResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-    PipelineResponse=PipelineResponse,
-    StackResponse=StackResponse,
-    PipelineBuildResponse=PipelineBuildResponse,
-    ScheduleResponse=ScheduleResponse,
-    CodeReferenceResponse=CodeReferenceResponse,
-)
-PipelineRunResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-    RunMetadataResponse=RunMetadataResponse,
-    StepRunResponse=StepRunResponse,
-)
-RunMetadataResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-RunMetadataResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-)
-ScheduleResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-ScheduleResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-)
-SecretResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-SecretResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-)
-ServiceConnectorResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-ServiceConnectorResponseMetadata.update_forward_refs(
-    ServiceConnectorTypeModel=ServiceConnectorTypeModel,
-    WorkspaceResponse=WorkspaceResponse,
-    ComponentResponse=ComponentResponse,
-)
-StackResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-)
-StackResponseMetadata.update_forward_refs(
-    ComponentResponse=ComponentResponse,
-    WorkspaceResponse=WorkspaceResponse,
-)
-StepRunRequest.update_forward_refs(
-    LogsRequest=LogsRequest,
-)
-StepRunResponseBody.update_forward_refs(
-    UserResponse=UserResponse,
-    ArtifactVersionResponse=ArtifactVersionResponse,
-)
-StepRunResponseMetadata.update_forward_refs(
-    WorkspaceResponse=WorkspaceResponse,
-    LogsResponse=LogsResponse,
-    RunMetadataResponse=RunMetadataResponse,
-)
+ActionResponseResources.model_rebuild()
+APIKeyResponseBody.model_rebuild()
+ArtifactVersionRequest.model_rebuild()
+ArtifactVersionResponseBody.model_rebuild()
+ArtifactVersionResponseMetadata.model_rebuild()
+CodeReferenceResponseBody.model_rebuild()
+CodeRepositoryResponseBody.model_rebuild()
+CodeRepositoryResponseMetadata.model_rebuild()
+ComponentResponseBody.model_rebuild()
+ComponentResponseMetadata.model_rebuild()
+ComponentResponseResources.model_rebuild()
+EventSourceResponseBody.model_rebuild()
+EventSourceResponseMetadata.model_rebuild()
+EventSourceResponseResources.model_rebuild()
+FlavorResponseBody.model_rebuild()
+FlavorResponseMetadata.model_rebuild()
+LazyArtifactVersionResponse.model_rebuild()
+ModelResponseBody.model_rebuild()
+ModelResponseMetadata.model_rebuild()
+ModelVersionResponseBody.model_rebuild()
+ModelVersionResponseMetadata.model_rebuild()
+ModelVersionResponseResources.model_rebuild()
+ModelVersionArtifactResponseBody.model_rebuild()
+ModelVersionPipelineRunResponseBody.model_rebuild()
+OAuthDeviceResponseBody.model_rebuild()
+PipelineResponseBody.model_rebuild()
+PipelineResponseMetadata.model_rebuild()
+PipelineResponseResources.model_rebuild()
+PipelineBuildBase.model_rebuild()
+PipelineBuildResponseBody.model_rebuild()
+PipelineBuildResponseMetadata.model_rebuild()
+PipelineDeploymentRequest.model_rebuild()
+PipelineDeploymentResponseBody.model_rebuild()
+PipelineDeploymentResponseMetadata.model_rebuild()
+PipelineDeploymentResponseResources.model_rebuild()
+PipelineRunResponseBody.model_rebuild()
+PipelineRunResponseMetadata.model_rebuild()
+PipelineRunResponseResources.model_rebuild()
+RunTemplateResponseBody.model_rebuild()
+RunTemplateResponseMetadata.model_rebuild()
+RunTemplateResponseResources.model_rebuild()
+RunTemplateResponseBody.model_rebuild()
+ScheduleResponseBody.model_rebuild()
+ScheduleResponseMetadata.model_rebuild()
+SecretResponseBody.model_rebuild()
+SecretResponseMetadata.model_rebuild()
+ServiceResponseBody.model_rebuild()
+ServiceResponseMetadata.model_rebuild()
+ServiceResponseResources.model_rebuild()
+ServiceConnectorResponseBody.model_rebuild()
+ServiceConnectorResponseMetadata.model_rebuild()
+StackRequest.model_rebuild()
+StackResponseBody.model_rebuild()
+StackResponseMetadata.model_rebuild()
+StackResponse.model_rebuild()
+StepRunRequest.model_rebuild()
+StepRunResponseBody.model_rebuild()
+StepRunResponseMetadata.model_rebuild()
+StepRunResponseResources.model_rebuild()
+TriggerExecutionResponseResources.model_rebuild()
+TriggerResponseBody.model_rebuild()
+TriggerResponseMetadata.model_rebuild()
+TriggerResponseResources.model_rebuild()
+TriggerResponseResources.model_rebuild()
+ComponentInfo.model_rebuild()
+ServiceConnectorInfo.model_rebuild()
+ServiceConnectorResourcesInfo.model_rebuild()
+ResourcesInfo.model_rebuild()
+
 
 __all__ = [
     # V2 Base
     "BaseRequest",
     "BaseResponse",
+    "BaseUpdate",
+    "BaseIdentifiedResponse",
     "BaseResponseBody",
     "BaseResponseMetadata",
+    "BaseResponseResources",
+    "BaseDatedResponseBody",
     "BaseZenModel",
+    "BasePluginFlavorResponse",
     "UserScopedRequest",
     "UserScopedFilter",
     "UserScopedResponse",
@@ -479,6 +496,7 @@ __all__ = [
     "WorkspaceScopedResponse",
     "WorkspaceScopedResponseBody",
     "WorkspaceScopedResponseMetadata",
+    "WorkspaceScopedResponseResources",
     "WorkspaceScopedTaggableFilter",
     "BaseFilter",
     "StrFilter",
@@ -487,6 +505,17 @@ __all__ = [
     "UUIDFilter",
     "Page",
     # V2 Core
+    "ActionFilter",
+    "ActionRequest",
+    "ActionResponse",
+    "ActionResponseBody",
+    "ActionResponseMetadata",
+    "ActionResponseResources",
+    "ActionUpdate",
+    "ActionFlavorResponse",
+    "ActionFlavorResponseBody",
+    "ActionFlavorResponseMetadata",
+    "ActionFlavorResponseResources",
     "APIKey",
     "APIKeyRequest",
     "APIKeyUpdate",
@@ -530,6 +559,11 @@ __all__ = [
     "ComponentResponse",
     "ComponentResponseBody",
     "ComponentResponseMetadata",
+    "ComponentResponseResources",
+    "EventSourceFlavorResponse",
+    "EventSourceFlavorResponseBody",
+    "EventSourceFlavorResponseMetadata",
+    "EventSourceFlavorResponseResources",
     "FlavorRequest",
     "FlavorUpdate",
     "FlavorFilter",
@@ -551,6 +585,7 @@ __all__ = [
     "ModelVersionResponse",
     "ModelVersionResponseBody",
     "ModelVersionResponseMetadata",
+    "ModelVersionResponseResources",
     "ModelVersionUpdate",
     "ModelVersionArtifactFilter",
     "ModelVersionArtifactRequest",
@@ -574,6 +609,7 @@ __all__ = [
     "PipelineResponse",
     "PipelineResponseBody",
     "PipelineResponseMetadata",
+    "PipelineResponseResources",
     "PipelineBuildBase",
     "PipelineBuildRequest",
     "PipelineBuildFilter",
@@ -592,11 +628,15 @@ __all__ = [
     "PipelineRunResponse",
     "PipelineRunResponseBody",
     "PipelineRunResponseMetadata",
+    "PipelineRunResponseResources",
+    "RunTemplateRequest",
+    "RunTemplateUpdate",
+    "RunTemplateResponse",
+    "RunTemplateResponseBody",
+    "RunTemplateResponseMetadata",
+    "RunTemplateResponseResources",
+    "RunTemplateFilter",
     "RunMetadataRequest",
-    "RunMetadataFilter",
-    "RunMetadataResponse",
-    "RunMetadataResponseBody",
-    "RunMetadataResponseMetadata",
     "ScheduleRequest",
     "ScheduleUpdate",
     "ScheduleFilter",
@@ -609,6 +649,19 @@ __all__ = [
     "SecretResponseBody",
     "SecretResponseMetadata",
     "SecretUpdate",
+    "ServiceResponse",
+    "ServiceResponseBody",
+    "ServiceResponseMetadata",
+    "ServiceUpdate",
+    "ServiceFilter",
+    "ServiceRequest",
+    "ServiceResponseResources",
+    "ServerActivationRequest",
+    "ServerSettingsResponse",
+    "ServerSettingsResponseResources",
+    "ServerSettingsResponseBody",
+    "ServerSettingsResponseMetadata",
+    "ServerSettingsUpdate",
     "ServiceAccountFilter",
     "ServiceAccountResponseBody",
     "ServiceAccountResponseMetadata",
@@ -633,6 +686,7 @@ __all__ = [
     "StepRunResponse",
     "StepRunResponseBody",
     "StepRunResponseMetadata",
+    "StepRunResponseResources",
     "TagFilter",
     "TagResourceResponse",
     "TagResourceResponseBody",
@@ -641,6 +695,26 @@ __all__ = [
     "TagResponseBody",
     "TagRequest",
     "TagUpdate",
+    "TriggerResponse",
+    "TriggerRequest",
+    "TriggerFilter",
+    "TriggerUpdate",
+    "TriggerResponseBody",
+    "TriggerResponseMetadata",
+    "TriggerResponseResources",
+    "TriggerExecutionRequest",
+    "TriggerExecutionFilter",
+    "TriggerExecutionResponse",
+    "TriggerExecutionResponseBody",
+    "TriggerExecutionResponseMetadata",
+    "TriggerExecutionResponseResources",
+    "EventSourceResponse",
+    "EventSourceRequest",
+    "EventSourceFilter",
+    "EventSourceUpdate",
+    "EventSourceResponseBody",
+    "EventSourceResponseMetadata",
+    "EventSourceResponseResources",
     "UserRequest",
     "UserUpdate",
     "UserFilter",
@@ -655,6 +729,7 @@ __all__ = [
     "WorkspaceResponseMetadata",
     # V2 Misc
     "AuthenticationMethodModel",
+    "DeployedStack",
     "ServiceConnectorResourcesModel",
     "ServiceConnectorTypeModel",
     "ServiceConnectorTypedResourcesModel",
@@ -664,14 +739,12 @@ __all__ = [
     "ExternalUserModel",
     "BuildItem",
     "LoadedVisualization",
-    "HubPluginRequestModel",
-    "HubPluginResponseModel",
-    "HubUserResponseModel",
-    "HubPluginBaseModel",
-    "PluginStatus",
+    "ServerLoadInfo",
     "ServerModel",
     "ServerDatabaseType",
     "ServerDeploymentType",
+    "StackDeploymentConfig",
+    "StackDeploymentInfo",
     "OAuthDeviceAuthorizationRequest",
     "OAuthDeviceAuthorizationResponse",
     "OAuthDeviceTokenRequest",
@@ -679,4 +752,10 @@ __all__ = [
     "OAuthDeviceVerificationRequest",
     "OAuthRedirectResponse",
     "OAuthTokenResponse",
+    "ComponentInfo",
+    "ServiceConnectorInfo",
+    "ServiceConnectorResourcesInfo",
+    "ResourcesInfo",
+    "RunMetadataEntry",
+    "RunMetadataResource",
 ]