Refactor doc db

1 files changed, 1 insertions, 1 deletions

diff --git a/rag/db/document.py b/rag/db/document.py
index c449158..f8c97c2 100644
--- a/rag/db/document.py
+++ b/rag/db/document.py
@@ -33,7 +33,7 @@ class DocumentDB:
         document = str.encode("".join([chunk.page_content for chunk in chunks]))
         return hashlib.sha256(document).hexdigest()
 
-    def add_document(self, chunks: List[Document]) -> bool:
+    def add(self, chunks: List[Document]) -> bool:
         log.debug("Inserting document hash into documents db...")
         with self.conn.cursor() as cur:
             hash = self.__hash(chunks)