diff --git a/constants.py b/constants.py index cae3c10..5a10f5d 100644 --- a/constants.py +++ b/constants.py @@ -6,6 +6,8 @@ load_dotenv() # Define the folder for storing database PERSIST_DIRECTORY = os.environ.get('PERSIST_DIRECTORY') +if PERSIST_DIRECTORY is None: + raise Exception("Please set the PERSIST_DIRECTORY environment variable") # Define the Chroma settings CHROMA_SETTINGS = Settings( diff --git a/ingest.py b/ingest.py index b24b302..d7a6635 100755 --- a/ingest.py +++ b/ingest.py @@ -24,13 +24,14 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter from langchain.vectorstores import Chroma from langchain.embeddings import HuggingFaceEmbeddings from langchain.docstore.document import Document + +if not load_dotenv(): + print("Could not load .env file or it is empty. Please check if it exists and is readable.") + exit(1) + from constants import CHROMA_SETTINGS import chromadb - -load_dotenv() - - # Load environment variables persist_directory = os.environ.get('PERSIST_DIRECTORY') source_directory = os.environ.get('SOURCE_DIRECTORY', 'source_documents') diff --git a/privateGPT.py b/privateGPT.py index e8bd587..75603c2 100755 --- a/privateGPT.py +++ b/privateGPT.py @@ -10,7 +10,9 @@ import os import argparse import time -load_dotenv() +if not load_dotenv(): + print("Could not load .env file or it is empty. Please check if it exists and is readable.") + exit(1) embeddings_model_name = os.environ.get("EMBEDDINGS_MODEL_NAME") persist_directory = os.environ.get('PERSIST_DIRECTORY')