"""Admin API Enums."""

from enum import Enum


class DocumentStatus(str, Enum):
    """Document status enum."""

    PENDING = "pending"
    AUTO_LABELING = "auto_labeling"
    LABELED = "labeled"
    EXPORTED = "exported"


class AutoLabelStatus(str, Enum):
    """Auto-labeling status enum."""

    RUNNING = "running"
    COMPLETED = "completed"
    FAILED = "failed"


class TrainingStatus(str, Enum):
    """Training task status enum."""

    PENDING = "pending"
    SCHEDULED = "scheduled"
    RUNNING = "running"
    COMPLETED = "completed"
    FAILED = "failed"
    CANCELLED = "cancelled"


class TrainingType(str, Enum):
    """Training task type enum."""

    TRAIN = "train"
    FINETUNE = "finetune"


class AnnotationSource(str, Enum):
    """Annotation source enum."""

    MANUAL = "manual"
    AUTO = "auto"
    IMPORTED = "imported"
