12 lines
394 B
Python
12 lines
394 B
Python
"""
|
|
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.") |