be06203
1
2
3
4
5
6
7
8
9
10
11
from dataclasses import dataclass from typing import Literal @dataclass class Message: """Class for keeping track of a chat message.""" origin: Literal["human", "ai"] message: str