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

fix(plugin): ensure node_errors is sent

Browse Source
This commit is contained in:
BennyKok 2023-12-29 16:53:32 +08:00
parent 62f8ba2c70
commit 6b99b8a66a
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
custom_routes.py
View File

@ -107,15 +107,30 @@ async def comfy_deploy_run(request):
stack_trace = traceback.format_exc().strip() stack_trace = traceback.format_exc().strip()
print(f"error: {error_type}, {e}") print(f"error: {error_type}, {e}")
print(f"stack trace: {stack_trace_short}") print(f"stack trace: {stack_trace_short}")
asyncio.create_task(update_run_with_output(prompt_id, { await update_run_with_output(prompt_id, {
"error_type": error_type, "error": {
"stack_trace": stack_trace "error_type": error_type,
})) "stack_trace": stack_trace
}
})
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 status = 400
await update_run_with_output(prompt_id, {
"error": {
**res
}
})
if "node_errors" in res:
status = 400
await update_run_with_output(prompt_id, {
"error": {
**res
}
})
return web.json_response(res, status=status) return web.json_response(res, status=status)