From eeadeb034edfe6a31a2fcf35fdaf6531ee8efd4a Mon Sep 17 00:00:00 2001 From: zhangsan Date: Tue, 26 Dec 2023 11:36:35 +0800 Subject: [PATCH] Fix env type mismatch for chroma datastore --- datastore/providers/chroma_datastore.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/datastore/providers/chroma_datastore.py b/datastore/providers/chroma_datastore.py index 59b802d63..f8d6a6508 100644 --- a/datastore/providers/chroma_datastore.py +++ b/datastore/providers/chroma_datastore.py @@ -26,7 +26,7 @@ ) from services.chunks import get_document_chunks -CHROMA_IN_MEMORY = os.environ.get("CHROMA_IN_MEMORY", "True") +CHROMA_IN_MEMORY = True if os.environ.get("CHROMA_IN_MEMORY", "True").lower() == "true" else False CHROMA_PERSISTENCE_DIR = os.environ.get("CHROMA_PERSISTENCE_DIR", "openai") CHROMA_HOST = os.environ.get("CHROMA_HOST", "http://127.0.0.1") CHROMA_PORT = os.environ.get("CHROMA_PORT", "8000")