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

revert to default

Browse source
This commit is contained in:
stellahsr 2023-12-15 00:41:13 +08:00
parent a42c4144a1
commit 6957c2df65
1 changed files with 36 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

72
metagpt/actions/write_analysis_code.py
View file

@ -191,42 +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,
# )
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
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,