From 7a69b578236ae8a29d3536f4f76fe96a068af4a2 Mon Sep 17 00:00:00 2001 From: Kalo Chin Date: Thu, 17 Jul 2025 23:16:47 +0900 Subject: [PATCH] Fix jinja2 variable naming inconsistencies (#22578) --- api/core/workflow/nodes/llm/node.py | 14 +++++++------- web/app/components/workflow/nodes/llm/panel.tsx | 4 ++-- .../components/workflow/nodes/llm/use-config.ts | 4 ++-- 3 files changed, 11 insertions(+), 11 deletions(-) diff --git a/api/core/workflow/nodes/llm/node.py b/api/core/workflow/nodes/llm/node.py index 9bfb402dc..be0675a0f 100644 --- a/api/core/workflow/nodes/llm/node.py +++ b/api/core/workflow/nodes/llm/node.py @@ -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: diff --git a/web/app/components/workflow/nodes/llm/panel.tsx b/web/app/components/workflow/nodes/llm/panel.tsx index 471d65ef2..b47542926 100644 --- a/web/app/components/workflow/nodes/llm/panel.tsx +++ b/web/app/components/workflow/nodes/llm/panel.tsx @@ -60,7 +60,7 @@ const Panel: FC> = ({ setStructuredOutputCollapsed, handleStructureOutputEnableChange, handleStructureOutputChange, - filterJinjia2InputVar, + filterJinja2InputVar, } = useConfig(id, data) const model = inputs.model @@ -166,7 +166,7 @@ const Panel: FC> = ({ list={inputs.prompt_config?.jinja2_variables || []} onChange={handleVarListChange} onVarNameChange={handleVarNameChange} - filterVar={filterJinjia2InputVar} + filterVar={filterJinja2InputVar} isSupportFileVar={false} /> diff --git a/web/app/components/workflow/nodes/llm/use-config.ts b/web/app/components/workflow/nodes/llm/use-config.ts index e4da65bed..b8516caed 100644 --- a/web/app/components/workflow/nodes/llm/use-config.ts +++ b/web/app/components/workflow/nodes/llm/use-config.ts @@ -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, } }