fix: sync LoadVideo 'file' media refs for remote workers

This commit is contained in:
Robert Wojciechowski
2026-02-28 04:53:20 +00:00
parent 9058cf061d
commit df606af80e
2 changed files with 7 additions and 2 deletions
+2 -2
View File
@@ -57,13 +57,13 @@ def convert_paths_for_platform(obj, target_separator):
def _find_media_references(prompt_obj):
"""Find media file references in image/video/audio inputs used by worker prompts."""
"""Find media file references in image/video/audio/file inputs used by worker prompts."""
media_refs = set()
for node in prompt_obj.values():
if not isinstance(node, dict):
continue
inputs = node.get("inputs", {})
for key in ("image", "video", "audio"):
for key in ("image", "video", "audio", "file"):
value = inputs.get(key)
if isinstance(value, str):
cleaned = re.sub(r"\s*\[\w+\]$", "", value).strip().replace("\\", "/")
+5
View File
@@ -174,6 +174,11 @@ class FindMediaReferencesTests(unittest.TestCase):
refs = ms._find_media_references(prompt)
self.assertIn("clip.mp4", refs)
def test_finds_file_input_for_load_video(self):
prompt = {"1": {"class_type": "LoadVideo", "inputs": {"file": "1 - Copy.mp4"}}}
refs = ms._find_media_references(prompt)
self.assertIn("1 - Copy.mp4", refs)
def test_finds_audio_input(self):
prompt = {"1": {"class_type": "LoadAudio", "inputs": {"audio": "track.wav"}}}
refs = ms._find_media_references(prompt)