UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-22 05:08:14 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: filter compatibility issue

Browse Source
This commit is contained in:
Timothy J. Baek 2024-08-22 01:08:59 +02:00
parent 85b4129219
commit 6fcd40d4d8
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
backend/main.py
View File

@ -525,9 +525,8 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
"chat_id": body.pop("chat_id", None), "chat_id": body.pop("chat_id", None),
"message_id": body.pop("id", None), "message_id": body.pop("id", None),
"session_id": body.pop("session_id", None), "session_id": body.pop("session_id", None),
"valves": body.pop("valves", None), "tool_ids": body.get("tool_ids", None),
"tool_ids": body.pop("tool_ids", None), "files": body.get("files", None),
"files": body.pop("files", None),
} }
body["metadata"] = metadata body["metadata"] = metadata
@ -560,6 +559,13 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
content={"detail": str(e)}, content={"detail": str(e)},
) )
metadata = {
**metadata,
"tool_ids": body.pop("tool_ids", None),
"files": body.pop("files", None),
}
body["metadata"] = metadata
try: try:
body, flags = await chat_completion_tools_handler(body, user, extra_params) body, flags = await chat_completion_tools_handler(body, user, extra_params)
contexts.extend(flags.get("contexts", [])) contexts.extend(flags.get("contexts", []))