diff --git a/ee/api/schemas_ee.py b/ee/api/schemas_ee.py index ceae7cfc5..60d92817b 100644 --- a/ee/api/schemas_ee.py +++ b/ee/api/schemas_ee.py @@ -79,7 +79,15 @@ class TrailSearchPayloadSchema(schemas._PaginatedSchema): user_id: Optional[int] = Field(default=None) query: Optional[str] = Field(default=None) action: Optional[str] = Field(default=None) - order: Literal["asc", "desc"] = Field(default="desc") + order: schemas.SortOrderType = Field(default=schemas.SortOrderType.desc) + + @root_validator(pre=True) + def transform_order(cls, values): + if values.get("order") is None: + values["order"] = schemas.SortOrderType.desc + else: + values["order"] = values["order"].upper() + return values class Config: alias_generator = schemas.attribute_to_camel_case