from datetime import datetime
from typing import Optional

from pydantic import BaseModel


class ChatMessageRequest(BaseModel):
    content: str


class ChatMessageResponse(BaseModel):
    role: str
    content: str
    timestamp: datetime


class ChatSessionResponse(BaseModel):
    id: str
    report_id: str
    messages: list[ChatMessageResponse]
    created_at: datetime