fix persist db directory at ingestion

This commit is contained in:
Andrea Pinto 2023-05-12 10:37:10 +02:00
parent 39df61ca07
commit 01f55441e7
2 changed files with 6 additions and 3 deletions

View file

@ -1,7 +1,11 @@
import os
from dotenv import load_dotenv
from chromadb.config import Settings
load_dotenv()
# Define the folder for storing database
PERSIST_DIRECTORY = 'db'
PERSIST_DIRECTORY = os.environ.get('PERSIST_DIRECTORY')
# Define the Chroma settings
CHROMA_SETTINGS = Settings(