from pydantic import BaseModel
from typing import Optional


class RagChatRequest(BaseModel):
    query: str
    top_k: int = 5


class RetrievedDoc(BaseModel):
    id: str
    doc_name: str
    clause_id: Optional[str] = None
    score: float
    content: str
    preview: str


class SourceInfo(BaseModel):
    name: str
    clause: Optional[str] = None


class QuickQuestion(BaseModel):
    id: str
    question: str
    category: str


class QuickQuestionsResponse(BaseModel):
    questions: list[QuickQuestion]