Timothy Jaeryang Baek 2 months ago
parent
commit
500e6e64fe
1 changed files with 3 additions and 0 deletions
  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