"""Safety API request contracts."""

from __future__ import annotations

from typing import Literal

from pydantic import BaseModel


class SafetyDecisionRequest(BaseModel):
    asset_id: str
    decision: Literal["sfw", "nsfw", "deferred"]