From 4eb1c14c27f94f2c3da2949642982faa16f03a0b 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=A0=D0=BE=D0=BC?=
 =?UTF-8?q?=D0=B0=D0=BD=D0=BE=D0=B2?= <kirilinka320@gmail.com>
Date: Thu, 4 Jun 2026 22:57:30 +0000
Subject: [PATCH] Add src/loader.py

---
 src/loader.py | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 src/loader.py

diff --git a/src/loader.py b/src/loader.py
new file mode 100644
index 0000000..28efef7
--- /dev/null
+++ b/src/loader.py
@@ -0,0 +1,29 @@
+"""Utility to load all .txt files from a directory into the knowledge base.
+
+This module is useful for initializing the vector store before interacting with the agent.
+"""
+
+from pathlib import Path
+
+from .vector_store import KnowledgeBase
+
+kb = KnowledgeBase()
+
+def load_directory(directory: str) -> None:
+    """Load all .txt files from *directory* into the knowledge base.
+
+    Parameters
+    ----------
+    directory: str
+        Path to the directory containing text files.
+    """
+    kb.load_from_directory(directory)
+
+if __name__ == "__main__":
+    import argparse
+
+    parser = argparse.ArgumentParser(description="Load text files into the knowledge base")
+    parser.add_argument("directory", type=str, help="Directory with .txt files")
+    args = parser.parse_args()
+    load_directory(args.directory)
+    print("Loading completed.")
\ No newline at end of file