from typing import Any, Dict, List

from pydantic import BaseModel


class QueryResponse(BaseModel):
    answer: str
    retrieved: List[Dict[str, Any]]


class UploadResponse(BaseModel):
    ok: bool
    message: str
    filename: str
    document_id: str


class PrecheckResponse(BaseModel):
    score: float
    recommendation: str
    rationale: str
    matched_topics: List[str]