Merge pull request #172 from Wenlin88/schemas.py-fix

schemas.py fix to support image input with latest open WebUI version

Author: tjbck

schemas.py

@@ -1,30 +1,21 @@
-from typing import List, Union, Optional
-from pydantic import BaseModel, RootModel, ConfigDict
-
-class ImageContent(BaseModel):
-    type: str
-    image_url: dict
-
-class TextContent(BaseModel):
-    type: str
-    text: str
-
-class MessageContent(RootModel):
-    root: Union[TextContent, ImageContent]
+from typing import List, Optional
+from pydantic import BaseModel, ConfigDict
 
 class OpenAIChatMessage(BaseModel):
     role: str
-    content: Union[str, List[MessageContent]]
+    content: str | List
 
     model_config = ConfigDict(extra="allow")
 
+
 class OpenAIChatCompletionForm(BaseModel):
     stream: bool = True
     model: str
     messages: List[OpenAIChatMessage]
 
     model_config = ConfigDict(extra="allow")
 
+
 class FilterForm(BaseModel):
     body: dict
     user: Optional[dict] = None