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