"""
自定义异常类
用于任务取消和中断处理
"""


class TaskCancelledException(Exception):
    """任务被取消异常"""

    def __init__(self, task_id: str = None, message: str = None):
        self.task_id = task_id
        if message is None:
            if task_id:
                message = f"任务 {task_id[:8]}... 被用户取消"
            else:
                message = "任务被用户取消"
        super().__init__(message)


class RetrievalInterruptedException(Exception):
    """检索过程被中断异常"""

    def __init__(self, stage: str = None, message: str = None):
        self.stage = stage
        if message is None:
            if stage:
                message = f"检索在 {stage} 阶段被中断"
            else:
                message = "检索过程被中断"
        super().__init__(message)


class WorkflowStoppedException(Exception):
    """工作流停止异常"""

    def __init__(self, node_name: str = None, message: str = None):
        self.node_name = node_name
        if message is None:
            if node_name:
                message = f"工作流在节点 {node_name} 处停止"
            else:
                message = "工作流被停止"
        super().__init__(message)