1

video-gen-api/app/enums/generation_task.py

@@ -3,6 +3,8 @@ from enum import Enum
 
 class GenerationMode(str, Enum):
     CHATAPI_ASYNC = "chatapi_async"
+    CHATAPI_MAIN = "chatapi_main"
+    CHATAPI_CHILD = "chatapi_child"
     HOT_OPENING_REPLICATE = "hot_opening_replicate"
     SHOT_REPLICATE = "shot_replicate"
 
@@ -19,6 +21,15 @@ class ChatGenerationTaskStatus(str, Enum):
     FAILED = "failed"
 
 
+class ChatGenerationDisplayStatus(str, Enum):
+    PENDING = "pending"
+    GENERATING = "generating"
+    COMPLETED = "completed"
+    FAILED = "failed"
+    DOWNLOAD_FAILED = "download_failed"
+    DELETED = "deleted"
+
+
 class ChatGenerationPipelineStage(str, Enum):
     QUEUED = "queued"
     PREPARING = "preparing"
@@ -36,6 +47,31 @@ class ChatGenerationPipelineStage(str, Enum):
 
 
 class ChatGenerationTaskEventType(str, Enum):
+    TASK_CREATED = "TASK_CREATED"
+    IDEMPOTENCY_HIT = "IDEMPOTENCY_HIT"
+    BATCH_CREATE_START = "BATCH_CREATE_START"
+    BATCH_MAIN_CREATED = "BATCH_MAIN_CREATED"
+    BATCH_CHILDREN_CREATED = "BATCH_CHILDREN_CREATED"
+    BATCH_BILLING_SUCCESS = "BATCH_BILLING_SUCCESS"
+    BATCH_COMMIT_SUCCESS = "BATCH_COMMIT_SUCCESS"
+    CHILD_ENQUEUE_START = "CHILD_ENQUEUE_START"
+    CHILD_ENQUEUE_SUCCESS = "CHILD_ENQUEUE_SUCCESS"
+    CHILD_ENQUEUE_FAILED = "CHILD_ENQUEUE_FAILED"
+    IMAGE_MAIN_CLAIM_ACQUIRED = "IMAGE_MAIN_CLAIM_ACQUIRED"
+    IMAGE_MAIN_CLAIM_REJECTED = "IMAGE_MAIN_CLAIM_REJECTED"
+    IMAGE_MAIN_CLAIM_EXPIRED = "IMAGE_MAIN_CLAIM_EXPIRED"
+    IMAGE_BATCH_PROVIDER_START = "IMAGE_BATCH_PROVIDER_START"
+    IMAGE_BATCH_PROVIDER_SUCCESS = "IMAGE_BATCH_PROVIDER_SUCCESS"
+    IMAGE_BATCH_PROVIDER_FAILED = "IMAGE_BATCH_PROVIDER_FAILED"
+    IMAGE_BATCH_SPLIT_START = "IMAGE_BATCH_SPLIT_START"
+    IMAGE_BATCH_SPLIT_SUCCESS = "IMAGE_BATCH_SPLIT_SUCCESS"
+    IMAGE_BATCH_SPLIT_FAILED = "IMAGE_BATCH_SPLIT_FAILED"
+    MAIN_STATUS_AGGREGATED = "MAIN_STATUS_AGGREGATED"
+    CHILD_RESOURCE_DELETE_START = "CHILD_RESOURCE_DELETE_START"
+    CHILD_RESOURCE_DELETE_SUCCESS = "CHILD_RESOURCE_DELETE_SUCCESS"
+    BATCH_GROUP_DELETE_SUCCESS = "BATCH_GROUP_DELETE_SUCCESS"
+    BATCH_RECOVERY_RECONCILED = "BATCH_RECOVERY_RECONCILED"
+
     PROMPT_CONCAT_START = "PROMPT_CONCAT_START"
     PROMPT_CONCAT_SUCCESS = "PROMPT_CONCAT_SUCCESS"
 
@@ -87,8 +123,23 @@ class ChatGenerationTaskEventType(str, Enum):
     TASK_FAILED = "TASK_FAILED"
 
 
-ALLOWED_GENERATION_MODES = {
+CHAT_TOP_LEVEL_MODES = {
     GenerationMode.CHATAPI_ASYNC.value,
+    GenerationMode.CHATAPI_MAIN.value,
+}
+
+CHAT_RESOURCE_MODES = {
+    GenerationMode.CHATAPI_ASYNC.value,
+    GenerationMode.CHATAPI_CHILD.value,
+}
+
+CHAT_EXECUTABLE_MODES = {
+    GenerationMode.CHATAPI_ASYNC.value,
+    GenerationMode.CHATAPI_CHILD.value,
+}
+
+ALLOWED_GENERATION_MODES = {
+    *CHAT_EXECUTABLE_MODES,
     GenerationMode.HOT_OPENING_REPLICATE.value,
     GenerationMode.SHOT_REPLICATE.value,
 }