from datetime import datetime, UTC
from enum import Enum
from typing import Optional

from pydantic import BaseModel

class AssetType(str, Enum):
    IMAGE = 'image'
    PROMPT = 'prompt'

class Asset(BaseModel):
    id: Optional[str] = None
    name: str
    type: AssetType
    linked_char_id: Optional[str] = None
    data: Optional[bytes] = None
    tg_doc_file_id: str
    tg_photo_file_id: Optional[str] = None
    created_at: datetime = datetime.now(UTC)