@prestyj/ai 4.12.1 → 4.13.3

package/dist/index.cjs

@@ -336,7 +336,7 @@ var EventStream = class {
     }
   }
 };
-var StreamResult = class {
+var StreamResult = class _StreamResult {
   response;
   buffer = [];
   done = false;
@@ -344,6 +344,18 @@ var StreamResult = class {
   resolveResponse;
   rejectResponse;
   resolveWait = null;
+  /**
+   * High-water mark: when the buffer exceeds this many unconsumed events,
+   * the pump pauses until the consumer drains below the low-water mark.
+   * Prevents unbounded memory growth when a consumer is slow.
+   * Only active when someone IS iterating — if nobody iterates (the `then()`
+   * path), backpressure is skipped so the pump can complete and resolve.
+   */
+  static HIGH_WATER = 5e3;
+  static LOW_WATER = 1e3;
+  iterating = false;
+  paused = false;
+  resolveDrain = null;
   constructor(generator, signal) {
     this.response = new Promise((resolve, reject) => {
       this.resolveResponse = resolve;
@@ -358,6 +370,13 @@ var StreamResult = class {
         this.buffer.push(next.value);
         this.resolveWait?.();
         this.resolveWait = null;
+        if (this.iterating && this.buffer.length > _StreamResult.HIGH_WATER) {
+          this.paused = true;
+          await new Promise((r) => {
+            this.resolveDrain = r;
+          });
+          this.paused = false;
+        }
         next = await this._nextWithAbort(generator, signal);
       }
       this.done = true;
@@ -396,11 +415,20 @@ var StreamResult = class {
     }
   }
   async *[Symbol.asyncIterator]() {
+    this.iterating = true;
     let index = 0;
     while (true) {
       while (index < this.buffer.length) {
         yield this.buffer[index++];
       }
+      if (this.paused && index > _StreamResult.LOW_WATER) {
+        this.resolveDrain?.();
+        this.resolveDrain = null;
+      }
+      if (index > 0 && !this.paused) {
+        this.buffer.splice(0, index);
+        index = 0;
+      }
       if (this.error) throw this.error;
       if (this.done) return;
       await new Promise((r) => {
@@ -413,6 +441,11 @@ var StreamResult = class {
     }
   }
   then(onfulfilled, onrejected) {
+    if (this.paused) {
+      this.paused = false;
+      this.resolveDrain?.();
+      this.resolveDrain = null;
+    }
     return this.response.then(onfulfilled, onrejected);
   }
 };