"""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 SSEEvent(BaseModel):
    event: str  # ack | processing | result | error | done
    data: dict