Merge branch 'data_analyst_ldy' into 'mgx_ops'

refine code

See merge request pub/MetaGPT!245
This commit is contained in:
林义章 2024-07-22 02:25:32 +00:00
commit 7b6636b223

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,