trauma/api/message/dto.py

from enum import Enum

from pydantic import BaseModel


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


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

class Feedback(BaseModel):
    isPositive: bool
    comment: str | None = None