task utils etc.

metagpt/schema.py

@@ -341,6 +341,18 @@ class Task(BaseModel):
     is_success: bool = False
     is_finished: bool = False
 
+    def reset(self):
+        self.code = ""
+        self.result = ""
+        self.is_success = False
+        self.is_finished = False
+
+    def update_task_result(self, task_result: TaskResult):
+        self.code_steps = task_result.code_steps
+        self.code = task_result.code
+        self.result = task_result.result
+        self.is_success = task_result.is_success
+
 
 class TaskResult(BaseModel):
     """Result of taking a task, with result and is_success required to be filled"""
@@ -434,10 +446,7 @@ class Plan(BaseModel):
         """
         if task_id in self.task_map:
             task = self.task_map[task_id]
-            task.code = ""
-            task.result = ""
-            task.is_success = False
-            task.is_finished = False
+            task.reset()
 
     def replace_task(self, new_task: Task):
         """
@@ -483,12 +492,6 @@ class Plan(BaseModel):
         self.task_map[new_task.task_id] = new_task
         self._update_current_task()
 
-    def update_task_result(self, task: Task, task_result: TaskResult):
-        task.code_steps = task_result.code_steps
-        task.code = task_result.code
-        task.result = task_result.result
-        task.is_success = task_result.is_success
-
     def has_task_id(self, task_id: str) -> bool:
         return task_id in self.task_map