theokit 0.73.0 → 0.74.1

package/dist/{chunk-EDBQM6RB.js → chunk-SHIXTMQE.js}

@@ -2239,20 +2239,52 @@ function encodeSse(text) {
 }
 function durableUiMessageStreamResponse(chunks, deps) {
   const { runId, cache } = deps;
+  const iterator = chunks[Symbol.asyncIterator]();
+  let finished = false;
+  function endCachedRun() {
+    try {
+      cache.end(runId);
+    } catch (err) {
+      console.error(
+        `[theokit] durable run ${runId}: cache.end failed, so a reconnect replay may be stale:`,
+        err
+      );
+    }
+  }
+  function finish(controller) {
+    if (finished) return;
+    finished = true;
+    endCachedRun();
+    try {
+      controller.enqueue(encodeSse(SSE_DONE_FRAME));
+      controller.close();
+    } catch (err) {
+      console.error(
+        `[theokit] durable run ${runId}: could not terminate a stream nobody is reading:`,
+        err
+      );
+    }
+  }
   const stream = new ReadableStream({
-    async start(controller) {
+    async pull(controller) {
+      if (finished) return;
       try {
-        for await (const chunk of chunks) {
-          const data = JSON.stringify(chunk);
-          const seq = cache.append(runId, data);
-          controller.enqueue(encodeSse(formatSseFrame(seq, data)));
+        const next = await iterator.next();
+        if (next.done === true) {
+          finish(controller);
+          return;
         }
+        const data = JSON.stringify(next.value);
+        const seq = cache.append(runId, data);
+        controller.enqueue(encodeSse(formatSseFrame(seq, data)));
       } catch {
-      } finally {
-        cache.end(runId);
-        controller.enqueue(encodeSse(SSE_DONE_FRAME));
-        controller.close();
+        finish(controller);
       }
+    },
+    async cancel(reason) {
+      finished = true;
+      endCachedRun();
+      await iterator.return?.(reason);
     }
   });
   return new Response(stream, {
@@ -2306,8 +2338,11 @@ function createInMemoryRunEventCache(opts = {}) {
         }
       }
       buf.listeners.clear();
-      buf.evictTimer = setTimeout(() => runs.delete(runId), evictAfterMs);
-      buf.evictTimer.unref();
+      const timer = setTimeout(() => runs.delete(runId), evictAfterMs);
+      buf.evictTimer = timer;
+      if (typeof timer === "object" && typeof timer.unref === "function") {
+        timer.unref();
+      }
     },
     attach(runId, afterSeq, onFrame, onEnd) {
       const buf = runs.get(runId);
@@ -2735,4 +2770,4 @@ export {
   CsrfReadinessStore,
   applySecurityHeaders
 };
-//# sourceMappingURL=chunk-EDBQM6RB.js.map
+//# sourceMappingURL=chunk-SHIXTMQE.js.map