Skip to content

Commit a2f0ebe

Browse files
Adding model_id in MD response
1 parent c66bdd6 commit a2f0ebe

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ads/aqua/modeldeployment/entities.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@ class AquaDeployment(DataClassSerializable):
4141
id: str = None
4242
display_name: str = None
4343
aqua_service_model: bool = None
44+
model_id: str = None
4445
aqua_model_name: str = None
4546
state: str = None
4647
description: str = None
@@ -97,7 +98,7 @@ def from_oci_model_deployment(
9798
else None
9899
),
99100
)
100-
101+
model_id = oci_model_deployment._model_deployment_configuration_details.model_configuration_details.model_id
101102
tags = {}
102103
tags.update(oci_model_deployment.freeform_tags or UNKNOWN_DICT)
103104
tags.update(oci_model_deployment.defined_tags or UNKNOWN_DICT)
@@ -110,6 +111,7 @@ def from_oci_model_deployment(
110111

111112
return AquaDeployment(
112113
id=oci_model_deployment.id,
114+
model_id=model_id,
113115
display_name=oci_model_deployment.display_name,
114116
aqua_service_model=aqua_service_model_tag is not None,
115117
aqua_model_name=aqua_model_name,

0 commit comments

Comments
 (0)