Skip to content

Commit 4917893

Browse files
authored
Add container info to MD telemetry. (#1300)
1 parent ef7d4f8 commit 4917893

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1211,7 +1211,10 @@ def _create_deployment(
12111211
progress_thread.start()
12121212

12131213
# we arbitrarily choose last 8 characters of OCID to identify MD in telemetry
1214-
telemetry_kwargs = {"ocid": get_ocid_substring(deployment_id, key_len=8)}
1214+
deployment_short_ocid = get_ocid_substring(deployment_id, key_len=8)
1215+
1216+
# Prepare telemetry kwargs
1217+
telemetry_kwargs = {"ocid": deployment_short_ocid}
12151218

12161219
if Tags.BASE_MODEL_CUSTOM in tags:
12171220
telemetry_kwargs["custom_base_model"] = True
@@ -1223,6 +1226,11 @@ def _create_deployment(
12231226
else:
12241227
telemetry_kwargs["deployment_type"] = DeploymentType.SINGLE
12251228

1229+
telemetry_kwargs["container"] = (
1230+
create_deployment_details.container_family
1231+
or create_deployment_details.container_image_uri
1232+
)
1233+
12261234
# tracks unique deployments that were created in the user compartment
12271235
self.telemetry.record_event_async(
12281236
category=f"aqua/{model_type}/deployment",
@@ -1236,6 +1244,7 @@ def _create_deployment(
12361244
action="shape",
12371245
detail=create_deployment_details.instance_shape,
12381246
value=model_name,
1247+
**{"ocid": deployment_short_ocid},
12391248
)
12401249

12411250
return AquaDeployment.from_oci_model_deployment(

0 commit comments

Comments
 (0)