From 890ffc9f020cb8f3d5e2bb37bdb7e78e0445c5fd Mon Sep 17 00:00:00 2001
From: puke <zijiao.wzj@alibaba-inc.com>
Date: Wed, 2 Jul 2025 11:19:46 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E8=A7=A3=E6=9E=90=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 oneapi.py | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/oneapi.py b/oneapi.py
index 2f2b9e5..3225471 100644
--- a/oneapi.py
+++ b/oneapi.py
@@ -1,6 +1,7 @@
 import os
 import json
 import traceback
+from typing import Sequence
 import uuid
 import time
 import copy
@@ -517,11 +518,17 @@ def _split_media_by_suffix(node_output, base_url):
     audios = []
     for media_key in ("images", "gifs", "audio"):
         for media_data in node_output.get(media_key, []):
-            if not isinstance(media_data, dict):
+            if (isinstance(media_data, list) or isinstance(media_data, tuple)) and len(media_data) == 2:
+                subfolder = ""
+                filename, media_type = media_data
+            elif isinstance(media_data, dict):
+                filename = media_data.get("filename")
+                subfolder = media_data.get("subfolder", "")
+                media_type = media_data.get("type", "output")
+            else:
+                print(f"Invalid media data: {media_key} | {media_data}")
                 continue
-            filename = media_data.get("filename")
-            subfolder = media_data.get("subfolder", "")
-            media_type = media_data.get("type", "output")
+            
             url = f"{base_url}/view?filename={filename}"
             if subfolder:
                 url += f"&subfolder={subfolder}"