UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-27 17:51:31 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added missing parameter for query_doc_with_hybrid_search.

Browse Source
This commit is contained in:
Marko Henning 2025-04-04 15:30:57 +02:00
parent be20e6dec0
commit 3b2b6e183d
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/open_webui/routers/retrieval.py
View File

@ -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