fix:chromadb persist_path compatible with old version

eosphoros-ai · Oct 9, 2023 · 813e226 · 813e226
1 parent 4aa825d
commit 813e226
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"