Better error message if .env is empty/does not exist.

This commit is contained in:
VaiTon
2023-07-06 00:16:11 +02:00
parent b1057afdf8
commit 28537b6a84
3 changed files with 11 additions and 6 deletions

View File

@@ -24,12 +24,13 @@ 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
load_dotenv()
# Load environment variables
persist_directory = os.environ.get('PERSIST_DIRECTORY')
source_directory = os.environ.get('SOURCE_DIRECTORY', 'source_documents')