diff options
Diffstat (limited to 'rag/retriever/vector.py')
-rw-r--r-- | rag/retriever/vector.py | 6 |
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( |