Update api/models.py

api/models.py

@@ -1,9 +1,14 @@
-from typing import List, Optional
+# api/models.py
+
+from typing import List, Optional, Union, Dict
 from pydantic import BaseModel
 
+class ImageContent(BaseModel):
+    image_url: Dict[str, str]
+
 class Message(BaseModel):
     role: str
-    content: str | list
+    content: Union[str, List[Dict[str, Any]]]
 
 class ChatRequest(BaseModel):
     model: str
@@ -12,4 +17,3 @@ class ChatRequest(BaseModel):
     temperature: Optional[float] = 0.5
     top_p: Optional[float] = 0.9
     max_tokens: Optional[int] = 1024
-    proxy: Optional[str] = None  # Added proxy field if necessary