Update main.py

main.py

@@ -1,26 +1,8 @@
 """
-Simple CLI to chat with the RAG agent.
+Entry point for the RAG agent.
 """
 
-import os
-import sys
-
-from agent import agent_executor
-
-def main():
-    print("Welcome to the RAG agent. Type 'exit' to quit.")
-    while True:
-        try:
-            user_input = input("\nQuery: ")
-        except (KeyboardInterrupt, EOFError):
-            print("\nExiting.")
-            break
-        if user_input.strip().lower() in {"exit", "quit", "q"}:
-            print("Goodbye!")
-            break
-        # Run agent
-        response = agent_executor.run(user_input)
-        print("\nAnswer:\n", response)
+from agent import main
 
 if __name__ == "__main__":
-    main()
+    main()