From 04686f497f120096435da72c6546306eb292846a Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 23 Apr 2024 00:50:25 +0200 Subject: Add delete all cli --- rag/retriever/vector.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'rag/retriever/vector.py') 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( -- cgit v1.2.3-70-g09d2