Browse Source

Added missing parameter for query_doc_with_hybrid_search.

Marko Henning 3 months ago
parent
commit
3b2b6e183d
1 changed files with 5 additions and 0 deletions
  1. 5 0
      backend/open_webui/routers/retrieval.py

+ 5 - 0
backend/open_webui/routers/retrieval.py

@@ -1534,8 +1534,13 @@ def query_doc_handler(
 ):
 ):
     try:
     try:
         if request.app.state.config.ENABLE_RAG_HYBRID_SEARCH:
         if request.app.state.config.ENABLE_RAG_HYBRID_SEARCH:
+            collection_results = {}
+            collection_results[form_data.collection_name] = VECTOR_DB_CLIENT.get(
+                collection_name=form_data.collection_name
+            )
             return query_doc_with_hybrid_search(
             return query_doc_with_hybrid_search(
                 collection_name=form_data.collection_name,
                 collection_name=form_data.collection_name,
+                collection_result=collection_results[form_data.collection_name],
                 query=form_data.query,
                 query=form_data.query,
                 embedding_function=lambda query, prefix: request.app.state.EMBEDDING_FUNCTION(
                 embedding_function=lambda query, prefix: request.app.state.EMBEDDING_FUNCTION(
                     query, prefix=prefix, user=user
                     query, prefix=prefix, user=user