apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add step plan

Browse source
This commit is contained in:
stellahsr 2023-12-12 16:10:05 +08:00
parent 9c426f73dc
commit 1da4409475
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
metagpt/roles/ml_engineer.py
View file

@ -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):