Browse Source

fix: Fix OpenAI batch embedding

Hwang In Tak 9 months ago
parent
commit
4fe1f2487d
1 changed files with 3 additions and 1 deletions
  1. 3 1
      backend/open_webui/apps/rag/main.py

+ 3 - 1
backend/open_webui/apps/rag/main.py

@@ -1112,7 +1112,9 @@ def store_docs_in_vector_db(
                 app.state.config.RAG_EMBEDDING_OPENAI_BATCH_SIZE,
             )
 
-            embedding_texts = list(map(lambda x: x.replace("\n", " "), texts))
+            embedding_texts = embedding_function(
+                list(map(lambda x: x.replace("\n", " "), texts))
+            )
 
             VECTOR_DB_CLIENT.insert(
                 collection_name=collection_name,