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

refine code

Browse source
This commit is contained in:
lidanyang 2024-07-19 18:31:06 +08:00
parent 9c655e8828
commit d2f6fc2594
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
metagpt/actions/di/write_analysis_code.py
View file

@ -31,12 +31,10 @@ class WriteAnalysisCode(Action):
rsp = await self._aask(reflection_prompt, system_msgs=[REFLECTION_SYSTEM_MSG])
reflection = json.loads(CodeParser.parse_code(text=rsp))
try:
improved_impl = CodeParser.parse_code(text=reflection["improved_impl"], lang="python")
except Exception as e:
improved_impl = reflection["improved_impl"]
if '```python' in reflection["improved_impl"]:
reflection["improved_impl"] = CodeParser.parse_code(text=reflection["improved_impl"], lang="python")
return improved_impl
return reflection["improved_impl"]
async def run(
self,