Add DATA_PATH env var support

2026-03-03 22:47:05 -05:00
parent 8d09d03fe8
commit 46afc4c256
1 changed files with 2 additions and 0 deletions
--- a/src/knowledge_rag/vector_store.py
+++ b/src/knowledge_rag/vector_store.py
@ -30,6 +30,8 @@ class KnowledgeVectorStore:
        if persist_dir is None:
            if os.path.exists("/data"):
                persist_dir = "/data/chroma_db"
+            elif os.environ.get("DATA_PATH"):
+                persist_dir = os.environ.get("DATA_PATH")
            else:
                persist_dir = "./data/chroma_db"