Update keys

2 files changed, 1 insertions, 2 deletions

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,