AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-19 19:59:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #12450 from mahenning/fix-missing-parameter-rag

Browse Source 
Fix: Add missing parameter for query_doc_with_hybrid_search
This commit is contained in:
Timothy Jaeryang Baek 2025-04-04 07:00:16 -07:00 committed by GitHub
commit a32bb850fa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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