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

fix: load $context correctly in step run llm node

Browse Source
This commit is contained in:
Stream
2026-01-28 06:35:01 +08:00
parent 403114eee9
commit 9d287647c1
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1404,12 +1404,18 @@ class LLMNode(Node[LLMNodeData]):
prompt_template = typed_node_data.prompt_template
variable_selectors = []
prompt_context_selectors: list[Sequence[str]] = []
if isinstance(prompt_template, list):
for prompt in prompt_template:
prompt: LLMNodeChatModelMessage | PromptMessageContext
if isinstance(prompt, LLMNodeChatModelMessage) and prompt.edition_type == "jinja2":
variable_template_parser = VariableTemplateParser(template=prompt.text)
variable_selectors.extend(variable_template_parser.extract_variable_selectors())
continue
if isinstance(prompt, PromptMessageContext):
if len(prompt.value_selector) < 2:
continue
prompt_context_selectors.append(prompt.value_selector)
elif isinstance(prompt_template, LLMNodeCompletionModelPromptTemplate):
if prompt_template.edition_type != "jinja2":
variable_template_parser = VariableTemplateParser(template=prompt_template.text)
@ -1421,6 +1427,10 @@ class LLMNode(Node[LLMNodeData]):
for variable_selector in variable_selectors:
variable_mapping[variable_selector.variable] = variable_selector.value_selector
for context_selector in prompt_context_selectors:
variable_key = f"#{'.'.join(context_selector)}#"
variable_mapping[variable_key] = list(context_selector)
memory = typed_node_data.memory
if memory and memory.query_prompt_template:
query_variable_selectors = VariableTemplateParser(