| # 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 | |