from pydantic import BaseModel
from datetime import datetime

class UserQuestion(BaseModel):
    question: str

# create a HistoryInput data model with a chat_history and question attributes.
class HistoryInput(BaseModel):
    chat_history: str
    question: str

# let's create a UserRequest data model with a question and username attribute. 
# This will be used to parse the input request.
class UserRequest(BaseModel):
    username: str
    question: str

# implement MessageBase as a schema mapping from the database model to the 
# FastAPI data model. Basically MessageBase should have the same attributes as models.Message
class MessageBase(BaseModel):
    # id: int
    # user_id: int
    message: str
    type: str
    timestamp: datetime
    # user: str