From de60899e49399ab094bc0f7ffd7105d4a1331c18 Mon Sep 17 00:00:00 2001 From: Lu Peng Date: Thu, 30 Oct 2025 19:44:10 -0400 Subject: [PATCH] Added deployment type in telemetry. --- ads/aqua/modeldeployment/constants.py | 1 + ads/aqua/modeldeployment/deployment.py | 7 +++++++ 2 files changed, 8 insertions(+) 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",