youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 10:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: union type handling

Browse Source
This commit is contained in:
Stream
2026-01-28 02:06:18 +08:00
parent c46856d5ac
commit 71d44ec52c
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1406,7 +1406,8 @@ class LLMNode(Node[LLMNodeData]):
variable_selectors = [] variable_selectors = []
if isinstance(prompt_template, list): if isinstance(prompt_template, list):
for prompt in prompt_template: for prompt in prompt_template:
if prompt.edition_type != "jinja2": prompt: LLMNodeChatModelMessage | PromptMessageContext
if isinstance(prompt, LLMNodeChatModelMessage) and prompt.edition_type == "jinja2":
variable_template_parser = VariableTemplateParser(template=prompt.text) variable_template_parser = VariableTemplateParser(template=prompt.text)
variable_selectors.extend(variable_template_parser.extract_variable_selectors()) variable_selectors.extend(variable_template_parser.extract_variable_selectors())
elif isinstance(prompt_template, LLMNodeCompletionModelPromptTemplate): elif isinstance(prompt_template, LLMNodeCompletionModelPromptTemplate):
@ -1442,12 +1443,13 @@ class LLMNode(Node[LLMNodeData]):
if isinstance(prompt_template, list): if isinstance(prompt_template, list):
for prompt in prompt_template: for prompt in prompt_template:
if prompt.edition_type == "jinja2": prompt: LLMNodeChatModelMessage | PromptMessageContext
if isinstance(prompt, LLMNodeChatModelMessage) and prompt.edition_type == "jinja2":
enable_jinja = True enable_jinja = True
break break
else: else:
if prompt_template.edition_type == "jinja2": prompt_template: LLMNodeCompletionModelPromptTemplate
enable_jinja = True enable_jinja = True
if enable_jinja: if enable_jinja:
for variable_selector in typed_node_data.prompt_config.jinja2_variables or []: for variable_selector in typed_node_data.prompt_config.jinja2_variables or []: