From e2df200857fb7744588a7745de0ce422744646d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=9A=D0=B8=D1=80=D0=B8=D0=BB=D0=BB=20=D0=9A=D1=83=D1=82?=
 =?UTF-8?q?=D0=BB=D0=B0=D1=85=D0=BC=D0=B5=D1=82=D0=BE=D0=B2?=
 <kirillkutlahmetov10@gmail.com>
Date: Thu, 28 May 2026 09:15:27 +0000
Subject: [PATCH] Add init_documents.py

---
 init_documents.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 init_documents.py

diff --git a/init_documents.py b/init_documents.py
new file mode 100644
index 0000000..09d65aa
--- /dev/null
+++ b/init_documents.py
@@ -0,0 +1,25 @@
+"""
+Script to load all .txt files from a directory into the knowledge base.
+"""
+
+import argparse
+from pathlib import Path
+from agent import agent
+from langchain_core.messages import HumanMessage
+
+async def init_docs(dir_path: Path):
+    for txt_file in dir_path.rglob("*.txt"):
+        content = txt_file.read_text(encoding="utf-8")
+        title = txt_file.stem
+        await agent.ainvoke(
+            {"messages": [HumanMessage(content=f"Add document: {title}")]},
+            {"configurable": {"thread_id": f"init-{txt_file.name}"}},
+        )
+    print("Initialization complete.")
+
+if __name__ == "__main__":  # pragma: no cover
+    parser = argparse.ArgumentParser(description="Initialize knowledge base from directory")
+    parser.add_argument("dir", type=Path, help="Directory with text files")
+    args = parser.parse_args()
+    import asyncio
+    asyncio.run(init_docs(args.dir))