UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-24 08:16:02 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac: embedding prefix

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-04-06 17:17:24 -07:00
parent bc17511a05
commit 65ed76abe1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/open_webui/retrieval/utils.py
View File

@ -357,7 +357,7 @@ def get_embedding_function(
): ):
if embedding_engine == "": if embedding_engine == "":
return lambda query, prefix=None, user=None: embedding_function.encode( return lambda query, prefix=None, user=None: embedding_function.encode(
query, prompt=prefix if prefix else None query, **({"prompt": prefix} if prefix else {})
).tolist() ).tolist()
elif embedding_engine in ["ollama", "openai"]: elif embedding_engine in ["ollama", "openai"]:
func = lambda query, prefix=None, user=None: generate_embeddings( func = lambda query, prefix=None, user=None: generate_embeddings(