From fd31cc065a74ce8b17765ab7b44ff51ce0adc833 Mon Sep 17 00:00:00 2001
From: lidanyang <lidanyang@fuzhi.ai>
Date: Tue, 12 Dec 2023 10:30:05 +0800
Subject: [PATCH] save jupyter file

---
 metagpt/actions/execute_code.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/metagpt/actions/execute_code.py b/metagpt/actions/execute_code.py
index 981aa894c..6fd980494 100644
--- a/metagpt/actions/execute_code.py
+++ b/metagpt/actions/execute_code.py
@@ -156,6 +156,11 @@ class ExecutePyCode(ExecuteCode, Action):
 
         return code, language
 
+    def save_notebook(self, path: str):
+        path = Path(path)
+        path.parent.mkdir(parents=True, exist_ok=True)
+        nbformat.write(self.nb, path)
+
     async def run(self, code: Union[str, Dict, Message], language: str = "python") -> Tuple[str, bool]:
         code, language = self._process_code(code, language)