chat_with_pdf / database.py
mercybabs's picture
update with pgvector setup and tested docker
5d11203
raw
history blame
305 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 = f"postgresql+psycopg://{user}:{password}@{host}:5432/{COLLECTION_NAME}"