fix output code format

This commit is contained in:
lidanyang 2024-07-19 17:03:37 +08:00
parent 976df80074
commit aa92662487

View file

@ -31,8 +31,12 @@ 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"]
return reflection["improved_impl"]
return improved_impl
async def run(
self,