diff --git a/ads/aqua/modeldeployment/constants.py b/ads/aqua/modeldeployment/constants.py index 1853c04b7..f6ae5a029 100644 --- a/ads/aqua/modeldeployment/constants.py +++ b/ads/aqua/modeldeployment/constants.py @@ -16,5 +16,6 @@ class DeploymentType(ExtendedEnum): + SINGLE = "SINGLE" STACKED = "STACKED" MULTI = "MULTI" diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index 65c4ff6dd..214d98d1f 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -1214,6 +1214,13 @@ def _create_deployment( if Tags.BASE_MODEL_CUSTOM in tags: telemetry_kwargs["custom_base_model"] = True + if Tags.MULTIMODEL_TYPE_TAG in tags: + telemetry_kwargs["deployment_type"] = DeploymentType.MULTI + elif Tags.STACKED_MODEL_TYPE_TAG in tags: + telemetry_kwargs["deployment_type"] = DeploymentType.STACKED + else: + telemetry_kwargs["deployment_type"] = DeploymentType.SINGLE + # tracks unique deployments that were created in the user compartment self.telemetry.record_event_async( category=f"aqua/{model_type}/deployment",