diracx-client 0.0.1a11__py3-none-any.whl → 0.0.1a13__py3-none-any.whl

diracx/client/operations/_operations.py

@@ -1,30 +1,42 @@
-# 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.9)
 # 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 typing import (
+    Any,
+    Callable,
+    Dict,
+    IO,
+    List,
+    Optional,
+    Type,
+    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 +105,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 +118,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 +189,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 +197,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,24 +274,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(
     *,
-    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 {})
@@ -277,15 +289,19 @@ def build_config_serve_config_request(
     _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)
 
@@ -339,6 +355,120 @@ def build_jobs_initiate_sandbox_upload_request(
     return HttpRequest(method="POST", url=_url, headers=_headers, **kwargs)
 
 
+def build_jobs_unassign_bulk_jobs_sandboxes_request(  # pylint: disable=name-too-long
+    *, jobs_ids: List[int], **kwargs: Any
+) -> HttpRequest:
+    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+    _params = case_insensitive_dict(kwargs.pop("params", {}) or {})
+
+    accept = _headers.pop("Accept", "application/json")
+
+    # Construct URL
+    _url = "/api/jobs/sandbox"
+
+    # Construct parameters
+    _params["jobs_ids"] = _SERIALIZER.query("jobs_ids", jobs_ids, "[int]")
+
+    # Construct headers
+    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+
+    return HttpRequest(
+        method="DELETE", url=_url, params=_params, headers=_headers, **kwargs
+    )
+
+
+def build_jobs_get_job_sandboxes_request(job_id: int, **kwargs: Any) -> HttpRequest:
+    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+
+    accept = _headers.pop("Accept", "application/json")
+
+    # Construct URL
+    _url = "/api/jobs/{job_id}/sandbox"
+    path_format_arguments = {
+        "job_id": _SERIALIZER.url("job_id", job_id, "int"),
+    }
+
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
+
+    # Construct headers
+    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+
+    return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
+
+
+def build_jobs_unassign_job_sandboxes_request(  # pylint: disable=name-too-long
+    job_id: int, **kwargs: Any
+) -> HttpRequest:
+    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+
+    accept = _headers.pop("Accept", "application/json")
+
+    # Construct URL
+    _url = "/api/jobs/{job_id}/sandbox"
+    path_format_arguments = {
+        "job_id": _SERIALIZER.url("job_id", job_id, "int"),
+    }
+
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
+
+    # Construct headers
+    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+
+    return HttpRequest(method="DELETE", url=_url, headers=_headers, **kwargs)
+
+
+def build_jobs_get_job_sandbox_request(
+    job_id: int, sandbox_type: Union[str, _models.SandboxType], **kwargs: Any
+) -> HttpRequest:
+    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+
+    accept = _headers.pop("Accept", "application/json")
+
+    # Construct URL
+    _url = "/api/jobs/{job_id}/sandbox/{sandbox_type}"
+    path_format_arguments = {
+        "job_id": _SERIALIZER.url("job_id", job_id, "int"),
+        "sandbox_type": _SERIALIZER.url("sandbox_type", sandbox_type, "str"),
+    }
+
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
+
+    # Construct headers
+    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+
+    return HttpRequest(method="GET", url=_url, headers=_headers, **kwargs)
+
+
+def build_jobs_assign_sandbox_to_job_request(
+    job_id: int, *, content: str, **kwargs: Any
+) -> HttpRequest:
+    _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+
+    content_type: Optional[str] = kwargs.pop(
+        "content_type", _headers.pop("Content-Type", None)
+    )
+    accept = _headers.pop("Accept", "application/json")
+
+    # Construct URL
+    _url = "/api/jobs/{job_id}/sandbox/output"
+    path_format_arguments = {
+        "job_id": _SERIALIZER.url("job_id", job_id, "int"),
+    }
+
+    _url: str = _url.format(**path_format_arguments)  # type: ignore
+
+    # Construct headers
+    if content_type is not None:
+        _headers["Content-Type"] = _SERIALIZER.header(
+            "content_type", content_type, "str"
+        )
+    _headers["Accept"] = _SERIALIZER.header("accept", accept, "str")
+
+    return HttpRequest(
+        method="PATCH", url=_url, headers=_headers, content=content, **kwargs
+    )
+
+
 def build_jobs_submit_bulk_jobs_request(**kwargs: Any) -> HttpRequest:
     _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
 
@@ -533,7 +663,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")
@@ -605,7 +735,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")
@@ -624,7 +754,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")
@@ -649,7 +779,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:
@@ -680,7 +810,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")
@@ -699,7 +829,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")
@@ -718,7 +848,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")
@@ -743,7 +873,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:
@@ -774,7 +904,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")
@@ -807,13 +937,13 @@ class WellKnownOperations:
     def openid_configuration(self, **kwargs: Any) -> Any:
         """Openid Configuration.
 
-        Openid Configuration.
+        OpenID Connect discovery endpoint.
 
         :return: any
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -826,22 +956,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
             )
@@ -850,21 +982,21 @@ 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:
         """Installation Metadata.
 
-        Installation Metadata.
+        Get metadata about the dirac installation.
 
         :return: Metadata
         :rtype: ~client.models.Metadata
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -877,22 +1009,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
             )
@@ -901,12 +1035,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.
@@ -926,6 +1060,7 @@ class AuthOperations:
         self._deserialize = (
             input_args.pop(0) if input_args else kwargs.pop("deserializer")
         )
+
         raise_if_not_implemented(
             self.__class__,
             [
@@ -952,7 +1087,7 @@ class AuthOperations:
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -965,23 +1100,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
             )
@@ -990,13 +1127,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.
 
@@ -1012,13 +1149,11 @@ 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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1031,25 +1166,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
             )
@@ -1060,9 +1196,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:
@@ -1082,7 +1218,7 @@ class AuthOperations:
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1095,24 +1231,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
             )
@@ -1121,21 +1259,21 @@ 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:
         """Finished.
 
-        Finished.
+        This is the final step of the device flow.
 
         :return: any
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1148,22 +1286,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
             )
@@ -1172,21 +1312,22 @@ 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]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1199,22 +1340,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
             )
@@ -1223,15 +1366,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
@@ -1239,7 +1383,7 @@ class AuthOperations:
         :rtype: str
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1252,23 +1396,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
             )
@@ -1277,17 +1423,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: MutableMapping[int, Type[HttpResponseError]] = {
+            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,
@@ -1296,14 +1495,20 @@ class AuthOperations:
     ) -> Any:
         """Authorization Flow.
 
-        Authorization Flow.
+        Initiate the authorization flow.
+        It will redirect to the actual OpenID server (IAM, CheckIn) to
+        perform a authorization code flow.
+
+        We set the user details obtained from the user authorize flow in a cookie
+        to be able to map the authorization flow with the corresponding
+        user authorize 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.
@@ -1316,7 +1521,7 @@ class AuthOperations:
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1329,7 +1534,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,
@@ -1340,18 +1545,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
             )
@@ -1360,9 +1567,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(
@@ -1370,7 +1577,13 @@ class AuthOperations:
     ) -> Any:
         """Authorization Flow Complete.
 
-        Authorization Flow Complete.
+        Complete the authorization flow.
+
+        The user is redirected back to the DIRAC auth service after completing the IAM's authorization
+        flow.
+        We retrieve the original flow details from the decrypted state and store the ID token requested
+        from the IAM.
+        The user is then redirected to the client's redirect URI.
 
         :keyword code: Required.
         :paramtype code: str
@@ -1380,7 +1593,7 @@ class AuthOperations:
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1393,24 +1606,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
             )
@@ -1419,60 +1634,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
-
-    @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 deserialized
+        return deserialized  # type: ignore
 
 
 class ConfigOperations:
@@ -1500,8 +1664,9 @@ class ConfigOperations:
     def serve_config(
         self,
         *,
-        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.
@@ -1513,20 +1678,29 @@ class ConfigOperations:
         If If-Modified-Since is given and is newer than latest,
             return 304: this is to avoid flip/flopping.
 
-        :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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             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 {}
@@ -1534,24 +1708,27 @@ class ConfigOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        request = build_config_serve_config_request(
-            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
             )
@@ -1560,9 +1737,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
@@ -1606,7 +1783,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: ~client.models.SandboxDownloadResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1619,23 +1796,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
             )
@@ -1644,9 +1823,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(
@@ -1678,7 +1857,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.
 
@@ -1691,7 +1870,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
@@ -1702,7 +1881,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.
 
@@ -1714,16 +1893,13 @@ 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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1747,25 +1923,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
             )
@@ -1774,9 +1952,305 @@ 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  # type: ignore
+
+    @distributed_trace
+    def unassign_bulk_jobs_sandboxes(
+        self, *, jobs_ids: List[int], **kwargs: Any
+    ) -> Any:
+        """Unassign Bulk Jobs Sandboxes.
+
+        Delete bulk jobs sandbox mapping.
+
+        :keyword jobs_ids: Required.
+        :paramtype jobs_ids: list[int]
+        :return: any
+        :rtype: any
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+            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[Any] = kwargs.pop("cls", None)
+
+        _request = build_jobs_unassign_bulk_jobs_sandboxes_request(
+            jobs_ids=jobs_ids,
+            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("object", pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
 
-        return deserialized
+        return deserialized  # type: ignore
+
+    @distributed_trace
+    def get_job_sandboxes(self, job_id: int, **kwargs: Any) -> Dict[str, List[Any]]:
+        """Get Job Sandboxes.
+
+        Get input and output sandboxes of given job.
+
+        :param job_id: Required.
+        :type job_id: int
+        :return: dict mapping str to list of any
+        :rtype: dict[str, list[any]]
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+            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[Dict[str, List[Any]]] = kwargs.pop("cls", None)
+
+        _request = build_jobs_get_job_sandboxes_request(
+            job_id=job_id,
+            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("{[object]}", pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
+
+    @distributed_trace
+    def unassign_job_sandboxes(self, job_id: int, **kwargs: Any) -> Any:
+        """Unassign Job Sandboxes.
+
+        Delete single job sandbox mapping.
+
+        :param job_id: Required.
+        :type job_id: int
+        :return: any
+        :rtype: any
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+            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[Any] = kwargs.pop("cls", None)
+
+        _request = build_jobs_unassign_job_sandboxes_request(
+            job_id=job_id,
+            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("object", pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
+
+    @distributed_trace
+    def get_job_sandbox(
+        self, job_id: int, sandbox_type: Union[str, _models.SandboxType], **kwargs: Any
+    ) -> List[Any]:
+        """Get Job Sandbox.
+
+        Get input or output sandbox of given job.
+
+        :param job_id: Required.
+        :type job_id: int
+        :param sandbox_type: Known values are: "input" and "output". Required.
+        :type sandbox_type: str or ~client.models.SandboxType
+        :return: list of any
+        :rtype: list[any]
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+            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[List[Any]] = kwargs.pop("cls", None)
+
+        _request = build_jobs_get_job_sandbox_request(
+            job_id=job_id,
+            sandbox_type=sandbox_type,
+            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("[object]", pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
+
+    @distributed_trace
+    def assign_sandbox_to_job(self, job_id: int, body: str, **kwargs: Any) -> Any:
+        """Assign Sandbox To Job.
+
+        Mapp the pfn as output sandbox to job.
+
+        :param job_id: Required.
+        :type job_id: int
+        :param body: Required.
+        :type body: str
+        :return: any
+        :rtype: any
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
+            401: ClientAuthenticationError,
+            404: ResourceNotFoundError,
+            409: ResourceExistsError,
+            304: ResourceNotModifiedError,
+        }
+        error_map.update(kwargs.pop("error_map", {}) or {})
+
+        _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+        _params = kwargs.pop("params", {}) or {}
+
+        content_type: str = kwargs.pop(
+            "content_type", _headers.pop("Content-Type", "application/json")
+        )
+        cls: ClsType[Any] = kwargs.pop("cls", None)
+
+        _content = self._serialize.body(body, "str")
+
+        _request = build_jobs_assign_sandbox_to_job_request(
+            job_id=job_id,
+            content_type=content_type,
+            content=_content,
+            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("object", pipeline_response)
+
+        if cls:
+            return cls(pipeline_response, deserialized, {})  # type: ignore
+
+        return deserialized  # type: ignore
 
     @overload
     def submit_bulk_jobs(
@@ -1798,14 +2272,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
@@ -1816,22 +2290,19 @@ 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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1855,25 +2326,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
             )
@@ -1882,9 +2355,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:
@@ -1898,7 +2371,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1911,23 +2384,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
             )
@@ -1936,9 +2411,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:
@@ -1952,7 +2427,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1965,23 +2440,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
             )
@@ -1990,9 +2467,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:
@@ -2010,7 +2487,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2023,23 +2500,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
             )
@@ -2048,9 +2527,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(
@@ -2066,7 +2545,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: dict[str, ~client.models.LimitedJobStatusReturn]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2081,23 +2560,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
             )
@@ -2106,9 +2587,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(
@@ -2138,7 +2619,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",
@@ -2149,7 +2630,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.
@@ -2163,7 +2644,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,
@@ -2172,18 +2653,15 @@ 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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2207,7 +2685,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,
@@ -2215,18 +2693,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
             )
@@ -2235,9 +2715,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(
@@ -2253,7 +2733,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: dict[str, list[~client.models.JobStatusReturn]]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2266,23 +2746,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
             )
@@ -2291,9 +2773,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:
@@ -2307,7 +2789,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2320,23 +2802,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
             )
@@ -2345,9 +2829,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:
@@ -2361,7 +2845,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2374,23 +2858,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
             )
@@ -2399,9 +2885,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(
@@ -2436,7 +2922,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,
@@ -2450,7 +2936,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.
@@ -2466,7 +2952,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,
@@ -2478,20 +2964,17 @@ 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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2518,7 +3001,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,
@@ -2527,18 +3010,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
             )
@@ -2547,9 +3032,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(
@@ -2575,14 +3060,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
@@ -2592,21 +3077,20 @@ 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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2630,25 +3114,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
             )
@@ -2657,9 +3143,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:
@@ -2673,7 +3159,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2686,23 +3172,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
             )
@@ -2711,9 +3199,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:
@@ -2727,7 +3215,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2740,23 +3228,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
             )
@@ -2765,9 +3255,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(
@@ -2787,7 +3277,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2805,7 +3295,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,
@@ -2813,18 +3303,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
             )
@@ -2833,9 +3325,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:
@@ -2849,7 +3341,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2862,23 +3354,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
             )
@@ -2887,9 +3381,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:
@@ -2907,7 +3401,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2920,23 +3414,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
             )
@@ -2945,9 +3441,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(
@@ -2963,7 +3459,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: dict[str, ~client.models.LimitedJobStatusReturn]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2978,23 +3474,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
             )
@@ -3003,9 +3501,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(
@@ -3039,7 +3537,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",
@@ -3052,7 +3550,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.
@@ -3067,7 +3565,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,
@@ -3078,18 +3576,15 @@ 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:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -3113,7 +3608,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,
@@ -3122,18 +3617,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
             )
@@ -3142,9 +3639,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(
@@ -3160,7 +3657,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: dict[str, list[~client.models.JobStatusReturn]]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -3173,23 +3670,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
             )
@@ -3198,6 +3697,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