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 code locally

Browse source
This commit is contained in:
stellahsr 2023-12-13 14:33:50 +08:00
parent 1da4409475
commit ea0b93d2b9
2 changed files with 21 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

26
metagpt/roles/ml_engineer.py
View file

@ -148,7 +148,7 @@ class MLEngineer(Role):
while self.plan.current_task:
task = self.plan.current_task
logger.info(f"ready to take on task {task}")
logger.info(f"ready to take on task: {task}")
# take on current task
code, result, success, code_steps = await self._write_and_exec_code()
@ -157,9 +157,11 @@ class MLEngineer(Role):
task_result_confirmed = await self._ask_review()
# 针对当前task进行单独plan
if not success or not task_result_confirmed:
# fixme: 增加对应plan
self.state.plan()
# if not success or not task_result_confirmed:
# # fixme: 增加对应plan
# logger.info(task.result)
# # import pdb;pdb.set_trace()
# # self.state.plan()
if success and task_result_confirmed:
# tick off this task and record progress
@ -175,13 +177,13 @@ class MLEngineer(Role):
# update plan according to user's feedback and to take on changed tasks
await self._update_plan()
finished_tasks = self.plan.get_finished_tasks()
if len(finished_tasks) == len(self.plan.tasks):
code_context = [task.code for task in finished_tasks]
code_context = "\n\n".join(code_context)
result, success = await self.execute_code.run(code_context)
# truncated the result
print(truncate(result))
# finished_tasks = self.plan.get_finished_tasks()
# if len(finished_tasks) == len(self.plan.tasks):
# code_context = [task.code for task in finished_tasks]
# code_context = "\n\n".join(code_context)
# result, success = await self.execute_code.run(code_context)
# # truncated the result
# print(truncate(result))
async def _generate_data_desc(self):
data_desc = await GenerateDataDesc().run(self.data_path)
@ -258,8 +260,6 @@ class MLEngineer(Role):
counter += 1
success = False
return code, result, success, code_steps
async def _ask_review(self):