from pydantic import BaseModel

class ChatRequest(BaseModel):
    message: str

class AskRequest(BaseModel):
    question: str

class ChatResponse(BaseModel):
    answer: str
    latency_ms: int

class AskResponse(BaseModel):
    answer: str
    sources: list[str]
    latency_ms: int