From ca2fa1b03db91b9070e29b0c2c7e67377aa19085 Mon Sep 17 00:00:00 2001 From: zhangheng Date: Mon, 18 May 2026 19:41:12 +0800 Subject: [PATCH] fix(image): tolerate shell-sourced REFERENCE_IMAGES --- gpt_image2_generate_multi.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/gpt_image2_generate_multi.py b/gpt_image2_generate_multi.py index d226087..bbbdca8 100644 --- a/gpt_image2_generate_multi.py +++ b/gpt_image2_generate_multi.py @@ -71,7 +71,15 @@ def split_reference_values(raw_value: str) -> list[str]: return [] if value.startswith("["): - parsed = json.loads(value) + try: + parsed = json.loads(value) + except json.JSONDecodeError: + inner = value[1:-1].strip() + if not inner: + return [] + parts = [part.strip().strip('"').strip("'") for part in inner.split(",")] + return [part for part in parts if part] + if not isinstance(parsed, list): raise RuntimeError("REFERENCE_IMAGES 必须是 JSON 数组") return [str(item).strip() for item in parsed if str(item).strip()]