wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 22:30:19 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Handle custom_nodes returning None responses more gracefully

Browse Source
This commit is contained in:
doctorpangloss 2024-10-15 11:01:21 -07:00
parent 89f6d3482e
commit 5412451def
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/vendor/aiohttp_server_instrumentation.py vendored
View File

@ -232,6 +232,10 @@ async def middleware(request, handler):
except web.HTTPException as ex: except web.HTTPException as ex:
set_status_code(span, ex.status_code) set_status_code(span, ex.status_code)
raise raise
except AttributeError:
# No response was returned or a NoneType response was returned, handle gracefully
set_status_code(span, 500)
raise
finally: finally:
duration = max((default_timer() - start) * 1000, 0) duration = max((default_timer() - start) * 1000, 0)
duration_histogram.record(duration, duration_attrs) duration_histogram.record(duration, duration_attrs)