message.py

from dataclasses import dataclass
from typing import Literal, Optional


@dataclass
class Message:
    """Class for keeping track of a chat message."""

    origin: Literal["human", "ai"]
    message: str
    metadata: Optional[dict] = None