43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
|
"""
|
||
|
|
自定义异常类
|
||
|
|
用于任务取消和中断处理
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
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)
|