Backend / app /schemas.py
bachephysicdun's picture
implemented rag and filtered_rag
8dbef5d
raw
history blame contribute delete
763 Bytes
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