Timothy Jaeryang Baek 4 kuukautta sitten
vanhempi
commit
359bcb837d
2 muutettua tiedostoa jossa 5 lisäystä ja 1 poistoa
  1. 4 1
      backend/open_webui/utils/chat.py
  2. 1 0
      src/lib/components/chat/Chat.svelte

+ 4 - 1
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(

+ 1 - 0
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,