OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: query_embedding param

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-09 23:33:43 -08:00
parent 8de91df1ff
commit 2db837cab4
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/open_webui/apps/retrieval/main.py
View File

@ -608,9 +608,7 @@ async def update_rag_config(form_data: ConfigUpdateForm, user=Depends(get_admin_
app.state.config.BRAVE_SEARCH_API_KEY = ( app.state.config.BRAVE_SEARCH_API_KEY = (
form_data.web.search.brave_search_api_key form_data.web.search.brave_search_api_key
) )
app.state.config.KAGI_SEARCH_API_KEY = ( app.state.config.KAGI_SEARCH_API_KEY = form_data.web.search.kagi_search_api_key
form_data.web.search.kagi_search_api_key
)
app.state.config.MOJEEK_SEARCH_API_KEY = ( app.state.config.MOJEEK_SEARCH_API_KEY = (
form_data.web.search.mojeek_search_api_key form_data.web.search.mojeek_search_api_key
) )
@ -1390,8 +1388,7 @@ def query_doc_handler(
else: else:
return query_doc( return query_doc(
collection_name=form_data.collection_name, collection_name=form_data.collection_name,
query=form_data.query, query_embedding=app.state.EMBEDDING_FUNCTION(form_data.query),
embedding_function=app.state.EMBEDDING_FUNCTION,
k=form_data.k if form_data.k else app.state.config.TOP_K, k=form_data.k if form_data.k else app.state.config.TOP_K,
) )
except Exception as e: except Exception as e: