Update keys

3 files changed, 3 insertions, 4 deletions

diff --git a/README.md b/README.md
index 5c4f217..ff86560 100644
--- a/README.md
+++ b/README.md
@@ -19,8 +19,8 @@ EMBEDDING_DIM = <EMBEDDING_DIM>
 
 GENERATOR_MODEL = <GENERATOR_MODEL>
 
-RAG_DB_NAME = <DOCUMENT_DB_NAME>
-RAG_DB_USER = <RAG_DB_USER>
+DOCUMENT_DB_NAME = <DOCUMENT_DB_NAME>
+DOCUMENT_DB_USER = <DOCUMENT_DB_USER>
 
 QDRANT_URL = <QDRANT_URL>
 QDRANT_COLLECTION_NAME = <QDRANT_COLLECTION_NAME>
diff --git a/rag/db/document.py b/rag/db/document.py
index b657e55..528a399 100644
--- a/rag/db/document.py
+++ b/rag/db/document.py
@@ -13,7 +13,7 @@ CREATE TABLE IF NOT EXISTS document (
 class DocumentDB:
     def __init__(self) -> None:
         self.conn = psycopg.connect(
-            f"dbname={os.environ['RAG_DB_NAME']} user={os.environ['RAG_DB_USER']}"
+            f"dbname={os.environ['DOCUMENT_DB_NAME']} user={os.environ['DOCUMENT_DB_USER']}"
         )
         self.__configure()
 
diff --git a/rag/parser/pdf.py b/rag/parser/pdf.py
index 22fc4e0..ed4dc8b 100644
--- a/rag/parser/pdf.py
+++ b/rag/parser/pdf.py
@@ -3,7 +3,6 @@ from pathlib import Path
 from typing import Iterator, Optional
 
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain_community.document_loaders import PyPDFLoader
 from langchain_core.documents import Document
 from langchain_community.document_loaders.parsers.pdf import (
     PyPDFParser,