summaryrefslogtreecommitdiff
path: root/rag/drop.py
diff options
context:
space:
mode:
Diffstat (limited to 'rag/drop.py')
-rw-r--r--rag/drop.py25
1 files changed, 0 insertions, 25 deletions
diff --git a/rag/drop.py b/rag/drop.py
deleted file mode 100644
index 5ff5983..0000000
--- a/rag/drop.py
+++ /dev/null
@@ -1,25 +0,0 @@
-import click
-from dotenv import load_dotenv
-from loguru import logger as log
-
-from rag.retriever.retriever import Retriever
-
-
-def drop():
- log.debug("Deleting all data...")
- retriever = Retriever()
- doc_db = retriever.doc_db
- doc_db.delete_all()
- vec_db = retriever.vec_db
- vec_db.delete_collection()
-
-
-@click.command()
-@click.confirmation_option(prompt="Are you sure you want to drop the db?")
-def main():
- drop()
-
-
-if __name__ == "__main__":
- load_dotenv()
- main()