Add delete all cli

1 files changed, 5 insertions, 1 deletions

diff --git a/rag/retriever/vector.py b/rag/retriever/vector.py
index e8d22c2..b72a3c1 100644
--- a/rag/retriever/vector.py
+++ b/rag/retriever/vector.py
@@ -22,7 +22,7 @@ class Document:
 
 
 class VectorDB:
-    def __init__(self, score_threshold: float = 0.6):
+    def __init__(self, score_threshold: float = 0.5):
         self.dim = int(os.environ["EMBEDDING_DIM"])
         self.collection_name = os.environ["QDRANT_COLLECTION_NAME"]
         self.client = QdrantClient(url=os.environ["QDRANT_URL"])
@@ -42,6 +42,10 @@ class VectorDB:
         else:
             log.debug(f"Collection {self.collection_name} already exists!")
 
+    def delete_collection(self):
+        log.info(f"Deleting collection {self.collection_name}")
+        self.client.delete_collection(self.collection_name)
+
     def add(self, points: List[Point]):
         log.debug(f"Inserting {len(points)} vectors into the vector db...")
         self.client.upload_points(