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

update for make tools test.

Browse source
This commit is contained in:
刘棒棒 2023-12-19 10:24:57 +08:00
parent 87821fc6cc
commit 4cb2028c72
1 changed files with 16 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

25
metagpt/roles/ml_engineer.py
View file

@ -48,7 +48,8 @@ class MLEngineer(Role):
self.plan = Plan(goal=goal)
self.use_tools = False
self.make_tools = True
self.make_udfs = False
self.use_udfs = False
self.use_code_steps = False
self.execute_code = ExecutePyCode()
self.auto_run = auto_run
@ -168,14 +169,19 @@ class MLEngineer(Role):
logger.info(f"new code \n{code}")
cause_by = DebugCode
elif not self.use_tools or self.plan.current_task.task_type == "other":
logger.info("Write code with pure generation")
code = await WriteCodeByGenerate().run(
context=context, plan=self.plan, temperature=0.0
)
debug_context = [self.get_useful_memories(task_exclude_field={'result', 'code_steps'})[0]]
cause_by = WriteCodeByGenerate
if self.make_tools:
# make and save tools.
if self.use_udfs:
# use user-defined function tools.
pass
else:
logger.info("Write code with pure generation")
code = await WriteCodeByGenerate().run(
context=context, plan=self.plan, temperature=0.0
)
debug_context = [self.get_useful_memories(task_exclude_field={'result', 'code_steps'})[0]]
cause_by = WriteCodeByGenerate
if self.make_udfs:
# make and save user-defined function tools.
make_tools = MakeTools()
tool_code = await make_tools.run(code)
make_tools.save(tool_code)
@ -291,6 +297,7 @@ if __name__ == "__main__":
async def main(requirement: str = requirement, auto_run: bool = True):
role = MLEngineer(goal=requirement, auto_run=auto_run)
role.make_udfs = True
await role.run(requirement)
fire.Fire(main)