bug fix for code runs

InterpreterAtomicFlow.py

@@ -111,6 +111,7 @@ class InterpreterAtomicFlow(AtomicFlow):
             for line in code_interpreter.run(code):
                 if "output" in line:
                     output += "\n" + line["output"]
+                    code_runs = line.get("code_runs", True)
                     # Truncate output
                     output = truncate_output(output, self.max_output)
                     output = output.strip()

code_interpreters/subprocess_code_interpreter.py

@@ -64,7 +64,7 @@ class SubprocessCodeInterpreter(BaseCodeInterpreter):
             if not self.process:
                 self.start_process()
         except:
-            yield {"output": traceback.format_exc()}
+            yield {"output": traceback.format_exc(), "code_runs": False}
             return
             
 
@@ -83,7 +83,7 @@ class SubprocessCodeInterpreter(BaseCodeInterpreter):
                     # For UX, I like to hide this if it happens once. Obviously feels better to not see errors
                     # Most of the time it doesn't matter, but we should figure out why it happens frequently with:
                     # applescript
-                    yield {"output": traceback.format_exc()}
+                    yield {"output": traceback.format_exc(), "code_runs": False}
                     yield {"output": f"Retrying... ({retry_count}/{max_retries})"}
                     yield {"output": "Restarting process."}
 
@@ -91,7 +91,7 @@ class SubprocessCodeInterpreter(BaseCodeInterpreter):
 
                 retry_count += 1
                 if retry_count > max_retries:
-                    yield {"output": "Maximum retries reached. Could not execute code."}
+                    yield {"output": "Maximum retries reached. Could not execute code.", "code_runs": False}
                     return
 
         while True: