trauma/api/chat/model.py

from datetime import datetime

from pydantic import Field

from trauma.api.account.model import AccountModel
from trauma.api.chat.dto import ModelType
from trauma.core.database import MongoBaseModel


class ChatModel(MongoBaseModel):
    title: str = 'New Chat'
    model: ModelType
    account: AccountModel | None = None
    datetimeInserted: datetime = Field(default_factory=datetime.now)
    datetimeUpdated: datetime = Field(default_factory=datetime.now)