From dfc59422d9a55221aa4f3be40f392b82f7fbb39d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B4=AA=E6=80=9D=E7=9D=BF?= <stellahong@fuzhi.ai>
Date: Mon, 2 Oct 2023 03:31:31 +0800
Subject: [PATCH] update player base

---
 metagpt/minecraft_team.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/metagpt/minecraft_team.py b/metagpt/minecraft_team.py
index 4f3e701a7..be2b959f6 100644
--- a/metagpt/minecraft_team.py
+++ b/metagpt/minecraft_team.py
@@ -214,6 +214,25 @@ class MinecraftPlayer(SoftwareCompany):
 
     def set_resume(self, resume: bool = False):
         self.game_memory.set_mc_resume(resume=resume)
+    
+    def check_complete_round(self):
+        complete_round = []
+        for role in self.environment.roles.values():
+            status = role.finish_step
+            complete_round.append(status)
+            #if not status:
+            #    return complete_round
+        #complete_round = True
+        complete_round_tag = all(complete_round)
+        logger.info(f"complete_round {complete_round}")
+        return complete_round_tag
+    
+    def update_round(self):
+        for role in self.environment.roles.values():
+            role.finish_step = False
+            role.round_id+=1
+            role._rc.todo = None
+            logger.info(f"round_id:{role.round_id}")
 
     def hire(self, roles: list[Role]):
         self.environment.add_roles(roles)