fix: Fixed dependency and client

Author: Robin-Van-de-Merghel

diracx-client/src/diracx/client/_generated/aio/operations/_operations.py

@@ -2775,52 +2775,14 @@ async def add_pilot_stamps(
 
         return deserialized  # type: ignore
 
-    @overload
-    async def delete_pilots(
-        self, body: List[str], *, content_type: str = "application/json", **kwargs: Any
-    ) -> None:
-        """Delete Pilots.
-
-        Endpoint to delete a pilot.
-
-        :param body: Required.
-        :type body: list[str]
-        :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 delete_pilots(
-        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
-    ) -> None:
-        """Delete Pilots.
-
-        Endpoint to delete a pilot.
-
-        :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 delete_pilots(
-        self, body: Union[List[str], IO[bytes]], **kwargs: Any
-    ) -> None:
+    async def delete_pilots(self, *, pilot_stamps: List[str], **kwargs: Any) -> None:
         """Delete Pilots.
 
         Endpoint to delete a pilot.
 
-        :param body: Is either a [str] type or a IO[bytes] type. Required.
-        :type body: list[str] or IO[bytes]
+        :keyword pilot_stamps: Stamps of the pilots we want to delete. Required.
+        :paramtype pilot_stamps: list[str]
         :return: None
         :rtype: None
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -2833,26 +2795,13 @@ async def delete_pilots(
         }
         error_map.update(kwargs.pop("error_map", {}) or {})
 
-        _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+        _headers = 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, "[str]")
-
         _request = build_pilots_delete_pilots_request(
-            content_type=content_type,
-            json=_json,
-            content=_content,
+            pilot_stamps=pilot_stamps,
             headers=_headers,
             params=_params,
         )
@@ -2876,56 +2825,21 @@ async def delete_pilots(
         if cls:
             return cls(pipeline_response, None, {})  # type: ignore
 
-    @overload
-    async def clear_pilots(
-        self,
-        body: _models.BodyPilotsClearPilots,
-        *,
-        content_type: str = "application/json",
-        **kwargs: Any,
-    ) -> None:
-        """Clear Pilots.
-
-        Delete all pilots that lived more than age_in_days.
-
-        :param body: Required.
-        :type body: ~_generated.models.BodyPilotsClearPilots
-        :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 clear_pilots(
-        self, body: IO[bytes], *, content_type: str = "application/json", **kwargs: Any
-    ) -> None:
-        """Clear Pilots.
-
-        Delete all pilots that lived more than age_in_days.
-
-        :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 clear_pilots(
-        self, body: Union[_models.BodyPilotsClearPilots, IO[bytes]], **kwargs: Any
+        self, *, age_in_days: int, delete_only_aborted: bool = True, **kwargs: Any
     ) -> None:
         """Clear Pilots.
 
         Delete all pilots that lived more than age_in_days.
 
-        :param body: Is either a BodyPilotsClearPilots type or a IO[bytes] type. Required.
-        :type body: ~_generated.models.BodyPilotsClearPilots or IO[bytes]
+        :keyword age_in_days: The number of days that define the maximum age of pilots to be
+         deleted.Pilots older than this age will be considered for deletion. Required.
+        :paramtype age_in_days: int
+        :keyword delete_only_aborted: Flag indicating whether to only delete pilots whose status is
+         'Aborted'.If set to True, only pilots with the 'Aborted' status will be deleted.It is set by
+         default as True to avoid any mistake. Default value is True.
+        :paramtype delete_only_aborted: bool
         :return: None
         :rtype: None
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -2938,26 +2852,14 @@ async def clear_pilots(
         }
         error_map.update(kwargs.pop("error_map", {}) or {})
 
-        _headers = case_insensitive_dict(kwargs.pop("headers", {}) or {})
+        _headers = 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, "BodyPilotsClearPilots")
-
         _request = build_pilots_clear_pilots_request(
-            content_type=content_type,
-            json=_json,
-            content=_content,
+            age_in_days=age_in_days,
+            delete_only_aborted=delete_only_aborted,
             headers=_headers,
             params=_params,
         )
@@ -3350,7 +3252,7 @@ async def associate_pilots_with_secrets(
     @overload
     async def update_pilot_fields(
         self,
-        body: List[_models.PilotFieldsMapping],
+        body: _models.BodyPilotsUpdatePilotFields,
         *,
         content_type: str = "application/json",
         **kwargs: Any,
@@ -3360,7 +3262,7 @@ async def update_pilot_fields(
         Modify a field of a pilot.
 
         :param body: Required.
-        :type body: list[~_generated.models.PilotFieldsMapping]
+        :type body: ~_generated.models.BodyPilotsUpdatePilotFields
         :keyword content_type: Body Parameter content-type. Content type parameter for JSON body.
          Default value is "application/json".
         :paramtype content_type: str
@@ -3389,14 +3291,14 @@ async def update_pilot_fields(
 
     @distributed_trace_async
     async def update_pilot_fields(
-        self, body: Union[List[_models.PilotFieldsMapping], IO[bytes]], **kwargs: Any
+        self, body: Union[_models.BodyPilotsUpdatePilotFields, IO[bytes]], **kwargs: Any
     ) -> None:
         """Update Pilot Fields.
 
         Modify a field of a pilot.
 
-        :param body: Is either a [PilotFieldsMapping] type or a IO[bytes] type. Required.
-        :type body: list[~_generated.models.PilotFieldsMapping] or IO[bytes]
+        :param body: Is either a BodyPilotsUpdatePilotFields type or a IO[bytes] type. Required.
+        :type body: ~_generated.models.BodyPilotsUpdatePilotFields or IO[bytes]
         :return: None
         :rtype: None
         :raises ~azure.core.exceptions.HttpResponseError:
@@ -3423,7 +3325,7 @@ async def update_pilot_fields(
         if isinstance(body, (IOBase, bytes)):
             _content = body
         else:
-            _json = self._serialize.body(body, "[PilotFieldsMapping]")
+            _json = self._serialize.body(body, "BodyPilotsUpdatePilotFields")
 
         _request = build_pilots_update_pilot_fields_request(
             content_type=content_type,

diracx-client/src/diracx/client/_generated/models/__init__.py

@@ -19,8 +19,8 @@
     BodyPilotsAddPilotStamps,
     BodyPilotsAssociatePilotWithJobs,
     BodyPilotsAssociatePilotsWithSecrets,
-    BodyPilotsClearPilots,
     BodyPilotsCreatePilotSecrets,
+    BodyPilotsUpdatePilotFields,
     GroupInfo,
     HTTPValidationError,
     HeartbeatData,
@@ -79,8 +79,8 @@
     "BodyPilotsAddPilotStamps",
     "BodyPilotsAssociatePilotWithJobs",
     "BodyPilotsAssociatePilotsWithSecrets",
-    "BodyPilotsClearPilots",
     "BodyPilotsCreatePilotSecrets",
+    "BodyPilotsUpdatePilotFields",
     "GroupInfo",
     "HTTPValidationError",
     "HeartbeatData",

diracx-client/src/diracx/client/_generated/models/_models.py

@@ -295,46 +295,6 @@ def __init__(
         self.pilot_jobs_ids = pilot_jobs_ids
 
 
-class BodyPilotsClearPilots(_serialization.Model):
-    """Body_pilots_clear_pilots.
-
-    All required parameters must be populated in order to send to server.
-
-    :ivar age_in_days: The number of days that define the maximum age of pilots to be
-     deleted.Pilots older than this age will be considered for deletion. Required.
-    :vartype age_in_days: int
-    :ivar delete_only_aborted: Flag indicating whether to only delete pilots whose status is
-     'Aborted'.If set to True, only pilots with the 'Aborted' status will be deleted.It is set by
-     default as True to avoid any mistake.
-    :vartype delete_only_aborted: bool
-    """
-
-    _validation = {
-        "age_in_days": {"required": True},
-    }
-
-    _attribute_map = {
-        "age_in_days": {"key": "age_in_days", "type": "int"},
-        "delete_only_aborted": {"key": "delete_only_aborted", "type": "bool"},
-    }
-
-    def __init__(
-        self, *, age_in_days: int, delete_only_aborted: bool = True, **kwargs: Any
-    ) -> None:
-        """
-        :keyword age_in_days: The number of days that define the maximum age of pilots to be
-         deleted.Pilots older than this age will be considered for deletion. Required.
-        :paramtype age_in_days: int
-        :keyword delete_only_aborted: Flag indicating whether to only delete pilots whose status is
-         'Aborted'.If set to True, only pilots with the 'Aborted' status will be deleted.It is set by
-         default as True to avoid any mistake.
-        :paramtype delete_only_aborted: bool
-        """
-        super().__init__(**kwargs)
-        self.age_in_days = age_in_days
-        self.delete_only_aborted = delete_only_aborted
-
-
 class BodyPilotsCreatePilotSecrets(_serialization.Model):
     """Body_pilots_create_pilot_secrets.
 
@@ -393,6 +353,41 @@ def __init__(
         self.pilot_secret_use_count_max = pilot_secret_use_count_max
 
 
+class BodyPilotsUpdatePilotFields(_serialization.Model):
+    """Body_pilots_update_pilot_fields.
+
+    All required parameters must be populated in order to send to server.
+
+    :ivar pilot_stamps_to_fields_mapping: (pilot_stamp, pilot_fields) mapping to change. Required.
+    :vartype pilot_stamps_to_fields_mapping: list[~_generated.models.PilotFieldsMapping]
+    """
+
+    _validation = {
+        "pilot_stamps_to_fields_mapping": {"required": True},
+    }
+
+    _attribute_map = {
+        "pilot_stamps_to_fields_mapping": {
+            "key": "pilot_stamps_to_fields_mapping",
+            "type": "[PilotFieldsMapping]",
+        },
+    }
+
+    def __init__(
+        self,
+        *,
+        pilot_stamps_to_fields_mapping: List["_models.PilotFieldsMapping"],
+        **kwargs: Any,
+    ) -> None:
+        """
+        :keyword pilot_stamps_to_fields_mapping: (pilot_stamp, pilot_fields) mapping to change.
+         Required.
+        :paramtype pilot_stamps_to_fields_mapping: list[~_generated.models.PilotFieldsMapping]
+        """
+        super().__init__(**kwargs)
+        self.pilot_stamps_to_fields_mapping = pilot_stamps_to_fields_mapping
+
+
 class GroupInfo(_serialization.Model):
     """GroupInfo.