mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-04-27 19:28:38 +00:00
17 lines
438 B
Python
17 lines
438 B
Python
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
|
|
)
|