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

update reflect on previous plan

Browse source
This commit is contained in:
yzlin 2023-12-04 14:29:47 +08:00
parent d3d08fe5f3
commit 8d7657f347
5 changed files with 44 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/roles/kaggle_manager.py
View file

@ -38,8 +38,8 @@ class DownloadData(Action):
run_command(f"kaggle competitions download {competition} --path {WORKSPACE_ROOT}")
# if not os.path.exists(data_path):
if True:
if not os.path.exists(data_path):
# if True:
# run_command(f"rm -r {data_path / '*'}")
run_command(f"unzip -o {WORKSPACE_ROOT / '*.zip'} -d {data_path}") # FIXME: not safe

19
metagpt/roles/ml_engineer.py
View file

@ -48,13 +48,11 @@ class MLEngineer(Role):
if latest_event == DownloadData:
self.plan.context = memories[-1].content
elif latest_event == SubmitResult:
# self reflect on previous plan outcomes and think about how to improve the plan, add to working memory
await self._reflect()
# get feedback for improvement from human, add to working memory
await self._ask_review(trigger=ReviewConst.TASK_REVIEW_TRIGGER)
# self reflect on previous plan outcomes and think about how to improve the plan, add to working memory
prev_plan_outcomes = memories[-1].content
reflection = await Reflect().run(context=prev_plan_outcomes)
self.working_memory.add(Message(content=reflection, role="assistant"))
### Common Procedure in both single- and multi-agent setting ###
# create initial plan and update until confirmation
@ -172,7 +170,16 @@ class MLEngineer(Role):
self.plan.replace_task(tasks[0])
else:
self.plan.add_tasks(tasks)
self.working_memory.clear()
self.working_memory.clear()
async def _reflect(self):
context = self.get_memories()
context = "\n".join([str(msg) for msg in context])
# print("*" * 10)
# print(context)
# print("*" * 10)
reflection = await Reflect().run(context=context)
self.working_memory.add(Message(content=reflection, role="assistant"))
def get_useful_memories(self) -> List[Message]:
"""find useful memories only to reduce context length and improve performance"""