start listening to close events before running the job

agents/src/ipc/job_proc_lazy_main.ts

@@ -126,6 +126,12 @@ const startJob = (
     }, 10000);
 
     try {
+      const closePromise = once(closeEvent, 'close').then((close) => {
+        logger.debug('shutting down');
+        shutdown = true;
+        safeSend({ case: 'exiting', value: { reason: close[1] } });
+      });
+
       // Run the job function within the AsyncLocalStorage context
       await runWithJobContextAsync(ctx, async () => {
         const { tracer, traceTypes } = await import('../telemetry/index.js');
@@ -138,14 +144,11 @@ const startJob = (
           },
           { name: 'job_entrypoint' },
         );
-      }).finally(() => {
+      }).finally(async () => {
         clearTimeout(unconnectedTimeout);
-      });
-
-      await once(closeEvent, 'close').then((close) => {
-        logger.debug('shutting down');
-        shutdown = true;
-        safeSend({ case: 'exiting', value: { reason: close[1] } });
+        if (!shutdown) {
+          await closePromise;
+        }
       });
     } catch (error) {
       logger.error({ error }, 'error in entry function');

agents/src/ipc/proc_pool.test.ts

@@ -1,6 +1,7 @@
 // SPDX-FileCopyrightText: 2026 LiveKit, Inc.
 //
 // SPDX-License-Identifier: Apache-2.0
+import { Throws } from '@livekit/throws-transformer/throws';
 import { describe, expect, it, vi } from 'vitest';
 import type { RunningJobInfo } from '../job.js';
 import { type JobExecutor, JobStatus } from './job_executor.js';
@@ -22,7 +23,9 @@ function createMockExecutor() {
 }
 
 describe('ProcPool warmed process lock handling', () => {
-  it('releases lock token from the dequeued warmed process entry', async () => {
+  it('releases lock token from the dequeued warmed process entry', async (): Promise<
+    Throws<void, Error>
+  > => {
     const pool = new ProcPool('agent', 1, 1000, 1000, undefined, 0, 0);
     const unlock = vi.fn();
     const executor = createMockExecutor();