@deepagents/context 7.1.4 → 7.2.0

package/dist/browser.js

@@ -2366,6 +2366,28 @@ var ContextStore = class {
   }
 };
 
+// packages/context/src/lib/stream/stream-store.ts
+function collectStreamFailures(chunks) {
+  const failures = /* @__PURE__ */ new Map();
+  for (const chunk of chunks) {
+    if (failures.has(chunk.streamId)) continue;
+    if (chunk.data.type === "error") {
+      failures.set(chunk.streamId, chunk.data.errorText);
+    } else if (chunk.data.type === "finish" && chunk.data.finishReason === "interrupted") {
+      failures.set(
+        chunk.streamId,
+        "Model stream interrupted before completion."
+      );
+    }
+  }
+  return [...failures].map(([streamId, error]) => ({ streamId, error }));
+}
+var StreamStore = class {
+  async listRunningStreamIds() {
+    return this.listStreamIds({ status: "running" });
+  }
+};
+
 // packages/context/src/lib/stream-buffer.ts
 async function persistedWriter(options) {
   const {
@@ -2377,7 +2399,7 @@ async function persistedWriter(options) {
   } = options;
   let seq = 0;
   let buffer = [];
-  let failedByErrorChunk = false;
+  let failedByChunk = false;
   let abortedByChunk = false;
   async function flush() {
     if (buffer.length === 0) return;
@@ -2393,15 +2415,9 @@ async function persistedWriter(options) {
       createdAt: Date.now()
     };
   }
-  function isStreamErrorPart(part) {
-    return part.type === "error";
-  }
   async function appendBatch(batch) {
-    const hasErrorPart = !failedByErrorChunk && batch.map((chunk) => chunk.data).some(isStreamErrorPart);
     await store.appendChunks(batch);
-    if (hasErrorPart) {
-      failedByErrorChunk = true;
-    }
+    failedByChunk ||= collectStreamFailures(batch).length > 0;
   }
   async function persistChunk(chunk) {
     if (chunk.data.type === "abort") {
@@ -2438,7 +2454,7 @@ async function persistedWriter(options) {
     flush,
     async complete() {
       await flush();
-      if (failedByErrorChunk) return;
+      if (failedByChunk) return;
       await store.updateStreamStatus(
         streamId,
         abortedByChunk ? "cancelled" : "completed"
@@ -2446,7 +2462,7 @@ async function persistedWriter(options) {
     },
     async fail(error) {
       await flush();
-      if (failedByErrorChunk) return;
+      if (failedByChunk) return;
       await store.updateStreamStatus(streamId, "failed", { error });
     },
     async cleanup() {
@@ -2455,22 +2471,6 @@ async function persistedWriter(options) {
   };
 }
 
-// packages/context/src/lib/stream/stream-store.ts
-function collectStreamFailures(chunks) {
-  const failures = /* @__PURE__ */ new Map();
-  for (const chunk of chunks) {
-    if (chunk.data.type === "error" && !failures.has(chunk.streamId)) {
-      failures.set(chunk.streamId, chunk.data.errorText);
-    }
-  }
-  return [...failures].map(([streamId, error]) => ({ streamId, error }));
-}
-var StreamStore = class {
-  async listRunningStreamIds() {
-    return this.listStreamIds({ status: "running" });
-  }
-};
-
 // packages/context/src/lib/visualize.ts
 function visualizeGraph(data) {
   if (data.nodes.length === 0) {