estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix jinja2 variable naming inconsistencies (#22578)

Browse Source
This commit is contained in:
Kalo Chin
2025-07-17 23:16:47 +09:00
committed by GitHub
parent 2423f97c72
commit 7a69b57823
3 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/core/workflow/nodes/llm/node.py
View File

@@ -849,7 +849,7 @@ class LLMNode(BaseNode[LLMNodeData]):
if message.edition_type == "jinja2":
result_text = _render_jinja2_message(
template=message.jinja2_text or "",
jinjia2_variables=jinja2_variables,
jinja2_variables=jinja2_variables,
variable_pool=variable_pool,
)
prompt_message = _combine_message_content_with_role(
@@ -1021,20 +1021,20 @@ def _combine_message_content_with_role(
def _render_jinja2_message(
*,
template: str,
jinjia2_variables: Sequence[VariableSelector],
jinja2_variables: Sequence[VariableSelector],
variable_pool: VariablePool,
):
if not template:
return ""
jinjia2_inputs = {}
for jinja2_variable in jinjia2_variables:
jinja2_inputs = {}
for jinja2_variable in jinja2_variables:
variable = variable_pool.get(jinja2_variable.value_selector)
jinjia2_inputs[jinja2_variable.variable] = variable.to_object() if variable else ""
jinja2_inputs[jinja2_variable.variable] = variable.to_object() if variable else ""
code_execute_resp = CodeExecutor.execute_workflow_code_template(
language=CodeLanguage.JINJA2,
code=template,
inputs=jinjia2_inputs,
inputs=jinja2_inputs,
)
result_text = code_execute_resp["result"]
return result_text
@@ -1130,7 +1130,7 @@ def _handle_completion_template(
if template.edition_type == "jinja2":
result_text = _render_jinja2_message(
template=template.jinja2_text or "",
jinjia2_variables=jinja2_variables,
jinja2_variables=jinja2_variables,
variable_pool=variable_pool,
)
else:

4
web/app/components/workflow/nodes/llm/panel.tsx
View File

@@ -60,7 +60,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
setStructuredOutputCollapsed,
handleStructureOutputEnableChange,
handleStructureOutputChange,
filterJinjia2InputVar,
filterJinja2InputVar,
} = useConfig(id, data)
const model = inputs.model
@@ -166,7 +166,7 @@ const Panel: FC<NodePanelProps<LLMNodeType>> = ({
list={inputs.prompt_config?.jinja2_variables || []}
onChange={handleVarListChange}
onVarNameChange={handleVarNameChange}
filterVar={filterJinjia2InputVar}
filterVar={filterJinja2InputVar}
isSupportFileVar={false}
/>
</Field>

4
web/app/components/workflow/nodes/llm/use-config.ts
View File

@@ -309,7 +309,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
return [VarType.number, VarType.string, VarType.secret, VarType.arrayString, VarType.arrayNumber, VarType.file, VarType.arrayFile].includes(varPayload.type)
}, [])
const filterJinjia2InputVar = useCallback((varPayload: Var) => {
const filterJinja2InputVar = useCallback((varPayload: Var) => {
return [VarType.number, VarType.string, VarType.secret, VarType.arrayString, VarType.arrayNumber].includes(varPayload.type)
}, [])
@@ -356,7 +356,7 @@ const useConfig = (id: string, payload: LLMNodeType) => {
structuredOutputCollapsed,
setStructuredOutputCollapsed,
handleStructureOutputEnableChange,
filterJinjia2InputVar,
filterJinja2InputVar,
}
}