Add tags, provisioning_state and creation_context to batch-deployment…

… list output (#29721)

Co-authored-by: Nancy Mejia Juarez <[email protected]>

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/batch/batch_deployment.py

@@ -12,6 +12,7 @@
 from azure.ai.ml._schema._deployment.batch.job_definition_schema import JobDefinitionSchema
 from azure.ai.ml._schema._deployment.deployment import DeploymentSchema
 from azure.ai.ml._schema.core.fields import ComputeField, ExperimentalField, NestedField, StringTransformedEnum
+from azure.ai.ml._schema.job.creation_context import CreationContextSchema
 from azure.ai.ml._schema.job_resource_configuration import JobResourceConfigurationSchema
 from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY
 from azure.ai.ml.constants._deployment import BatchDeploymentOutputAction, BatchDeploymentType
@@ -56,6 +57,8 @@ class BatchDeploymentSchema(DeploymentSchema):
     type = StringTransformedEnum(allowed_values=[BatchDeploymentType.COMPONENT, BatchDeploymentType.MODEL])
 
     job_definition = ExperimentalField(NestedField(JobDefinitionSchema))
+    creation_context = NestedField(CreationContextSchema, dump_only=True)
+    provisioning_state = fields.Str(dump_only=True)
 
     @post_load
     def make(self, data: Any, **kwargs: Any) -> Any:

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/batch_deployment.py

@@ -24,6 +24,7 @@
 from azure.ai.ml.entities._deployment.deployment_settings import BatchRetrySettings
 from azure.ai.ml.entities._job.resource_configuration import ResourceConfiguration
 from azure.ai.ml.entities._util import load_from_dict
+from azure.ai.ml.entities._system_data import SystemData
 from azure.ai.ml.exceptions import ErrorCategory, ErrorTarget, ValidationErrorType, ValidationException
 
 from ..._vendor.azure_resources.flatten_json import flatten, unflatten
@@ -117,6 +118,7 @@ def __init__(
 
         self.deployment_type = kwargs.pop("type", "Model")
         self.job_definition = kwargs.pop("job_definition", None)
+        self._provisioning_state = kwargs.pop("provisioning_state", None)
 
         super(BatchDeployment, self).__init__(
             name=name,
@@ -160,6 +162,15 @@ def __init__(
     def instance_count(self) -> int:
         return self.resources.instance_count if self.resources else None
 
+    @property
+    def provisioning_state(self) -> Optional[str]:
+        """Batch deployment provisioning state, readonly.
+
+        :return: Batch deployment provisioning state.
+        :rtype: Optional[str]
+        """
+        return self._provisioning_state
+
     @instance_count.setter
     def instance_count(self, value: int) -> None:
         if not self.resources:
@@ -264,6 +275,8 @@ def _from_rest_object(cls, deployment: BatchDeploymentData):  # pylint: disable=
             max_concurrency_per_instance=deployment.properties.max_concurrency_per_instance,
             endpoint_name=_parse_endpoint_name_from_deployment_id(deployment.id),
             properties=deployment.properties.properties,
+            creation_context=SystemData._from_rest_object(deployment.system_data),
+            provisioning_state=deployment.properties.provisioning_state,
         )
 
         # Job definition is in private preview. If private preview environment is