fix: Fix OpenAI batch embedding

2025-12-16 11:57:51 +01:00 · 2024-09-26 20:39:40 +09:00
parent be41994d40
commit a4bc0b2829
3 changed files with 3 additions and 7 deletions
--- a/backend/open_webui/apps/rag/main.py
+++ b/backend/open_webui/apps/rag/main.py
@@ -1112,13 +1112,15 @@ def store_docs_in_vector_db(
                app.state.config.RAG_EMBEDDING_OPENAI_BATCH_SIZE,
            )

+            embedding_texts = list(map(lambda x: x.replace("\n", " "), texts))
+
            VECTOR_DB_CLIENT.insert(
                collection_name=collection_name,
                items=[
                    {
                        "id": str(uuid.uuid4()),
                        "text": text,
-                        "vector": embedding_function(text.replace("\n", " ")),
+                        "vector": embedding_texts[idx],
                        "metadata": metadatas[idx],
                    }
                    for idx, text in enumerate(texts)