소스 검색

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 달 전
부모
커밋
5e37ab60d8
1개의 변경된 파일5개의 추가작업 그리고 2개의 파일을 삭제
  1. 5 2
      api/core/helper/code_executor/template_transformer.py

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

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