app/models.py

from pydantic import BaseModel
from typing import Optional, Any
from enum import Enum

class StatusEnum(str, Enum):
    processing = "processing"
    complete = "complete"
    failed = "failed"

class TaskStatus(BaseModel):
    status: StatusEnum
    result: Optional[Any] = None