49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
from enum import Enum
|
|
|
|
|
|
class GenerationStatus(str, Enum):
|
|
"""生成状态。"""
|
|
prompt_optimized = "prompt_optimized"
|
|
generating = "generating"
|
|
completed = "completed"
|
|
failed = "failed"
|
|
|
|
|
|
class GenerationRecordPipelineStage(str, Enum):
|
|
"""GenerationRecord 图片/视频生成、下载与超分流水线阶段。"""
|
|
|
|
QUEUED = "queued"
|
|
PREPARING = "preparing"
|
|
CREATING_PROVIDER_TASK = "creating_provider_task"
|
|
PROVIDER_RESULT_STAGED = "provider_result_staged"
|
|
WAITING_REMOTE = "waiting_remote"
|
|
POLLING = "polling"
|
|
RESULT_READY = "result_ready"
|
|
DOWNLOAD_QUEUED = "download_queued"
|
|
DOWNLOADING = "downloading"
|
|
RETRY_WAITING = "retry_waiting"
|
|
RECOVERY_INCONSISTENT = "recovery_inconsistent"
|
|
UPSCALE_QUEUED = "upscale_queued"
|
|
UPSCALE_PROCESSING = "upscale_processing"
|
|
UPSCALE_POLLING = "upscale_polling"
|
|
UPSCALE_DOWNLOADING = "upscale_downloading"
|
|
UPSCALE_FINALIZING = "upscale_finalizing"
|
|
UPSCALE_RETRY_WAITING = "upscale_retry_waiting"
|
|
UPSCALE_FAILED = "upscale_failed"
|
|
DONE = "done"
|
|
FAILED = "failed"
|
|
TIMEOUT = "timeout"
|
|
DOWNLOAD_FAILED = "download_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 = ["1K", "2K", "4K"] |