changed rstrip for strip and added case for empty string

changed rstrip for strip and added case for empty string in agent.py
javay · Apr 17, 2023 · 286edbb · 286edbb
1 parent 10cd0f3
commit 286edbb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/autogpt/agent/agent.py b/autogpt/agent/agent.py
@@ -115,9 +115,12 @@ def start_interaction_loop(self):
                     console_input = clean_input(
                         Fore.MAGENTA + "Input:" + Style.RESET_ALL
                     )
-                    if console_input.lower().rstrip() == "y":
+                    if console_input.lower().strip() == "y":
                         user_input = "GENERATE NEXT COMMAND JSON"
                         break
+                    elif console_input.lower().strip() == "":
+                        print("Invalid input format.")
+                        break
                     elif console_input.lower().startswith("y -"):
                         try:
                             self.next_action_count = abs(