from typing import List

from pydantic import BaseModel


class QueryRequest(BaseModel):
    question: str
    mode: str
    top_k: int | None = None
    temperature: float | None = None


class PresentationRequest(BaseModel):
    ideas: List[str]
    title: str | None = None
    audience: str | None = None
    language: str | None = None
    num_slides: int | None = None