From 0519a498150fe5cb596c318cc55f44fc9cc34a7c Mon Sep 17 00:00:00 2001
From: Taha Yassine Kraiem <tahayk2@gmail.com>
Date: Mon, 23 Jan 2023 16:56:14 +0100
Subject: [PATCH] feat(chalice): fixed trail-schema

---
 ee/api/schemas_ee.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

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