diff --git a/rag/enio/base_task.go b/rag/enio/base_task.go index 0da9f59..6364fba 100644 --- a/rag/enio/base_task.go +++ b/rag/enio/base_task.go @@ -9,7 +9,7 @@ type BaseTask struct { beans.MongoBaseDO `bson:",inline"` // 任务信息 TaskType string `bson:"taskType" json:"taskType"` - Status string `bson:"status" json:"status"` + Status TaskStatus `bson:"status" json:"status"` Priority TaskPriority `bson:"priority,omitempty" json:"priority,omitempty"` // 进度 TotalItems int64 `bson:"totalItems" json:"totalItems"` diff --git a/rag/enio/task_status.go b/rag/enio/task_status.go new file mode 100644 index 0000000..d6d2479 --- /dev/null +++ b/rag/enio/task_status.go @@ -0,0 +1,12 @@ +package enio + +// TaskStatus 任务状态 +type TaskStatus string + +const ( + TaskStatusPending TaskStatus = "pending" // 待处理 + TaskStatusRunning TaskStatus = "running" // 运行中 + TaskStatusCompleted TaskStatus = "completed" // 已完成 + TaskStatusFailed TaskStatus = "failed" // 失败 + TaskStatusCancelled TaskStatus = "cancelled" // 已取消 +)