diracx-client 0.0.1a10__py3-none-any.whl → 0.0.1a12__py3-none-any.whl

diracx/client/operations/_operations.py

@@ -1,30 +1,31 @@
-# pylint: disable=too-many-lines
+# pylint: disable=too-many-lines,too-many-statements
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.1, generator: @autorest/python@6.4.11)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.13.2)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 from io import IOBase
 import sys
 from typing import Any, Callable, Dict, IO, List, Optional, TypeVar, Union, overload
 
+from azure.core import MatchConditions
 from azure.core.exceptions import (
     ClientAuthenticationError,
     HttpResponseError,
     ResourceExistsError,
+    ResourceModifiedError,
     ResourceNotFoundError,
     ResourceNotModifiedError,
     map_error,
 )
 from azure.core.pipeline import PipelineResponse
-from azure.core.pipeline.transport import HttpResponse
-from azure.core.rest import HttpRequest
+from azure.core.rest import HttpRequest, HttpResponse
 from azure.core.tracing.decorator import distributed_trace
 from azure.core.utils import case_insensitive_dict
 
 from .. import models as _models
 from .._serialization import Serializer
-from .._vendor import _format_url_section, raise_if_not_implemented
+from .._vendor import prep_if_match, prep_if_none_match, raise_if_not_implemented
 
 if sys.version_info >= (3, 9):
     from collections.abc import MutableMapping
@@ -93,7 +94,7 @@ def build_auth_do_device_flow_request(*, user_code: str, **kwargs: Any) -> HttpR
 
 
 def build_auth_initiate_device_flow_request(
-    *, client_id: str, scope: str, audience: str, **kwargs: Any
+    *, client_id: str, scope: str, **kwargs: Any
 ) -> HttpRequest:
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
     _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
@@ -106,7 +107,6 @@ def build_auth_initiate_device_flow_request(
     # Construct parameters
     _params["client_id"] = _SERIALIZER.query("client_id", client_id, "str")
     _params["scope"] = _SERIALIZER.query("scope", scope, "str")
-    _params["audience"] = _SERIALIZER.query("audience", audience, "str")
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -178,7 +178,7 @@ def build_auth_revoke_refresh_token_request(jti: str, **kwargs: Any) -> HttpRequ
         "jti": _SERIALIZER.url("jti", jti, "str"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -186,11 +186,25 @@ def build_auth_revoke_refresh_token_request(jti: str, **kwargs: Any) -> HttpRequ
     return HttpRequest(method="DELETE", url=_url, headers=_headers, **kwargs)
 
 
+def build_auth_userinfo_request(**kwargs: Any) -> HttpRequest:
+    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+
+    accept = _headers.pop("Accept", "application/json")
+
+    # Construct URL
+    _url = "/api/auth/userinfo"
+
+    # Construct headers
+    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+
+    return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
+
+
 def build_auth_authorization_flow_request(
     *,
-    response_type: Union[str, _models.Enum3],
+    response_type: Union[str, _models.Enum0],
     code_challenge: str,
-    code_challenge_method: Union[str, _models.Enum4],
+    code_challenge_method: Union[str, _models.Enum1],
     client_id: str,
     redirect_uri: str,
     scope: str,
@@ -249,25 +263,11 @@ def build_auth_authorization_flow_complete_request(  # pylint: disable=name-too-
     )
 
 
-def build_auth_userinfo_request(**kwargs: Any) -> HttpRequest:
-    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
-
-    accept = _headers.pop("Accept", "application/json")
-
-    # Construct URL
-    _url = "/api/auth/userinfo"
-
-    # Construct headers
-    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
-
-    return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
-
-
 def build_config_serve_config_request(
-    vo: str,
     *,
-    if_none_match: Optional[str] = None,
     if_modified_since: Optional[str] = None,
+    etag: Optional[str] = None,
+    match_condition: Optional[MatchConditions] = None,
     **kwargs: Any,
 ) -> HttpRequest:
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
@@ -275,23 +275,22 @@ def build_config_serve_config_request(
     accept = _headers.pop("Accept", "application/json")
 
     # Construct URL
-    _url = "/api/config/{vo}"
-    path_format_arguments = {
-        "vo": _SERIALIZER.url("vo", vo, "str"),
-    }
-
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url = "/api/config/"
 
     # Construct headers
-    if if_none_match is not None:
-        _headers["if-none-match"] = _SERIALIZER.header(
-            "if_none_match", if_none_match, "str"
-        )
     if if_modified_since is not None:
         _headers["if-modified-since"] = _SERIALIZER.header(
             "if_modified_since", if_modified_since, "str"
         )
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+    if_match = prep_if_match(etag, match_condition)
+    if if_match is not None:
+        _headers["If-Match"] = _SERIALIZER.header("if_match", if_match, "str")
+    if_none_match = prep_if_none_match(etag, match_condition)
+    if if_none_match is not None:
+        _headers["If-None-Match"] = _SERIALIZER.header(
+            "if_none_match", if_none_match, "str"
+        )
 
     return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
 
@@ -539,7 +538,7 @@ def build_jobs_reschedule_single_job_request(job_id: int, **kwargs: Any) -> Http
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -611,7 +610,7 @@ def build_jobs_get_single_job_request(job_id: int, **kwargs: Any) -> HttpRequest
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -630,7 +629,7 @@ def build_jobs_delete_single_job_request(job_id: int, **kwargs: Any) -> HttpRequ
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -655,7 +654,7 @@ def build_jobs_set_single_job_properties_request(  # pylint: disable=name-too-lo
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     if update_timestamp is not None:
@@ -686,7 +685,7 @@ def build_jobs_kill_single_job_request(job_id: int, **kwargs: Any) -> HttpReques
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -705,7 +704,7 @@ def build_jobs_remove_single_job_request(job_id: int, **kwargs: Any) -> HttpRequ
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -724,7 +723,7 @@ def build_jobs_get_single_job_status_request(job_id: int, **kwargs: Any) -> Http
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -749,7 +748,7 @@ def build_jobs_set_single_job_status_request(
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct parameters
     if force is not None:
@@ -780,7 +779,7 @@ def build_jobs_get_single_job_status_history_request(  # pylint: disable=name-to
         "job_id": _SERIALIZER.url("job_id", job_id, "int"),
     }
 
-    _url: str = _format_url_section(_url, **path_format_arguments)  # type: ignore
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
 
     # Construct headers
     _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
@@ -832,22 +831,24 @@ class WellKnownOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_well_known_openid_configuration_request(
+        _request = build_well_known_openid_configuration_request(
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -856,9 +857,9 @@ class WellKnownOperations:
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def installation_metadata(self, **kwargs: Any) -> _models.Metadata:
@@ -883,22 +884,24 @@ class WellKnownOperations:
 
         cls: ClsType[_models.Metadata] = kwargs.pop("cls", None)
 
-        request = build_well_known_installation_metadata_request(
+        _request = build_well_known_installation_metadata_request(
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -907,12 +910,12 @@ class WellKnownOperations:
         deserialized = self._deserialize("Metadata", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
 
-class AuthOperations:
+class AuthOperations:  # pylint: disable=abstract-class-instantiated
     """
     .. warning::
         **DO NOT** instantiate this class directly.
@@ -932,6 +935,7 @@ class AuthOperations:
         self._deserialize = (
             input_args.pop(0) if input_args else kwargs.pop("deserializer")
         )
+
         raise_if_not_implemented(
             self.__class__,
             [
@@ -971,23 +975,25 @@ class AuthOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_auth_do_device_flow_request(
+        _request = build_auth_do_device_flow_request(
             user_code=user_code,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -996,13 +1002,13 @@ class AuthOperations:
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def initiate_device_flow(
-        self, *, client_id: str, scope: str, audience: str, **kwargs: Any
+        self, *, client_id: str, scope: str, **kwargs: Any
     ) -> _models.InitiateDeviceFlowResponse:
         """Initiate Device Flow.
 
@@ -1018,8 +1024,6 @@ class AuthOperations:
         :paramtype client_id: str
         :keyword scope: Required.
         :paramtype scope: str
-        :keyword audience: Required.
-        :paramtype audience: str
         :return: InitiateDeviceFlowResponse
         :rtype: ~client.models.InitiateDeviceFlowResponse
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -1037,25 +1041,26 @@ class AuthOperations:
 
         cls: ClsType[_models.InitiateDeviceFlowResponse] = kwargs.pop("cls", None)
 
-        request = build_auth_initiate_device_flow_request(
+        _request = build_auth_initiate_device_flow_request(
             client_id=client_id,
             scope=scope,
-            audience=audience,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1066,9 +1071,9 @@ class AuthOperations:
         )
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def finish_device_flow(self, *, code: str, state: str, **kwargs: Any) -> Any:
@@ -1101,24 +1106,26 @@ class AuthOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_auth_finish_device_flow_request(
+        _request = build_auth_finish_device_flow_request(
             code=code,
             state=state,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1127,9 +1134,9 @@ class AuthOperations:
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def finished(self, **kwargs: Any) -> Any:
@@ -1154,22 +1161,24 @@ class AuthOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_auth_finished_request(
+        _request = build_auth_finished_request(
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1178,15 +1187,16 @@ class AuthOperations:
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def get_refresh_tokens(self, **kwargs: Any) -> List[Any]:
         """Get Refresh Tokens.
 
-        Get Refresh Tokens.
+        Get all refresh tokens for the user. If the user has the ``proxy_management`` property, then
+        the subject is not used to filter the refresh tokens.
 
         :return: list of any
         :rtype: list[any]
@@ -1205,22 +1215,24 @@ class AuthOperations:
 
         cls: ClsType[List[Any]] = kwargs.pop("cls", None)
 
-        request = build_auth_get_refresh_tokens_request(
+        _request = build_auth_get_refresh_tokens_request(
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1229,15 +1241,16 @@ class AuthOperations:
         deserialized = self._deserialize("[object]", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def revoke_refresh_token(self, jti: str, **kwargs: Any) -> str:
         """Revoke Refresh Token.
 
-        Revoke Refresh Token.
+        Revoke a refresh token. If the user has the ``proxy_management`` property, then
+        the subject is not used to filter the refresh tokens.
 
         :param jti: Required.
         :type jti: str
@@ -1258,23 +1271,25 @@ class AuthOperations:
 
         cls: ClsType[str] = kwargs.pop("cls", None)
 
-        request = build_auth_revoke_refresh_token_request(
+        _request = build_auth_revoke_refresh_token_request(
             jti=jti,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1283,17 +1298,70 @@ class AuthOperations:
         deserialized = self._deserialize("str", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
+
+    @distributed_trace
+    def userinfo(self, **kwargs: Any) -> _models.UserInfoResponse:
+        """Userinfo.
+
+        Get information about the user's identity.
+
+        :return: UserInfoResponse
+        :rtype: ~client.models.UserInfoResponse
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map = {
+            401: ClientAuthenticationError,
+            404: ResourceNotFoundError,
+            409: ResourceExistsError,
+            304: ResourceNotModifiedError,
+        }
+        error_map.update(kwargs.pop("error_map", {}) or {})
+
+        _headers = kwargs.pop("headers", {}) or {}
+        _params = kwargs.pop("params", {}) or {}
+
+        cls: ClsType[_models.UserInfoResponse] = kwargs.pop("cls", None)
+
+        _request = build_auth_userinfo_request(
+            headers=_headers,
+            params=_params,
+        )
+        _request.url = self._client.format_url(_request.url)
+
+        _stream = False
+        pipeline_response: PipelineResponse = (
+            self._client._pipeline.run(  # pylint: disable=protected-access
+                _request, stream=_stream, **kwargs
+            )
+        )
+
+        response = pipeline_response.http_response
+
+        if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
+            map_error(
+                status_code=response.status_code, response=response, error_map=error_map
+            )
+            raise HttpResponseError(response=response)
+
+        deserialized = self._deserialize("UserInfoResponse", pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def authorization_flow(
         self,
         *,
-        response_type: Union[str, _models.Enum3],
+        response_type: Union[str, _models.Enum0],
         code_challenge: str,
-        code_challenge_method: Union[str, _models.Enum4],
+        code_challenge_method: Union[str, _models.Enum1],
         client_id: str,
         redirect_uri: str,
         scope: str,
@@ -1305,11 +1373,11 @@ class AuthOperations:
         Authorization Flow.
 
         :keyword response_type: "code" Required.
-        :paramtype response_type: str or ~client.models.Enum3
+        :paramtype response_type: str or ~client.models.Enum0
         :keyword code_challenge: Required.
         :paramtype code_challenge: str
         :keyword code_challenge_method: "S256" Required.
-        :paramtype code_challenge_method: str or ~client.models.Enum4
+        :paramtype code_challenge_method: str or ~client.models.Enum1
         :keyword client_id: Required.
         :paramtype client_id: str
         :keyword redirect_uri: Required.
@@ -1335,7 +1403,7 @@ class AuthOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_auth_authorization_flow_request(
+        _request = build_auth_authorization_flow_request(
             response_type=response_type,
             code_challenge=code_challenge,
             code_challenge_method=code_challenge_method,
@@ -1346,18 +1414,20 @@ class AuthOperations:
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1366,9 +1436,9 @@ class AuthOperations:
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def authorization_flow_complete(
@@ -1399,24 +1469,26 @@ class AuthOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_auth_authorization_flow_complete_request(
+        _request = build_auth_authorization_flow_complete_request(
             code=code,
             state=state,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1425,60 +1497,9 @@ class AuthOperations:
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
-
-        return deserialized
-
-    @distributed_trace
-    def userinfo(self, **kwargs: Any) -> _models.UserInfoResponse:
-        """Userinfo.
-
-        Userinfo.
-
-        :return: UserInfoResponse
-        :rtype: ~client.models.UserInfoResponse
-        :raises ~azure.core.exceptions.HttpResponseError:
-        """
-        error_map = {
-            401: ClientAuthenticationError,
-            404: ResourceNotFoundError,
-            409: ResourceExistsError,
-            304: ResourceNotModifiedError,
-        }
-        error_map.update(kwargs.pop("error_map", {}) or {})
-
-        _headers = kwargs.pop("headers", {}) or {}
-        _params = kwargs.pop("params", {}) or {}
-
-        cls: ClsType[_models.UserInfoResponse] = kwargs.pop("cls", None)
-
-        request = build_auth_userinfo_request(
-            headers=_headers,
-            params=_params,
-        )
-        request.url = self._client.format_url(request.url)
-
-        _stream = False
-        pipeline_response: PipelineResponse = (
-            self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **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
-            )
-            raise HttpResponseError(response=response)
-
-        deserialized = self._deserialize("UserInfoResponse", pipeline_response)
-
-        if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
 
 class ConfigOperations:
@@ -1505,15 +1526,14 @@ class ConfigOperations:
     @distributed_trace
     def serve_config(
         self,
-        vo: str,
         *,
-        if_none_match: Optional[str] = None,
         if_modified_since: Optional[str] = None,
+        etag: Optional[str] = None,
+        match_condition: Optional[MatchConditions] = None,
         **kwargs: Any,
     ) -> Any:
         """Serve Config.
 
-        "
         Get the latest view of the config.
 
         If If-None-Match header is given and matches the latest ETag, return 304
@@ -1521,12 +1541,13 @@ class ConfigOperations:
         If If-Modified-Since is given and is newer than latest,
             return 304: this is to avoid flip/flopping.
 
-        :param vo: Required.
-        :type vo: str
-        :keyword if_none_match: Default value is None.
-        :paramtype if_none_match: str
         :keyword if_modified_since: Default value is None.
         :paramtype if_modified_since: str
+        :keyword etag: check if resource is changed. Set None to skip checking etag. Default value is
+         None.
+        :paramtype etag: str
+        :keyword match_condition: The match condition to use upon the etag. Default value is None.
+        :paramtype match_condition: ~azure.core.MatchConditions
         :return: any
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -1537,6 +1558,12 @@ class ConfigOperations:
             409: ResourceExistsError,
             304: ResourceNotModifiedError,
         }
+        if match_condition == MatchConditions.IfNotModified:
+            error_map[412] = ResourceModifiedError
+        elif match_condition == MatchConditions.IfPresent:
+            error_map[412] = ResourceNotFoundError
+        elif match_condition == MatchConditions.IfMissing:
+            error_map[412] = ResourceExistsError
         error_map.update(kwargs.pop("error_map", {}) or {})
 
         _headers = kwargs.pop("headers", {}) or {}
@@ -1544,25 +1571,27 @@ class ConfigOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_config_serve_config_request(
-            vo=vo,
-            if_none_match=if_none_match,
+        _request = build_config_serve_config_request(
             if_modified_since=if_modified_since,
+            etag=etag,
+            match_condition=match_condition,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1571,9 +1600,9 @@ class ConfigOperations:
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
 
 class JobsOperations:  # pylint: disable=too-many-public-methods
@@ -1630,23 +1659,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[_models.SandboxDownloadResponse] = kwargs.pop("cls", None)
 
-        request = build_jobs_get_sandbox_file_request(
+        _request = build_jobs_get_sandbox_file_request(
             pfn=pfn,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1655,9 +1686,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("SandboxDownloadResponse", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def initiate_sandbox_upload(
@@ -1689,7 +1720,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
     @overload
     def initiate_sandbox_upload(
-        self, body: IO, *, content_type: str = "application/json", **kwargs: Any
+        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> _models.SandboxUploadResponse:
         """Initiate Sandbox Upload.
 
@@ -1702,7 +1733,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         should be used to upload the sandbox to the storage backend.
 
         :param body: Required.
-        :type body: IO
+        :type body: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -1713,7 +1744,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
     @distributed_trace
     def initiate_sandbox_upload(
-        self, body: Union[_models.SandboxInfo, IO], **kwargs: Any
+        self, body: Union[_models.SandboxInfo, IO[bytes]], **kwargs: Any
     ) -> _models.SandboxUploadResponse:
         """Initiate Sandbox Upload.
 
@@ -1725,11 +1756,8 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         If the sandbox does not exist in the database then the "url" and "fields"
         should be used to upload the sandbox to the storage backend.
 
-        :param body: Is either a SandboxInfo type or a IO type. Required.
-        :type body: ~client.models.SandboxInfo or IO
-        :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
-         Default value is None.
-        :paramtype content_type: str
+        :param body: Is either a SandboxInfo type or a IO[bytes] type. Required.
+        :type body: ~client.models.SandboxInfo or IO[bytes]
         :return: SandboxUploadResponse
         :rtype: ~client.models.SandboxUploadResponse
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -1758,25 +1786,27 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         else:
             _json = self._serialize.body(body, "SandboxInfo")
 
-        request = build_jobs_initiate_sandbox_upload_request(
+        _request = build_jobs_initiate_sandbox_upload_request(
             content_type=content_type,
             json=_json,
             content=_content,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1785,9 +1815,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("SandboxUploadResponse", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def submit_bulk_jobs(
@@ -1809,14 +1839,14 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
     @overload
     def submit_bulk_jobs(
-        self, body: IO, *, content_type: str = "application/json", **kwargs: Any
+        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> List[_models.InsertedJob]:
         """Submit Bulk Jobs.
 
         Submit Bulk Jobs.
 
         :param body: Required.
-        :type body: IO
+        :type body: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -1827,17 +1857,14 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
     @distributed_trace
     def submit_bulk_jobs(
-        self, body: Union[List[str], IO], **kwargs: Any
+        self, body: Union[List[str], IO[bytes]], **kwargs: Any
     ) -> List[_models.InsertedJob]:
         """Submit Bulk Jobs.
 
         Submit Bulk Jobs.
 
-        :param body: Is either a [str] type or a IO type. Required.
-        :type body: list[str] or IO
-        :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
-         Default value is None.
-        :paramtype content_type: str
+        :param body: Is either a [str] type or a IO[bytes] type. Required.
+        :type body: list[str] or IO[bytes]
         :return: list of InsertedJob
         :rtype: list[~client.models.InsertedJob]
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -1866,25 +1893,27 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         else:
             _json = self._serialize.body(body, "[str]")
 
-        request = build_jobs_submit_bulk_jobs_request(
+        _request = build_jobs_submit_bulk_jobs_request(
             content_type=content_type,
             json=_json,
             content=_content,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1893,9 +1922,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("[InsertedJob]", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def delete_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
@@ -1922,23 +1951,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_delete_bulk_jobs_request(
+        _request = build_jobs_delete_bulk_jobs_request(
             job_ids=job_ids,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -1947,9 +1978,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def kill_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
@@ -1976,23 +2007,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_kill_bulk_jobs_request(
+        _request = build_jobs_kill_bulk_jobs_request(
             job_ids=job_ids,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2001,9 +2034,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def remove_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
@@ -2034,23 +2067,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_remove_bulk_jobs_request(
+        _request = build_jobs_remove_bulk_jobs_request(
             job_ids=job_ids,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2059,9 +2094,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def get_job_status_bulk(
@@ -2092,23 +2127,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
             "cls", None
         )
 
-        request = build_jobs_get_job_status_bulk_request(
+        _request = build_jobs_get_job_status_bulk_request(
             job_ids=job_ids,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2117,9 +2154,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("{LimitedJobStatusReturn}", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def set_job_status_bulk(
@@ -2149,7 +2186,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
     @overload
     def set_job_status_bulk(
         self,
-        body: IO,
+        body: IO[bytes],
         *,
         force: bool = False,
         content_type: str = "application/json",
@@ -2160,7 +2197,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         Set Job Status Bulk.
 
         :param body: Required.
-        :type body: IO
+        :type body: IO[bytes]
         :keyword force: Default value is False.
         :paramtype force: bool
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
@@ -2174,7 +2211,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
     @distributed_trace
     def set_job_status_bulk(
         self,
-        body: Union[Dict[str, Dict[str, _models.JobStatusUpdate]], IO],
+        body: Union[Dict[str, Dict[str, _models.JobStatusUpdate]], IO[bytes]],
         *,
         force: bool = False,
         **kwargs: Any,
@@ -2183,13 +2220,10 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         Set Job Status Bulk.
 
-        :param body: Is either a {str: {str: JobStatusUpdate}} type or a IO type. Required.
-        :type body: dict[str, dict[str, ~client.models.JobStatusUpdate]] or IO
+        :param body: Is either a {str: {str: JobStatusUpdate}} type or a IO[bytes] type. Required.
+        :type body: dict[str, dict[str, ~client.models.JobStatusUpdate]] or IO[bytes]
         :keyword force: Default value is False.
         :paramtype force: bool
-        :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
-         Default value is None.
-        :paramtype content_type: str
         :return: dict mapping str to SetJobStatusReturn
         :rtype: dict[str, ~client.models.SetJobStatusReturn]
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -2218,7 +2252,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         else:
             _json = self._serialize.body(body, "{{JobStatusUpdate}}")
 
-        request = build_jobs_set_job_status_bulk_request(
+        _request = build_jobs_set_job_status_bulk_request(
             force=force,
             content_type=content_type,
             json=_json,
@@ -2226,18 +2260,20 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2246,9 +2282,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("{SetJobStatusReturn}", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def get_job_status_history_bulk(
@@ -2277,23 +2313,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Dict[str, List[_models.JobStatusReturn]]] = kwargs.pop("cls", None)
 
-        request = build_jobs_get_job_status_history_bulk_request(
+        _request = build_jobs_get_job_status_history_bulk_request(
             job_ids=job_ids,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2302,9 +2340,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("{[JobStatusReturn]}", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def reschedule_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
@@ -2331,23 +2369,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_reschedule_bulk_jobs_request(
+        _request = build_jobs_reschedule_bulk_jobs_request(
             job_ids=job_ids,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2356,9 +2396,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def reschedule_single_job(self, job_id: int, **kwargs: Any) -> Any:
@@ -2385,23 +2425,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_reschedule_single_job_request(
+        _request = build_jobs_reschedule_single_job_request(
             job_id=job_id,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2410,9 +2452,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def search(
@@ -2447,7 +2489,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
     @overload
     def search(
         self,
-        body: Optional[IO] = None,
+        body: Optional[IO[bytes]] = None,
         *,
         page: int = 0,
         per_page: int = 100,
@@ -2461,7 +2503,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         **TODO: Add more docs**.
 
         :param body: Default value is None.
-        :type body: IO
+        :type body: IO[bytes]
         :keyword page: Default value is 0.
         :paramtype page: int
         :keyword per_page: Default value is 100.
@@ -2477,7 +2519,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
     @distributed_trace
     def search(
         self,
-        body: Optional[Union[_models.JobSearchParams, IO]] = None,
+        body: Optional[Union[_models.JobSearchParams, IO[bytes]]] = None,
         *,
         page: int = 0,
         per_page: int = 100,
@@ -2489,15 +2531,12 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         **TODO: Add more docs**.
 
-        :param body: Is either a JobSearchParams type or a IO type. Default value is None.
-        :type body: ~client.models.JobSearchParams or IO
+        :param body: Is either a JobSearchParams type or a IO[bytes] type. Default value is None.
+        :type body: ~client.models.JobSearchParams or IO[bytes]
         :keyword page: Default value is 0.
         :paramtype page: int
         :keyword per_page: Default value is 100.
         :paramtype per_page: int
-        :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
-         Default value is None.
-        :paramtype content_type: str
         :return: list of JSON
         :rtype: list[JSON]
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -2529,7 +2568,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
             else:
                 _json = None
 
-        request = build_jobs_search_request(
+        _request = build_jobs_search_request(
             page=page,
             per_page=per_page,
             content_type=content_type,
@@ -2538,18 +2577,20 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2558,9 +2599,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("[object]", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def summary(
@@ -2586,14 +2627,14 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
     @overload
     def summary(
-        self, body: IO, *, content_type: str = "application/json", **kwargs: Any
+        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> Any:
         """Summary.
 
         Show information suitable for plotting.
 
         :param body: Required.
-        :type body: IO
+        :type body: IO[bytes]
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -2603,16 +2644,15 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         """
 
     @distributed_trace
-    def summary(self, body: Union[_models.JobSummaryParams, IO], **kwargs: Any) -> Any:
+    def summary(
+        self, body: Union[_models.JobSummaryParams, IO[bytes]], **kwargs: Any
+    ) -> Any:
         """Summary.
 
         Show information suitable for plotting.
 
-        :param body: Is either a JobSummaryParams type or a IO type. Required.
-        :type body: ~client.models.JobSummaryParams or IO
-        :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
-         Default value is None.
-        :paramtype content_type: str
+        :param body: Is either a JobSummaryParams type or a IO[bytes] type. Required.
+        :type body: ~client.models.JobSummaryParams or IO[bytes]
         :return: any
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -2641,25 +2681,27 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         else:
             _json = self._serialize.body(body, "JobSummaryParams")
 
-        request = build_jobs_summary_request(
+        _request = build_jobs_summary_request(
             content_type=content_type,
             json=_json,
             content=_content,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2668,9 +2710,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def get_single_job(self, job_id: int, **kwargs: Any) -> Any:
@@ -2697,23 +2739,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_get_single_job_request(
+        _request = build_jobs_get_single_job_request(
             job_id=job_id,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2722,9 +2766,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def delete_single_job(self, job_id: int, **kwargs: Any) -> Any:
@@ -2751,23 +2795,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_delete_single_job_request(
+        _request = build_jobs_delete_single_job_request(
             job_id=job_id,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2776,9 +2822,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def set_single_job_properties(
@@ -2816,7 +2862,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         _json = self._serialize.body(body, "object")
 
-        request = build_jobs_set_single_job_properties_request(
+        _request = build_jobs_set_single_job_properties_request(
             job_id=job_id,
             update_timestamp=update_timestamp,
             content_type=content_type,
@@ -2824,18 +2870,20 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2844,9 +2892,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def kill_single_job(self, job_id: int, **kwargs: Any) -> Any:
@@ -2873,23 +2921,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_kill_single_job_request(
+        _request = build_jobs_kill_single_job_request(
             job_id=job_id,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2898,9 +2948,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def remove_single_job(self, job_id: int, **kwargs: Any) -> Any:
@@ -2931,23 +2981,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_jobs_remove_single_job_request(
+        _request = build_jobs_remove_single_job_request(
             job_id=job_id,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -2956,9 +3008,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("object", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def get_single_job_status(
@@ -2989,23 +3041,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
             "cls", None
         )
 
-        request = build_jobs_get_single_job_status_request(
+        _request = build_jobs_get_single_job_status_request(
             job_id=job_id,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -3014,9 +3068,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("{LimitedJobStatusReturn}", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @overload
     def set_single_job_status(
@@ -3050,7 +3104,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
     def set_single_job_status(
         self,
         job_id: int,
-        body: IO,
+        body: IO[bytes],
         *,
         force: bool = False,
         content_type: str = "application/json",
@@ -3063,7 +3117,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :param job_id: Required.
         :type job_id: int
         :param body: Required.
-        :type body: IO
+        :type body: IO[bytes]
         :keyword force: Default value is False.
         :paramtype force: bool
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
@@ -3078,7 +3132,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
     def set_single_job_status(
         self,
         job_id: int,
-        body: Union[Dict[str, _models.JobStatusUpdate], IO],
+        body: Union[Dict[str, _models.JobStatusUpdate], IO[bytes]],
         *,
         force: bool = False,
         **kwargs: Any,
@@ -3089,13 +3143,10 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         :param job_id: Required.
         :type job_id: int
-        :param body: Is either a {str: JobStatusUpdate} type or a IO type. Required.
-        :type body: dict[str, ~client.models.JobStatusUpdate] or IO
+        :param body: Is either a {str: JobStatusUpdate} type or a IO[bytes] type. Required.
+        :type body: dict[str, ~client.models.JobStatusUpdate] or IO[bytes]
         :keyword force: Default value is False.
         :paramtype force: bool
-        :keyword content_type: Body Parameter content-type. Known values are: 'application/json'.
-         Default value is None.
-        :paramtype content_type: str
         :return: dict mapping str to SetJobStatusReturn
         :rtype: dict[str, ~client.models.SetJobStatusReturn]
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -3124,7 +3175,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         else:
             _json = self._serialize.body(body, "{JobStatusUpdate}")
 
-        request = build_jobs_set_single_job_status_request(
+        _request = build_jobs_set_single_job_status_request(
             job_id=job_id,
             force=force,
             content_type=content_type,
@@ -3133,18 +3184,20 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -3153,9 +3206,9 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("{SetJobStatusReturn}", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
 
     @distributed_trace
     def get_single_job_status_history(
@@ -3184,23 +3237,25 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         cls: ClsType[Dict[str, List[_models.JobStatusReturn]]] = kwargs.pop("cls", None)
 
-        request = build_jobs_get_single_job_status_history_request(
+        _request = build_jobs_get_single_job_status_history_request(
             job_id=job_id,
             headers=_headers,
             params=_params,
         )
-        request.url = self._client.format_url(request.url)
+        _request.url = self._client.format_url(_request.url)
 
         _stream = False
         pipeline_response: PipelineResponse = (
             self._client._pipeline.run(  # pylint: disable=protected-access
-                request, stream=_stream, **kwargs
+                _request, stream=_stream, **kwargs
             )
         )
 
         response = pipeline_response.http_response
 
         if response.status_code not in [200]:
+            if _stream:
+                response.read()  # Load the body in memory and close the socket
             map_error(
                 status_code=response.status_code, response=response, error_map=error_map
             )
@@ -3209,6 +3264,6 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         deserialized = self._deserialize("{[JobStatusReturn]}", pipeline_response)
 
         if cls:
-            return cls(pipeline_response, deserialized, {})
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore