UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-22 08:07:55 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: embedding metadata issue

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-13 03:25:11 -07:00
parent 92605fd59f
commit 797afd0b72
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -678,7 +678,19 @@ def save_docs_to_vector_db(
raise ValueError(ERROR_MESSAGES.EMPTY_CONTENT)
texts = [doc.page_content for doc in docs]
metadatas = [{**doc.metadata, **(metadata if metadata else {})} for doc in docs]
metadatas = [
{
**doc.metadata,
**(metadata if metadata else {}),
"embedding_config": json.dumps(
{
"engine": app.state.config.RAG_EMBEDDING_ENGINE,
"model": app.state.config.RAG_EMBEDDING_MODEL,
}
),
}
for doc in docs
]
# ChromaDB does not like datetime formats
# for meta-data so convert them to string.
@ -717,13 +729,7 @@ def save_docs_to_vector_db(
"id": str(uuid.uuid4()),
"text": text,
"vector": embeddings[idx],
"metadata": {
**metadatas[idx],
"embedding": {
"engine": app.state.config.RAG_EMBEDDING_ENGINE,
"model": app.state.config.RAG_EMBEDDING_MODEL,
},
},
"metadata": metadatas[idx],
}
for idx, text in enumerate(texts)
]