import os | |
from dotenv import load_dotenv | |
from chromadb.config import Settings | |
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( | |
persist_directory=PERSIST_DIRECTORY, | |
anonymized_telemetry=False | |
) | |