chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(plugin): set status to failed if there are node_errors

Browse Source
This commit is contained in:
BennyKok 2023-12-30 02:48:48 +08:00
parent 824ce16b4e
commit 744a222e26
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
custom_routes.py
View File

@ -116,7 +116,16 @@ async def comfy_deploy_run(request):
return web.Response(status=500, reason=f"{error_type}: {e}, {stack_trace_short}") return web.Response(status=500, reason=f"{error_type}: {e}, {stack_trace_short}")
status = 200 status = 200
if "error" in res: # if "error" in res:
# status = 400
# await update_run_with_output(prompt_id, {
# "error": {
# **res
# }
# })
if "node_errors" in res and res["node_errors"]:
# Even tho there are node_errors it can still be run
status = 400 status = 400
await update_run_with_output(prompt_id, { await update_run_with_output(prompt_id, {
"error": { "error": {
@ -124,13 +133,9 @@ async def comfy_deploy_run(request):
} }
}) })
if "node_errors" in res and res["node_errors"]: # When there are critical errors, the prompt is actually not run
status = 400 if "error" in res:
await update_run_with_output(prompt_id, { await update_run(prompt_id, Status.FAILED)
"error": {
**res
}
})
return web.json_response(res, status=status) return web.json_response(res, status=status)