diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-24 22:02:35 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-24 22:02:35 +0200 |
| commit | 80555333e9df94e8a16cbe58e8e122f80c114563 (patch) | |
| tree | 885e1a49a6b4f4e0451fc5641751a9e03d0605e9 | |
| parent | 0fd41b05692085e8e4dab75a59f4c56ba4ac5ea7 (diff) | |
Fix drop script
| -rw-r--r-- | rag/drop.py | 3 |
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() |