# api/models.py

from typing import List, Optional, Union, Dict, Any  # Ensure 'Any' is imported
from pydantic import BaseModel

class ImageContent(BaseModel):
    image_url: Dict[str, str]

class Message(BaseModel):
    role: str
    content: Union[str, List[Dict[str, Any]]]  # 'Any' is now defined

class ChatRequest(BaseModel):
    model: str
    messages: List[Message]
    stream: Optional[bool] = False
    temperature: Optional[float] = 0.5
    top_p: Optional[float] = 0.9
    max_tokens: Optional[int] = 1024