12 lines
335 B
Python
12 lines
335 B
Python
import os
|
|
from dotenv import load_dotenv
|
|
from src.vectorstore import create_vectorstore, load_documents
|
|
|
|
def main():
|
|
load_dotenv()
|
|
vectorstore = create_vectorstore(persist_directory="./chroma_db")
|
|
load_documents("./documents", vectorstore)
|
|
print("Documents loaded into ChromaDB.")
|
|
|
|
if __name__ == "__main__":
|
|
main() |