diff --git a/init_db.py b/init_db.py new file mode 100644 index 0000000..838ca84 --- /dev/null +++ b/init_db.py @@ -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.") \ No newline at end of file