From 1c078bdb55b67f6ad65591c8229eb4d4bda3bd44 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sun, 1 Dec 2024 13:36:36 -0800 Subject: [PATCH] fix: api query collection --- backend/open_webui/apps/retrieval/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/open_webui/apps/retrieval/main.py b/backend/open_webui/apps/retrieval/main.py index 7ad4af064..341f4f500 100644 --- a/backend/open_webui/apps/retrieval/main.py +++ b/backend/open_webui/apps/retrieval/main.py @@ -1399,7 +1399,7 @@ def query_collection_handler( if app.state.config.ENABLE_RAG_HYBRID_SEARCH: return query_collection_with_hybrid_search( collection_names=form_data.collection_names, - query=form_data.query, + queries=[form_data.query], embedding_function=app.state.EMBEDDING_FUNCTION, k=form_data.k if form_data.k else app.state.config.TOP_K, reranking_function=app.state.sentence_transformer_rf, @@ -1410,7 +1410,7 @@ def query_collection_handler( else: return query_collection( collection_names=form_data.collection_names, - query=form_data.query, + queries=[form_data.query], embedding_function=app.state.EMBEDDING_FUNCTION, k=form_data.k if form_data.k else app.state.config.TOP_K, )