run job termination task asynchronously

livekit · Aug 22, 2024 · 3a4517e · 3a4517e
1 parent 47c00d3
commit 3a4517e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/livekit-agents/livekit/agents/worker.py b/livekit-agents/livekit/agents/worker.py
@@ -486,7 +486,9 @@ async def _recv_task():
                 elif which == "assignment":
                     self._handle_assignment(msg.assignment)
                 elif which == "termination":
-                    await self._handle_termination(msg.termination)
+                    user_task = self._loop.create_task(self._handle_termination(msg.termination), name="termination")
+                    self._tasks.add(user_task)
+                    user_task.add_done_callback(self._tasks.discard)
 
         tasks = [
             asyncio.create_task(_load_task()),