diff options
| -rw-r--r-- | README.md | 4 | ||||
| -rw-r--r-- | rag/db/document.py | 2 | ||||
| -rw-r--r-- | rag/parser/pdf.py | 1 |
3 files changed, 3 insertions, 4 deletions
@@ -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, |