| import nbformat | |
| from nbconvert.preprocessors import ExecutePreprocessor | |
| from nbconvert.preprocessors.execute import CellExecutionError | |
| nb = nbformat.v4.new_notebook() | |
| # Add a cell with your code | |
| code_cell = nbformat.v4.new_code_cell(source=f'import os\nprint(os.getcwd())') | |
| nb.cells.append(code_cell) | |
| # Execute the notebook | |
| ep = ExecutePreprocessor(timeout=600, kernel_name='python3') | |
| output_str, error_str = None, None | |
| ep.preprocess(nb) | |
| if nb.cells[0].outputs: # Check if there are any outputs | |
| output = nb.cells[-1].outputs[0] | |
| print(output) | |
| # Repo path :: /home/seungyoun/llama_code_interpreter\n | |