src/models/_messages.py

from datetime import datetime
from beanie import Document, Link
from pydantic import Field

from ._conversations import Conversation


class Message(Document):
    conversation: Link[Conversation]
    role: str
    content: str
    created_at: datetime = Field(default_factory=datetime.now)
    updated_at: datetime = Field(default_factory=datetime.now)

    class Settings:
        name = "messages"