src/kittykat_agent/queue/models.py

from enum import Enum
from typing import Literal, Optional
from pydantic import BaseModel
from datetime import datetime


class QueueStatus(str, Enum):
    PROCESSING = "processing"
    COMPELTED = "completed"
    FAILED = "failed"


class QueueItem(BaseModel):
    title: str
    description: str
    type: Literal["video", "image", "product_extract",  "duplication"]
    status: QueueStatus
    metadata: Optional[dict] = None


class QueueItemDeleteRequest(BaseModel):
    ids: list[str]