"""Pydantic models for NanoClaude API."""

from __future__ import annotations

from enum import Enum

from pydantic import BaseModel


class JobStatus(str, Enum):
    queued = "queued"
    processing = "processing"
    completed = "completed"
    failed = "failed"
    cancelled = "cancelled"


class ChatRequest(BaseModel):
    message: str


class ChatResponse(BaseModel):
    job_id: str


class CancelResponse(BaseModel):
    status: str


class JobStatusResponse(BaseModel):
    job_id: str
    status: JobStatus
    created_at: float
    pipeline: bool
    queue_position: int | None = None


class SSEEvent(BaseModel):
    event: str  # ack | processing | rewrite | result | error | done | queued
    data: dict