fix: Missing notification of bgtask cancellation or failure when shut…

…ting down the server (#2579)

Co-authored-by: Joongi Kim <joongi@lablup.com>
Backported-from: main (24.12)
Backported-to: 24.09
Backport-of: 2579

changes/2579.fix.md

@@ -0,0 +1 @@
+Fix missing notification of cancellation or failure of background tasks when shutting down the server

src/ai/backend/common/bgtask.py

@@ -168,7 +168,11 @@ async def push_bgtask_events(
                             else:
                                 await resp.send("{}", event="bgtask_done")
                             await resp.send("{}", event="server_close")
-                        case BgtaskCancelledEvent() | BgtaskFailedEvent():
+                        case BgtaskCancelledEvent():
+                            await resp.send(json.dumps(body), event="bgtask_cancelled")
+                            await resp.send("{}", event="server_close")
+                        case BgtaskFailedEvent():
+                            await resp.send(json.dumps(body), event="bgtask_failed")
                             await resp.send("{}", event="server_close")
             except:
                 log.exception("")