assistant-stream 0.2.45 → 0.2.47

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

@@ -1,27 +1,25 @@
-// src/core/utils/stream/LineDecoderStream.ts
-var LineDecoderStream = class extends TransformStream {
-  buffer = "";
-  constructor() {
-    super({
-      transform: (chunk, controller) => {
-        this.buffer += chunk;
-        const lines = this.buffer.split("\n");
-        for (let i = 0; i < lines.length - 1; i++) {
-          controller.enqueue(lines[i]);
-        }
-        this.buffer = lines[lines.length - 1] || "";
-      },
-      flush: () => {
-        if (this.buffer) {
-          throw new Error(
-            `Stream ended with an incomplete line: "${this.buffer}"`
-          );
-        }
-      }
-    });
-  }
-};
-export {
-  LineDecoderStream
-};
+export class LineDecoderStream extends TransformStream {
+    buffer = "";
+    constructor() {
+        super({
+            transform: (chunk, controller) => {
+                this.buffer += chunk;
+                const lines = this.buffer.split("\n");
+                // Process all complete lines
+                for (let i = 0; i < lines.length - 1; i++) {
+                    controller.enqueue(lines[i]);
+                }
+                // Keep the last incomplete line in the buffer
+                this.buffer = lines[lines.length - 1] || "";
+            },
+            flush: () => {
+                // If there's content in the buffer when the stream ends, it means
+                // the stream ended with an incomplete line (no trailing newline)
+                if (this.buffer) {
+                    throw new Error(`Stream ended with an incomplete line: "${this.buffer}"`);
+                }
+            },
+        });
+    }
+}
 //# sourceMappingURL=LineDecoderStream.js.map

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/core/utils/stream/LineDecoderStream.ts"],"sourcesContent":["export class LineDecoderStream extends TransformStream<string, string> {\n  private buffer = \"\";\n\n  constructor() {\n    super({\n      transform: (chunk, controller) => {\n        this.buffer += chunk;\n        const lines = this.buffer.split(\"\\n\");\n\n        // Process all complete lines\n        for (let i = 0; i < lines.length - 1; i++) {\n          controller.enqueue(lines[i]);\n        }\n\n        // Keep the last incomplete line in the buffer\n        this.buffer = lines[lines.length - 1] || \"\";\n      },\n      flush: () => {\n        // If there's content in the buffer when the stream ends, it means\n        // the stream ended with an incomplete line (no trailing newline)\n        if (this.buffer) {\n          throw new Error(\n            `Stream ended with an incomplete line: \"${this.buffer}\"`,\n          );\n        }\n      },\n    });\n  }\n}\n"],"mappings":";AAAO,IAAM,oBAAN,cAAgC,gBAAgC;AAAA,EAC7D,SAAS;AAAA,EAEjB,cAAc;AACZ,UAAM;AAAA,MACJ,WAAW,CAAC,OAAO,eAAe;AAChC,aAAK,UAAU;AACf,cAAM,QAAQ,KAAK,OAAO,MAAM,IAAI;AAGpC,iBAAS,IAAI,GAAG,IAAI,MAAM,SAAS,GAAG,KAAK;AACzC,qBAAW,QAAQ,MAAM,CAAC,CAAC;AAAA,QAC7B;AAGA,aAAK,SAAS,MAAM,MAAM,SAAS,CAAC,KAAK;AAAA,MAC3C;AAAA,MACA,OAAO,MAAM;AAGX,YAAI,KAAK,QAAQ;AACf,gBAAM,IAAI;AAAA,YACR,0CAA0C,KAAK,MAAM;AAAA,UACvD;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;","names":[]}
+{"version":3,"file":"LineDecoderStream.js","sourceRoot":"","sources":["../../../../src/core/utils/stream/LineDecoderStream.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,iBAAkB,SAAQ,eAA+B;IAC5D,MAAM,GAAG,EAAE,CAAC;IAEpB;QACE,KAAK,CAAC;YACJ,SAAS,EAAE,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;gBAC/B,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC;gBACrB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEtC,6BAA6B;gBAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC1C,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/B,CAAC;gBAED,8CAA8C;gBAC9C,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YAC9C,CAAC;YACD,KAAK,EAAE,GAAG,EAAE;gBACV,kEAAkE;gBAClE,iEAAiE;gBACjE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;oBAChB,MAAM,IAAI,KAAK,CACb,0CAA0C,IAAI,CAAC,MAAM,GAAG,CACzD,CAAC;gBACJ,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

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

@@ -1,15 +1,11 @@
-// src/core/utils/stream/PipeableTransformStream.ts
-var PipeableTransformStream = class extends TransformStream {
-  constructor(transform) {
-    super();
-    const readable = transform(super.readable);
-    Object.defineProperty(this, "readable", {
-      value: readable,
-      writable: false
-    });
-  }
-};
-export {
-  PipeableTransformStream
-};
+export class PipeableTransformStream extends TransformStream {
+    constructor(transform) {
+        super();
+        const readable = transform(super.readable);
+        Object.defineProperty(this, "readable", {
+            value: readable,
+            writable: false,
+        });
+    }
+}
 //# sourceMappingURL=PipeableTransformStream.js.map

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/core/utils/stream/PipeableTransformStream.ts"],"sourcesContent":["export class PipeableTransformStream<I, O> extends TransformStream<I, O> {\n  constructor(transform: (readable: ReadableStream<I>) => ReadableStream<O>) {\n    super();\n    const readable = transform(super.readable as unknown as ReadableStream<I>);\n    Object.defineProperty(this, \"readable\", {\n      value: readable,\n      writable: false,\n    });\n  }\n}\n"],"mappings":";AAAO,IAAM,0BAAN,cAA4C,gBAAsB;AAAA,EACvE,YAAY,WAA+D;AACzE,UAAM;AACN,UAAM,WAAW,UAAU,MAAM,QAAwC;AACzE,WAAO,eAAe,MAAM,YAAY;AAAA,MACtC,OAAO;AAAA,MACP,UAAU;AAAA,IACZ,CAAC;AAAA,EACH;AACF;","names":[]}
+{"version":3,"file":"PipeableTransformStream.js","sourceRoot":"","sources":["../../../../src/core/utils/stream/PipeableTransformStream.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,uBAA8B,SAAQ,eAAqB;IACtE,YAAY,SAA6D;QACvE,KAAK,EAAE,CAAC;QACR,MAAM,QAAQ,GAAG,SAAS,CAAC,KAAK,CAAC,QAAwC,CAAC,CAAC;QAC3E,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,UAAU,EAAE;YACtC,KAAK,EAAE,QAAQ;YACf,QAAQ,EAAE,KAAK;SAChB,CAAC,CAAC;IACL,CAAC;CACF"}

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

@@ -1,4 +1,4 @@
-import { PipeableTransformStream } from "./PipeableTransformStream";
+import { PipeableTransformStream } from "./PipeableTransformStream.js";
 export declare class SSEEncoder<T> extends PipeableTransformStream<T, Uint8Array<ArrayBuffer>> {
     static readonly headers: Headers;
     headers: Headers;

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

@@ -1 +1 @@
-{"version":3,"file":"SSE.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/SSE.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAGpE,qBAAa,UAAU,CAAC,CAAC,CAAE,SAAQ,uBAAuB,CACxD,CAAC,EACD,UAAU,CAAC,WAAW,CAAC,CACxB;IACC,MAAM,CAAC,QAAQ,CAAC,OAAO,UAIpB;IAEH,OAAO,UAAsB;;CAe9B;AAoED,qBAAa,UAAU,CAAC,CAAC,CAAE,SAAQ,uBAAuB,CACxD,UAAU,CAAC,WAAW,CAAC,EACvB,CAAC,CACF;;CAsBA"}
+{"version":3,"file":"SSE.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/SSE.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,qCAAkC;AAGpE,qBAAa,UAAU,CAAC,CAAC,CAAE,SAAQ,uBAAuB,CACxD,CAAC,EACD,UAAU,CAAC,WAAW,CAAC,CACxB;IACC,MAAM,CAAC,QAAQ,CAAC,OAAO,UAIpB;IAEH,OAAO,UAAsB;;CAe9B;AAoED,qBAAa,UAAU,CAAC,CAAC,CAAE,SAAQ,uBAAuB,CACxD,UAAU,CAAC,WAAW,CAAC,EACvB,CAAC,CACF;;CAsBA"}

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

@@ -1,102 +1,94 @@
-// src/core/utils/stream/SSE.ts
 import { PipeableTransformStream } from "./PipeableTransformStream.js";
 import { LineDecoderStream } from "./LineDecoderStream.js";
-var SSEEncoder = class _SSEEncoder extends PipeableTransformStream {
-  static headers = new Headers({
-    "Content-Type": "text/event-stream",
-    "Cache-Control": "no-cache",
-    Connection: "keep-alive"
-  });
-  headers = _SSEEncoder.headers;
-  constructor() {
-    super(
-      (readable) => readable.pipeThrough(
-        new TransformStream({
-          transform(chunk, controller) {
-            controller.enqueue(`data: ${JSON.stringify(chunk)}
-
-`);
-          }
-        })
-      ).pipeThrough(new TextEncoderStream())
-    );
-  }
-};
-var SSEEventStream = class extends TransformStream {
-  constructor() {
-    let eventBuffer = {};
-    let dataLines = [];
-    super({
-      start() {
-        eventBuffer = {};
-        dataLines = [];
-      },
-      transform(line, controller) {
-        if (line.startsWith(":")) return;
-        if (line === "") {
-          if (dataLines.length > 0) {
-            controller.enqueue({
-              event: eventBuffer.event || "message",
-              data: dataLines.join("\n"),
-              id: eventBuffer.id,
-              retry: eventBuffer.retry
-            });
-          }
-          eventBuffer = {};
-          dataLines = [];
-          return;
-        }
-        const [field, ...rest] = line.split(":");
-        const value = rest.join(":").trimStart();
-        switch (field) {
-          case "event":
-            eventBuffer.event = value;
-            break;
-          case "data":
-            dataLines.push(value);
-            break;
-          case "id":
-            eventBuffer.id = value;
-            break;
-          case "retry":
-            eventBuffer.retry = Number(value);
-            break;
-        }
-      },
-      flush(controller) {
-        if (dataLines.length > 0) {
-          controller.enqueue({
-            event: eventBuffer.event || "message",
-            data: dataLines.join("\n"),
-            id: eventBuffer.id,
-            retry: eventBuffer.retry
-          });
-        }
-      }
+export class SSEEncoder extends PipeableTransformStream {
+    static headers = new Headers({
+        "Content-Type": "text/event-stream",
+        "Cache-Control": "no-cache",
+        Connection: "keep-alive",
     });
-  }
-};
-var SSEDecoder = class extends PipeableTransformStream {
-  constructor() {
-    super(
-      (readable) => readable.pipeThrough(new TextDecoderStream()).pipeThrough(new LineDecoderStream()).pipeThrough(new SSEEventStream()).pipeThrough(
-        new TransformStream({
-          transform(event, controller) {
-            switch (event.event) {
-              case "message":
-                controller.enqueue(JSON.parse(event.data));
-                break;
-              default:
-                throw new Error(`Unknown SSE event type: ${event.event}`);
-            }
-          }
-        })
-      )
-    );
-  }
-};
-export {
-  SSEDecoder,
-  SSEEncoder
-};
+    headers = SSEEncoder.headers;
+    constructor() {
+        super((readable) => readable
+            .pipeThrough(new TransformStream({
+            transform(chunk, controller) {
+                controller.enqueue(`data: ${JSON.stringify(chunk)}\n\n`);
+            },
+        }))
+            .pipeThrough(new TextEncoderStream()));
+    }
+}
+class SSEEventStream extends TransformStream {
+    constructor() {
+        let eventBuffer = {};
+        let dataLines = [];
+        super({
+            start() {
+                eventBuffer = {};
+                dataLines = [];
+            },
+            transform(line, controller) {
+                if (line.startsWith(":"))
+                    return; // Ignore comments
+                if (line === "") {
+                    if (dataLines.length > 0) {
+                        controller.enqueue({
+                            event: eventBuffer.event || "message",
+                            data: dataLines.join("\n"),
+                            id: eventBuffer.id,
+                            retry: eventBuffer.retry,
+                        });
+                    }
+                    eventBuffer = {};
+                    dataLines = [];
+                    return;
+                }
+                const [field, ...rest] = line.split(":");
+                const value = rest.join(":").trimStart();
+                switch (field) {
+                    case "event":
+                        eventBuffer.event = value;
+                        break;
+                    case "data":
+                        dataLines.push(value);
+                        break;
+                    case "id":
+                        eventBuffer.id = value;
+                        break;
+                    case "retry":
+                        eventBuffer.retry = Number(value);
+                        break;
+                }
+            },
+            flush(controller) {
+                if (dataLines.length > 0) {
+                    controller.enqueue({
+                        event: eventBuffer.event || "message",
+                        data: dataLines.join("\n"),
+                        id: eventBuffer.id,
+                        retry: eventBuffer.retry,
+                    });
+                }
+            },
+        });
+    }
+}
+export class SSEDecoder extends PipeableTransformStream {
+    constructor() {
+        super((readable) => readable
+            .pipeThrough(new TextDecoderStream())
+            .pipeThrough(new LineDecoderStream())
+            .pipeThrough(new SSEEventStream())
+            .pipeThrough(new TransformStream({
+            transform(event, controller) {
+                switch (event.event) {
+                    case "message":
+                        controller.enqueue(JSON.parse(event.data));
+                        break;
+                    default:
+                        throw new Error(`Unknown SSE event type: ${event.event}`);
+                }
+            },
+        })));
+    }
+}
 //# sourceMappingURL=SSE.js.map

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../../src/core/utils/stream/SSE.ts"],"sourcesContent":["import { PipeableTransformStream } from \"./PipeableTransformStream\";\nimport { LineDecoderStream } from \"./LineDecoderStream\";\n\nexport class SSEEncoder<T> extends PipeableTransformStream<\n  T,\n  Uint8Array<ArrayBuffer>\n> {\n  static readonly headers = new Headers({\n    \"Content-Type\": \"text/event-stream\",\n    \"Cache-Control\": \"no-cache\",\n    Connection: \"keep-alive\",\n  });\n\n  headers = SSEEncoder.headers;\n\n  constructor() {\n    super((readable) =>\n      readable\n        .pipeThrough(\n          new TransformStream<T, string>({\n            transform(chunk, controller) {\n              controller.enqueue(`data: ${JSON.stringify(chunk)}\\n\\n`);\n            },\n          }),\n        )\n        .pipeThrough(new TextEncoderStream()),\n    );\n  }\n}\n\ntype SSEEvent = {\n  event: string;\n  data: string;\n  id?: string | undefined;\n  retry?: number | undefined;\n};\n\nclass SSEEventStream extends TransformStream<string, SSEEvent> {\n  constructor() {\n    let eventBuffer: Partial<SSEEvent> = {};\n    let dataLines: string[] = [];\n\n    super({\n      start() {\n        eventBuffer = {};\n        dataLines = [];\n      },\n      transform(line, controller) {\n        if (line.startsWith(\":\")) return; // Ignore comments\n\n        if (line === \"\") {\n          if (dataLines.length > 0) {\n            controller.enqueue({\n              event: eventBuffer.event || \"message\",\n              data: dataLines.join(\"\\n\"),\n              id: eventBuffer.id,\n              retry: eventBuffer.retry,\n            });\n          }\n          eventBuffer = {};\n          dataLines = [];\n          return;\n        }\n\n        const [field, ...rest] = line.split(\":\");\n        const value = rest.join(\":\").trimStart();\n\n        switch (field) {\n          case \"event\":\n            eventBuffer.event = value;\n            break;\n          case \"data\":\n            dataLines.push(value);\n            break;\n          case \"id\":\n            eventBuffer.id = value;\n            break;\n          case \"retry\":\n            eventBuffer.retry = Number(value);\n            break;\n        }\n      },\n      flush(controller) {\n        if (dataLines.length > 0) {\n          controller.enqueue({\n            event: eventBuffer.event || \"message\",\n            data: dataLines.join(\"\\n\"),\n            id: eventBuffer.id,\n            retry: eventBuffer.retry,\n          });\n        }\n      },\n    });\n  }\n}\n\nexport class SSEDecoder<T> extends PipeableTransformStream<\n  Uint8Array<ArrayBuffer>,\n  T\n> {\n  constructor() {\n    super((readable) =>\n      readable\n        .pipeThrough(new TextDecoderStream())\n        .pipeThrough(new LineDecoderStream())\n        .pipeThrough(new SSEEventStream())\n        .pipeThrough(\n          new TransformStream<SSEEvent, T>({\n            transform(event, controller) {\n              switch (event.event) {\n                case \"message\":\n                  controller.enqueue(JSON.parse(event.data));\n                  break;\n                default:\n                  throw new Error(`Unknown SSE event type: ${event.event}`);\n              }\n            },\n          }),\n        ),\n    );\n  }\n}\n"],"mappings":";AAAA,SAAS,+BAA+B;AACxC,SAAS,yBAAyB;AAE3B,IAAM,aAAN,MAAM,oBAAsB,wBAGjC;AAAA,EACA,OAAgB,UAAU,IAAI,QAAQ;AAAA,IACpC,gBAAgB;AAAA,IAChB,iBAAiB;AAAA,IACjB,YAAY;AAAA,EACd,CAAC;AAAA,EAED,UAAU,YAAW;AAAA,EAErB,cAAc;AACZ;AAAA,MAAM,CAAC,aACL,SACG;AAAA,QACC,IAAI,gBAA2B;AAAA,UAC7B,UAAU,OAAO,YAAY;AAC3B,uBAAW,QAAQ,SAAS,KAAK,UAAU,KAAK,CAAC;AAAA;AAAA,CAAM;AAAA,UACzD;AAAA,QACF,CAAC;AAAA,MACH,EACC,YAAY,IAAI,kBAAkB,CAAC;AAAA,IACxC;AAAA,EACF;AACF;AASA,IAAM,iBAAN,cAA6B,gBAAkC;AAAA,EAC7D,cAAc;AACZ,QAAI,cAAiC,CAAC;AACtC,QAAI,YAAsB,CAAC;AAE3B,UAAM;AAAA,MACJ,QAAQ;AACN,sBAAc,CAAC;AACf,oBAAY,CAAC;AAAA,MACf;AAAA,MACA,UAAU,MAAM,YAAY;AAC1B,YAAI,KAAK,WAAW,GAAG,EAAG;AAE1B,YAAI,SAAS,IAAI;AACf,cAAI,UAAU,SAAS,GAAG;AACxB,uBAAW,QAAQ;AAAA,cACjB,OAAO,YAAY,SAAS;AAAA,cAC5B,MAAM,UAAU,KAAK,IAAI;AAAA,cACzB,IAAI,YAAY;AAAA,cAChB,OAAO,YAAY;AAAA,YACrB,CAAC;AAAA,UACH;AACA,wBAAc,CAAC;AACf,sBAAY,CAAC;AACb;AAAA,QACF;AAEA,cAAM,CAAC,OAAO,GAAG,IAAI,IAAI,KAAK,MAAM,GAAG;AACvC,cAAM,QAAQ,KAAK,KAAK,GAAG,EAAE,UAAU;AAEvC,gBAAQ,OAAO;AAAA,UACb,KAAK;AACH,wBAAY,QAAQ;AACpB;AAAA,UACF,KAAK;AACH,sBAAU,KAAK,KAAK;AACpB;AAAA,UACF,KAAK;AACH,wBAAY,KAAK;AACjB;AAAA,UACF,KAAK;AACH,wBAAY,QAAQ,OAAO,KAAK;AAChC;AAAA,QACJ;AAAA,MACF;AAAA,MACA,MAAM,YAAY;AAChB,YAAI,UAAU,SAAS,GAAG;AACxB,qBAAW,QAAQ;AAAA,YACjB,OAAO,YAAY,SAAS;AAAA,YAC5B,MAAM,UAAU,KAAK,IAAI;AAAA,YACzB,IAAI,YAAY;AAAA,YAChB,OAAO,YAAY;AAAA,UACrB,CAAC;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEO,IAAM,aAAN,cAA4B,wBAGjC;AAAA,EACA,cAAc;AACZ;AAAA,MAAM,CAAC,aACL,SACG,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,IAAI,eAAe,CAAC,EAChC;AAAA,QACC,IAAI,gBAA6B;AAAA,UAC/B,UAAU,OAAO,YAAY;AAC3B,oBAAQ,MAAM,OAAO;AAAA,cACnB,KAAK;AACH,2BAAW,QAAQ,KAAK,MAAM,MAAM,IAAI,CAAC;AACzC;AAAA,cACF;AACE,sBAAM,IAAI,MAAM,2BAA2B,MAAM,KAAK,EAAE;AAAA,YAC5D;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACJ;AAAA,EACF;AACF;","names":[]}
+{"version":3,"file":"SSE.js","sourceRoot":"","sources":["../../../../src/core/utils/stream/SSE.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,qCAAkC;AACpE,OAAO,EAAE,iBAAiB,EAAE,+BAA4B;AAExD,MAAM,OAAO,UAAc,SAAQ,uBAGlC;IACC,MAAM,CAAU,OAAO,GAAG,IAAI,OAAO,CAAC;QACpC,cAAc,EAAE,mBAAmB;QACnC,eAAe,EAAE,UAAU;QAC3B,UAAU,EAAE,YAAY;KACzB,CAAC,CAAC;IAEH,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;IAE7B;QACE,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE,CACjB,QAAQ;aACL,WAAW,CACV,IAAI,eAAe,CAAY;YAC7B,SAAS,CAAC,KAAK,EAAE,UAAU;gBACzB,UAAU,CAAC,OAAO,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC3D,CAAC;SACF,CAAC,CACH;aACA,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC,CACxC,CAAC;IACJ,CAAC;;AAUH,MAAM,cAAe,SAAQ,eAAiC;IAC5D;QACE,IAAI,WAAW,GAAsB,EAAE,CAAC;QACxC,IAAI,SAAS,GAAa,EAAE,CAAC;QAE7B,KAAK,CAAC;YACJ,KAAK;gBACH,WAAW,GAAG,EAAE,CAAC;gBACjB,SAAS,GAAG,EAAE,CAAC;YACjB,CAAC;YACD,SAAS,CAAC,IAAI,EAAE,UAAU;gBACxB,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;oBAAE,OAAO,CAAC,kBAAkB;gBAEpD,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;oBAChB,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACzB,UAAU,CAAC,OAAO,CAAC;4BACjB,KAAK,EAAE,WAAW,CAAC,KAAK,IAAI,SAAS;4BACrC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;4BAC1B,EAAE,EAAE,WAAW,CAAC,EAAE;4BAClB,KAAK,EAAE,WAAW,CAAC,KAAK;yBACzB,CAAC,CAAC;oBACL,CAAC;oBACD,WAAW,GAAG,EAAE,CAAC;oBACjB,SAAS,GAAG,EAAE,CAAC;oBACf,OAAO;gBACT,CAAC;gBAED,MAAM,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBACzC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC;gBAEzC,QAAQ,KAAK,EAAE,CAAC;oBACd,KAAK,OAAO;wBACV,WAAW,CAAC,KAAK,GAAG,KAAK,CAAC;wBAC1B,MAAM;oBACR,KAAK,MAAM;wBACT,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBACtB,MAAM;oBACR,KAAK,IAAI;wBACP,WAAW,CAAC,EAAE,GAAG,KAAK,CAAC;wBACvB,MAAM;oBACR,KAAK,OAAO;wBACV,WAAW,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;wBAClC,MAAM;gBACV,CAAC;YACH,CAAC;YACD,KAAK,CAAC,UAAU;gBACd,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,UAAU,CAAC,OAAO,CAAC;wBACjB,KAAK,EAAE,WAAW,CAAC,KAAK,IAAI,SAAS;wBACrC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;wBAC1B,EAAE,EAAE,WAAW,CAAC,EAAE;wBAClB,KAAK,EAAE,WAAW,CAAC,KAAK;qBACzB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,OAAO,UAAc,SAAQ,uBAGlC;IACC;QACE,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE,CACjB,QAAQ;aACL,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;aACpC,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;aACpC,WAAW,CAAC,IAAI,cAAc,EAAE,CAAC;aACjC,WAAW,CACV,IAAI,eAAe,CAAc;YAC/B,SAAS,CAAC,KAAK,EAAE,UAAU;gBACzB,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;oBACpB,KAAK,SAAS;wBACZ,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;wBAC3C,MAAM;oBACR;wBACE,MAAM,IAAI,KAAK,CAAC,2BAA2B,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC;SACF,CAAC,CACH,CACJ,CAAC;IACJ,CAAC;CACF"}

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

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

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

@@ -1 +1 @@
-{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
+{"version":3,"file":"UnderlyingReadable.js","sourceRoot":"","sources":["../../../../src/core/utils/stream/UnderlyingReadable.ts"],"names":[],"mappings":""}

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

@@ -1,4 +1,4 @@
-import { AssistantStreamChunk } from "../../AssistantStreamChunk";
+import { AssistantStreamChunk } from "../../AssistantStreamChunk.js";
 export declare const createMergeStream: () => {
     readable: ReadableStream<AssistantStreamChunk>;
     isSealed(): boolean;

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,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAQlE,eAAO,MAAM,iBAAiB;;;;sBA0ER,cAAc,CAAC,oBAAoB,CAAC;mBAUvC,oBAAoB;CAWtC,CAAC"}
+{"version":3,"file":"merge.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/merge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,sCAAmC;AAQlE,eAAO,MAAM,iBAAiB;;;;sBA0ER,cAAc,CAAC,oBAAoB,CAAC;mBAUvC,oBAAoB;CAWtC,CAAC"}

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

@@ -1,85 +1,173 @@
-// src/core/utils/stream/merge.ts
 import { promiseWithResolvers } from "../../../utils/promiseWithResolvers.js";
-var 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);
+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;
+            });
         }
-        currentPull?.resolve();
-        currentPull = void 0;
-      }).catch((e) => {
-        console.error(e);
-        list.forEach((item2) => {
-          item2.reader.cancel();
-        });
-        list.length = 0;
-        controller.error(e);
-        currentPull?.reject(e);
-        currentPull = void 0;
-      });
-    }
-  };
-  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();
-          }
-        })
-      );
-    }
-  };
-};
-export {
-  createMergeStream
+    };
+    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();
+                },
+            }));
+        },
+    };
 };
+// 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();
+//   }
+// }
 //# sourceMappingURL=merge.js.map