From fdf16f55352102d002223af6ee4d054622be0e79 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= Date: Tue, 19 Dec 2023 14:38:08 +0800 Subject: [PATCH] add code_prompt for make tools. --- metagpt/roles/ml_engineer.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py index 75c403226..96e21c8c8 100644 --- a/metagpt/roles/ml_engineer.py +++ b/metagpt/roles/ml_engineer.py @@ -180,10 +180,12 @@ class MLEngineer(Role): debug_context = [self.get_useful_memories(task_exclude_field={'result', 'code_steps'})[0]] cause_by = WriteCodeByGenerate - if self.make_udfs: + if self.make_udfs and len(code.split('\n')) > 2: # make and save user-defined function tools. make_tools = MakeTools() - tool_code = await make_tools.run(code) + code_prompt = f"The following code is about {self.plan.current_task.instruction},\ + convert it to be a General Function, {code}" + tool_code = await make_tools.run(code_prompt) make_tools.save(tool_code) else: logger.info("Write code with tools")