diff --git a/backend/open_webui/retrieval/utils.py b/backend/open_webui/retrieval/utils.py index e39025978..5f181fba0 100644 --- a/backend/open_webui/retrieval/utils.py +++ b/backend/open_webui/retrieval/utils.py @@ -154,15 +154,18 @@ def merge_get_results(get_results: list[dict]) -> dict: # Initialize lists to store combined data combined_documents = [] combined_metadatas = [] + combined_ids = [] for data in get_results: combined_documents.extend(data["documents"][0]) combined_metadatas.extend(data["metadatas"][0]) + combined_ids.extend(data["ids"][0]) # Create the output dictionary result = { "documents": [combined_documents], "metadatas": [combined_metadatas], + "ids": [combined_ids], } return result