from pydantic import BaseModel


class QueryQuestionRequest(BaseModel):
    message: str


class QueryKnowledgeRequest(BaseModel):
    subject: str
    module: str
    name: str


class QueryExamRequest(BaseModel):
    subject: str
    total: int