import datetime as dt
import enum

from pydantic import BaseModel


class RequestInput(BaseModel):
    author: str
    content: str


class EntriesResult(BaseModel):
    id: str
    author: str
    summary: str
    source_snippet: str
    tags: list[str]
    date: dt.datetime


class JobInput(BaseModel):
    id: str
    author: str
    content: str


class JobStatus(str, enum.Enum):
    pending = "pending"
    done = "done"
    failed = "failed"
    cancelled = "cancelled"
    not_found = "not found"


class JobStatusResult(BaseModel):
    id: str
    status: JobStatus
    last_updated: dt.datetime | None