Prechádzať zdrojové kódy

Merge pull request #12236 from Phlogi/fix_prefix

Fix middleware.py missing prefix on embedding function
Timothy Jaeryang Baek 1 mesiac pred
rodič
commit
3dc40030a1
1 zmenil súbory, kde vykonal 2 pridanie a 2 odobranie
  1. 2 2
      backend/open_webui/utils/middleware.py

+ 2 - 2
backend/open_webui/utils/middleware.py

@@ -595,8 +595,8 @@ async def chat_completion_files_handler(
                         request=request,
                         request=request,
                         files=files,
                         files=files,
                         queries=queries,
                         queries=queries,
-                        embedding_function=lambda query: request.app.state.EMBEDDING_FUNCTION(
-                            query, user=user
+                        embedding_function=lambda query, prefix: request.app.state.EMBEDDING_FUNCTION(
+                            query, prefix=prefix, user=user
                         ),
                         ),
                         k=request.app.state.config.TOP_K,
                         k=request.app.state.config.TOP_K,
                         reranking_function=request.app.state.rf,
                         reranking_function=request.app.state.rf,