Update main.py

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()