Update api/models.py

api/models.py

@@ -1,17 +1,13 @@
 from typing import List, Optional, Union
-from pydantic import BaseModel, Field
+from pydantic import BaseModel
 
-class ImageResponse(BaseModel):
-    images: Union[str, List[str]]  # Can handle both URLs and Base64 data
-    alt: Optional[str] = ""
-
-class MessageContent(BaseModel):
+class Message(BaseModel):
     role: str
-    content: Union[str, List[dict]]  # Handle both text and complex structures like images
+    content: Union[str, List[dict]]  # Allowing for both text and complex structures like images
 
 class ChatRequest(BaseModel):
     model: str
-    messages: List[MessageContent]
+    messages: List[Message]
     stream: Optional[bool] = False
     temperature: Optional[float] = 0.7
     top_p: Optional[float] = 0.9