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

update code prompt for make tools.

Browse source
This commit is contained in:
刘棒棒 2023-12-21 17:16:33 +08:00
parent 6d36511249
commit bb7f4c3310
2 changed files with 14 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
metagpt/actions/write_analysis_code.py
View file

@ -270,9 +270,18 @@ class MakeTools(WriteCodeByGenerate):
saved_path.write_text(tool_code, encoding='utf-8')
@retry(stop=stop_after_attempt(3), wait=wait_fixed(1))
async def run(self, code_message: List[Message | Dict], **kwargs) -> str:
msgs = self.process_msg(code_message, self.DEFAULT_SYSTEM_MSG)
async def run(self, code: str, code_desc: str = None, **kwargs) -> str:
# 拼接code prompt
code_prompt = f"The following code is about {code_desc}, convert it to be a General Function, {code}"
msgs = self.process_msg(code_prompt, self.DEFAULT_SYSTEM_MSG)
logger.info(f"\n\nAsk to Make tools:\n{'-'*60}\n {msgs[-1]}")
# 更新kwargs
if 'code' in kwargs:
kwargs.pop('code')
if 'code_desc' in kwargs:
kwargs.pop('code_desc')
tool_code = await self.llm.aask_code(msgs, **kwargs)
max_tries, current_try = 3, 1
func_name = self.parse_function_name(tool_code['code'])

7
metagpt/roles/ml_engineer.py
View file

@ -291,15 +291,14 @@ class MLEngineer(Role):
logger.warning(f"Making tools for task_id {self.plan.current_task_id}: \
`{self.plan.current_task.instruction}` \n code: \n {code}")
make_tools = MakeTools()
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)
tool_code = await make_tools.run(code, self.plan.current_task.instruction)
# check tool_code by execute_code
logger.info(f"Checking task_id {self.plan.current_task_id} tool code by executor...")
_, success = await self.execute_code.run(tool_code)
make_tool_retries, make_tool_current_retry = 3, 1
while not success:
tool_code = await make_tools.run(code_prompt)
# tool_code = await make_tools.run(code_prompt)
tool_code = await make_tools.run(code)
_, success = await self.execute_code.run(tool_code)
if make_tool_current_retry > make_tool_retries:
logger.error(f"We have tried the maximum number of attempts {make_tool_retries}\