From 6bca8996f30d5e72bfd0dcb705e9153bb19a1bbf Mon Sep 17 00:00:00 2001
From: lonpatovaadelina <lonpatovaadelina@mail.ru>
Date: Wed, 3 Jun 2026 09:08:29 +0000
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D1=80=D0=B0=D0=B1=D0=BE=D1=82?=
 =?UTF-8?q?=D0=BA=D0=B0:=201.=20=D0=92=20vectorstore.py=20=D0=B7=D0=B0?=
 =?UTF-8?q?=D0=BC=D0=B5=D0=BD=D0=B8=D1=82=D1=8C=20=D0=B8=D0=BC=D0=BF=D0=BE?=
 =?UTF-8?q?=D1=80=D1=82=20Ollama=20=D0=BD=D0=B0=20from=20langchain?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/init_db.py b/init_db.py
new file mode 100644
index 0000000..1241777
--- /dev/null
+++ b/init_db.py
@@ -0,0 +1,18 @@
+import os
+from dotenv import load_dotenv
+from vectorstore import create_vectorstore, load_documents
+
+def main():
+    load_dotenv()  # загружаем TAVILY_API_KEY и другие переменные
+    persist_dir = os.getenv("CHROMA_PERSIST_DIR", "./chroma_db")
+    vectorstore = create_vectorstore(persist_directory=persist_dir)
+    docs_dir = os.path.join(os.path.dirname(__file__), "documents")
+    if not os.path.isdir(docs_dir):
+        print(f"Каталог {docs_dir} не найден.")
+        return
+    print(f"Загрузка документов из {docs_dir} в ChromaDB ({persist_dir})...")
+    load_documents(docs_dir, vectorstore)
+    print("Загрузка завершена.")
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file