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

feat: add auto_run.

Browse source
This commit is contained in:
刘棒棒 2023-11-28 13:40:28 +08:00
parent 311cb5e8b4
commit 460e373dae
1 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
metagpt/roles/ml_engineer.py
View file

@ -38,13 +38,14 @@ class WriteTaskGuide(Action):
return ""
class MLEngineer(Role):
def __init__(self, name="ABC", profile="MLEngineer", goal=""):
def __init__(self, name="ABC", profile="MLEngineer", goal="", auto_run: bool = False):
super().__init__(name=name, profile=profile, goal=goal)
self._set_react_mode(react_mode="plan_and_act")
self.plan = Plan(goal=goal)
self.use_tools = False
self.use_task_guide = False
self.execute_code = ExecutePyCode()
self.auto_run = auto_run
async def _plan_and_act(self):
@ -104,11 +105,13 @@ class MLEngineer(Role):
return code, result, success
async def _ask_review(self):
context = self.get_useful_memories()
review, confirmed = await AskReview().run(context=context[-5:], plan=self.plan)
if review.lower() not in ("confirm", "y", "yes"):
self._rc.memory.add(Message(content=review, role="user", cause_by=AskReview))
return confirmed
if not self.auto_run:
context = self.get_useful_memories()
review, confirmed = await AskReview().run(context=context[-5:], plan=self.plan)
if review.lower() not in ("confirm", "y", "yes"):
self._rc.memory.add(Message(content=review, role="user", cause_by=AskReview))
return confirmed
return True
async def _update_plan(self, max_tasks: int = 3):
current_plan = str([task.json() for task in self.plan.tasks])