from pydantic import BaseModel, Field class TextInput(BaseModel): text: str = Field(..., example="Это просто пиииииииииииииииииииздец") class ToxicityOutput(BaseModel): toxicity_score: float = Field(..., example=0.85)