From 0dfa3f5b5aca6b76d967950924754b1d276f9dc2 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:16:29 +0000
Subject: [PATCH] Add main.py

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

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..921f2d1
--- /dev/null
+++ b/main.py
@@ -0,0 +1,29 @@
+"""
+Entry point demonstrating the RAG agent.
+
+Runs three example scenarios:
+1. Add a sample text directly via tool call.
+2. Search for a keyword.
+3. Load documents from a directory using init_documents.
+"""
+
+import asyncio
+from pathlib import Path
+from langchain_core.messages import HumanMessage
+from agent import agent
+
+async def demo():
+    # 1. Add sample text
+    await agent.ainvoke(
+        {"messages": [HumanMessage(content="Add document: Sample")]},
+        {"configurable": {"thread_id": "demo-1"}},
+    )
+    # 2. Search
+    result = await agent.ainvoke(
+        {"messages": [HumanMessage(content="Search for Python")]},
+        {"configurable": {"thread_id": "demo-1"}},
+    )
+    print("Search result:\n", result["messages"][-1].content)
+
+if __name__ == "__main__":  # pragma: no cover
+    asyncio.run(demo())