From d2f6fc2594042226d31d4c28cc2405aba3ffea21 Mon Sep 17 00:00:00 2001 From: lidanyang Date: Fri, 19 Jul 2024 18:31:06 +0800 Subject: [PATCH] refine code --- metagpt/actions/di/write_analysis_code.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/metagpt/actions/di/write_analysis_code.py b/metagpt/actions/di/write_analysis_code.py index 39de18987..ddfbe956d 100644 --- a/metagpt/actions/di/write_analysis_code.py +++ b/metagpt/actions/di/write_analysis_code.py @@ -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,