fix: Use also VO as parameter (in case DIRAC has a bad VO)

Author: Robin-Van-de-Merghel

diracx-client/src/diracx/client/patches/pilots/common.py

@@ -98,6 +98,7 @@ class AddPilotStampsBody(TypedDict, total=False):
     grid_site: str
     pilot_references: dict[str, str]
     pilot_status: PilotStatus
+    vo: str
 
 class AddPilotStampsKwargs(AddPilotStampsBody, ResponseExtra): ...
 
@@ -111,7 +112,7 @@ def make_add_pilot_stamps_body(**kwargs: Unpack[AddPilotStampsKwargs]) -> Underl
     for key in AddPilotStampsBody.__optional_keys__:
         if key not in kwargs:
             continue
-        key = cast(Literal["pilot_stamps", "grid_type", "grid_site", "pilot_references", "pilot_status"], key)
+        key = cast(Literal["pilot_stamps", "grid_type", "grid_site", "pilot_references", "pilot_status", "vo"], key)
         value = kwargs.pop(key)
         if value is not None:
             body[key] = value

diracx-routers/src/diracx/routers/pilots/management.py

@@ -55,6 +55,7 @@ async def add_pilot_stamps(
     pilot_status: Annotated[
         PilotStatus, Body(description="Status of the pilots.")
     ] = PilotStatus.SUBMITTED,
+    vo: str | None = None,
 ):
     """Endpoint where a you can create pilots with their references.
 
@@ -79,7 +80,7 @@ async def add_pilot_stamps(
         await register_new_pilots(
             pilot_db=pilot_db,
             pilot_stamps=pilot_stamps,
-            vo=user_info.vo,
+            vo=vo if vo else user_info.vo,
             grid_type=grid_type,
             grid_site=grid_site,
             destination_site=destination_site,