diff --git a/metagpt/actions/di/execute_nb_code.py b/metagpt/actions/di/execute_nb_code.py index f9330884d..873c11106 100644 --- a/metagpt/actions/di/execute_nb_code.py +++ b/metagpt/actions/di/execute_nb_code.py @@ -25,7 +25,7 @@ from rich.syntax import Syntax from metagpt.actions import Action from metagpt.const import DEFAULT_WORKSPACE_ROOT -from metagpt.logs import logger +from metagpt.logs import ToolLogItem, log_tool_output, logger class ExecuteNbCode(Action): @@ -208,6 +208,10 @@ class ExecuteNbCode(Action): if "!pip" in code: success = False + file_path = DEFAULT_WORKSPACE_ROOT / "code.ipynb" + nbformat.write(self.nb, file_path) + log_tool_output(ToolLogItem(name="file_path", value=file_path), tool_name="ExecuteNbCode") + return outputs, success elif language == "markdown":