Support enable sso for compute instance in python sdk v2 (#36358)

* re-add workspace default app insights, fix project ARM template (#36325) (#36336)

* re-add workspace default app insights, fix project ARM template

* move cl lines to right place

* cl wording

* add method to enable sso

* Update CHANGELOG.md

* Update CHANGELOG.md

* Update CHANGELOG.md

---------

Co-authored-by: MilesHolland <108901744+MilesHolland@users.noreply.github.com>

Author: tanshaocong

sdk/ml/azure-ai-ml/CHANGELOG.md

@@ -10,6 +10,7 @@
 ### Features Added
 
 - Expose `public_ip_address` in `AmlComputeNodeInfo`, to get the public ip address with the ssh port when calling `ml_client.compute.list_nodes`
+- Support `update_sso_settings` in `ComputeOperations`, to enable or disable single sign-on settings of a compute instance.
 
 ### Bugs Fixed
 - InputTypes exported in constants module

sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_04_01_preview/aio/operations/_compute_operations.py

@@ -21,7 +21,7 @@
 
 from ... import models as _models
 from ..._vendor import _convert_request
-from ...operations._compute_operations import build_create_or_update_request_initial, build_delete_request_initial, build_get_allowed_resize_sizes_request, build_get_request, build_list_keys_request, build_list_nodes_request, build_list_request, build_resize_request_initial, build_restart_request_initial, build_start_request_initial, build_stop_request_initial, build_update_custom_services_request, build_update_data_mounts_request, build_update_idle_shutdown_setting_request, build_update_request_initial
+from ...operations._compute_operations import build_create_or_update_request_initial, build_delete_request_initial, build_get_allowed_resize_sizes_request, build_get_request, build_list_keys_request, build_list_nodes_request, build_list_request, build_resize_request_initial, build_restart_request_initial, build_start_request_initial, build_stop_request_initial, build_update_custom_services_request, build_update_data_mounts_request, build_update_idle_shutdown_setting_request, build_update_sso_settings_request, build_update_request_initial
 T = TypeVar('T')
 ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
 
@@ -1276,6 +1276,72 @@ async def update_idle_shutdown_setting(  # pylint: disable=inconsistent-return-s
     update_idle_shutdown_setting.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/updateIdleShutdownSetting"}  # type: ignore
 
 
+    @distributed_trace_async
+    async def update_sso_settings(  # pylint: disable=inconsistent-return-statements
+        self,
+        resource_group_name: str,
+        workspace_name: str,
+        compute_name: str,
+        parameters: "_models.SsoSetting",
+        **kwargs: Any
+    ) -> None:
+        """Update single sign-on settings of the compute instance.
+
+        :param resource_group_name: The name of the resource group. The name is case insensitive.
+        :type resource_group_name: str
+        :param workspace_name: Name of Azure Machine Learning workspace.
+        :type workspace_name: str
+        :param compute_name: Name of the Azure Machine Learning compute.
+        :type compute_name: str
+        :param parameters: The object for updating sso setting of specified ComputeInstance.
+        :type parameters: ~azure.mgmt.machinelearningservices.models.SsoSetting
+        :keyword callable cls: A custom type or function that will be passed the direct response
+        :return: None, or the result of cls(response)
+        :rtype: None
+        :raises: ~azure.core.exceptions.HttpResponseError
+        """
+        cls = kwargs.pop('cls', None)  # type: ClsType[None]
+        error_map = {
+            401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
+        }
+        error_map.update(kwargs.pop('error_map', {}))
+
+        api_version = kwargs.pop('api_version', "2024-04-01-preview")  # type: str
+        content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
+
+        _json = self._serialize.body(parameters, 'SsoSetting')
+
+        request = build_update_sso_settings_request(
+            subscription_id=self._config.subscription_id,
+            resource_group_name=resource_group_name,
+            workspace_name=workspace_name,
+            compute_name=compute_name,
+            api_version=api_version,
+            content_type=content_type,
+            json=_json,
+            template_url=self.update_sso_settings.metadata['url'],
+        )
+        request = _convert_request(request)
+        request.url = self._client.format_url(request.url)
+
+        pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
+            request,
+            stream=False,
+            **kwargs
+        )
+        response = pipeline_response.http_response
+
+        if response.status_code not in [200]:
+            map_error(status_code=response.status_code, response=response, error_map=error_map)
+            error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response)
+            raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
+
+        if cls:
+            return cls(pipeline_response, None, {})
+
+    update_sso_settings.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/enableSso"}  # type: ignore
+
+
     @distributed_trace_async
     async def get_allowed_resize_sizes(
         self,

sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_04_01_preview/models/__init__.py

@@ -594,6 +594,7 @@
     from ._models_py3 import SpeechEndpointDeploymentResourceProperties
     from ._models_py3 import SpeechEndpointResourceProperties
     from ._models_py3 import SslConfiguration
+    from ._models_py3 import SsoSetting
     from ._models_py3 import StackEnsembleSettings
     from ._models_py3 import StaticInputData
     from ._models_py3 import StatusMessage
@@ -1265,6 +1266,7 @@
     from ._models import SpeechEndpointDeploymentResourceProperties  # type: ignore
     from ._models import SpeechEndpointResourceProperties  # type: ignore
     from ._models import SslConfiguration  # type: ignore
+    from ._models import SsoSetting  # type: ignore
     from ._models import StackEnsembleSettings  # type: ignore
     from ._models import StaticInputData  # type: ignore
     from ._models import StatusMessage  # type: ignore
@@ -2151,6 +2153,7 @@
     'SpeechEndpointDeploymentResourceProperties',
     'SpeechEndpointResourceProperties',
     'SslConfiguration',
+    'SsoSetting',
     'StackEnsembleSettings',
     'StaticInputData',
     'StatusMessage',

sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_04_01_preview/models/_models.py

@@ -16369,6 +16369,29 @@ def __init__(
         self.idle_time_before_shutdown = kwargs.get('idle_time_before_shutdown', None)
 
 
+class SsoSetting(msrest.serialization.Model):
+    """Single sign-on settings of the compute instance.
+
+    :ivar enable_sso: The value of sso settings.
+    :vartype enable_sso: bool
+    """
+
+    _attribute_map = {
+        'enable_sso': {'key': 'enableSSO', 'type': 'bool'},
+    }
+
+    def __init__(
+        self,
+        **kwargs
+    ):
+        """
+        :keyword enable_sso: The value of sso settings.
+        :paramtype enable_sso: bool
+        """
+        super(SsoSetting, self).__init__(**kwargs)
+        self.enable_sso = kwargs.get('enable_sso', True)
+
+
 class Image(msrest.serialization.Model):
     """Image.
 

sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_04_01_preview/models/_models_py3.py

@@ -17662,6 +17662,31 @@ def __init__(
         self.idle_time_before_shutdown = idle_time_before_shutdown
 
 
+class SsoSetting(msrest.serialization.Model):
+    """Single sign-on settings of the compute instance.
+
+    :ivar enable_sso: The value of sso settings.
+    :vartype enable_sso: bool
+    """
+
+    _attribute_map = {
+        'enable_sso': {'key': 'enableSSO', 'type': 'bool'},
+    }
+
+    def __init__(
+        self,
+        *,
+        enable_sso: Optional[bool] = True,
+        **kwargs
+    ):
+        """
+        :keyword enable_sso: The value of sso settings.
+        :paramtype enable_sso: bool
+        """
+        super(SsoSetting, self).__init__(**kwargs)
+        self.enable_sso = enable_sso
+
+
 class Image(msrest.serialization.Model):
     """Image.
 

sdk/ml/azure-ai-ml/azure/ai/ml/_restclient/v2024_04_01_preview/operations/_compute_operations.py

@@ -558,6 +558,48 @@ def build_update_idle_shutdown_setting_request(
     )
 
 
+def build_update_sso_settings_request(
+    subscription_id,  # type: str
+    resource_group_name,  # type: str
+    workspace_name,  # type: str
+    compute_name,  # type: str
+    **kwargs  # type: Any
+):
+    # type: (...) -> HttpRequest
+    api_version = kwargs.pop('api_version', "2024-04-01-preview")  # type: str
+    content_type = kwargs.pop('content_type', None)  # type: Optional[str]
+
+    accept = "application/json"
+    # Construct URL
+    _url = kwargs.pop("template_url", "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/enableSso")  # pylint: disable=line-too-long
+    path_format_arguments = {
+        "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str', min_length=1),
+        "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str', max_length=90, min_length=1),
+        "workspaceName": _SERIALIZER.url("workspace_name", workspace_name, 'str', pattern=r'^[a-zA-Z0-9][a-zA-Z0-9_-]{2,32}$'),
+        "computeName": _SERIALIZER.url("compute_name", compute_name, 'str', pattern=r'^[a-zA-Z](?![a-zA-Z0-9-]*-\d+$)[a-zA-Z0-9\-]{2,23}$'),
+    }
+
+    _url = _format_url_section(_url, **path_format_arguments)
+
+    # Construct parameters
+    _query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
+    _query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
+
+    # Construct headers
+    _header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
+    if content_type is not None:
+        _header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
+    _header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
+
+    return HttpRequest(
+        method="POST",
+        url=_url,
+        params=_query_parameters,
+        headers=_header_parameters,
+        **kwargs
+    )
+
+
 def build_get_allowed_resize_sizes_request(
     subscription_id,  # type: str
     resource_group_name,  # type: str
@@ -1909,6 +1951,73 @@ def update_idle_shutdown_setting(  # pylint: disable=inconsistent-return-stateme
     update_idle_shutdown_setting.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/updateIdleShutdownSetting"}  # type: ignore
 
 
+    @distributed_trace
+    def update_sso_settings(  # pylint: disable=inconsistent-return-statements
+        self,
+        resource_group_name,  # type: str
+        workspace_name,  # type: str
+        compute_name,  # type: str
+        parameters,  # type: "_models.SsoSetting"
+        **kwargs  # type: Any
+    ):
+        # type: (...) -> None
+        """Update single sign-on settings of the compute instance.
+
+        :param resource_group_name: The name of the resource group. The name is case insensitive.
+        :type resource_group_name: str
+        :param workspace_name: Name of Azure Machine Learning workspace.
+        :type workspace_name: str
+        :param compute_name: Name of the Azure Machine Learning compute.
+        :type compute_name: str
+        :param parameters: The object for updating sso setting of specified ComputeInstance.
+        :type parameters: ~azure.mgmt.machinelearningservices.models.SsoSetting
+        :keyword callable cls: A custom type or function that will be passed the direct response
+        :return: None, or the result of cls(response)
+        :rtype: None
+        :raises: ~azure.core.exceptions.HttpResponseError
+        """
+        cls = kwargs.pop('cls', None)  # type: ClsType[None]
+        error_map = {
+            401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
+        }
+        error_map.update(kwargs.pop('error_map', {}))
+
+        api_version = kwargs.pop('api_version', "2024-01-01-preview")  # type: str
+        content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
+
+        _json = self._serialize.body(parameters, 'SsoSetting')
+
+        request = build_update_sso_settings_request(
+            subscription_id=self._config.subscription_id,
+            resource_group_name=resource_group_name,
+            workspace_name=workspace_name,
+            compute_name=compute_name,
+            api_version=api_version,
+            content_type=content_type,
+            json=_json,
+            template_url=self.update_sso_settings.metadata['url'],
+        )
+        request = _convert_request(request)
+        request.url = self._client.format_url(request.url)
+
+        pipeline_response = self._client._pipeline.run(  # pylint: disable=protected-access
+            request,
+            stream=False,
+            **kwargs
+        )
+        response = pipeline_response.http_response
+
+        if response.status_code not in [200]:
+            map_error(status_code=response.status_code, response=response, error_map=error_map)
+            error = self._deserialize.failsafe_deserialize(_models.ErrorResponse, pipeline_response)
+            raise HttpResponseError(response=response, model=error, error_format=ARMErrorFormat)
+
+        if cls:
+            return cls(pipeline_response, None, {})
+
+    update_sso_settings.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.MachineLearningServices/workspaces/{workspaceName}/computes/{computeName}/enableSso"}  # type: ignore
+
+
     @distributed_trace
     def get_allowed_resize_sizes(
         self,