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