From 5412451def2e0df8cc34399965bfe65de30f4d87 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Tue, 15 Oct 2024 11:01:21 -0700
Subject: [PATCH] Handle custom_nodes returning None responses more gracefully

---
 comfy/vendor/aiohttp_server_instrumentation.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/vendor/aiohttp_server_instrumentation.py b/comfy/vendor/aiohttp_server_instrumentation.py
index 61b355237..5e334c49f 100644
--- a/comfy/vendor/aiohttp_server_instrumentation.py
+++ b/comfy/vendor/aiohttp_server_instrumentation.py
@@ -232,6 +232,10 @@ async def middleware(request, handler):
         except web.HTTPException as ex:
             set_status_code(span, ex.status_code)
             raise
+        except AttributeError:
+            # No response was returned or a NoneType response was returned, handle gracefully
+            set_status_code(span, 500)
+            raise
         finally:
             duration = max((default_timer() - start) * 1000, 0)
             duration_histogram.record(duration, duration_attrs)