jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
John Wang 2023-06-27 15:51:55 +08:00 • committed by GitHub
commit 835bf9fd8d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
api/core/prompt/prompt_template.py
View file

@ -14,6 +14,7 @@ class JinjaPromptTemplate(PromptTemplate):
def from_template(cls, template: str, **kwargs: Any) -> PromptTemplate: def from_template(cls, template: str, **kwargs: Any) -> PromptTemplate:
"""Load a prompt template from a template.""" """Load a prompt template from a template."""
env = Environment() env = Environment()
template = template.replace("{{}}", "{}")
ast = env.parse(template) ast = env.parse(template)
input_variables = meta.find_undeclared_variables(ast) input_variables = meta.find_undeclared_variables(ast)