Add delete all cli

1 files changed, 24 insertions, 0 deletions

diff --git a/rag/drop.py b/rag/drop.py
new file mode 100644
index 0000000..89ae755
--- /dev/null
+++ b/rag/drop.py
@@ -0,0 +1,24 @@
+import click
+from dotenv import load_dotenv
+from loguru import logger as log
+
+from rag.retriever.retriever import Retriever
+
+
+def drop():
+    log.debug("Dropping documents")
+    retriever = Retriever()
+    doc_db = retriever.doc_db
+    doc_db.delete_all()
+    vec_db = retriever.vec_db
+    vec_db.delete_collection()
+
+
+@click.confirmation_option(prompt="Are you sure you want to drop the db?")
+def main():
+    drop()
+
+
+if __name__ == "__main__":
+    load_dotenv()
+    main()