Browse Source

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/0.14.2
-LAN- 10 months ago
parent
commit
5e37ab60d8
No account linked to committer's email address
1 changed files with 5 additions and 2 deletions
  1. 5
    2
      api/core/helper/code_executor/template_transformer.py

+ 5
- 2
api/core/helper/code_executor/template_transformer.py View File

@@ -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

Loading…
Cancel
Save