"""생성 추적 API 스키마."""

from typing import Optional

from pydantic import BaseModel


class GenerationTraceResponse(BaseModel):
    id: str
    image_asset_id: Optional[str] = None
    still_id: Optional[str] = None
    entity_id: Optional[str] = None
    attempt_number: int
    prompt_used: str
    prompt_version: str
    model_name: Optional[str] = None
    status: str
    block_reason: Optional[str] = None
    block_categories: str
    response_time_ms: Optional[int] = None
    sanitizer_feedback: Optional[str] = None
    created_at: str


class RejectionStatsResponse(BaseModel):
    total: int
    success: int
    blocked: int
    error: int
    block_reasons: dict