This commit is contained in:
lidanyang 2024-07-22 09:48:16 +08:00
parent d2f6fc2594
commit 70bddcfd3c

View file

@ -31,7 +31,7 @@ class WriteAnalysisCode(Action):
rsp = await self._aask(reflection_prompt, system_msgs=[REFLECTION_SYSTEM_MSG])
reflection = json.loads(CodeParser.parse_code(text=rsp))
if '```python' in reflection["improved_impl"]:
if "```python" in reflection["improved_impl"]:
reflection["improved_impl"] = CodeParser.parse_code(text=reflection["improved_impl"], lang="python")
return reflection["improved_impl"]