from pydantic.v1 import BaseModel
from datetime import datetime
from typing import Optional

class UserQuestion(BaseModel):
    question: str

class UserRequest(UserQuestion):
    username: str

class HistoryInput(BaseModel):
    chat_history: str
    question: str

class RagInput(HistoryInput):
    hybrid_search: bool = False

class MessageBase(BaseModel):
    id: Optional[int] = None
    user_id: Optional[int] = None
    message: str
    type: str
    timestamp: Optional[datetime] = None

    class Config:
        orm_mode = True