diff options
Diffstat (limited to 'rag')
| -rw-r--r-- | rag/db/document.py | 2 | ||||
| -rw-r--r-- | rag/parser/pdf.py | 1 |
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, |