o2o-infinith-backend/app/models/file.py

30 lines
518 B
Python

from enum import Enum
from pydantic import BaseModel
class FileType(str, Enum):
IMAGE = "image"
VIDEO = "video"
AUDIO = "audio"
DOCUMENT = "document"
FILE = "file"
class FileUploadResponse(BaseModel):
id: int
analysis_run_id: str
file_type: FileType
file_name: str
file_url: str
size_bytes: int | None = None
class FileListItem(BaseModel):
id: int
file_type: FileType
file_name: str
file_url: str
size_bytes: int | None = None
created_at: str