From 5d4ff852281cf18dac9955442b60737e25447257 Mon Sep 17 00:00:00 2001 From: Firat Birlik Date: Sun, 3 Mar 2024 21:25:00 -0600 Subject: [PATCH] recreate rag collection instead of falling back to stale version --- backend/apps/rag/main.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/backend/apps/rag/main.py b/backend/apps/rag/main.py index 2a8b2a49..0a5fde2c 100644 --- a/backend/apps/rag/main.py +++ b/backend/apps/rag/main.py @@ -118,6 +118,10 @@ def store_data_in_vector_db(data, collection_name) -> bool: metadatas = [doc.metadata for doc in docs] try: + for collection in CHROMA_CLIENT.list_collections(): + if collection_name == collection.name: + CHROMA_CLIENT.delete_collection(name=collection_name) + collection = CHROMA_CLIENT.create_collection( name=collection_name, embedding_function=app.state.sentence_transformer_ef,