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

diracx/client/operations/_operations.py

@@ -1,12 +1,23 @@
 # pylint: disable=too-many-lines,too-many-statements
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.13.2)
+# 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 (
@@ -344,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 {})
 
@@ -812,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,
@@ -865,13 +990,13 @@ class WellKnownOperations:
     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,
@@ -962,7 +1087,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1028,7 +1153,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: ~client.models.InitiateDeviceFlowResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1093,7 +1218,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1142,13 +1267,13 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
     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,
@@ -1202,7 +1327,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: list[any]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1258,7 +1383,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: str
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1312,7 +1437,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: ~client.models.UserInfoResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1370,7 +1495,13 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
     ) -> 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.Enum0
@@ -1390,7 +1521,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1446,7 +1577,13 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
     ) -> 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
@@ -1456,7 +1593,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1552,7 +1689,7 @@ class ConfigOperations:
         :rtype: any
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1646,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,
@@ -1762,7 +1899,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: ~client.models.SandboxUploadResponse
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1819,6 +1956,302 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
 
         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  # 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(
         self, body: List[str], *, content_type: str = "application/json", **kwargs: Any
@@ -1869,7 +2302,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: list[~client.models.InsertedJob]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -1938,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,
@@ -1994,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,
@@ -2054,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,
@@ -2112,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,
@@ -2228,7 +2661,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :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,
@@ -2300,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,
@@ -2356,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,
@@ -2412,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,
@@ -2541,7 +2974,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :rtype: list[JSON]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
-        error_map = {
+        error_map: MutableMapping[int, Type[HttpResponseError]] = {
             401: ClientAuthenticationError,
             404: ResourceNotFoundError,
             409: ResourceExistsError,
@@ -2657,7 +3090,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,
@@ -2726,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,
@@ -2782,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,
@@ -2844,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,
@@ -2908,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,
@@ -2968,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,
@@ -3026,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,
@@ -3151,7 +3584,7 @@ class JobsOperations:  # pylint: disable=too-many-public-methods
         :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,
@@ -3224,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,