from datetime import datetime

from pydantic import Field

from trauma.api.data.model import EntityModelExtended
from trauma.api.message.dto import Author, Feedback
from trauma.core.database import MongoBaseModel


class MessageModel(MongoBaseModel):
    chatId: str
    author: Author
    text: str
    entities: list[EntityModelExtended] | None = None
    feedback: Feedback | None = None
    datetimeInserted: datetime = Field(default_factory=datetime.now)
    datetimeUpdated: datetime = Field(default_factory=datetime.now)