message.py

from dataclasses import dataclass
from typing import Literal, Optional, List

from langchain.schema import Document


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

    origin: Literal["human", "ai"]
    message: str
    documents: Optional[List[Document]] = None