add notebook log

This commit is contained in:
yzlin 2024-04-09 14:53:25 +08:00
parent 4d828bb664
commit b8929be564

View file

@ -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":