diff --git a/main.py b/main.py new file mode 100644 index 0000000..a160896 --- /dev/null +++ b/main.py @@ -0,0 +1,26 @@ +""" +Simple CLI to chat with 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) + +if __name__ == "__main__": + main() \ No newline at end of file