from pydantic import BaseModel


class ChatMessage(BaseModel):
    role: str
    content: str


class ChatHistory(BaseModel):
    messages: list[ChatMessage]