File size: 661 Bytes
afe6333 6f793f1 afe6333 6f793f1 afe6333 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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):
pass
|