diff --git a/examples/di/data_visualization.py b/examples/di/data_visualization.py index 4c88ffe00..184e04f26 100644 --- a/examples/di/data_visualization.py +++ b/examples/di/data_visualization.py @@ -1,10 +1,11 @@ import asyncio + from metagpt.logs import logger -from metagpt.utils.recovery_util import save_history from metagpt.roles.di.data_interpreter import DataInterpreter +from metagpt.utils.recovery_util import save_history + async def main(requirement: str = ""): - di = DataInterpreter() rsp = await di.run(requirement) logger.info(rsp) @@ -12,7 +13,5 @@ async def main(requirement: str = ""): if __name__ == "__main__": - requirement = "Run data analysis on sklearn Iris dataset, include a plot" asyncio.run(main(requirement)) -