ernie/knowledge-base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add DATA_PATH env var support

Browse Source
This commit is contained in:
Ernie Cook
2026-03-03 22:47:05 -05:00
parent 8d09d03fe8
commit 46afc4c256
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/knowledge_rag/vector_store.py
View File

@ -30,6 +30,8 @@ class KnowledgeVectorStore:
if persist_dir is None: if persist_dir is None:
if os.path.exists("/data"): if os.path.exists("/data"):
persist_dir = "/data/chroma_db" persist_dir = "/data/chroma_db"
elif os.environ.get("DATA_PATH"):
persist_dir = os.environ.get("DATA_PATH")
else: else:
persist_dir = "./data/chroma_db" persist_dir = "./data/chroma_db"