refactor(mothership): wait on a workflow tool's run with the shared interruptible sleep

Author: waleedlatif1

apps/sim/lib/mothership/tools/handlers/workflow/mutations.ts

@@ -1,4 +1,5 @@
 import { createLogger } from '@sim/logger'
+import { interruptibleSleep } from '@sim/utils/helpers'
 import { generateId } from '@sim/utils/id'
 import { filterUndefined, isPlainRecord, isRecordLike } from '@sim/utils/object'
 import { createCopilotWorkspaceApiKey } from '@/lib/api-key/application/create-api-key'
@@ -277,18 +278,16 @@ async function answerWithinWait(
   })
   const settled = run({ ...lifecycle, executionId, onDispatch: () => markDispatched() })
   retainSandboxExecution(settled)
-  let timer: ReturnType<typeof setTimeout> | undefined
-  const waited = new Promise<null>((resolve) => {
-    timer = setTimeout(() => resolve(null), waitMs)
-  })
+  const stopWaiting = new AbortController()
+  const waited = interruptibleSleep(waitMs, stopWaiting.signal).then(() => null)
   try {
     const result = await Promise.race([settled, waited])
     if (result) return result
     // Only a run handed to the executor is answered running; one still in preflight settles first.
     const started = await Promise.race([settled, dispatched])
     if (started) return started
   } finally {
-    clearTimeout(timer)
+    stopWaiting.abort()
   }
   logger.info('Workflow tool run outlasted its wait; answering it still running', {
     toolCallId: context.toolCallId,