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