mirror of
https://github.com/imartinez/privateGPT.git
synced 2025-04-28 11:36:25 +00:00
18 lines
479 B
Python
18 lines
479 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(
|
|
chroma_db_impl='duckdb+parquet',
|
|
persist_directory=PERSIST_DIRECTORY,
|
|
anonymized_telemetry=False
|
|
)
|