from pydantic import BaseModel

class AssistantRequest(BaseModel):
    customer_id: int
    question: str

class AssistantResponse(BaseModel):
    answer: str