assistant-stream 0.3.14 → 0.3.16

package/dist/core/utils/stream/UnderlyingReadable.d.ts

@@ -1,6 +1,9 @@
-export type UnderlyingReadable<TController> = {
-    start?: (controller: TController) => void;
-    pull?: (controller: TController) => void | PromiseLike<void>;
-    cancel?: UnderlyingSourceCancelCallback;
+//#region src/core/utils/stream/UnderlyingReadable.d.ts
+type UnderlyingReadable<TController> = {
+  start?: (controller: TController) => void;
+  pull?: (controller: TController) => void | PromiseLike<void>;
+  cancel?: UnderlyingSourceCancelCallback;
 };
+//#endregion
+export { UnderlyingReadable };
 //# sourceMappingURL=UnderlyingReadable.d.ts.map

package/dist/core/utils/stream/UnderlyingReadable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"UnderlyingReadable.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/UnderlyingReadable.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,kBAAkB,CAAC,WAAW,IAAI;IAC5C,KAAK,CAAC,EAAE,CAAC,UAAU,EAAE,WAAW,KAAK,IAAI,CAAC;IAC1C,IAAI,CAAC,EAAE,CAAC,UAAU,EAAE,WAAW,KAAK,IAAI,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IAC7D,MAAM,CAAC,EAAE,8BAA8B,CAAC;CACzC,CAAC"}
+{"version":3,"file":"UnderlyingReadable.d.ts","names":[],"sources":["../../../../src/core/utils/stream/UnderlyingReadable.ts"],"mappings":";KAAY,kBAAA;EACV,KAAA,IAAS,UAAA,EAAY,WAAA;EACrB,IAAA,IAAQ,UAAA,EAAY,WAAA,YAAuB,WAAA;EAC3C,MAAA,GAAS,8BAAA;AAAA"}

package/dist/core/utils/stream/UnderlyingReadable.js

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=UnderlyingReadable.js.map

package/dist/core/utils/stream/merge.d.ts

@@ -1,9 +1,13 @@
-import type { AssistantStreamChunk } from "../../AssistantStreamChunk.js";
-export declare const createMergeStream: () => {
-    readable: ReadableStream<AssistantStreamChunk>;
-    isSealed(): boolean;
-    seal(): void;
-    addStream(stream: ReadableStream<AssistantStreamChunk>): void;
-    enqueue(chunk: AssistantStreamChunk): void;
+import { AssistantStreamChunk } from "../../AssistantStreamChunk.js";
+
+//#region src/core/utils/stream/merge.d.ts
+declare const createMergeStream: () => {
+  readable: ReadableStream<AssistantStreamChunk>;
+  isSealed(): boolean;
+  seal(): void;
+  addStream(stream: ReadableStream<AssistantStreamChunk>): void;
+  enqueue(chunk: AssistantStreamChunk): void;
 };
+//#endregion
+export { createMergeStream };
 //# sourceMappingURL=merge.d.ts.map

package/dist/core/utils/stream/merge.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"merge.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/merge.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,sCAAmC;AAQvE,eAAO,MAAM,iBAAiB;;;;sBA0ER,cAAc,CAAC,oBAAoB,CAAC;mBAUvC,oBAAoB;CAWtC,CAAC"}
+{"version":3,"file":"merge.d.ts","names":[],"sources":["../../../../src/core/utils/stream/merge.ts"],"mappings":";;;cAQa,iBAAA;;;;oBA0ES,cAAA,CAAe,oBAAA;iBAUlB,oBAAA;AAAA"}

package/dist/core/utils/stream/merge.js

@@ -1,173 +1,71 @@
 import { promiseWithResolvers } from "../../../utils/promiseWithResolvers.js";
-export const createMergeStream = () => {
-    const list = [];
-    let sealed = false;
-    let controller;
-    let currentPull;
-    const handlePull = (item) => {
-        if (!item.promise) {
-            // TODO for most streams, we can directly pipeTo to avoid the microTask queue
-            // add an option to eagerly pipe the stream to the merge stream
-            // ideally, using assistant-stream w sync run method + piping to a sync WritableStream runs in the same microtask
-            // this is useful because we often use AssistantStreams internally as a serialization utility, e. g. AssistantTransformStream
-            // idea: avoid reader.read() by instead using a WritableStream & if (!hasPendingPull) await waitForPull()?
-            item.promise = item.reader
-                .read()
-                .then(({ done, value }) => {
-                item.promise = undefined;
-                if (done) {
-                    list.splice(list.indexOf(item), 1);
-                    if (sealed && list.length === 0) {
-                        controller.close();
-                    }
-                }
-                else {
-                    controller.enqueue(value);
-                }
-                currentPull?.resolve();
-                currentPull = undefined;
-            })
-                .catch((e) => {
-                console.error(e);
-                list.forEach((item) => {
-                    item.reader.cancel();
-                });
-                list.length = 0;
-                controller.error(e);
-                currentPull?.reject(e);
-                currentPull = undefined;
-            });
-        }
-    };
-    const readable = new ReadableStream({
-        start(c) {
-            controller = c;
-        },
-        pull() {
-            currentPull = promiseWithResolvers();
-            list.forEach((item) => {
-                handlePull(item);
-            });
-            return currentPull.promise;
-        },
-        cancel() {
-            list.forEach((item) => {
-                item.reader.cancel();
-            });
-            list.length = 0;
-        },
-    });
-    return {
-        readable,
-        isSealed() {
-            return sealed;
-        },
-        seal() {
-            sealed = true;
-            if (list.length === 0)
-                controller.close();
-        },
-        addStream(stream) {
-            if (sealed)
-                throw new Error("Cannot add streams after the run callback has settled.");
-            const item = { reader: stream.getReader() };
-            list.push(item);
-            handlePull(item);
-        },
-        enqueue(chunk) {
-            this.addStream(new ReadableStream({
-                start(c) {
-                    c.enqueue(chunk);
-                    c.close();
-                },
-            }));
-        },
-    };
+//#region src/core/utils/stream/merge.ts
+const createMergeStream = () => {
+	const list = [];
+	let sealed = false;
+	let controller;
+	let currentPull;
+	const handlePull = (item) => {
+		if (!item.promise) item.promise = item.reader.read().then(({ done, value }) => {
+			item.promise = void 0;
+			if (done) {
+				list.splice(list.indexOf(item), 1);
+				if (sealed && list.length === 0) controller.close();
+			} else controller.enqueue(value);
+			currentPull?.resolve();
+			currentPull = void 0;
+		}).catch((e) => {
+			console.error(e);
+			list.forEach((item) => {
+				item.reader.cancel();
+			});
+			list.length = 0;
+			controller.error(e);
+			currentPull?.reject(e);
+			currentPull = void 0;
+		});
+	};
+	return {
+		readable: new ReadableStream({
+			start(c) {
+				controller = c;
+			},
+			pull() {
+				currentPull = promiseWithResolvers();
+				list.forEach((item) => {
+					handlePull(item);
+				});
+				return currentPull.promise;
+			},
+			cancel() {
+				list.forEach((item) => {
+					item.reader.cancel();
+				});
+				list.length = 0;
+			}
+		}),
+		isSealed() {
+			return sealed;
+		},
+		seal() {
+			sealed = true;
+			if (list.length === 0) controller.close();
+		},
+		addStream(stream) {
+			if (sealed) throw new Error("Cannot add streams after the run callback has settled.");
+			const item = { reader: stream.getReader() };
+			list.push(item);
+			handlePull(item);
+		},
+		enqueue(chunk) {
+			this.addStream(new ReadableStream({ start(c) {
+				c.enqueue(chunk);
+				c.close();
+			} }));
+		}
+	};
 };
-// TODO
-// export class SpanContainerMerger {
-//   public get isSealed() {
-//     return this.mergeStream.isSealed();
-//   }
-//   public get readable() {
-//     return this.mergeStream.readable;
-//   }
-//   private subAllocator = new Counter();
-//   private mergeStream = createMergeStream();
-//   constructor() {
-//     // id 0 is auto allocated
-//     this.subAllocator.up();
-//   }
-//   add(stream: ReadableStream<AssistantStreamChunk>) {
-//     this.mergeStream.addStream(
-//       stream.pipeThrough(new SpanParentEncoder(this.subAllocator)),
-//     );
-//   }
-//   enqueue(chunk: AssistantStreamChunk & { parentId: 0 }) {
-//     this.mergeStream.addStream(
-//       new ReadableStream({
-//         start(c) {
-//           c.enqueue(chunk);
-//           c.close();
-//         },
-//       }),
-//     );
-//   }
-//   seal() {
-//     this.mergeStream.seal();
-//   }
-// }
-// export class SpanContainerSplitter {
-//   public writable;
-//   private isSealed = false;
-//   private writers = new Map<
-//     number,
-//     WritableStreamDefaultWriter<AssistantStreamChunk>
-//   >();
-//   private closeTasks: Promise<void>[] = [];
-//   private allocator = new Counter();
-//   private subAllocator = new Counter();
-//   constructor() {
-//     // id 0 is auto-allocated
-//     this.allocator.up();
-//     this.writable = new WritableStream({
-//       write: (chunk) => {
-//         const { type, parentId } = chunk;
-//         const writer = this.writers.get(parentId);
-//         if (writer === undefined) throw new Error("Parent id not found");
-//         writer.write(chunk);
-//         if (type === "span") {
-//           // allocate a new span id
-//           this.writers.set(this.allocator.up(), writer);
-//         }
-//         if (type === "finish") {
-//           this.writers.delete(parentId);
-//           writer.close();
-//           if (this.writers.size === 0) {
-//             const closeTask = this.writable.close();
-//             this.closeTasks.push(closeTask);
-//             closeTask.then(() => {
-//               this.closeTasks.splice(this.closeTasks.indexOf(closeTask), 1);
-//             });
-//           }
-//         }
-//       },
-//       close: async () => {
-//         if (this.writers.size > 0) throw new Error("Not all writers closed");
-//         // await and throw on any errors
-//         await Promise.all(this.closeTasks);
-//       },
-//     });
-//   }
-//   add(stream: WritableStream<AssistantStreamChunk>) {
-//     if (this.isSealed) throw new Error("Cannot add streams after sealing");
-//     const decoder = new SpanParentDecoder(this.subAllocator);
-//     decoder.readable.pipeTo(stream);
-//     this.writers.set(this.allocator.up(), decoder.writable.getWriter());
-//   }
-//   seal() {
-//     this.isSealed = true;
-//     if (this.writers.size === 0) this.writable.close();
-//   }
-// }
+//#endregion
+export { createMergeStream };
+
 //# sourceMappingURL=merge.js.map

package/dist/core/utils/stream/merge.js.map

@@ -1 +1 @@
-{"version":3,"file":"merge.js","sourceRoot":"","sources":["../../../../src/core/utils/stream/merge.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,+CAA4C;AAO3E,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAG,EAAE;IACpC,MAAM,IAAI,GAAsB,EAAE,CAAC;IACnC,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,IAAI,UAAiE,CAAC;IACtE,IAAI,WAAsE,CAAC;IAE3E,MAAM,UAAU,GAAG,CAAC,IAAqB,EAAE,EAAE;QAC3C,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,6EAA6E;YAC7E,+DAA+D;YAC/D,iHAAiH;YACjH,6HAA6H;YAC7H,0GAA0G;YAC1G,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,MAAM;iBACvB,IAAI,EAAE;iBACN,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;gBACxB,IAAI,CAAC,OAAO,GAAG,SAAS,CAAC;gBACzB,IAAI,IAAI,EAAE,CAAC;oBACT,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;oBACnC,IAAI,MAAM,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAChC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC5B,CAAC;gBAED,WAAW,EAAE,OAAO,EAAE,CAAC;gBACvB,WAAW,GAAG,SAAS,CAAC;YAC1B,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;gBACX,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAEjB,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;oBACpB,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;gBACvB,CAAC,CAAC,CAAC;gBACH,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;gBAEhB,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBAEpB,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;gBACvB,WAAW,GAAG,SAAS,CAAC;YAC1B,CAAC,CAAC,CAAC;QACP,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,QAAQ,GAAG,IAAI,cAAc,CAAuB;QACxD,KAAK,CAAC,CAAC;YACL,UAAU,GAAG,CAAC,CAAC;QACjB,CAAC;QACD,IAAI;YACF,WAAW,GAAG,oBAAoB,EAAE,CAAC;YACrC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACpB,UAAU,CAAC,IAAI,CAAC,CAAC;YACnB,CAAC,CAAC,CAAC;YAEH,OAAO,WAAW,CAAC,OAAO,CAAC;QAC7B,CAAC;QACD,MAAM;YACJ,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBACpB,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;QAClB,CAAC;KACF,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ;QACR,QAAQ;YACN,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI;YACF,MAAM,GAAG,IAAI,CAAC;YACd,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,UAAU,CAAC,KAAK,EAAE,CAAC;QAC5C,CAAC;QACD,SAAS,CAAC,MAA4C;YACpD,IAAI,MAAM;gBACR,MAAM,IAAI,KAAK,CACb,wDAAwD,CACzD,CAAC;YAEJ,MAAM,IAAI,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC;YAC5C,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,UAAU,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;QACD,OAAO,CAAC,KAA2B;YACjC,IAAI,CAAC,SAAS,CACZ,IAAI,cAAc,CAAC;gBACjB,KAAK,CAAC,CAAC;oBACL,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBACjB,CAAC,CAAC,KAAK,EAAE,CAAC;gBACZ,CAAC;aACF,CAAC,CACH,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AAEF,OAAO;AACP,qCAAqC;AACrC,4BAA4B;AAC5B,0CAA0C;AAC1C,MAAM;AAEN,4BAA4B;AAC5B,wCAAwC;AACxC,MAAM;AAEN,0CAA0C;AAC1C,+CAA+C;AAE/C,oBAAoB;AACpB,gCAAgC;AAChC,8BAA8B;AAC9B,MAAM;AAEN,wDAAwD;AACxD,kCAAkC;AAClC,sEAAsE;AACtE,SAAS;AACT,MAAM;AAEN,6DAA6D;AAC7D,kCAAkC;AAClC,6BAA6B;AAC7B,qBAAqB;AACrB,8BAA8B;AAC9B,uBAAuB;AACvB,aAAa;AACb,YAAY;AACZ,SAAS;AACT,MAAM;AAEN,aAAa;AACb,+BAA+B;AAC/B,MAAM;AACN,IAAI;AAEJ,uCAAuC;AACvC,qBAAqB;AAErB,8BAA8B;AAC9B,+BAA+B;AAC/B,cAAc;AACd,wDAAwD;AACxD,SAAS;AAET,8CAA8C;AAE9C,uCAAuC;AACvC,0CAA0C;AAE1C,oBAAoB;AACpB,gCAAgC;AAChC,2BAA2B;AAE3B,2CAA2C;AAC3C,4BAA4B;AAC5B,4CAA4C;AAE5C,qDAAqD;AACrD,4EAA4E;AAE5E,+BAA+B;AAE/B,iCAAiC;AACjC,sCAAsC;AACtC,2DAA2D;AAC3D,YAAY;AACZ,mCAAmC;AACnC,2CAA2C;AAC3C,4BAA4B;AAE5B,2CAA2C;AAC3C,uDAAuD;AACvD,+CAA+C;AAC/C,qCAAqC;AACrC,+EAA+E;AAC/E,kBAAkB;AAClB,cAAc;AACd,YAAY;AACZ,WAAW;AACX,6BAA6B;AAC7B,gFAAgF;AAEhF,2CAA2C;AAC3C,8CAA8C;AAC9C,WAAW;AACX,UAAU;AACV,MAAM;AAEN,wDAAwD;AACxD,8EAA8E;AAE9E,gEAAgE;AAChE,uCAAuC;AAEvC,2EAA2E;AAC3E,MAAM;AAEN,aAAa;AACb,4BAA4B;AAC5B,0DAA0D;AAC1D,MAAM;AACN,IAAI"}
+{"version":3,"file":"merge.js","names":[],"sources":["../../../../src/core/utils/stream/merge.ts"],"sourcesContent":["import type { AssistantStreamChunk } from \"../../AssistantStreamChunk\";\nimport { promiseWithResolvers } from \"../../../utils/promiseWithResolvers\";\n\ntype MergeStreamItem = {\n  reader: ReadableStreamDefaultReader<AssistantStreamChunk>;\n  promise?: Promise<unknown> | undefined;\n};\n\nexport const createMergeStream = () => {\n  const list: MergeStreamItem[] = [];\n  let sealed = false;\n  let controller: ReadableStreamDefaultController<AssistantStreamChunk>;\n  let currentPull: ReturnType<typeof promiseWithResolvers<void>> | undefined;\n\n  const handlePull = (item: MergeStreamItem) => {\n    if (!item.promise) {\n      // TODO for most streams, we can directly pipeTo to avoid the microTask queue\n      // add an option to eagerly pipe the stream to the merge stream\n      // ideally, using assistant-stream w sync run method + piping to a sync WritableStream runs in the same microtask\n      // this is useful because we often use AssistantStreams internally as a serialization utility, e. g. AssistantTransformStream\n      // idea: avoid reader.read() by instead using a WritableStream & if (!hasPendingPull) await waitForPull()?\n      item.promise = item.reader\n        .read()\n        .then(({ done, value }) => {\n          item.promise = undefined;\n          if (done) {\n            list.splice(list.indexOf(item), 1);\n            if (sealed && list.length === 0) {\n              controller.close();\n            }\n          } else {\n            controller.enqueue(value);\n          }\n\n          currentPull?.resolve();\n          currentPull = undefined;\n        })\n        .catch((e) => {\n          console.error(e);\n\n          list.forEach((item) => {\n            item.reader.cancel();\n          });\n          list.length = 0;\n\n          controller.error(e);\n\n          currentPull?.reject(e);\n          currentPull = undefined;\n        });\n    }\n  };\n\n  const readable = new ReadableStream<AssistantStreamChunk>({\n    start(c) {\n      controller = c;\n    },\n    pull() {\n      currentPull = promiseWithResolvers();\n      list.forEach((item) => {\n        handlePull(item);\n      });\n\n      return currentPull.promise;\n    },\n    cancel() {\n      list.forEach((item) => {\n        item.reader.cancel();\n      });\n      list.length = 0;\n    },\n  });\n\n  return {\n    readable,\n    isSealed() {\n      return sealed;\n    },\n    seal() {\n      sealed = true;\n      if (list.length === 0) controller.close();\n    },\n    addStream(stream: ReadableStream<AssistantStreamChunk>) {\n      if (sealed)\n        throw new Error(\n          \"Cannot add streams after the run callback has settled.\",\n        );\n\n      const item = { reader: stream.getReader() };\n      list.push(item);\n      handlePull(item);\n    },\n    enqueue(chunk: AssistantStreamChunk) {\n      this.addStream(\n        new ReadableStream({\n          start(c) {\n            c.enqueue(chunk);\n            c.close();\n          },\n        }),\n      );\n    },\n  };\n};\n\n// TODO\n// export class SpanContainerMerger {\n//   public get isSealed() {\n//     return this.mergeStream.isSealed();\n//   }\n\n//   public get readable() {\n//     return this.mergeStream.readable;\n//   }\n\n//   private subAllocator = new Counter();\n//   private mergeStream = createMergeStream();\n\n//   constructor() {\n//     // id 0 is auto allocated\n//     this.subAllocator.up();\n//   }\n\n//   add(stream: ReadableStream<AssistantStreamChunk>) {\n//     this.mergeStream.addStream(\n//       stream.pipeThrough(new SpanParentEncoder(this.subAllocator)),\n//     );\n//   }\n\n//   enqueue(chunk: AssistantStreamChunk & { parentId: 0 }) {\n//     this.mergeStream.addStream(\n//       new ReadableStream({\n//         start(c) {\n//           c.enqueue(chunk);\n//           c.close();\n//         },\n//       }),\n//     );\n//   }\n\n//   seal() {\n//     this.mergeStream.seal();\n//   }\n// }\n\n// export class SpanContainerSplitter {\n//   public writable;\n\n//   private isSealed = false;\n//   private writers = new Map<\n//     number,\n//     WritableStreamDefaultWriter<AssistantStreamChunk>\n//   >();\n\n//   private closeTasks: Promise<void>[] = [];\n\n//   private allocator = new Counter();\n//   private subAllocator = new Counter();\n\n//   constructor() {\n//     // id 0 is auto-allocated\n//     this.allocator.up();\n\n//     this.writable = new WritableStream({\n//       write: (chunk) => {\n//         const { type, parentId } = chunk;\n\n//         const writer = this.writers.get(parentId);\n//         if (writer === undefined) throw new Error(\"Parent id not found\");\n\n//         writer.write(chunk);\n\n//         if (type === \"span\") {\n//           // allocate a new span id\n//           this.writers.set(this.allocator.up(), writer);\n//         }\n//         if (type === \"finish\") {\n//           this.writers.delete(parentId);\n//           writer.close();\n\n//           if (this.writers.size === 0) {\n//             const closeTask = this.writable.close();\n//             this.closeTasks.push(closeTask);\n//             closeTask.then(() => {\n//               this.closeTasks.splice(this.closeTasks.indexOf(closeTask), 1);\n//             });\n//           }\n//         }\n//       },\n//       close: async () => {\n//         if (this.writers.size > 0) throw new Error(\"Not all writers closed\");\n\n//         // await and throw on any errors\n//         await Promise.all(this.closeTasks);\n//       },\n//     });\n//   }\n\n//   add(stream: WritableStream<AssistantStreamChunk>) {\n//     if (this.isSealed) throw new Error(\"Cannot add streams after sealing\");\n\n//     const decoder = new SpanParentDecoder(this.subAllocator);\n//     decoder.readable.pipeTo(stream);\n\n//     this.writers.set(this.allocator.up(), decoder.writable.getWriter());\n//   }\n\n//   seal() {\n//     this.isSealed = true;\n//     if (this.writers.size === 0) this.writable.close();\n//   }\n// }\n"],"mappings":";;AAQA,MAAa,0BAA0B;CACrC,MAAM,OAA0B,CAAC;CACjC,IAAI,SAAS;CACb,IAAI;CACJ,IAAI;CAEJ,MAAM,cAAc,SAA0B;EAC5C,IAAI,CAAC,KAAK,SAMR,KAAK,UAAU,KAAK,OACjB,KAAK,EACL,MAAM,EAAE,MAAM,YAAY;GACzB,KAAK,UAAU,KAAA;GACf,IAAI,MAAM;IACR,KAAK,OAAO,KAAK,QAAQ,IAAI,GAAG,CAAC;IACjC,IAAI,UAAU,KAAK,WAAW,GAC5B,WAAW,MAAM;GAErB,OACE,WAAW,QAAQ,KAAK;GAG1B,aAAa,QAAQ;GACrB,cAAc,KAAA;EAChB,CAAC,EACA,OAAO,MAAM;GACZ,QAAQ,MAAM,CAAC;GAEf,KAAK,SAAS,SAAS;IACrB,KAAK,OAAO,OAAO;GACrB,CAAC;GACD,KAAK,SAAS;GAEd,WAAW,MAAM,CAAC;GAElB,aAAa,OAAO,CAAC;GACrB,cAAc,KAAA;EAChB,CAAC;CAEP;CAsBA,OAAO;EACL,UAAA,IArBmB,eAAqC;GACxD,MAAM,GAAG;IACP,aAAa;GACf;GACA,OAAO;IACL,cAAc,qBAAqB;IACnC,KAAK,SAAS,SAAS;KACrB,WAAW,IAAI;IACjB,CAAC;IAED,OAAO,YAAY;GACrB;GACA,SAAS;IACP,KAAK,SAAS,SAAS;KACrB,KAAK,OAAO,OAAO;IACrB,CAAC;IACD,KAAK,SAAS;GAChB;EACF,CAGS;EACP,WAAW;GACT,OAAO;EACT;EACA,OAAO;GACL,SAAS;GACT,IAAI,KAAK,WAAW,GAAG,WAAW,MAAM;EAC1C;EACA,UAAU,QAA8C;GACtD,IAAI,QACF,MAAM,IAAI,MACR,wDACF;GAEF,MAAM,OAAO,EAAE,QAAQ,OAAO,UAAU,EAAE;GAC1C,KAAK,KAAK,IAAI;GACd,WAAW,IAAI;EACjB;EACA,QAAQ,OAA6B;GACnC,KAAK,UACH,IAAI,eAAe,EACjB,MAAM,GAAG;IACP,EAAE,QAAQ,KAAK;IACf,EAAE,MAAM;GACV,EACF,CAAC,CACH;EACF;CACF;AACF"}

package/dist/core/utils/stream/path-utils.d.ts

@@ -1,12 +1,16 @@
-import type { AssistantStreamChunk } from "../../AssistantStreamChunk.js";
+import { AssistantStreamChunk } from "../../AssistantStreamChunk.js";
 import { Counter } from "../Counter.js";
-export declare class PathAppendEncoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
-    constructor(idx: number);
+
+//#region src/core/utils/stream/path-utils.d.ts
+declare class PathAppendEncoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
+  constructor(idx: number);
 }
-export declare class PathAppendDecoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
-    constructor(idx: number);
+declare class PathAppendDecoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
+  constructor(idx: number);
 }
-export declare class PathMergeEncoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
-    constructor(counter: Counter);
+declare class PathMergeEncoder extends TransformStream<AssistantStreamChunk, AssistantStreamChunk> {
+  constructor(counter: Counter);
 }
+//#endregion
+export { PathAppendDecoder, PathAppendEncoder, PathMergeEncoder };
 //# sourceMappingURL=path-utils.d.ts.map

package/dist/core/utils/stream/path-utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"path-utils.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/path-utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,sCAAmC;AACvE,OAAO,EAAE,OAAO,EAAE,sBAAmB;AAErC,qBAAa,iBAAkB,SAAQ,eAAe,CACpD,oBAAoB,EACpB,oBAAoB,CACrB;gBACa,GAAG,EAAE,MAAM;CAUxB;AAED,qBAAa,iBAAkB,SAAQ,eAAe,CACpD,oBAAoB,EACpB,oBAAoB,CACrB;gBACa,GAAG,EAAE,MAAM;CAiBxB;AAED,qBAAa,gBAAiB,SAAQ,eAAe,CACnD,oBAAoB,EACpB,oBAAoB,CACrB;gBACa,OAAO,EAAE,OAAO;CAwB7B"}
+{"version":3,"file":"path-utils.d.ts","names":[],"sources":["../../../../src/core/utils/stream/path-utils.ts"],"mappings":";;;;cAGa,iBAAA,SAA0B,eAAA,CACrC,oBAAA,EACA,oBAAA;cAEY,GAAA;AAAA;AAAA,cAYD,iBAAA,SAA0B,eAAA,CACrC,oBAAA,EACA,oBAAA;cAEY,GAAA;AAAA;AAAA,cAmBD,gBAAA,SAAyB,eAAA,CACpC,oBAAA,EACA,oBAAA;cAEY,OAAA,EAAS,OAAA;AAAA"}

package/dist/core/utils/stream/path-utils.js

@@ -1,54 +1,48 @@
 import { Counter } from "../Counter.js";
-export class PathAppendEncoder extends TransformStream {
-    constructor(idx) {
-        super({
-            transform(chunk, controller) {
-                controller.enqueue({
-                    ...chunk,
-                    path: [idx, ...chunk.path],
-                });
-            },
-        });
-    }
-}
-export class PathAppendDecoder extends TransformStream {
-    constructor(idx) {
-        super({
-            transform(chunk, controller) {
-                const { path: [idx2, ...path], } = chunk;
-                if (idx !== idx2)
-                    throw new Error(`Path mismatch: expected ${idx}, got ${idx2}`);
-                controller.enqueue({
-                    ...chunk,
-                    path,
-                });
-            },
-        });
-    }
-}
-export class PathMergeEncoder extends TransformStream {
-    constructor(counter) {
-        const innerCounter = new Counter();
-        const mapping = new Map();
-        super({
-            transform(chunk, controller) {
-                if (chunk.type === "part-start" && chunk.path.length === 0) {
-                    mapping.set(innerCounter.up(), counter.up());
-                }
-                const [idx, ...path] = chunk.path;
-                if (idx === undefined) {
-                    controller.enqueue(chunk);
-                    return;
-                }
-                const mappedIdx = mapping.get(idx);
-                if (mappedIdx === undefined)
-                    throw new Error("Path not found");
-                controller.enqueue({
-                    ...chunk,
-                    path: [mappedIdx, ...path],
-                });
-            },
-        });
-    }
-}
+//#region src/core/utils/stream/path-utils.ts
+var PathAppendEncoder = class extends TransformStream {
+	constructor(idx) {
+		super({ transform(chunk, controller) {
+			controller.enqueue({
+				...chunk,
+				path: [idx, ...chunk.path]
+			});
+		} });
+	}
+};
+var PathAppendDecoder = class extends TransformStream {
+	constructor(idx) {
+		super({ transform(chunk, controller) {
+			const { path: [idx2, ...path] } = chunk;
+			if (idx !== idx2) throw new Error(`Path mismatch: expected ${idx}, got ${idx2}`);
+			controller.enqueue({
+				...chunk,
+				path
+			});
+		} });
+	}
+};
+var PathMergeEncoder = class extends TransformStream {
+	constructor(counter) {
+		const innerCounter = new Counter();
+		const mapping = /* @__PURE__ */ new Map();
+		super({ transform(chunk, controller) {
+			if (chunk.type === "part-start" && chunk.path.length === 0) mapping.set(innerCounter.up(), counter.up());
+			const [idx, ...path] = chunk.path;
+			if (idx === void 0) {
+				controller.enqueue(chunk);
+				return;
+			}
+			const mappedIdx = mapping.get(idx);
+			if (mappedIdx === void 0) throw new Error("Path not found");
+			controller.enqueue({
+				...chunk,
+				path: [mappedIdx, ...path]
+			});
+		} });
+	}
+};
+//#endregion
+export { PathAppendDecoder, PathAppendEncoder, PathMergeEncoder };
+
 //# sourceMappingURL=path-utils.js.map

package/dist/core/utils/stream/path-utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"path-utils.js","sourceRoot":"","sources":["../../../../src/core/utils/stream/path-utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,sBAAmB;AAErC,MAAM,OAAO,iBAAkB,SAAQ,eAGtC;IACC,YAAY,GAAW;QACrB,KAAK,CAAC;YACJ,SAAS,CAAC,KAAK,EAAE,UAAU;gBACzB,UAAU,CAAC,OAAO,CAAC;oBACjB,GAAG,KAAK;oBACR,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC;iBAC3B,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,OAAO,iBAAkB,SAAQ,eAGtC;IACC,YAAY,GAAW;QACrB,KAAK,CAAC;YACJ,SAAS,CAAC,KAAK,EAAE,UAAU;gBACzB,MAAM,EACJ,IAAI,EAAE,CAAC,IAAI,EAAE,GAAG,IAAI,CAAC,GACtB,GAAG,KAAK,CAAC;gBAEV,IAAI,GAAG,KAAK,IAAI;oBACd,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,SAAS,IAAI,EAAE,CAAC,CAAC;gBAEjE,UAAU,CAAC,OAAO,CAAC;oBACjB,GAAG,KAAK;oBACR,IAAI;iBACL,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,OAAO,gBAAiB,SAAQ,eAGrC;IACC,YAAY,OAAgB;QAC1B,MAAM,YAAY,GAAG,IAAI,OAAO,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC1C,KAAK,CAAC;YACJ,SAAS,CAAC,KAAK,EAAE,UAAU;gBACzB,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC3D,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC/C,CAAC;gBAED,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC;gBAClC,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;oBACtB,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC1B,OAAO;gBACT,CAAC;gBACD,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACnC,IAAI,SAAS,KAAK,SAAS;oBAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;gBAE/D,UAAU,CAAC,OAAO,CAAC;oBACjB,GAAG,KAAK;oBACR,IAAI,EAAE,CAAC,SAAS,EAAE,GAAG,IAAI,CAAC;iBAC3B,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}
+{"version":3,"file":"path-utils.js","names":[],"sources":["../../../../src/core/utils/stream/path-utils.ts"],"sourcesContent":["import type { AssistantStreamChunk } from \"../../AssistantStreamChunk\";\nimport { Counter } from \"../Counter\";\n\nexport class PathAppendEncoder extends TransformStream<\n  AssistantStreamChunk,\n  AssistantStreamChunk\n> {\n  constructor(idx: number) {\n    super({\n      transform(chunk, controller) {\n        controller.enqueue({\n          ...chunk,\n          path: [idx, ...chunk.path],\n        });\n      },\n    });\n  }\n}\n\nexport class PathAppendDecoder extends TransformStream<\n  AssistantStreamChunk,\n  AssistantStreamChunk\n> {\n  constructor(idx: number) {\n    super({\n      transform(chunk, controller) {\n        const {\n          path: [idx2, ...path],\n        } = chunk;\n\n        if (idx !== idx2)\n          throw new Error(`Path mismatch: expected ${idx}, got ${idx2}`);\n\n        controller.enqueue({\n          ...chunk,\n          path,\n        });\n      },\n    });\n  }\n}\n\nexport class PathMergeEncoder extends TransformStream<\n  AssistantStreamChunk,\n  AssistantStreamChunk\n> {\n  constructor(counter: Counter) {\n    const innerCounter = new Counter();\n    const mapping = new Map<number, number>();\n    super({\n      transform(chunk, controller) {\n        if (chunk.type === \"part-start\" && chunk.path.length === 0) {\n          mapping.set(innerCounter.up(), counter.up());\n        }\n\n        const [idx, ...path] = chunk.path;\n        if (idx === undefined) {\n          controller.enqueue(chunk);\n          return;\n        }\n        const mappedIdx = mapping.get(idx);\n        if (mappedIdx === undefined) throw new Error(\"Path not found\");\n\n        controller.enqueue({\n          ...chunk,\n          path: [mappedIdx, ...path],\n        });\n      },\n    });\n  }\n}\n"],"mappings":";;AAGA,IAAa,oBAAb,cAAuC,gBAGrC;CACA,YAAY,KAAa;EACvB,MAAM,EACJ,UAAU,OAAO,YAAY;GAC3B,WAAW,QAAQ;IACjB,GAAG;IACH,MAAM,CAAC,KAAK,GAAG,MAAM,IAAI;GAC3B,CAAC;EACH,EACF,CAAC;CACH;AACF;AAEA,IAAa,oBAAb,cAAuC,gBAGrC;CACA,YAAY,KAAa;EACvB,MAAM,EACJ,UAAU,OAAO,YAAY;GAC3B,MAAM,EACJ,MAAM,CAAC,MAAM,GAAG,UACd;GAEJ,IAAI,QAAQ,MACV,MAAM,IAAI,MAAM,2BAA2B,IAAI,QAAQ,MAAM;GAE/D,WAAW,QAAQ;IACjB,GAAG;IACH;GACF,CAAC;EACH,EACF,CAAC;CACH;AACF;AAEA,IAAa,mBAAb,cAAsC,gBAGpC;CACA,YAAY,SAAkB;EAC5B,MAAM,eAAe,IAAI,QAAQ;EACjC,MAAM,0BAAU,IAAI,IAAoB;EACxC,MAAM,EACJ,UAAU,OAAO,YAAY;GAC3B,IAAI,MAAM,SAAS,gBAAgB,MAAM,KAAK,WAAW,GACvD,QAAQ,IAAI,aAAa,GAAG,GAAG,QAAQ,GAAG,CAAC;GAG7C,MAAM,CAAC,KAAK,GAAG,QAAQ,MAAM;GAC7B,IAAI,QAAQ,KAAA,GAAW;IACrB,WAAW,QAAQ,KAAK;IACxB;GACF;GACA,MAAM,YAAY,QAAQ,IAAI,GAAG;GACjC,IAAI,cAAc,KAAA,GAAW,MAAM,IAAI,MAAM,gBAAgB;GAE7D,WAAW,QAAQ;IACjB,GAAG;IACH,MAAM,CAAC,WAAW,GAAG,IAAI;GAC3B,CAAC;EACH,EACF,CAAC;CACH;AACF"}