iamstacyok/SE_FastAPI_Example

new_file
48 строк2 KB
KateBel
create: new_file
03 май 2026, 22:12
03 май 2026, 22:12d5f9e11
from pydantic import BaseModel, Field, validator
from typing import Optional


class PredictRequest(BaseModel):
    """Модель запроса для предсказания тональности"""
    text: str = Field(
        ..., 
        min_length=1, 
        max_length=1000,
        description="Текст для анализа тональности"
    )
    
    @validator('text')
    def text_not_empty(cls, v):
        """Проверяет, что текст не пустой и не состоит из пробелов"""
        if not v or not v.strip():
            raise ValueError('Текст не может быть пустым')
        return v.strip()
    
    @validator('text')
    def text_min_length(cls, v):
        """Проверяет минимальную длину текста"""
        if len(v) < 3:
            raise ValueError('Текст должен содержать минимум 3 символа')
        return v


class PredictResponse(BaseModel):
    """Модель ответа для предсказания тональности"""
    sentiment: str = Field(..., description="Тональность: POSITIVE или NEGATIVE")
    confidence: float = Field(..., ge=0, le=1, description="Уверенность модели (0-1)")
    text_length: int = Field(..., description="Длина обработанного текста")
    
    class Config:
        json_schema_extra = {
            "example": {
                "sentiment": "POSITIVE",
                "confidence": 0.987,
                "text_length": 42
            }
        }


class HealthResponse(BaseModel):
    """Модель ответа для health check"""
    status: str
    version: str