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

update: 按code step 逐一生成代码

Browse source
This commit is contained in:
stellahsr 2023-12-14 10:46:43 +08:00
parent 8d694d47d9
commit e4ee3efeb8
1 changed files with 28 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

30
metagpt/actions/write_analysis_code.py
View file

@ -202,8 +202,34 @@ class WriteCodeWithTools(BaseWriteAnalysisCode):
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,