Update api/models.py

api/models.py

@@ -1,6 +1,10 @@
 from typing import List, Optional, Union
 from pydantic import BaseModel, Field
 
+class ImageResponse(BaseModel):
+    images: Union[str, List[str]]  # Can handle both URLs and Base64 data
+    alt: Optional[str] = ""
+
 class MessageContent(BaseModel):
     role: str
     content: Union[str, List[dict]]  # Handle both text and complex structures like images