"""
Chat-related models
"""

from pydantic import BaseModel, Field


class ChatRequest(BaseModel):
    """Basic chat request model"""
    message: str = Field(..., description="User's chat message")


class ChatResponse(BaseModel):
    """Basic chat response model"""
    reply: str = Field(..., description="Assistant's reply to the user")