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

Merge branch 'dev_ldy' into 'dev'

Browse source 
Dev ldy

See merge request agents/data_agents_opt!13
This commit is contained in:
林义章 2023-12-01 07:57:54 +00:00
commit 20a918bf39
5 changed files with 165 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/roles/ml_engineer.py
View file

@ -125,7 +125,7 @@ class MLEngineer(Role):
# print("*" * 10)
# breakpoint()
if not self.use_tools or self.plan.current_task.task_type == "":
if not self.use_tools or self.plan.current_task.task_type == "other":
# code = "print('abc')"
code = await WriteCodeByGenerate().run(
context=context, plan=self.plan, task_guide=task_guide, temperature=0.0
@ -171,7 +171,9 @@ class MLEngineer(Role):
plan_confirmed = False
while not plan_confirmed:
context = self.get_useful_memories()
rsp = await WritePlan().run(context, max_tasks=max_tasks)
rsp = await WritePlan().run(
context, max_tasks=max_tasks, use_tools=self.use_tools
)
self.working_memory.add(
Message(content=rsp, role="assistant", cause_by=WritePlan)
)