summaryrefslogtreecommitdiff
path: root/rag/db
diff options
context:
space:
mode:
Diffstat (limited to 'rag/db')
-rw-r--r--rag/db/document.py2
1 files changed, 1 insertions, 1 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()