From 95f47c4900a96d91daaef93bf87094ed3d4da43c Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 6 Apr 2024 13:24:28 +0200 Subject: Update keys --- rag/db/document.py | 2 +- rag/parser/pdf.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) (limited to 'rag') 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, -- cgit v1.2.3-70-g09d2