Spaces:
Sleeping
Sleeping
| 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 |