From 333317a7ce399399c1c753da982ed172530f1899 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sat, 12 Oct 2024 18:30:21 -0700
Subject: [PATCH] refac: add embedding metadata

---
 backend/open_webui/apps/retrieval/main.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/backend/open_webui/apps/retrieval/main.py b/backend/open_webui/apps/retrieval/main.py
index c80b2011d9..2fe60ea707 100644
--- 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)
         ]