@axiom-lattice/core 2.1.0 → 2.1.2

package/dist/index.mjs

@@ -2524,7 +2524,11 @@ var ThreadStatus = /* @__PURE__ */ ((ThreadStatus2) => {
 })(ThreadStatus || {});
 
 // src/chunk_buffer_lattice/InMemoryChunkBuffer.ts
-import { ReplaySubject } from "rxjs";
+import {
+  ReplaySubject,
+  observeOn,
+  asyncScheduler
+} from "rxjs";
 var InMemoryChunkBuffer = class extends ChunkBuffer {
   constructor(config) {
     super();
@@ -2582,13 +2586,12 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
    */
   getOrCreateBuffer(threadId) {
     let buffer = this.getBufferIfValid(threadId);
-    4;
     if (!buffer) {
       const now = Date.now();
       buffer = {
         threadId,
         chunks$: new ReplaySubject(
-          this.config.maxChunks ?? 1e4
+          this.config.maxChunks ?? 1e5
         ),
         status: "active" /* ACTIVE */,
         createdAt: now,
@@ -2600,7 +2603,7 @@ var InMemoryChunkBuffer = class extends ChunkBuffer {
     if (buffer.status !== "active" /* ACTIVE */) {
       buffer.status = "active" /* ACTIVE */;
       buffer.chunks$ = new ReplaySubject(
-        this.config.maxChunks ?? 1e4
+        this.config.maxChunks ?? 1e5
       );
       buffer.updatedAt = Date.now();
       buffer.expiresAt = Date.now() + this.config.ttl;
@@ -2704,7 +2707,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(observeOn(asyncScheduler)).subscribe({
       next: (chunk) => {
         queue.push(chunk);
         if (resolveNext) {
@@ -2714,10 +2718,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: () => {
@@ -2729,8 +2738,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) => {
@@ -2738,16 +2751,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;
           }
         }
       }