UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-16 21:42:50 +00:00
Code Issues Actions3 Packages Projects Releases Wiki Activity

fix: rag duplicate collection issue

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-21 03:53:53 +02:00
parent aed8e2156f
commit ee6b1376c3
1 changed files with 26 additions and 26 deletions

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

@ -1099,6 +1099,10 @@ def store_docs_in_vector_db(
log.info(f"deleting existing collection {collection_name}")
VECTOR_DB_CLIENT.delete_collection(collection_name=collection_name)
if VECTOR_DB_CLIENT.has_collection(collection_name=collection_name):
log.info(f"collection {collection_name} already exists")
return True
else:
embedding_function = get_embedding_function(
app.state.config.RAG_EMBEDDING_ENGINE,
app.state.config.RAG_EMBEDDING_MODEL,
@ -1123,11 +1127,7 @@ def store_docs_in_vector_db(
return True
except Exception as e:
if e.__class__.__name__ == "UniqueConstraintError":
return True
log.exception(e)
return False