ensure job status updates contain the correct status (#1319)

Co-authored-by: Théo Monnom <theo.8bits@gmail.com>
livekit · Jan 2, 2025 · f483f04 · f483f04
1 parent 55e6f0e
commit f483f04
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.changeset/lazy-buckets-obey.md b/.changeset/lazy-buckets-obey.md
@@ -0,0 +1,5 @@
+---
+"livekit-agents": patch
+---
+
+ensure job status updates contain the correct status
diff --git a/livekit-agents/livekit/agents/ipc/job_proc_executor.py b/livekit-agents/livekit/agents/ipc/job_proc_executor.py
@@ -98,6 +98,11 @@ async def _main_task(self, ipc_ch: aio.ChanReceiver[channel.Message]) -> None:
         finally:
             await aio.gracefully_cancel(*self._inference_tasks)
 
+    @log_exceptions(logger=logger)
+    async def _supervise_task(self) -> None:
+        try:
+            await super()._supervise_task()
+        finally:
             self._job_status = (
                 JobStatus.SUCCESS if self.exitcode == 0 else JobStatus.FAILED
             )