diff --git a/api/core/helper/code_executor/code_executor.py b/api/core/helper/code_executor/code_executor.py index b70f57680d..063a21b192 100644 --- a/api/core/helper/code_executor/code_executor.py +++ b/api/core/helper/code_executor/code_executor.py @@ -6,7 +6,7 @@ from yarl import URL from config import get_env from core.helper.code_executor.javascript_transformer import NodeJsTemplateTransformer -from core.helper.code_executor.jina2_transformer import Jinja2TemplateTransformer +from core.helper.code_executor.jinja2_transformer import Jinja2TemplateTransformer from core.helper.code_executor.python_transformer import PythonTemplateTransformer # Code Executor diff --git a/api/core/helper/code_executor/jina2_transformer.py b/api/core/helper/code_executor/jinja2_transformer.py similarity index 99% rename from api/core/helper/code_executor/jina2_transformer.py rename to api/core/helper/code_executor/jinja2_transformer.py index 8d97a28e85..27a3579493 100644 --- a/api/core/helper/code_executor/jina2_transformer.py +++ b/api/core/helper/code_executor/jinja2_transformer.py @@ -55,6 +55,7 @@ if __name__ == '__main__': """ + class Jinja2TemplateTransformer(TemplateTransformer): @classmethod def transform_caller(cls, code: str, inputs: dict) -> tuple[str, str]: