from enum import Enum

from pydantic import BaseModel


class Author(Enum):
    User = "user"
    Assistant = "assistant"


class File(BaseModel):
    name: str
    url: str