ソースを参照

refac: add embedding metadata

Timothy J. Baek 7 ヶ月 前
コミット
333317a7ce
1 ファイル変更7 行追加1 行削除
  1. 7 1
      backend/open_webui/apps/retrieval/main.py

+ 7 - 1
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)
         ]