From 835bf9fd8d65ddb797f9c2e69d0fbd473704f1f4 Mon Sep 17 00:00:00 2001 From: John Wang Date: Tue, 27 Jun 2023 15:51:55 +0800 Subject: [PATCH] fix: template parse error when pre prompt include {{}} (#462) --- api/core/prompt/prompt_template.py | 1 + 1 file changed, 1 insertion(+) diff --git a/api/core/prompt/prompt_template.py b/api/core/prompt/prompt_template.py index 4337389786..c51c4700c1 100644 --- a/api/core/prompt/prompt_template.py +++ b/api/core/prompt/prompt_template.py @@ -14,6 +14,7 @@ class JinjaPromptTemplate(PromptTemplate): def from_template(cls, template: str, **kwargs: Any) -> PromptTemplate: """Load a prompt template from a template.""" env = Environment() + template = template.replace("{{}}", "{}") ast = env.parse(template) input_variables = meta.find_undeclared_variables(ast)