@axiom-lattice/core 2.1.0 → 2.1.2

package/dist/index.js

@@ -2617,13 +2617,12 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
    */
   getOrCreateBuffer(threadId) {
     let buffer = this.getBufferIfValid(threadId);
-    4;
     if (!buffer) {
       const now = Date.now();
       buffer = {
         threadId,
         chunks$: new import_rxjs.ReplaySubject(
-          this.config.maxChunks ?? 1e4
+          this.config.maxChunks ?? 1e5
         ),
         status: "active" /* ACTIVE */,
         createdAt: now,
@@ -2635,7 +2634,7 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
     if (buffer.status !== "active" /* ACTIVE */) {
       buffer.status = "active" /* ACTIVE */;
       buffer.chunks$ = new import_rxjs.ReplaySubject(
-        this.config.maxChunks ?? 1e4
+        this.config.maxChunks ?? 1e5
       );
       buffer.updatedAt = Date.now();
       buffer.expiresAt = Date.now() + this.config.ttl;
@@ -2739,7 +2738,8 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
     let resolveNext = null;
     let errorNext = null;
     let isCompleted = false;
-    const subscription = buffer.chunks$.subscribe({
+    let pendingError = null;
+    const subscription = buffer.chunks$.pipe((0, import_rxjs.observeOn)(import_rxjs.asyncScheduler)).subscribe({
       next: (chunk) => {
         queue.push(chunk);
         if (resolveNext) {
@@ -2749,10 +2749,15 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
         }
       },
       error: (err) => {
+        pendingError = err;
         if (errorNext) {
           const reject = errorNext;
           errorNext = null;
           reject(err);
+        } else if (resolveNext) {
+          const resolve = resolveNext;
+          resolveNext = null;
+          resolve();
         }
       },
       complete: () => {
@@ -2764,8 +2769,12 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
         }
       }
     });
+    let startYieldChunk = false;
     try {
       while (true) {
+        if (pendingError) {
+          throw pendingError;
+        }
         if (queue.length === 0) {
           if (isCompleted) break;
           await new Promise((resolve, reject) => {
@@ -2773,16 +2782,18 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
             errorNext = reject;
           });
         }
+        if (pendingError) {
+          throw pendingError;
+        }
         while (queue.length > 0) {
           const chunk = queue.shift();
           if (!chunk) continue;
           if (chunk.data?.id === messageId) {
+            startYieldChunk = true;
             accumulatedContent += chunk.data?.content || "";
           }
-          if (accumulatedContent.length > knownContent.length) {
-            if (accumulatedContent.startsWith(knownContent)) {
-              yield chunk;
-            }
+          if (startYieldChunk) {
+            yield chunk;
           }
         }
       }