From 7ca08838e9199965db1f2f5ee6914b671effad3f 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: Tue, 2 Jun 2026 07:20:26 +0000
Subject: [PATCH] Update main.py

---
 main.py | 21 ++++++---------------
 1 file changed, 6 insertions(+), 15 deletions(-)

diff --git a/main.py b/main.py
index 78d317c..4e8b5ac 100644
--- a/main.py
+++ b/main.py
@@ -1,19 +1,10 @@
-"""Entry point for the RAG agent.
-
-This script simply imports and runs the chat loop defined in ``agent.py``.
+"""
+Entry point for the RAG agent. Delegates to agent.py.
 """
 
-from agent import agent_executor
+import sys
+
+from agent import __main__ as agent_main
 
 if __name__ == "__main__":
-    print("RAG Agent ready. Type 'exit' to quit.")
-    while True:
-        user_input = input("Запрос: ")
-        if user_input.lower() in {"exit", "quit", "q"}:
-            print("Bye!")
-            break
-        try:
-            result = agent_executor.invoke({"input": user_input})
-            print(result["output"])
-        except Exception as e:
-            print(f"Error: {e}")
+    agent_main()