From aa9c5acc537c1ed22244faf14d9481e9f38bfa14 Mon Sep 17 00:00:00 2001
From: Dmitrii Cherkasov <dmitrii.cherkasov@oracle.com>
Date: Thu, 13 Nov 2025 18:26:40 -0800
Subject: [PATCH] Add container info to MD telemetry.

---
 ads/aqua/modeldeployment/deployment.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py
index e5d4d1fb5..6f4ac2070 100644
--- a/ads/aqua/modeldeployment/deployment.py
+++ b/ads/aqua/modeldeployment/deployment.py
@@ -1211,7 +1211,10 @@ def _create_deployment(
         progress_thread.start()
 
         # we arbitrarily choose last 8 characters of OCID to identify MD in telemetry
-        telemetry_kwargs = {"ocid": get_ocid_substring(deployment_id, key_len=8)}
+        deployment_short_ocid = get_ocid_substring(deployment_id, key_len=8)
+        
+        # Prepare telemetry kwargs
+        telemetry_kwargs = {"ocid": deployment_short_ocid}
 
         if Tags.BASE_MODEL_CUSTOM in tags:
             telemetry_kwargs["custom_base_model"] = True
@@ -1223,6 +1226,11 @@ def _create_deployment(
         else:
             telemetry_kwargs["deployment_type"] = DeploymentType.SINGLE
 
+        telemetry_kwargs["container"] = (
+            create_deployment_details.container_family
+            or create_deployment_details.container_image_uri
+        )
+
         # tracks unique deployments that were created in the user compartment
         self.telemetry.record_event_async(
             category=f"aqua/{model_type}/deployment",
@@ -1236,6 +1244,7 @@ def _create_deployment(
             action="shape",
             detail=create_deployment_details.instance_shape,
             value=model_name,
+            **{"ocid": deployment_short_ocid},
         )
 
         return AquaDeployment.from_oci_model_deployment(