Просмотр исходного кода

recreate rag collection instead of falling back to stale version

Firat Birlik 1 год назад
Родитель
Сommit
5d4ff85228
1 измененных файлов с 4 добавлено и 0 удалено
  1. 4 0
      backend/apps/rag/main.py

+ 4 - 0
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,