This commit is contained in:
Richard Yu 2025-12-02 19:33:22 -08:00
parent b874f469ce
commit 380b6aec8e
2 changed files with 10 additions and 21 deletions

View File

@ -133,22 +133,16 @@ def get_outputs_summary(outputs):
for item in items: for item in items:
count += 1 count += 1
if preview_output is not None: if preview_output is None and is_previewable(media_type, item):
continue enriched = {
**item,
if not is_previewable(media_type, item): 'nodeId': node_id,
continue 'mediaType': media_type
}
enriched = { if item.get('type') == 'output':
**item, preview_output = enriched
'nodeId': node_id, elif fallback_preview is None:
'mediaType': media_type fallback_preview = enriched
}
if item.get('type') == 'output':
preview_output = enriched
elif fallback_preview is None:
fallback_preview = enriched
return count, preview_output or fallback_preview return count, preview_output or fallback_preview

View File

@ -1,12 +1,7 @@
"""Unit tests for comfy_execution/jobs.py""" """Unit tests for comfy_execution/jobs.py"""
import pytest
from pytest import fixture
from unittest.mock import MagicMock
from comfy_execution.jobs import ( from comfy_execution.jobs import (
JobStatus, JobStatus,
PREVIEWABLE_MEDIA_TYPES,
THREE_D_EXTENSIONS,
is_previewable, is_previewable,
normalize_queue_item, normalize_queue_item,
normalize_history_item, normalize_history_item,