diracx-client 0.0.1a28__py3-none-any.whl → 0.0.1a31__py3-none-any.whl

diracx/client/generated/__init__.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 # pylint: disable=wrong-import-position

diracx/client/generated/_client.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 

diracx/client/generated/_configuration.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 

diracx/client/generated/_patch.py

@@ -8,27 +8,7 @@ Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python
 """
 from __future__ import annotations
 
-import os
-from datetime import datetime, timezone
-import importlib.util
-import json
-import jwt
-import httpx
-
-from pathlib import Path
-from typing import Any, Dict, List, Optional, cast
-from urllib import parse
-from azure.core.credentials import AccessToken
-from azure.core.credentials import TokenCredential
-from azure.core.pipeline import PipelineRequest
-from azure.core.pipeline.policies import BearerTokenCredentialPolicy
-
-from diracx.core.preferences import DiracxPreferences, get_diracx_preferences
-
-
-import sys
-import importlib
-from importlib.abc import MetaPathFinder, Loader
+from typing import List
 
 __all__: List[str] = [
     "DiracClient",

diracx/client/generated/_serialization.py

@@ -1,4 +1,4 @@
-# pylint: disable=too-many-lines
+# pylint: disable=line-too-long,useless-suppression,too-many-lines
 # --------------------------------------------------------------------------
 #
 # Copyright (c) Microsoft Corporation. All rights reserved.
@@ -437,7 +437,7 @@ class Model:
         :param function key_extractors: A key extractor function.
         :param str content_type: JSON by default, set application/xml if XML.
         :returns: An instance of this model
-        :raises: DeserializationError if something went wrong
+        :raises DeserializationError: if something went wrong
         :rtype: Self
         """
         deserializer = Deserializer(cls._infer_class_models())
@@ -1452,7 +1452,7 @@ def xml_key_extractor(
         # Iter and wrapped, should have found one node only (the wrap one)
         if len(children) != 1:
             raise DeserializationError(
-                "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format(  # pylint: disable=line-too-long
+                "Tried to deserialize an array not wrapped, and found several nodes '{}'. Maybe you should declare this array as wrapped?".format(
                     xml_name
                 )
             )

diracx/client/generated/_vendor.py

@@ -1,5 +1,5 @@
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 

diracx/client/generated/aio/__init__.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 # pylint: disable=wrong-import-position

diracx/client/generated/aio/_client.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 

diracx/client/generated/aio/_configuration.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 

diracx/client/generated/aio/_vendor.py

@@ -1,5 +1,5 @@
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 

diracx/client/generated/aio/operations/__init__.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 # pylint: disable=wrong-import-position

diracx/client/generated/aio/operations/_operations.py

@@ -1,7 +1,7 @@
 # pylint: disable=too-many-lines
 # coding=utf-8
 # --------------------------------------------------------------------------
-# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.2, generator: @autorest/python@6.28.3)
+# Code generated by Microsoft (R) AutoRest Code Generator (autorest: 3.10.4, generator: @autorest/python@6.31.0)
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 from io import IOBase
@@ -26,12 +26,12 @@ from azure.core.utils import case_insensitive_dict
 from ... import models as _models
 from ..._serialization import Deserializer, Serializer
 from ...operations._operations import (
-    build_auth_authorization_flow_complete_request,
-    build_auth_authorization_flow_request,
+    build_auth_complete_authorization_flow_request,
     build_auth_do_device_flow_request,
     build_auth_finish_device_flow_request,
     build_auth_finished_request,
     build_auth_get_refresh_tokens_request,
+    build_auth_initiate_authorization_flow_request,
     build_auth_initiate_device_flow_request,
     build_auth_revoke_refresh_token_request,
     build_auth_userinfo_request,
@@ -41,16 +41,17 @@ from ...operations._operations import (
     build_jobs_get_job_sandboxes_request,
     build_jobs_get_sandbox_file_request,
     build_jobs_initiate_sandbox_upload_request,
-    build_jobs_remove_bulk_jobs_request,
-    build_jobs_reschedule_bulk_jobs_request,
+    build_jobs_patch_metadata_request,
+    build_jobs_remove_jobs_request,
+    build_jobs_reschedule_jobs_request,
     build_jobs_search_request,
     build_jobs_set_job_statuses_request,
-    build_jobs_submit_bulk_jdl_jobs_request,
+    build_jobs_submit_jdl_jobs_request,
     build_jobs_summary_request,
     build_jobs_unassign_bulk_jobs_sandboxes_request,
     build_jobs_unassign_job_sandboxes_request,
-    build_well_known_installation_metadata_request,
-    build_well_known_openid_configuration_request,
+    build_well_known_get_installation_metadata_request,
+    build_well_known_get_openid_configuration_request,
 )
 from .._configuration import DiracConfiguration
 from .._vendor import raise_if_not_implemented
@@ -63,7 +64,6 @@ T = TypeVar("T")
 ClsType = Optional[
     Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]
 ]
-JSON = MutableMapping[str, Any]  # pylint: disable=unsubscriptable-object
 
 
 class WellKnownOperations:
@@ -94,13 +94,15 @@ class WellKnownOperations:
         )
 
     @distributed_trace_async
-    async def openid_configuration(self, **kwargs: Any) -> Any:
-        """Openid Configuration.
+    async def get_openid_configuration(
+        self, **kwargs: Any
+    ) -> _models.OpenIDConfiguration:
+        """Get Openid Configuration.
 
         OpenID Connect discovery endpoint.
 
-        :return: any
-        :rtype: any
+        :return: OpenIDConfiguration
+        :rtype: ~generated.models.OpenIDConfiguration
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         error_map: MutableMapping = {
@@ -114,9 +116,9 @@ class WellKnownOperations:
         _headers = kwargs.pop("headers", {}) or {}
         _params = kwargs.pop("params", {}) or {}
 
-        cls: ClsType[Any] = kwargs.pop("cls", None)
+        cls: ClsType[_models.OpenIDConfiguration] = kwargs.pop("cls", None)
 
-        _request = build_well_known_openid_configuration_request(
+        _request = build_well_known_get_openid_configuration_request(
             headers=_headers,
             params=_params,
         )
@@ -137,7 +139,9 @@ class WellKnownOperations:
             )
             raise HttpResponseError(response=response)
 
-        deserialized = self._deserialize("object", pipeline_response.http_response)
+        deserialized = self._deserialize(
+            "OpenIDConfiguration", pipeline_response.http_response
+        )
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
@@ -145,8 +149,8 @@ class WellKnownOperations:
         return deserialized  # type: ignore
 
     @distributed_trace_async
-    async def installation_metadata(self, **kwargs: Any) -> _models.Metadata:
-        """Installation Metadata.
+    async def get_installation_metadata(self, **kwargs: Any) -> _models.Metadata:
+        """Get Installation Metadata.
 
         Get metadata about the dirac installation.
 
@@ -167,7 +171,7 @@ class WellKnownOperations:
 
         cls: ClsType[_models.Metadata] = kwargs.pop("cls", None)
 
-        _request = build_well_known_installation_metadata_request(
+        _request = build_well_known_get_installation_metadata_request(
             headers=_headers,
             params=_params,
         )
@@ -226,7 +230,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         raise_if_not_implemented(
             self.__class__,
             [
-                "token",
+                "get_oidc_token",
             ],
         )
 
@@ -647,7 +651,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         return deserialized  # type: ignore
 
     @distributed_trace_async
-    async def authorization_flow(
+    async def initiate_authorization_flow(
         self,
         *,
         response_type: str,
@@ -659,7 +663,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         state: str,
         **kwargs: Any,
     ) -> Any:
-        """Authorization Flow.
+        """Initiate Authorization Flow.
 
         Initiate the authorization flow.
         It will redirect to the actual OpenID server (IAM, CheckIn) to
@@ -719,7 +723,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        _request = build_auth_authorization_flow_request(
+        _request = build_auth_initiate_authorization_flow_request(
             response_type=response_type,
             code_challenge=code_challenge,
             code_challenge_method=code_challenge_method,
@@ -755,10 +759,10 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
         return deserialized  # type: ignore
 
     @distributed_trace_async
-    async def authorization_flow_complete(
+    async def complete_authorization_flow(
         self, *, code: str, state: str, **kwargs: Any
     ) -> Any:
-        """Authorization Flow Complete.
+        """Complete Authorization Flow.
 
         Complete the authorization flow.
 
@@ -789,7 +793,7 @@ class AuthOperations:  # pylint: disable=abstract-class-instantiated
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        _request = build_auth_authorization_flow_complete_request(
+        _request = build_auth_complete_authorization_flow_request(
             code=code,
             state=state,
             headers=_headers,
@@ -1436,8 +1440,8 @@ class JobsOperations:
         return deserialized  # type: ignore
 
     @distributed_trace_async
-    async def remove_bulk_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
-        """Remove Bulk Jobs.
+    async def remove_jobs(self, *, job_ids: List[int], **kwargs: Any) -> Any:
+        """Remove Jobs.
 
         Fully remove a list of jobs from the WMS databases.
 
@@ -1465,7 +1469,7 @@ class JobsOperations:
 
         cls: ClsType[Any] = kwargs.pop("cls", None)
 
-        _request = build_jobs_remove_bulk_jobs_request(
+        _request = build_jobs_remove_jobs_request(
             job_ids=job_ids,
             headers=_headers,
             params=_params,
@@ -1623,19 +1627,19 @@ class JobsOperations:
         return deserialized  # type: ignore
 
     @distributed_trace_async
-    async def reschedule_bulk_jobs(
+    async def reschedule_jobs(
         self, *, job_ids: List[int], reset_jobs: bool = False, **kwargs: Any
-    ) -> Any:
-        """Reschedule Bulk Jobs.
+    ) -> Dict[str, Any]:
+        """Reschedule Jobs.
 
-        Reschedule Bulk Jobs.
+        Reschedule Jobs.
 
         :keyword job_ids: Required.
         :paramtype job_ids: list[int]
         :keyword reset_jobs: Default value is False.
         :paramtype reset_jobs: bool
-        :return: any
-        :rtype: any
+        :return: dict mapping str to any
+        :rtype: dict[str, any]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         error_map: MutableMapping = {
@@ -1649,9 +1653,9 @@ class JobsOperations:
         _headers = kwargs.pop("headers", {}) or {}
         _params = kwargs.pop("params", {}) or {}
 
-        cls: ClsType[Any] = kwargs.pop("cls", None)
+        cls: ClsType[Dict[str, Any]] = kwargs.pop("cls", None)
 
-        _request = build_jobs_reschedule_bulk_jobs_request(
+        _request = build_jobs_reschedule_jobs_request(
             job_ids=job_ids,
             reset_jobs=reset_jobs,
             headers=_headers,
@@ -1674,13 +1678,118 @@ class JobsOperations:
             )
             raise HttpResponseError(response=response)
 
-        deserialized = self._deserialize("object", pipeline_response.http_response)
+        deserialized = self._deserialize("{object}", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, {})  # type: ignore
 
         return deserialized  # type: ignore
 
+    @overload
+    async def patch_metadata(
+        self,
+        body: Dict[str, Dict[str, Any]],
+        *,
+        content_type: str = "application/json",
+        **kwargs: Any,
+    ) -> None:
+        """Patch Metadata.
+
+        Patch Metadata.
+
+        :param body: Required.
+        :type body: dict[str, dict[str, any]]
+        :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
+         Default value is "application/json".
+        :paramtype content_type: str
+        :return: None
+        :rtype: None
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @overload
+    async def patch_metadata(
+        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
+    ) -> None:
+        """Patch Metadata.
+
+        Patch Metadata.
+
+        :param body: Required.
+        :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
+        :return: None
+        :rtype: None
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+
+    @distributed_trace_async
+    async def patch_metadata(
+        self, body: Union[Dict[str, Dict[str, Any]], IO[bytes]], **kwargs: Any
+    ) -> None:
+        """Patch Metadata.
+
+        Patch Metadata.
+
+        :param body: Is either a {str: {str: Any}} type or a IO[bytes] type. Required.
+        :type body: dict[str, dict[str, any]] or IO[bytes]
+        :return: None
+        :rtype: None
+        :raises ~azure.core.exceptions.HttpResponseError:
+        """
+        error_map: MutableMapping = {
+            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: Optional[str] = kwargs.pop(
+            "content_type", _headers.pop("Content-Type", None)
+        )
+        cls: ClsType[None] = kwargs.pop("cls", None)
+
+        content_type = content_type or "application/json"
+        _json = None
+        _content = None
+        if isinstance(body, (IOBase, bytes)):
+            _content = body
+        else:
+            _json = self._serialize.body(body, "{{object}}")
+
+        _request = build_jobs_patch_metadata_request(
+            content_type=content_type,
+            json=_json,
+            content=_content,
+            headers=_headers,
+            params=_params,
+        )
+        _request.url = self._client.format_url(_request.url)
+
+        _stream = False
+        pipeline_response: PipelineResponse = (
+            await self._client._pipeline.run(  # pylint: disable=protected-access
+                _request, stream=_stream, **kwargs
+            )
+        )
+
+        response = pipeline_response.http_response
+
+        if response.status_code not in [204]:
+            map_error(
+                status_code=response.status_code, response=response, error_map=error_map
+            )
+            raise HttpResponseError(response=response)
+
+        if cls:
+            return cls(pipeline_response, None, {})  # type: ignore
+
     @overload
     async def search(
         self,
@@ -1690,7 +1799,7 @@ class JobsOperations:
         per_page: int = 100,
         content_type: str = "application/json",
         **kwargs: Any,
-    ) -> List[JSON]:
+    ) -> List[Dict[str, Any]]:
         """Search.
 
         Retrieve information about jobs.
@@ -1706,8 +1815,8 @@ class JobsOperations:
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
-        :return: list of JSON
-        :rtype: list[JSON]
+        :return: list of dict mapping str to any
+        :rtype: list[dict[str, any]]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -1720,7 +1829,7 @@ class JobsOperations:
         per_page: int = 100,
         content_type: str = "application/json",
         **kwargs: Any,
-    ) -> List[JSON]:
+    ) -> List[Dict[str, Any]]:
         """Search.
 
         Retrieve information about jobs.
@@ -1736,8 +1845,8 @@ class JobsOperations:
         :keyword content_type: Body Parameter content-type. Content type parameter for binary body.
          Default value is "application/json".
         :paramtype content_type: str
-        :return: list of JSON
-        :rtype: list[JSON]
+        :return: list of dict mapping str to any
+        :rtype: list[dict[str, any]]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
 
@@ -1749,7 +1858,7 @@ class JobsOperations:
         page: int = 1,
         per_page: int = 100,
         **kwargs: Any,
-    ) -> List[JSON]:
+    ) -> List[Dict[str, Any]]:
         """Search.
 
         Retrieve information about jobs.
@@ -1762,8 +1871,8 @@ class JobsOperations:
         :paramtype page: int
         :keyword per_page: Default value is 100.
         :paramtype per_page: int
-        :return: list of JSON
-        :rtype: list[JSON]
+        :return: list of dict mapping str to any
+        :rtype: list[dict[str, any]]
         :raises ~azure.core.exceptions.HttpResponseError:
         """
         error_map: MutableMapping = {
@@ -1780,7 +1889,7 @@ class JobsOperations:
         content_type: Optional[str] = kwargs.pop(
             "content_type", _headers.pop("Content-Type", None)
         )
-        cls: ClsType[List[JSON]] = kwargs.pop("cls", None)
+        cls: ClsType[List[Dict[str, Any]]] = kwargs.pop("cls", None)
 
         content_type = content_type or "application/json"
         _json = None
@@ -1825,7 +1934,7 @@ class JobsOperations:
                 "str", response.headers.get("Content-Range")
             )
 
-        deserialized = self._deserialize("[object]", pipeline_response.http_response)
+        deserialized = self._deserialize("[{object}]", pipeline_response.http_response)
 
         if cls:
             return cls(pipeline_response, deserialized, response_headers)  # type: ignore
@@ -1942,12 +2051,12 @@ class JobsOperations:
         return deserialized  # type: ignore
 
     @overload
-    async def submit_bulk_jdl_jobs(
+    async def submit_jdl_jobs(
         self, body: List[str], *, content_type: str = "application/json", **kwargs: Any
     ) -> List[_models.InsertedJob]:
-        """Submit Bulk Jdl Jobs.
+        """Submit Jdl Jobs.
 
-        Submit Bulk Jdl Jobs.
+        Submit a list of jobs in JDL format.
 
         :param body: Required.
         :type body: list[str]
@@ -1960,12 +2069,12 @@ class JobsOperations:
         """
 
     @overload
-    async def submit_bulk_jdl_jobs(
+    async def submit_jdl_jobs(
         self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
     ) -> List[_models.InsertedJob]:
-        """Submit Bulk Jdl Jobs.
+        """Submit Jdl Jobs.
 
-        Submit Bulk Jdl Jobs.
+        Submit a list of jobs in JDL format.
 
         :param body: Required.
         :type body: IO[bytes]
@@ -1978,12 +2087,12 @@ class JobsOperations:
         """
 
     @distributed_trace_async
-    async def submit_bulk_jdl_jobs(
+    async def submit_jdl_jobs(
         self, body: Union[List[str], IO[bytes]], **kwargs: Any
     ) -> List[_models.InsertedJob]:
-        """Submit Bulk Jdl Jobs.
+        """Submit Jdl Jobs.
 
-        Submit Bulk Jdl Jobs.
+        Submit a list of jobs in JDL format.
 
         :param body: Is either a [str] type or a IO[bytes] type. Required.
         :type body: list[str] or IO[bytes]
@@ -2015,7 +2124,7 @@ class JobsOperations:
         else:
             _json = self._serialize.body(body, "[str]")
 
-        _request = build_jobs_submit_bulk_jdl_jobs_request(
+        _request = build_jobs_submit_jdl_jobs_request(
             content_type=content_type,
             json=_json,
             content=_content,

diracx/client/generated/aio/operations/_patch.py

@@ -8,7 +8,7 @@ Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python
 """
 import io
 import json
-from typing import Any, List
+from typing import Any, Dict, List
 
 from azure.core.rest import HttpRequest
 from azure.core.exceptions import map_error, HttpResponseError
@@ -21,7 +21,6 @@ from ._operations import (
     AuthOperations as AuthOperationsGenerated,
     JobsOperations as JobsOperationsGenerated,
     _models,
-    JSON,
 )
 
 __all__: List[str] = [
@@ -54,7 +53,7 @@ def build_token_request(**kwargs: Any) -> HttpRequest:
 
 class AuthOperations(AuthOperationsGenerated):
     @distributed_trace_async
-    async def token(
+    async def get_oidc_token(
         self, device_code: str, client_id: str, **kwargs
     ) -> _models.TokenResponse | _models.DeviceFlowErrorResponse:
         request = build_token_request(
@@ -93,7 +92,7 @@ class JobsOperations(JobsOperationsGenerated):
         search: list[str] | None = None,
         sort: list[str] | None = None,
         **kwargs: Any,
-    ) -> List[JSON]:
+    ) -> List[Dict[str, Any]]:
         """TODO"""
         body = {}
         if parameters is not None:
@@ -114,7 +113,7 @@ class JobsOperations(JobsOperationsGenerated):
         grouping: list[str] | None = None,
         search: list[str] | None = None,
         **kwargs: Any,
-    ) -> List[JSON]:
+    ) -> List[Dict[str, Any]]:
         """TODO"""
         body = {}
         if grouping is not None: