Преглед на файлове

refac: embedding prefix

Timothy Jaeryang Baek преди 5 месеца
родител
ревизия
65ed76abe1
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      backend/open_webui/retrieval/utils.py

+ 1 - 1
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(