From 3077bfdff3587aac5ef91a8fb5730c11c8bb5273 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sat, 6 Apr 2024 01:53:41 +0200
Subject: Refactor doc db

---
 rag/rag.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'rag/rag.py')

diff --git a/rag/rag.py b/rag/rag.py
index 7242db9..87b44c5 100644
--- a/rag/rag.py
+++ b/rag/rag.py
@@ -22,7 +22,7 @@ class RAG:
 
     def add_pdf(self, filepath: Path):
         chunks = pdf.parser(filepath)
-        added = self.document_db.add_document(chunks)
+        added = self.document_db.add(chunks)
         if added:
             log.debug(f"Adding pdf with filepath: {filepath} to vector db")
             points = self.encoder.encode_document(chunks)
-- 
cgit v1.2.3-70-g09d2