chat_with_pdf / database.py
mercybabs's picture
update all files
bb07d5b
raw
history blame
392 Bytes
import os
from dotenv import load_dotenv
load_dotenv()
host = os.getenv("PG_VECTOR_HOST")
user = os.getenv("PG_VECTOR_USER")
password = os.getenv("PG_VECTOR_PASSWORD")
COLLECTION_NAME = os.getenv("PGDATABASE")
CONNECTION_STRING = "postgresql+psycopg://langchain:langchain@localhost:6024/langchain"
#CONNECTION_STRING = f"postgresql+psycopg://{user}:{password}@{host}:6042/{COLLECTION_NAME}"