Add init_db.py

2026-05-31 16:22:56 +00:00
parent f12163d5b9
commit 26344152fa
1 changed files with 12 additions and 0 deletions
@@ -0,0 +1,12 @@
 """
 Script to initialise the ChromaDB with documents from ./documents.
 """
 import os
 from vectorstore import create_vectorstore, load_documents
 if __name__ == "__main__":
    db_dir = os.getenv("CHROMA_DB_DIR", "./chroma_db")
    store = create_vectorstore(db_dir)
    docs_path = os.getenv("DOCS_PATH", "./documents")
    load_documents(docs_path, store)
    print("Initialisation complete.")