From 359bcb837d150165193c047ef39e3857031a4ea9 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 17 May 2025 00:08:03 +0400 Subject: [PATCH] refac --- backend/open_webui/utils/chat.py | 5 ++++- src/lib/components/chat/Chat.svelte | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/utils/chat.py b/backend/open_webui/utils/chat.py index f6a98b5f3..ce86811d4 100644 --- a/backend/open_webui/utils/chat.py +++ b/backend/open_webui/utils/chat.py @@ -309,6 +309,7 @@ async def chat_completed(request: Request, form_data: dict, user: Any): metadata = { "chat_id": data["chat_id"], "message_id": data["id"], + "filter_ids": data.get("filter_ids", []), "session_id": data["session_id"], "user_id": user.id, } @@ -330,7 +331,9 @@ async def chat_completed(request: Request, form_data: dict, user: Any): try: filter_functions = [ Functions.get_function_by_id(filter_id) - for filter_id in get_sorted_filter_ids(request, model) + for filter_id in get_sorted_filter_ids( + request, model, metadata.get("filter_ids", []) + ) ] result, _ = await process_filter_functions( diff --git a/src/lib/components/chat/Chat.svelte b/src/lib/components/chat/Chat.svelte index e47a9fd19..98e5c9dd2 100644 --- a/src/lib/components/chat/Chat.svelte +++ b/src/lib/components/chat/Chat.svelte @@ -894,6 +894,7 @@ ...(m.usage ? { usage: m.usage } : {}), ...(m.sources ? { sources: m.sources } : {}) })), + filter_ids: selectedFilterIds.length > 0 ? selectedFilterIds : undefined, model_item: $models.find((m) => m.id === modelId), chat_id: chatId, session_id: $socket?.id,