from pydantic import BaseModel from models.enums import ActionModel, SubActionModel class MessageModel(BaseModel): text: str class QueryModel(BaseModel): sender: str message: MessageModel class AppModel(BaseModel): version: str class InputModel(BaseModel): _version: int query: QueryModel app: AppModel class OutputModel(BaseModel): action: ActionModel sub_action: SubActionModel class ReportMessageModel(BaseModel): timestamp: int sender: str message: str class ReportMessagesModel(BaseModel): messages: list[ReportMessageModel] class ReportInputModel(BaseModel): _version: int classification: ReportMessagesModel app: AppModel