merge main

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

@@ -7,4 +7,6 @@ from app.enums.user import *
 from app.enums.credit_record import *
 from app.enums.token_usage import *
 from app.enums.generation_task import *
-from app.enums.recent_generation import *
+from app.enums.recent_generation import *
+from app.enums.generation_status import *
+from app.enums.sms import *

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

@@ -0,0 +1,49 @@
+from enum import Enum
+
+
+class GenerationMode(str, Enum):
+    """生成模式。"""
+    STANDARD = "standard"
+    FAST = "fast"
+
+
+class GenerationType(str, Enum):
+    """生成类型。"""
+    video = "video"
+    image = "image"
+
+
+class ChatGenerationTaskStatus(str, Enum):
+    """聊天生成任务状态。"""
+    pending = "pending"
+    processing = "processing"
+    completed = "completed"
+    failed = "failed"
+
+
+class ChatGenerationPipelineStage(str, Enum):
+    """聊天生成任务阶段。"""
+    waiting = "waiting"
+    prompt_optimization = "prompt_optimization"
+    video_generation = "video_generation"
+    post_processing = "post_processing"
+    completed = "completed"
+    failed = "failed"
+
+
+class ChatGenerationTaskEventType(str, Enum):
+    """聊天生成任务事件类型。"""
+    TASK_CREATED = "TASK_CREATED"
+    TASK_DELETED = "TASK_DELETED"
+    TASK_CANCELLED = "TASK_CANCELLED"
+    PROMPT_OPT_STARTED = "PROMPT_OPT_STARTED"
+    PROMPT_OPT_COMPLETED = "PROMPT_OPT_COMPLETED"
+    PROMPT_OPT_FAILED = "PROMPT_OPT_FAILED"
+    VIDEO_GEN_STARTED = "VIDEO_GEN_STARTED"
+    VIDEO_GEN_COMPLETED = "VIDEO_GEN_COMPLETED"
+    VIDEO_GEN_FAILED = "VIDEO_GEN_FAILED"
+    POST_PROCESSING_STARTED = "POST_PROCESSING_STARTED"
+    POST_PROCESSING_COMPLETED = "POST_PROCESSING_COMPLETED"
+    POST_PROCESSING_FAILED = "POST_PROCESSING_FAILED"
+    TASK_COMPLETED = "TASK_COMPLETED"
+    TASK_FAILED = "TASK_FAILED"

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

@@ -0,0 +1,22 @@
+from enum import Enum
+
+
+class GenerationStatus(str, Enum):
+    """生成状态。"""
+    prompt_optimized = "prompt_optimized"
+    generating = "generating"
+    completed = "completed"
+    failed = "failed"
+
+
+class GenerationType(str, Enum):
+    """生成类型。"""
+    video = "video"
+    image = "image"
+
+
+# 生成配置常量
+DURATIONS = [4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
+ASPECT_RATIOS = ["16:9", "9:16", "1:1", "4:3", "3:4", "21:9"]
+RESOLUTIONS = ["480p", "720p", "1080p"]
+IMAGE_SIZES = ["2K", "4K"]

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

@@ -0,0 +1,9 @@
+from enum import Enum
+
+
+class SmsScene(str, Enum):
+    """短信场景。"""
+    register = "register"
+    login = "login"
+    common = "common"
+    set_password = "set_password"