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

add only_code arg for WriteCodeByGenerate.

Browse source
This commit is contained in:
刘棒棒 2024-01-15 16:59:49 +08:00
parent b69f2be165
commit 4f93c5fad3
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
metagpt/actions/write_analysis_code.py
View file

@ -88,8 +88,14 @@ class WriteCodeByGenerate(BaseWriteAnalysisCode):
) -> str:
# context.append(Message(content=self.REUSE_CODE_INSTRUCTION, role="user"))
prompt = self.process_msg(context, system_msg)
is_only_code = kwargs.pop("only_code", True)
code_content = await self.llm.aask_code(prompt, **kwargs)
return code_content["code"]
if is_only_code:
return code_content["code"]
else:
return code_content
class WriteCodeWithTools(BaseWriteAnalysisCode):