else

fal-ai · Nov 12, 2024 · ec2d1f6 · ec2d1f6
1 parent 532b5dc
commit ec2d1f6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/projects/fal/src/fal/app.py b/projects/fal/src/fal/app.py
@@ -383,7 +383,10 @@ async def _unset_at_end():
 
             try:
                 response: fastapi.responses.Response = await call_next(request)
-
+            except BaseException:
+                await _unset_at_end()
+                raise
+            else:
                 # We need to wait for the entire response to be sent before
                 # we can set the logger labels back to the default.
                 background_tasks = fastapi.BackgroundTasks()
@@ -394,9 +397,6 @@ async def _unset_at_end():
                 response.background = background_tasks
 
                 return response
-            except BaseException:
-                await _unset_at_end()
-                raise
 
         @app.exception_handler(RequestCancelledException)
         async def value_error_exception_handler(