File size: 519 Bytes
6525f4e 6c37ac6 14ebd6f 6525f4e 14ebd6f 6c37ac6 14ebd6f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# 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
|