delete collection before inserting docs

aimakerspace/vectordatabase.py

@@ -34,6 +34,24 @@ class VectorDatabase:
             f"Inserted {len(texts)} documents into collection '{self.collection_name}'."
         )
 
+    def _delete_collection(self):
+        """
+        Delete a collection from the Qdrant database.
+
+        Args:
+            collection_name (str): Name of the collection to delete.
+        """
+        # Check if the collection exists
+        collections = self.client.get_collections()
+        collection_names = [collection.name for collection in collections.collections]
+
+        if self.collection_name in collection_names:
+            # Delete the collection
+            self.client.delete_collection(collection_name)
+            print(f"Collection '{collection_name}' deleted.")
+        else:
+            print(f"Collection '{collection_name}' does not exist.")
+
     def search_similar(self, query_text: str):
         search_result = self.client.query(
             collection_name=self.collection_name,

app.py

@@ -31,6 +31,7 @@ class RetrievalAugmentedQAPipeline:
     def __init__(self, llm: ChatOpenAI(), vector_db_retriever: VectorDatabase) -> None:
         self.llm = llm
         self.vector_db_retriever = vector_db_retriever
+        self.vector_db_retriever.delete_collection()
 
     async def arun_pipeline(self, user_query: str):
         context_data = self.vector_db_retriever.search_similar(user_query)