Execute Python code and capture stdout as well as the full stack trace on error. Forces __name__ = "__main__" so that if __name__ == "__main__": blocks will run. Returns: (output, error) - output: string containing everything that was printed to stdout - err
(code)
| 217 | return run_code(utils_functions + '\n' + code) |
| 218 | |
| 219 | def run_code(code): |
| 220 | """ |
| 221 | Execute Python code and capture stdout as well as the full stack trace on error. |
| 222 | Forces __name__ = "__main__" so that if __name__ == "__main__": blocks will run. |
| 223 | |
| 224 | Returns: |
| 225 | (output, error) |
| 226 | - output: string containing everything that was printed to stdout |
| 227 | - error: string containing the full traceback if an exception occurred; None otherwise |
| 228 | """ |
| 229 | stdout_capture = io.StringIO() |
| 230 | # Provide a globals dict specifying that __name__ is "__main__" |
| 231 | exec_globals = {"__name__": "__main__"} |
| 232 | |
| 233 | with contextlib.redirect_stdout(stdout_capture): |
| 234 | try: |
| 235 | exec(code, exec_globals) |
| 236 | error = None |
| 237 | except Exception: |
| 238 | # Capture the entire stack trace |
| 239 | error = traceback.format_exc() |
| 240 | |
| 241 | output = stdout_capture.getvalue() |
| 242 | return output, error |
| 243 | |
| 244 | |
| 245 | def run_code_from_agent(agent, msg, num_retries=1): |
no outgoing calls
no test coverage detected