diff --git a/pilot/vector_store/chroma_store.py b/pilot/vector_store/chroma_store.py index f51bc9917..884b29b05 100644 --- a/pilot/vector_store/chroma_store.py +++ b/pilot/vector_store/chroma_store.py @@ -17,7 +17,8 @@ def __init__(self, ctx: {}) -> None: self.ctx = ctx chroma_path = ctx.get( - "CHROMA_PERSIST_PATH", os.getenv("CHROMA_PERSIST_PATH", os.getcwd()) + "CHROMA_PERSIST_PATH", + os.path.join(os.path.dirname(os.path.dirname(__file__)), "data"), ) self.persist_dir = os.path.join( chroma_path, ctx["vector_store_name"] + ".vectordb"