from pydantic import BaseModel
from typing import Optional, Union, List, Dict

class APIResponse(BaseModel):
    success: bool
    data: Optional[Union[Dict, List[Dict], str]] = None # Allow str for raw JSON
    error: Optional[str] = None

class ChatMessage(BaseModel):
    role: str # e.g., "user", "assistant"
    content: str

class ChatMemoryRequest(BaseModel):
    model: str
    messages: List[ChatMessage]