"""Modèles de données pour la synthèse IA."""

from __future__ import annotations

from datetime import date

from pydantic import BaseModel, Field

from pronote_sync.models.agenda import SchoolEvent
from pronote_sync.models.diff import AgendaDiff
from pronote_sync.models.message import Message


class SynthesisInput(BaseModel):
    """Entrée pour la synthèse IA."""

    agenda_diff: AgendaDiff | None = Field(None, description="Différences d'agenda")
    messages: list[Message] = Field(default_factory=list, description="Messages importants")
    school_events: list[SchoolEvent] = Field(
        default_factory=list, description="Événements scolaires"
    )
    target_date: date = Field(..., description="Date cible")


class SynthesisResult(BaseModel):
    """Résultat de la synthèse IA."""

    text: str | None = Field(None, description="Texte de la synthèse IA")