UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 21:42:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: allow openai list message format

Browse Source
This commit is contained in:
Timothy J. Baek 2024-11-03 01:34:45 -08:00
parent 0925c6b33b
commit 4e7951d5fc
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/open_webui/apps/ollama/main.py
View File

@ -817,7 +817,7 @@ class OpenAIChatMessageContent(BaseModel):
class OpenAIChatMessage(BaseModel): class OpenAIChatMessage(BaseModel):
role: str role: str
content: Union[str, OpenAIChatMessageContent] content: Union[str, list[OpenAIChatMessageContent]]
model_config = ConfigDict(extra="allow") model_config = ConfigDict(extra="allow")
@ -836,7 +836,15 @@ async def generate_openai_chat_completion(
url_idx: Optional[int] = None, url_idx: Optional[int] = None,
user=Depends(get_verified_user), user=Depends(get_verified_user),
): ):
completion_form = OpenAIChatCompletionForm(**form_data) try:
completion_form = OpenAIChatCompletionForm(**form_data)
except Exception as e:
log.exception(e)
raise HTTPException(
status_code=400,
detail=str(e),
)
payload = {**completion_form.model_dump(exclude_none=True, exclude=["metadata"])} payload = {**completion_form.model_dump(exclude_none=True, exclude=["metadata"])}
if "metadata" in payload: if "metadata" in payload:
del payload["metadata"] del payload["metadata"]