INTERPRETER_ACTIVE_LINE_DETECTION control and INTERPRETER_TERMINAL_INPUT_PATIENCE control

Author: KillianLucas

interpreter/core/computer/terminal/languages/jupyter_language.py

@@ -149,9 +149,12 @@ def iopub_message_listener():
                     self.finish_flag = True
                     return
                 try:
+                    input_patience = int(
+                        os.environ.get("INTERPRETER_TERMINAL_INPUT_PATIENCE", 15)
+                    )
                     if (
-                        time.time() - self.last_output_time > 15
-                        and time.time() - self.last_output_message_time > 15
+                        time.time() - self.last_output_time > input_patience
+                        and time.time() - self.last_output_message_time > input_patience
                     ):
                         self.last_output_message_time = time.time()
 
@@ -364,7 +367,11 @@ def preprocess_python(code):
 
     # Add print commands that tell us what the active line is
     # but don't do this if any line starts with ! or %
-    if not any(line.strip().startswith(("!", "%")) for line in code.split("\n")):
+    if (
+        not any(line.strip().startswith(("!", "%")) for line in code.split("\n"))
+        and os.environ.get("INTERPRETER_ACTIVE_LINE_DETECTION", "True").lower()
+        == "true"
+    ):
         code = add_active_line_prints(code)
 
     # Wrap in a try except (DISABLED)

interpreter/core/computer/terminal/languages/shell.py

@@ -45,7 +45,11 @@ def preprocess_shell(code):
 
     # Add commands that tell us what the active line is
     # if it's multiline, just skip this. soon we should make it work with multiline
-    if not has_multiline_commands(code):
+    if (
+        not has_multiline_commands(code)
+        and os.environ.get("INTERPRETER_ACTIVE_LINE_DETECTION", "True").lower()
+        == "true"
+    ):
         code = add_active_line_prints(code)
 
     # Add end command (we'll be listening for this so we know when it ends)

interpreter/core/utils/truncate_output.py

@@ -4,7 +4,7 @@ def truncate_output(data, max_output_chars=2800, add_scrollbars=False):
 
     needs_truncation = False
 
-    message = f"Output truncated. Showing the last {max_output_chars} characters.\n\n"
+    message = f"Output truncated. Showing the last {max_output_chars} characters. You should try again and use computer.ai.summarize(output) over the output, or break it down into smaller steps.\n\n"
 
     # This won't work because truncated code is stored in interpreter.messages :/
     # If the full code was stored, we could do this: