Update main.py

main.py

@@ -1,8 +1,19 @@
-"""
-Entry point for the RAG agent.
+"""Entry point for the RAG agent.
+
+This script simply imports and runs the chat loop defined in ``agent.py``.
 """
 
-from agent import main
+from agent import agent_executor
 
 if __name__ == "__main__":
-    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}")