Timothy Jaeryang Baek преди 2 месеца
родител
ревизия
500e6e64fe
променени са 1 файла, в които са добавени 3 реда и са изтрити 0 реда
  1. 3 0
      backend/open_webui/retrieval/utils.py

+ 3 - 0
backend/open_webui/retrieval/utils.py

@@ -611,6 +611,9 @@ def get_sources_from_items(
         elif item.get("collection_name"):
             # Direct Collection Name
             collection_names.append(item["collection_name"])
+        elif item.get("collection_names"):
+            # Collection Names List
+            collection_names.extend(item["collection_names"])
 
         # If query_result is None
         # Fallback to collection names and vector search the collections