Forráskód Böngészése

fix: query_embedding param

Timothy Jaeryang Baek 4 hónapja
szülő
commit
2db837cab4
1 módosított fájl, 2 hozzáadás és 5 törlés
  1. 2 5
      backend/open_webui/apps/retrieval/main.py

+ 2 - 5
backend/open_webui/apps/retrieval/main.py

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