from pydantic import BaseModel
from typing import List, Optional


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


class ChatRequest(BaseModel):
    system_prompt: str
    messages: List[ChatMessage]
    user_id: Optional[str] = None


class ChatResponse(BaseModel):
    message: ChatMessage