Ver código fonte

fix: validate response type in transform_response method (#11931)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.14.2
-LAN- 10 meses atrás
pai
commit
5e37ab60d8
Nenhuma conta vinculada ao e-mail do autor do commit

+ 5
- 2
api/core/helper/code_executor/template_transformer.py Ver arquivo

@@ -33,13 +33,16 @@ class TemplateTransformer(ABC):
return result

@classmethod
def transform_response(cls, response: str) -> dict:
def transform_response(cls, response: str):
"""
Transform response to dict
:param response: response
:return:
"""
return json.loads(cls.extract_result_str_from_response(response))
result = json.loads(cls.extract_result_str_from_response(response))
if not isinstance(result, dict):
raise ValueError("Result must be a dict")
return result

@classmethod
@abstractmethod

Carregando…
Cancelar
Salvar