from __future__ import annotations
from typing import Literal
from pydantic import BaseModel


class ParseRequest(BaseModel):
    query: str
    lang: Literal["de", "en", "es"]


class ParseResponse(BaseModel):
    personNames: list[str]
    personRole: Literal["sender", "receiver", "any"]
    dateFrom: str | None
    dateTo: str | None
    keywords: list[str]
    rawQuery: str