Timothy Jaeryang Baek hai 2 meses
pai
achega
500e6e64fe
Modificáronse 1 ficheiros con 3 adicións e 0 borrados
  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