from datetime import datetime

from pydantic import BaseModel

from models.Asset import Asset


class AssetResponse(BaseModel):
    id: str
    name: str
    type: str # uploaded / generated
    content_type: str # image / prompt
    linked_char_id: str | None = None
    created_at: datetime
    url: str | None = None

class AssetsResponse(BaseModel):
    assets: list[AssetResponse]
    total_count: int