From 9c043637ff953e018769e45982aa8b97d67a6a99 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=93=D0=BB=D0=B5=D0=B1=20=D0=9D=D0=B8=D0=BA=D0=B8=D1=88?=
 =?UTF-8?q?=D0=B8=D0=BD?= <glevelll@yandex.ru>
Date: Fri, 5 Jun 2026 14:55:45 +0000
Subject: [PATCH] feat: init_db script for documents/ folder

---
 init_db.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 init_db.py

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})")