OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-12 16:40:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source 
schemas.py fix to support image input with latest open WebUI version
This commit is contained in:
Timothy Jaeryang Baek 2024-10-19 14:00:32 -07:00 committed by GitHub
commit 52ffa648c4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
schemas.py
View File

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