From 80555333e9df94e8a16cbe58e8e122f80c114563 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 24 Apr 2024 22:02:35 +0200
Subject: Fix drop script

---
 rag/drop.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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()
-- 
cgit v1.2.3-70-g09d2