Prechádzať zdrojové kódy

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 mesiacov pred
rodič
commit
5e37ab60d8

+ 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