diff --git a/src/knowledge_rag/vector_store.py b/src/knowledge_rag/vector_store.py
index 3f5f57b..911fe63 100644
--- 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"