summaryrefslogtreecommitdiff
path: root/rag/rag.py
diff options
context:
space:
mode:
Diffstat (limited to 'rag/rag.py')
-rw-r--r--rag/rag.py2
1 files changed, 1 insertions, 1 deletions
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)