fix: template parse error when pre prompt include {{}} (#462)

2023-06-27 15:51:55 +08:00 · 2023-06-27 15:51:55 +08:00 · 835bf9fd8d
commit 835bf9fd8d
parent c720f831af
1 changed files with 1 additions and 0 deletions
--- 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)