apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-07-23 17:01:08 +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'])