trauma/api/chat/model.py

from datetime import datetime

from pydantic import Field

from trauma.api.chat.dto import ModelType, EntityData
from trauma.core.database import MongoBaseModel


class ChatModel(MongoBaseModel):
    title: str = 'New Chat'
    model: ModelType
    entityData: EntityData = EntityData()
    datetimeInserted: datetime = Field(default_factory=datetime.now)
    datetimeUpdated: datetime = Field(default_factory=datetime.now)