Added feature to interrupt y -N continuous commands. (Significant-Gra…

…vitas#4230) * Added feature to interrupt y -N continuous commands. * Fixed formatting. --------- Co-authored-by: Nicholas Tindle <[email protected]>
jryals579 · May 16, 2023 · c04f221 · c04f221
1 parent 6c4426d
commit c04f221
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/autogpt/agent/agent.py b/autogpt/agent/agent.py
@@ -1,3 +1,5 @@
+import signal
+import sys
 from datetime import datetime
 
 from colorama import Fore, Style
@@ -88,6 +90,20 @@ def start_interaction_loop(self):
         arguments = None
         user_input = ""
 
+        # Signal handler for interrupting y -N
+        def signal_handler(signum, frame):
+            if self.next_action_count == 0:
+                sys.exit()
+            else:
+                print(
+                    Fore.RED
+                    + "Interrupt signal received. Stopping continuous command execution."
+                    + Style.RESET_ALL
+                )
+                self.next_action_count = 0
+
+        signal.signal(signal.SIGINT, signal_handler)
+
         while True:
             # Discontinue if continuous limit is reached
             self.cycle_count += 1