from pydantic import BaseModel, Field

class UserRequest(BaseModel):
    question: str = Field(..., description="The user's question or request.")
class OutputResponse(BaseModel):
    answer: str = Field(..., description="The answer generated by the model.")
    justification: str = Field(
        ..., description="Why this query is relevant to the user's request."
    )