diff --git a/nodes/image_modal_nodes.py b/nodes/image_modal_nodes.py index fa8af13..03742fd 100644 --- a/nodes/image_modal_nodes.py +++ b/nodes/image_modal_nodes.py @@ -306,16 +306,17 @@ class ModalMidJourneyGenerateImage: raise Exception("生成失败, 可能因为风控") job_id = job_resp["data"] start_time = time() - for _ in range(0, timeout // interval, interval): - logger.info(f"已等待 {time() - start_time} 秒,{interval} 秒后查询...") + while start_time + timeout > time(): + logger.info(f"已等待 {(int(time() - start_time))} 秒,{interval} 秒后查询...") sleep(interval) - logger.info(f"查询结果 {job_id}") + logger.info(f"开始查询结果 {job_id}") resp = send_request( "get", f"https://{endpoint}/api/custom/task/status?task_id={job_id}", timeout=30, ) resp.raise_for_status() + logger.info(f"查询结果 {resp.json()['msg']}") if resp.json()["status"] == "running": logger.info(f"任务正在运行 {job_id}") continue