from enum import Enum


class CeleryQueue(str, Enum):
    GEN_CHATAPI_CREATE = "gen_chatapi_create"
    GEN_PROVIDER_POLL = "gen_provider_poll"
    GEN_RESULT_DOWNLOAD = "gen_result_download"
    GEN_RECOVERY = "gen_recovery"
    DEFAULT = "default"


class CeleryTaskName(str, Enum):
    CHATAPI_CREATE = "generation.chatapi_create_generation_task"
    POLL_GENERATION = "generation.poll_generation_task"
    DOWNLOAD_GENERATION_RESULT = "generation.download_generation_result_task"
    RECOVER_DOWNLOAD = "generation.recover_download_tasks_once"
    RECOVER_GENERATION = "generation.recover_generation_tasks_once"
    DISPATCH_DUE_POLL = "generation.dispatch_due_poll_tasks"
    STARTUP_RECOVERY = "recovery.startup_recovery_once"
    MODULE_ASYNC_RECOVERY = "module_async.recover_module_async_tasks_once"
    SHOT_SPLIT_RECOVERY = "shot_replicate.recover_split_tasks_once"