From ecda6fc464ad9564cb9de563a6a1611e61a8a0c4 Mon Sep 17 00:00:00 2001 From: Devbyteai Date: Fri, 26 Dec 2025 20:27:32 +0200 Subject: [PATCH] refactor: move imports to top of generated script, remove redundancy --- .../helper/code_executor/jinja2/jinja2_transformer.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/api/core/helper/code_executor/jinja2/jinja2_transformer.py b/api/core/helper/code_executor/jinja2/jinja2_transformer.py index 571db7d24a..5e4807401e 100644 --- a/api/core/helper/code_executor/jinja2/jinja2_transformer.py +++ b/api/core/helper/code_executor/jinja2/jinja2_transformer.py @@ -36,18 +36,17 @@ class Jinja2TemplateTransformer(TemplateTransformer): @classmethod def get_runner_script(cls) -> str: runner_script = dedent(f""" + import jinja2 + import json + from base64 import b64decode + # declare main function def main(**inputs): - import jinja2 - from base64 import b64decode # Decode base64-encoded template to handle special characters safely template_code = b64decode('{cls._template_b64_placeholder}').decode('utf-8') template = jinja2.Template(template_code) return template.render(**inputs) - import json - from base64 import b64decode - # decode and prepare input dict inputs_obj = json.loads(b64decode('{cls._inputs_placeholder}').decode('utf-8'))