first commit

AIEC-RAG/exceptions.py

@@ -0,0 +1,43 @@
+"""
+自定义异常类
+用于任务取消和中断处理
+"""
+
+
+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)