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

feat: add support for file outputs in LLMNode, including auto-collection and deduplication

Browse Source
This commit is contained in:
Novice
2026-03-11 16:30:23 +08:00
parent f0c82f28cb
commit 8513fa2897
5 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -573,7 +573,11 @@ class LLMNode(Node[LLMNodeData]):
"tool_calls": [self._serialize_tool_call(item) for item in generation_data.tool_calls],
"sequence": generation_data.sequence,
}
files_to_output = generation_data.files
files_to_output = list(generation_data.files)
# Merge auto-collected/structured-output files from self._file_outputs
if self._file_outputs:
existing_ids = {f.id for f in files_to_output}
files_to_output.extend(f for f in self._file_outputs if f.id not in existing_ids)
else:
# Classical runtime: use pre-computed generation-specific text pair,
# falling back to native model reasoning if no <think> tags were found.
@ -2126,6 +2130,12 @@ class LLMNode(Node[LLMNodeData]):
result = yield from self._process_tool_outputs(outputs)
# Auto-collect sandbox output/ files, deduplicate by id
collected_files = session.collect_output_files()
if collected_files:
existing_ids = {f.id for f in self._file_outputs}
self._file_outputs.extend(f for f in collected_files if f.id not in existing_ids)
if result is None:
raise LLMNodeError("SandboxSession exited unexpectedly")