Add delete all cli

1 files changed, 10 insertions, 0 deletions

diff --git a/rag/retriever/document.py b/rag/retriever/document.py
index 8a50f01..132ec4b 100644
--- a/rag/retriever/document.py
+++ b/rag/retriever/document.py
@@ -34,6 +34,16 @@ class DocumentDB:
         log.debug("Hashing document...")
         return hashlib.sha256(blob.as_bytes()).hexdigest()
 
+    def delete_all(self):
+        with self.conn.cursor() as cur:
+            cur.execute(
+                """
+                        TRUNCATE TABLE
+                        document
+                        """
+            )
+            self.conn.commit()
+
     def add(self, blob: Blob) -> bool:
         with self.conn.cursor() as cur:
             hash = self.__hash(blob)