refac: add embedding metadata

2024-11-06 16:59:42 +00:00 · 2024-10-12 18:30:21 -07:00 · 2024-10-12 18:30:21 -07:00 · 333317a7ce
commit 333317a7ce
parent 5c48fce382
1 changed files with 7 additions and 1 deletions
--- a/backend/open_webui/apps/retrieval/main.py
+++ b/backend/open_webui/apps/retrieval/main.py
@ -698,7 +698,13 @@ def save_docs_to_vector_db(
                "id": str(uuid.uuid4()),
                "text": text,
                "vector": embeddings[idx],
-                "metadata": metadatas[idx],
+                "metadata": {
+                    **metadatas[idx],
+                    "embedding": {
+                        "engine": app.state.config.RAG_EMBEDDING_ENGINE,
+                        "model": app.state.config.RAG_EMBEDDING_MODEL,
+                    },
+                },
            }
            for idx, text in enumerate(texts)
        ]