diff --git a/solution/services/backend/api/v1/generate/schemas.py b/solution/services/backend/api/v1/generate/schemas.py
index 66326e2..a5abfad 100644
--- a/solution/services/backend/api/v1/generate/schemas.py
+++ b/solution/services/backend/api/v1/generate/schemas.py
@@ -1,4 +1,5 @@
-from typing import Any
+from typing import Any, Literal
+from uuid import UUID
 
 from ninja import Schema
 
@@ -9,6 +10,14 @@ class GenerateAdTextIn(Schema):
 
 
 class Promise(Schema):
-    task_id: str
-    status: str
+    task_id: UUID
+    status: Literal[
+        "PENDING",
+        "RECEIVED",
+        "STARTED",
+        "SUCCESS",
+        "FAILURE",
+        "RETRY",
+        "REVOKED",
+    ]
     result: Any