From 1da4409475579705e5c7a44e1a873e337d02eb83 Mon Sep 17 00:00:00 2001 From: stellahsr Date: Tue, 12 Dec 2023 16:10:05 +0800 Subject: [PATCH] add step plan --- metagpt/roles/ml_engineer.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py index 45fe728dd..8ad75b399 100644 --- a/metagpt/roles/ml_engineer.py +++ b/metagpt/roles/ml_engineer.py @@ -156,6 +156,11 @@ class MLEngineer(Role): # ask for acceptance, users can other refuse and change tasks in the plan task_result_confirmed = await self._ask_review() + # 针对当前task进行单独plan + if not success or not task_result_confirmed: + # fixme: 增加对应plan + self.state.plan() + if success and task_result_confirmed: # tick off this task and record progress task.code = code @@ -203,8 +208,6 @@ class MLEngineer(Role): if counter == 0: context = self.get_useful_memories() else: - # context = self.get_useful_memories() - # logger.info(f"context {context}") improve_code = await DebugCode().run(plan=self.plan.current_task.instruction, finished_code=code_context, finished_code_result=code_result, @@ -255,6 +258,8 @@ class MLEngineer(Role): counter += 1 + success = False + return code, result, success, code_steps async def _ask_review(self):