src/api_models.py

from pydantic import BaseModel, Field


class ResponseModel(BaseModel):
    message: str
    data: dict
    code: int


class ResponseGuessWord(BaseModel):
    word: str


class RequestSemanticCalculation(BaseModel):
    supposed_word: str = Field(
        description="The word that the user is trying to guess",
        example="ніс"
    )
    guessed_word: str = Field(
        description="The word that the user guessed",
        example="око"
    )


class SemanticCalculation(BaseModel):
    score: float
    rating: int
    percentage: float
    closest_word: str


class ResponseSemanticCalculation(BaseModel):
    word_exist: bool
    metadata: SemanticCalculation | None


class ResponseMessage(BaseModel):
    message: str