from datetime import datetime
from typing import Optional, List
from pydantic import BaseModel, Field


class JobCreate(BaseModel):
    notebook_id: Optional[str]
    doc_id: Optional[str]
    source: str = "manual"
    blocks: Optional[List[str]] = None


class JobStatus(BaseModel):
    id: str
    status: str
    notebook_id: Optional[str]
    doc_id: Optional[str]
    created_at: datetime
    updated_at: datetime
    error_message: Optional[str]


class JobResponse(BaseModel):
    job: JobStatus