from pydantic.v1 import BaseModel


class UserQuestion(BaseModel):
    question: str

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

# TODO: 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

# TODO: 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):
    message: str
    type: str