From 65ed76abe1fc9417586ac1994b6614dd8a688995 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sun, 6 Apr 2025 17:17:24 -0700 Subject: [PATCH] refac: embedding prefix --- backend/open_webui/retrieval/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/open_webui/retrieval/utils.py b/backend/open_webui/retrieval/utils.py index f2d2c61de..12d48f869 100644 --- a/backend/open_webui/retrieval/utils.py +++ b/backend/open_webui/retrieval/utils.py @@ -357,7 +357,7 @@ def get_embedding_function( ): if embedding_engine == "": return lambda query, prefix=None, user=None: embedding_function.encode( - query, prompt=prefix if prefix else None + query, **({"prompt": prefix} if prefix else {}) ).tolist() elif embedding_engine in ["ollama", "openai"]: func = lambda query, prefix=None, user=None: generate_embeddings(