File size: 457 Bytes
4a13938
 
5b9fb0b
4a13938
 
 
 
 
 
 
 
 
 
 
5b9fb0b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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))