models/models.py

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