Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 19:59:50 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

refactor(prompt): improve handling of variable templates in advanced prompt transform

Browse Source
This commit is contained in:
-LAN- 2024-10-14 12:15:56 +08:00
parent 03018823d8
commit 0d310b503b
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
api/core/prompt/advanced_prompt_transform.py
View File

@ -145,11 +145,20 @@ class AdvancedPromptTransform(PromptTransform):
raw_prompt = prompt_item.text
if prompt_item.edition_type == "basic" or not prompt_item.edition_type:
vp = VariablePool()
for k, v in inputs.items():
vp.add(k[1:-1].split("."), v)
raw_prompt.replace("{{#context#}}", context or "")
prompt = vp.convert_template(raw_prompt).text
if self.with_variable_tmpl:
vp = VariablePool()
for k, v in inputs.items():
if k.startswith("#"):
vp.add(k[1:-1].split("."), v)
raw_prompt.replace("{{#context#}}", context or "")
prompt = vp.convert_template(raw_prompt).text
else:
parser = PromptTemplateParser(template=raw_prompt, with_variable_tmpl=self.with_variable_tmpl)
prompt_inputs = {k: inputs[k] for k in parser.variable_keys if k in inputs}
prompt_inputs = self._set_context_variable(
context=context, parser=parser, prompt_inputs=prompt_inputs
)
prompt = parser.format(prompt_inputs)
elif prompt_item.edition_type == "jinja2":
prompt = raw_prompt
prompt_inputs = inputs