1  import asyncio
 2  
 3  from metagpt.logs import logger
 4  from metagpt.roles.di.data_interpreter import DataInterpreter
 5  from metagpt.utils.recovery_util import save_history
 6  
 7  
 8  async def main(requirement: str = ""):
 9      di = DataInterpreter()
10      rsp = await di.run(requirement)
11      logger.info(rsp)
12      save_history(role=di)
13  
14  
15  if __name__ == "__main__":
16      requirement = "Run data analysis on sklearn Iris dataset, include a plot"
17      asyncio.run(main(requirement))