from pydantic import BaseModel


class TaskBaseModelSchema(BaseModel):
    task_id: str

class TaskDetailInfoSchema(TaskBaseModelSchema):
    status: str
    result: str | None = None
    
class TaskResultSchema(BaseModel):
    active_tasks: list[str]
    scheduled_tasks: list[str]
    completed_tasks: list[str]
    failed_tasks: list[str]