fix: 🐛 missing keyname

This commit is contained in:
Teddy Xiong
2026-03-26 07:27:57 +08:00
parent 3538a3de0c
commit 5e9fe5b700
+6 -4
View File
@@ -32,15 +32,17 @@ ASYNC_OUTPUT_EXTRA_KEY = "ASYNC_OUTPUT"
ASYNC_OUTPUT_INIT_SINGAL_KEY = "INIT_SIGNAL"
def onprompt(json_data):
data_info = json_data.workflow.extra
if ASYNC_OUTPUT_EXTRA_KEY not in json_data.extra:
json_data.extra[ASYNC_OUTPUT_EXTRA_KEY] = {}
if ASYNC_OUTPUT_EXTRA_KEY not in data_info:
data_info[ASYNC_OUTPUT_EXTRA_KEY] = {}
ASYNC_OUTPUT_EXTRA_DICT = json_data.extra[ASYNC_OUTPUT_EXTRA_KEY]
ASYNC_OUTPUT_EXTRA_DICT = data_info[ASYNC_OUTPUT_EXTRA_KEY]
if ASYNC_OUTPUT_INIT_SINGAL_KEY not in ASYNC_OUTPUT_EXTRA_DICT:
ASYNC_OUTPUT_EXTRA_DICT[ASYNC_OUTPUT_INIT_SINGAL_KEY] = False
json_data.extra[ASYNC_OUTPUT_EXTRA_KEY][ASYNC_OUTPUT_INIT_SINGAL_KEY] = False
data_info[ASYNC_OUTPUT_EXTRA_KEY][ASYNC_OUTPUT_INIT_SINGAL_KEY] = False
return json_data