diff --git a/main.py b/main.py index aee1500..e23e137 100644 --- a/main.py +++ b/main.py @@ -140,7 +140,11 @@ if __name__ == "__main__": if args.query: user_query = " ".join(args.query) else: - user_query = input("Enter your query: ") + # Loop until user provides non-empty input + while True: + user_query = input("Enter your query: ") + if user_query is not None and user_query != "": + break # Run the agent workflow with the provided parameters ask(