@holokai/neuron-sdk 0.4.0 → 0.5.0

package/dist/index.mjs

@@ -551,6 +551,7 @@ var TerminalFailure = class extends Error {
     this.name = "TerminalFailure";
   }
 };
+var DEFAULT_PROGRESS_TIMEOUT_MS = 5e3;
 function validateInput(zod, validator, jsonSchema, data) {
   if (zod) {
     const result = zod.safeParse(data);
@@ -573,7 +574,8 @@ async function runLease(opts) {
     transport,
     validator,
     logger,
-    signal
+    signal,
+    progressTimeoutMs = DEFAULT_PROGRESS_TIMEOUT_MS
   } = opts;
   const inputCheck = validateInput(zodInput, validator, inputSchema, lease.task.input);
   if (!inputCheck.ok) {
@@ -588,13 +590,13 @@ async function runLease(opts) {
     return;
   }
   const parsedInput = inputCheck.value;
-  const progressFn = makeProgressFn(transport, lease.leaseId, logger);
+  const reporter = makeProgressReporter(transport, lease.leaseId, logger, signal, progressTimeoutMs);
   const ctx = {
     task: lease.task,
     leaseId: lease.leaseId,
     signal,
     log: scopedLogger(logger, lease),
-    progress: progressFn,
+    progress: reporter.progress,
     fail: (reason, opts2) => {
       throw new TerminalFailure(reason, opts2?.code);
     }
@@ -603,7 +605,9 @@ async function runLease(opts) {
   try {
     output = await handler(parsedInput, ctx);
   } catch (err) {
-    if (signal.aborted || err.name === "AbortError") {
+    const cancelled = signal.aborted || err.name === "AbortError";
+    if (!cancelled) await reporter.drained();
+    if (cancelled) {
       await sendNack(
         transport,
         lease.leaseId,
@@ -637,6 +641,7 @@ async function runLease(opts) {
     );
     return;
   }
+  await reporter.drained();
   const outputCheck = validateInput(zodOutput, validator, outputSchema, output);
   if (!outputCheck.ok) {
     await sendNack(
@@ -678,20 +683,37 @@ async function sendNack(transport, leaseId, kind, reason, extras = {}, logger) {
     );
   });
 }
-function makeProgressFn(transport, leaseId, logger) {
-  return (update) => {
-    void transport.post(POST_PATH.progress, {
-      type: "progress",
-      leaseId,
-      percent: update.percent,
-      message: update.message,
-      data: update.data,
-      sentAt: (/* @__PURE__ */ new Date()).toISOString()
-    }).catch((err) => {
-      logger.warn({ err, leaseId }, "progress POST failed; continuing");
-    });
+function makeProgressReporter(transport, leaseId, logger, signal, timeoutMs) {
+  let tail = Promise.resolve();
+  return {
+    progress: (update) => {
+      const frame = {
+        type: "progress",
+        leaseId,
+        percent: update.percent,
+        message: update.message,
+        data: update.data,
+        sentAt: (/* @__PURE__ */ new Date()).toISOString()
+      };
+      tail = tail.then(async () => {
+        if (signal.aborted) return;
+        try {
+          await withTimeout(transport.post(POST_PATH.progress, frame), timeoutMs);
+        } catch (err) {
+          logger.warn({ err, leaseId }, "progress POST failed; continuing");
+        }
+      });
+    },
+    drained: () => tail
   };
 }
+function withTimeout(work, ms) {
+  let timer;
+  const timeout = new Promise((_resolve, reject) => {
+    timer = setTimeout(() => reject(new Error(`timed out after ${ms}ms`)), ms);
+  });
+  return Promise.race([work, timeout]).finally(() => clearTimeout(timer));
+}
 function scopedLogger(logger, lease) {
   const scope = { leaseId: lease.leaseId, taskId: lease.task.taskId };
   return {