[WIP][AQUA] Group Model Deployment

ads/aqua/app.py

@@ -290,6 +290,7 @@ def create_model_catalog(
         model_taxonomy_metadata: Union[ModelTaxonomyMetadata, Dict],
         compartment_id: str,
         project_id: str,
+        freeform_tags: Dict = None,
         defined_tags: Dict = None,
         **kwargs,
     ) -> DataScienceModel:
@@ -303,6 +304,7 @@ def create_model_catalog(
             .with_custom_metadata_list(model_custom_metadata)
             .with_defined_metadata_list(model_taxonomy_metadata)
             .with_provenance_metadata(ModelProvenanceMetadata(training_id=UNKNOWN))
+            .with_freeform_tags(**(freeform_tags or {}))
             .with_defined_tags(
                 **(defined_tags or {})
             )  # Create defined tags when a model is created.

ads/aqua/common/enums.py

@@ -44,6 +44,7 @@ class Tags(ExtendedEnum):
     MODEL_FORMAT = "model_format"
     MODEL_ARTIFACT_FILE = "model_file"
     MULTIMODEL_TYPE_TAG = "aqua_multimodel"
+    AQUA_FINE_TUNE_MODEL_VERSION = "fine_tune_model_version"
 
 
 class InferenceContainerType(ExtendedEnum):

ads/aqua/constants.py

@@ -45,6 +45,7 @@
 AQUA_TROUBLESHOOTING_LINK = "https://github.com/oracle-samples/oci-data-science-ai-samples/blob/main/ai-quick-actions/troubleshooting-tips.md"
 MODEL_FILE_DESCRIPTION_VERSION = "1.0"
 MODEL_FILE_DESCRIPTION_TYPE = "modelOSSReferenceDescription"
+AQUA_FINE_TUNE_MODEL_VERSION = "v2"
 
 TRAINING_METRICS_FINAL = "training_metrics_final"
 VALIDATION_METRICS_FINAL = "validation_metrics_final"

ads/aqua/finetuning/constants.py

@@ -14,6 +14,7 @@ class FineTuneCustomMetadata(ExtendedEnum):
     SERVICE_MODEL_ARTIFACT_LOCATION = "artifact_location"
     SERVICE_MODEL_DEPLOYMENT_CONTAINER = "deployment-container"
     SERVICE_MODEL_FINE_TUNE_CONTAINER = "finetune-container"
+    FINE_TUNE_INCLUDE_BASE_MODEL_ARTIFACT = "include_base_model_artifact"
 
 
 class FineTuningRestrictedParams(ExtendedEnum):

ads/aqua/finetuning/finetuning.py

@@ -304,6 +304,9 @@ def create(
             "val_set_size": create_fine_tuning_details.validation_set_size,
             "training_data": ft_dataset_path,
         }
+        # needs to add 'fine_tune_model_version' tag when creating the ft model for the
+        # ft container to block merging base model artifact with ft model artifact.
+        ft_model_freeform_tags = {Tags.AQUA_FINE_TUNE_MODEL_VERSION: "v2"}
 
         ft_model = self.create_model_catalog(
             display_name=create_fine_tuning_details.ft_name,
@@ -314,6 +317,7 @@ def create(
             compartment_id=target_compartment,
             project_id=target_project,
             model_by_reference=True,
+            freeform_tags=ft_model_freeform_tags,
             defined_tags=create_fine_tuning_details.defined_tags,
         )
         defined_metadata_dict = {}
@@ -446,6 +450,7 @@ def create(
 
         model_freeform_tags = {
             **model_freeform_tags,
+            **(ft_model.freeform_tags or {}),
             Tags.READY_TO_FINE_TUNE: "false",
             Tags.AQUA_TAG: UNKNOWN,
             Tags.AQUA_FINE_TUNED_MODEL_TAG: f"{source.id}#{source.display_name}",