Updated pr.

Author: lu-ohai

ads/aqua/modeldeployment/entities.py

@@ -50,7 +50,7 @@ class AquaDeployment(DataClassSerializable):
     private_endpoint_id: str = None
     console_link: str = None
     lifecycle_details: str = None
-    shape_info: field(default_factory=ShapeInfo) = None
+    shape_info: Optional[ShapeInfo] = None
     tags: dict = None
     environment_variables: dict = None
 
@@ -99,9 +99,7 @@ def from_oci_model_deployment(
         freeform_tags = oci_model_deployment.freeform_tags or UNKNOWN_DICT
         aqua_service_model_tag = freeform_tags.get(Tags.AQUA_SERVICE_MODEL_TAG, None)
         aqua_model_name = freeform_tags.get(Tags.AQUA_MODEL_NAME_TAG, UNKNOWN)
-        private_endpoint_id = (
-            getattr(oci_model_deployment.model_deployment_configuration_details.model_configuration_details.instance_configuration, "private_endpoint_id", UNKNOWN)
-        )
+        private_endpoint_id = getattr(instance_configuration, "private_endpoint_id", UNKNOWN)
 
         return AquaDeployment(
             id=oci_model_deployment.id,

ads/model/deployment/model_deployment.py

@@ -1652,6 +1652,7 @@ def _build_model_deployment_configuration_details(self) -> Dict:
             if not hasattr(
                 oci.data_science.models.InstanceConfiguration, "private_endpoint_id"
             ):
+                # TODO: add oci version with private endpoint support.
                 raise EnvironmentError(
                     "Private endpoint is not supported in the current OCI SDK installed."
                 )