Skip to content

Commit 81f3af8

Browse files
adding env_vars in get MD api
1 parent 7a5e1ce commit 81f3af8

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

ads/aqua/modeldeployment/entities.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55
from dataclasses import dataclass, field
66
from typing import Union
77

8-
from oci.data_science.models import ModelDeployment, ModelDeploymentSummary
8+
from oci.data_science.models import (
9+
ModelDeployment,
10+
ModelDeploymentSummary,
11+
)
912

1013
from ads.aqua.common.enums import Tags
1114
from ads.aqua.constants import UNKNOWN, UNKNOWN_DICT
@@ -48,6 +51,7 @@ class AquaDeployment(DataClassSerializable):
4851
lifecycle_details: str = None
4952
shape_info: field(default_factory=ShapeInfo) = None
5053
tags: dict = None
54+
environment_variables: dict = None
5155

5256
@classmethod
5357
def from_oci_model_deployment(
@@ -75,6 +79,7 @@ def from_oci_model_deployment(
7579
instance_configuration.model_deployment_instance_shape_config_details
7680
)
7781
instance_count = oci_model_deployment.model_deployment_configuration_details.model_configuration_details.scaling_policy.instance_count
82+
environment_variables = oci_model_deployment.model_deployment_configuration_details.environment_configuration_details.environment_variables
7883
shape_info = ShapeInfo(
7984
instance_shape=instance_configuration.instance_shape_name,
8085
instance_count=instance_count,
@@ -114,6 +119,7 @@ def from_oci_model_deployment(
114119
region=region,
115120
),
116121
tags=freeform_tags,
122+
environment_variables=environment_variables,
117123
)
118124

119125

0 commit comments

Comments
 (0)