[Model Watch Tower] Adding ModelVersion and ModelVersionLink entities

.gitignore

@@ -195,3 +195,5 @@ zenml_tutorial/
 
 # script for testing
 mlstacks_reset.sh
+
+.local/

src/zenml/constants.py

@@ -222,6 +222,7 @@ def handle_int_env_var(var: str, default: int = 0) -> int:
 SERVICE_CONNECTOR_RESOURCES = "/resources"
 SERVICE_CONNECTOR_CLIENT = "/client"
 MODELS = "/models"
+MODEL_VERSIONS = "/model_versions"
 
 # mandatory stack component attributes
 MANDATORY_COMPONENT_ATTRIBUTES = ["name", "uuid"]

src/zenml/model/__init__.py

@@ -0,0 +1,23 @@
+#  Copyright (c) ZenML GmbH 2023. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at:
+#
+#       https://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+#  or implied. See the License for the specific language governing
+#  permissions and limitations under the License.
+
+"""Initialization of ZenML model.
+ZenML model support Model WatchTower feature.
+"""
+
+from zenml.model.model_stages import ModelStages
+
+__all__ = [
+    "ModelStages",
+]

src/zenml/model/model_stages.py

@@ -0,0 +1,28 @@
+#  Copyright (c) ZenML GmbH 2023. All Rights Reserved.
+#
+#  Licensed under the Apache License, Version 2.0 (the "License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at:
+#
+#       https://www.apache.org/licenses/LICENSE-2.0
+#
+#  Unless required by applicable law or agreed to in writing, software
+#  distributed under the License is distributed on an "AS IS" BASIS,
+#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
+#  or implied. See the License for the specific language governing
+#  permissions and limitations under the License.
+"""ModelStages lists supported stages of a Model Version."""
+
+from zenml.utils.enum_utils import StrEnum
+
+
+class ModelStages(StrEnum):
+    """All possible stages of a Model Version."""
+
+    NONE = "none"
+    STAGING = "staging"
+    PRODUCTION = "production"
+    ARCHIVED = "archived"
+    # technical stages
+    LATEST = "latest"
+    RUNNING = "running"

src/zenml/models/__init__.py

@@ -151,12 +151,19 @@
     ModelResponseModel,
     ModelRequestModel,
     ModelUpdateModel,
-    ModelConfigBaseModel,
-    ModelConfigResponseModel,
-    ModelConfigRequestModel,
     ModelVersionBaseModel,
     ModelVersionResponseModel,
     ModelVersionRequestModel,
+    ModelVersionArtifactBaseModel,
+    ModelVersionArtifactFilterModel,
+    ModelVersionArtifactRequestModel,
+    ModelVersionArtifactResponseModel,
+    ModelVersionPipelineRunBaseModel,
+    ModelVersionPipelineRunFilterModel,
+    ModelVersionPipelineRunRequestModel,
+    ModelVersionPipelineRunResponseModel,
+    ModelVersionFilterModel,
+    ModelVersionUpdateModel,
 )
 
 ComponentResponseModel.update_forward_refs(
@@ -277,22 +284,29 @@
     WorkspaceResponseModel=WorkspaceResponseModel,
 )
 
-ModelConfigRequestModel.update_forward_refs(
+ModelVersionRequestModel.update_forward_refs(
     UserResponseModel=UserResponseModel,
     WorkspaceResponseModel=WorkspaceResponseModel,
 )
 
-ModelConfigResponseModel.update_forward_refs(
+ModelVersionResponseModel.update_forward_refs(
     UserResponseModel=UserResponseModel,
     WorkspaceResponseModel=WorkspaceResponseModel,
 )
 
-ModelVersionRequestModel.update_forward_refs(
+ModelVersionArtifactRequestModel.update_forward_refs(
     UserResponseModel=UserResponseModel,
     WorkspaceResponseModel=WorkspaceResponseModel,
 )
-
-ModelVersionResponseModel.update_forward_refs(
+ModelVersionArtifactResponseModel.update_forward_refs(
+    UserResponseModel=UserResponseModel,
+    WorkspaceResponseModel=WorkspaceResponseModel,
+)
+ModelVersionPipelineRunRequestModel.update_forward_refs(
+    UserResponseModel=UserResponseModel,
+    WorkspaceResponseModel=WorkspaceResponseModel,
+)
+ModelVersionPipelineRunResponseModel.update_forward_refs(
     UserResponseModel=UserResponseModel,
     WorkspaceResponseModel=WorkspaceResponseModel,
 )
@@ -398,6 +412,16 @@
     "ModelConfigRequestModel",
     "ModelConfigResponseModel",
     "ModelVersionBaseModel",
+    "ModelVersionFilterModel",
     "ModelVersionRequestModel",
     "ModelVersionResponseModel",
+    "ModelVersionUpdateModel",
+    "ModelVersionArtifactBaseModel",
+    "ModelVersionArtifactFilterModel",
+    "ModelVersionArtifactRequestModel",
+    "ModelVersionArtifactResponseModel",
+    "ModelVersionPipelineRunBaseModel",
+    "ModelVersionPipelineRunFilterModel",
+    "ModelVersionPipelineRunRequestModel",
+    "ModelVersionPipelineRunResponseModel",
 ]