UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-06 16:59:42 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac: add embedding metadata

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-12 18:30:21 -07:00
parent 5c48fce382
commit 333317a7ce
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/apps/retrieval/main.py
View File

@ -698,7 +698,13 @@ def save_docs_to_vector_db(
"id": str(uuid.uuid4()), "id": str(uuid.uuid4()),
"text": text, "text": text,
"vector": embeddings[idx], "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) for idx, text in enumerate(texts)
] ]