Fix stack dump on shutdown (#1023)

livekit · Nov 1, 2024 · 2ff2493 · 2ff2493
1 parent 95a90f0
commit 2ff2493
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.changeset/heavy-turtles-rule.md b/.changeset/heavy-turtles-rule.md
@@ -0,0 +1,5 @@
+---
+"livekit-agents": patch
+---
+
+Fix stack dump on closed stream
diff --git a/livekit-agents/livekit/agents/ipc/job_main.py b/livekit-agents/livekit/agents/ipc/job_main.py
@@ -208,7 +208,11 @@ async def _async_main(
     async def _read_ipc_task():
         nonlocal job_task
         while True:
-            msg = await channel.arecv_message(cch, proto.IPC_MESSAGES)
+            try:
+                msg = await channel.arecv_message(cch, proto.IPC_MESSAGES)
+            except duplex_unix.DuplexClosed:
+                break
+
             with contextlib.suppress(utils.aio.SleepFinished):
                 no_msg_timeout.reset()