From 7013635988fe4aa9483e0b0cd7c737a9b729437d Mon Sep 17 00:00:00 2001 From: ITQ Date: Sun, 16 Feb 2025 12:29:44 +0300 Subject: [PATCH] chore: improved type annotaions for Promise --- .../services/backend/api/v1/generate/schemas.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) 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