long_articles_agentic_supply/supply/domain/task/config.py

"""Task 配置与状态常量 —— 从 supply.jobs.task_config 迁移"""

from dataclasses import dataclass


@dataclass
class TaskConfig:
    """单个任务的运行时配置"""

    timeout: int  # 超时时间（秒）
    max_concurrent: int = 5  # 最大并发数
    retry_times: int = 0  # 重试次数
    retryable: bool = True  # 是否可重试
    alert_on_failure: bool = True  # 失败时是否告警


class TaskStatus:
    """任务状态常量"""

    INIT = 0
    PROCESSING = 1
    SUCCESS = 2
    FAILED = 99


class TaskConstants:
    """任务系统常量"""

    DEFAULT_TIMEOUT = 1800
    DEFAULT_MAX_CONCURRENT = 5
    DEFAULT_RETRY_TIMES = 0


def get_default_task_config() -> TaskConfig:
    """获取默认任务配置"""
    return TaskConfig(
        timeout=TaskConstants.DEFAULT_TIMEOUT,
        max_concurrent=TaskConstants.DEFAULT_MAX_CONCURRENT,
        retry_times=TaskConstants.DEFAULT_RETRY_TIMES,
    )