Update server.py

This commit is contained in:
donlinglok
2023-07-31 22:02:31 +08:00
committed by GitHub
parent 17d622d7bf
commit c19e6d3881

View File

@@ -265,7 +265,7 @@ if "chromadb" in modules:
posthog.capture = lambda *args, **kwargs: None
if args.chroma_host is None:
if args.chroma_persist:
chromadb_client = chromadb.PersistentClient(persist_directory=args.chroma_folder, chroma_db_impl='duckdb+parquet', settings=Settings(anonymized_telemetry=False))
chromadb_client = chromadb.PersistentClient(path=args.chroma_folder, chroma_db_impl='duckdb+parquet', settings=Settings(anonymized_telemetry=False))
print(f"ChromaDB is running in-memory with persistence. Persistence is stored in {args.chroma_folder}. Can be cleared by deleting the folder or purging db.")
else:
chromadb_client = chromadb.EphemeralClient(settings=Settings(anonymized_telemetry=False))