from pydantic import BaseModel


class RAGRequest(BaseModel):
    query: str
    protect_enabled: bool = False


class RAGResponse(BaseModel):
    response: str