diff --git a/init_db.py b/init_db.py new file mode 100644 index 0000000..1f8c259 --- /dev/null +++ b/init_db.py @@ -0,0 +1,20 @@ +"""Инициализация ChromaDB: загрузка документов из папки documents/.""" +from pathlib import Path +from vectorstore import create_vectorstore, load_documents + +CHROMA_DIR = "./chroma_db" +DOCS_DIR = "./documents" + +if not Path(DOCS_DIR).exists(): + Path(DOCS_DIR).mkdir(parents=True) + sample = Path(DOCS_DIR) / "intro.txt" + sample.write_text("LangChain — фреймворк для создания AI-приложений с LLM. +" + "RAG (Retrieval Augmented Generation) позволяет давать LLM доступ к внешним документам. +" + "ChromaDB — векторная база данных с открытым исходным кодом.", encoding="utf-8") + print(f"Создан тестовый файл: {sample}") + +vs = create_vectorstore(CHROMA_DIR) +count = load_documents(DOCS_DIR, vs) +print(f"Загружено {count} чанков из {DOCS_DIR}/ в ChromaDB ({CHROMA_DIR})")