1  import asyncio
 2  
 3  from metagpt.environment.mgx.mgx_env import MGXEnv
 4  from metagpt.logs import logger
 5  from metagpt.roles.architect import Architect
 6  from metagpt.roles.di.team_leader import TeamLeader
 7  from metagpt.schema import Message
 8  
 9  
10  async def main():
11      msg = "Write a TRD for a snake game"
12      env = MGXEnv()
13      env.add_roles([TeamLeader(), Architect()])
14      env.publish_message(Message(content=msg, role="user"))
15      tl = env.get_role("Mike")
16      await tl.run()
17  
18      role = env.get_role("Bob")
19      result = await role.run(msg)
20      logger.info(result)
21  
22  
23  if __name__ == "__main__":
24      asyncio.run(main())