From ef6e4a1b77a21cefeb165301dd1d47b5c273fdbb Mon Sep 17 00:00:00 2001 From: lidanyang Date: Thu, 14 Dec 2023 13:46:27 +0800 Subject: [PATCH] debug only when use_tools --- metagpt/roles/ml_engineer.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py index 51faf1e0d..3755e7bac 100644 --- a/metagpt/roles/ml_engineer.py +++ b/metagpt/roles/ml_engineer.py @@ -147,7 +147,6 @@ class MLEngineer(Role): ) counter = 0 - improve_code = "" success = False debug_context = [] @@ -158,17 +157,14 @@ class MLEngineer(Role): # print(context) # print("*" * 10) # breakpoint() - if counter > 0: - improve_code = await DebugCode().run( + if counter > 0 and self.use_tools: + code = await DebugCode().run( plan=self.plan.current_task.instruction, code=code, runtime_result=self.working_memory.get(), context=debug_context ) - - if improve_code != "": - code = improve_code - logger.info(f"new code \n{improve_code}") + 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")