prevent spurious exception

bananaml · Dec 7, 2023 · 90404a2 · 90404a2
1 parent 9241ee9
commit 90404a2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/potassium/potassium.py b/potassium/potassium.py
@@ -172,12 +172,12 @@ def wrapper(context, request):
                 # send in app's stateful context if GPU, and the request
                 out = func(context, request)
 
-                if type(out) != Response:
-                    raise Exception("Potassium Response object not returned")
-
-                if type(out.body) != bytes and type(out.body) != GeneratorType:
-                    raise Exception(
-                        "Potassium Response object body must be bytes", type(out.body))
+                if handler_type == HandlerType.HANDLER:
+                    if type(out) != Response:
+                        raise Exception("Potassium Response object not returned")
+                    if type(out.body) != bytes and type(out.body) != GeneratorType:
+                        raise Exception(
+                            "Potassium Response object body must be bytes", type(out.body))
 
                 return out