From a42c4144a11058f3d44c2f8f154437de449cd506 Mon Sep 17 00:00:00 2001 From: stellahsr Date: Fri, 15 Dec 2023 00:37:01 +0800 Subject: [PATCH] iterative step to code --- metagpt/actions/write_analysis_code.py | 46 ++++++++++++++++++++------ 1 file changed, 36 insertions(+), 10 deletions(-) diff --git a/metagpt/actions/write_analysis_code.py b/metagpt/actions/write_analysis_code.py index 6970fb4f0..0d548b806 100644 --- a/metagpt/actions/write_analysis_code.py +++ b/metagpt/actions/write_analysis_code.py @@ -191,16 +191,42 @@ class WriteCodeWithTools(BaseWriteAnalysisCode): module_name = ML_MODULE_MAP[task_type] - prompt = TOOL_USAGE_PROMPT.format( - user_requirement=plan.goal, - history_code=code_context, - current_task=plan.current_task.instruction, - column_info=column_info, - special_prompt=special_prompt, - code_steps=code_steps, - module_name=module_name, - tool_catalog=tool_catalog, - ) + # prompt = TOOL_USAGE_PROMPT.format( + # user_requirement=plan.goal, + # history_code=code_context, + # current_task=plan.current_task.instruction, + # column_info=column_info, + # special_prompt=special_prompt, + # code_steps=code_steps, + # module_name=module_name, + # tool_catalog=tool_catalog, + # ) + code_steps_ = eval(code_steps) + print(code_steps_) + + new_code = "" + tool_context = "" + for idx, (step_id, step_instruction) in enumerate(code_steps_.items()): + prompt = TOOL_USAGE_PROMPT.format( + user_requirement=plan.goal, + history_code=code_context, + current_task=plan.current_task.instruction, + column_info=column_info, + special_prompt=special_prompt, + code_steps=step_instruction, + module_name=module_name, + tool_catalog=tool_catalog, + ) + + tool_config = create_func_config(CODE_GENERATOR_WITH_TOOLS) + + rsp = await self.llm.aask_code(prompt, **tool_config) + logger.info(f"rsp is: {rsp}") + new_code = new_code + "\n\n" + rsp["code"] + code_context = code_context + "\n\n" + new_code + tool_context = tool_context + "\n\n" + prompt + context = [Message(content=tool_context, role="user")] + return context, new_code else: prompt = GENERATE_CODE_PROMPT.format( user_requirement=plan.goal,