from pydantic import BaseModel


class TaskState(BaseModel):
    task_id: str
    status: str
    user_message: str
    workspace: str | None = None
    debug: bool = False
    final_response: str | None = None
    created_at: str
    updated_at: str