backend/models/message.py

from odmantic import Model, Field
from enum import Enum
from datetime import datetime, timezone, timedelta
from typing import Optional

class Role(str, Enum):
    user = "user"
    bot = "bot"
    image = "image"

class Message(Model):
    role: Role = Field(...)
    text: Optional[str] = None
    src: Optional[str] = None
    timestamp: datetime = Field(default_factory=lambda: datetime.now(timezone(timedelta(hours=5, minutes=30))).replace(tzinfo=None))