32 lines
597 B
Python
32 lines
597 B
Python
|
from datetime import datetime
|
||
|
|
from enum import Enum
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class JobStatus(str, Enum):
|
||
|
|
pending = 'pending'
|
||
|
|
processing = 'processing'
|
||
|
|
completed = 'completed'
|
||
|
|
failed = 'failed'
|
||
|
|
cancelled = 'cancelled'
|
||
|
|
|
||
|
|
|
||
|
|
class JobCreateResponse(BaseModel):
|
||
|
|
job_id: str
|
||
|
|
status: JobStatus
|
||
|
|
|
||
|
|
|
||
|
|
class JobStatusResponse(BaseModel):
|
||
|
|
job_id: str
|
||
|
|
status: JobStatus
|
||
|
|
created_at: datetime
|
||
|
|
updated_at: Optional[datetime] = None
|
||
|
|
error: Optional[str] = None
|
||
|
|
|
||
|
|
|
||
|
|
class JobDeleteResponse(BaseModel):
|
||
|
|
job_id: str
|
||
|
|
deleted: bool
|