fix: code

This commit is contained in:
Yeuoly
2024-03-17 21:08:25 +08:00
parent 73c2b35dfe
commit d8ab611480
9 changed files with 24 additions and 7 deletions

View File

@ -72,7 +72,7 @@ class CodeExecutor:
response = response.json()
except:
raise CodeExecutionException('Failed to parse response')
response = CodeExecutionResponse(**response)
if response.code != 0:

View File

@ -48,7 +48,7 @@ class PythonTemplateTransformer(TemplateTransformer):
:return:
"""
# extract result
result = re.search(r'<<RESULT>>(.*)<<RESULT>>', response, re.DOTALL)
result = re.search(r'<<RESULT>>(.*?)<<RESULT>>', response, re.DOTALL)
if not result:
raise ValueError('Failed to parse result')
result = result.group(1)