summaryrefslogtreecommitdiff
path: root/rag
diff options
context:
space:
mode:
Diffstat (limited to 'rag')
-rw-r--r--rag/drop.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/rag/drop.py b/rag/drop.py
index 89ae755..5ff5983 100644
--- a/rag/drop.py
+++ b/rag/drop.py
@@ -6,7 +6,7 @@ from rag.retriever.retriever import Retriever
def drop():
- log.debug("Dropping documents")
+ log.debug("Deleting all data...")
retriever = Retriever()
doc_db = retriever.doc_db
doc_db.delete_all()
@@ -14,6 +14,7 @@ def drop():
vec_db.delete_collection()
+@click.command()
@click.confirmation_option(prompt="Are you sure you want to drop the db?")
def main():
drop()