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