apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

updata di example and Change the location of the .ipynb locations

Browse source
This commit is contained in:
luxiangtao 2024-03-24 15:01:45 +08:00
parent 893501fd35
commit 83e5eb133a
2 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
examples/di/data_visualization.py
View file

@ -1,14 +1,21 @@
import asyncio
import os
import sys
import metagpt.utils
from metagpt.logs import logger
from metagpt.utils.recovery_util import save_history
from metagpt.roles.di.data_interpreter import DataInterpreter
async def main(requirement: str = ""):
di = DataInterpreter()
await di.run(requirement)
rsp = await di.run(requirement)
logger.info(rsp)
save_history(role=di)
if __name__ == "__main__":
requirement = "Run data analysis on sklearn Iris dataset, include a plot"
requirement = "Run data analysis on sklearn Iris dataset, include a plot"
asyncio.run(main(requirement))

2
metagpt/utils/recovery_util.py
View file

@ -54,5 +54,5 @@ def save_history(role: Role, save_dir: str = ""):
with open(save_path / "plan.json", "w", encoding="utf-8") as plan_file:
json.dump(plan, plan_file, indent=4, ensure_ascii=False)
save_code_file(name=Path(record_time) / "history_nb", code_context=role.execute_code.nb, file_format="ipynb")
save_code_file(name=Path(record_time), code_context=role.execute_code.nb, file_format="ipynb")
return save_path