From 46afc4c256d363e18f7bfa16cb3af340a43652a2 Mon Sep 17 00:00:00 2001
From: Ernie Cook <erniecook9@gmail.com>
Date: Tue, 3 Mar 2026 22:47:05 -0500
Subject: [PATCH] Add DATA_PATH env var support

---
 src/knowledge_rag/vector_store.py | 2 ++
 1 file changed, 2 insertions(+)

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"