assistant-stream 0.3.9 → 0.3.10

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

@@ -1 +1 @@
-{"version":3,"file":"LineDecoderStream.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/LineDecoderStream.ts"],"names":[],"mappings":"AAAA,qBAAa,iBAAkB,SAAQ,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC;IACpE,OAAO,CAAC,MAAM,CAAM;;CA2BrB"}
+{"version":3,"file":"LineDecoderStream.d.ts","sourceRoot":"","sources":["../../../../src/core/utils/stream/LineDecoderStream.ts"],"names":[],"mappings":"AAAA,qBAAa,iBAAkB,SAAQ,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC;IACpE,OAAO,CAAC,MAAM,CAAM;;CA6BrB"}

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

@@ -7,7 +7,9 @@ export class LineDecoderStream extends TransformStream {
                 const lines = this.buffer.split("\n");
                 // Process all complete lines
                 for (let i = 0; i < lines.length - 1; i++) {
-                    controller.enqueue(lines[i]);
+                    // Strip trailing \r to handle \r\n (CRLF) line endings per the SSE spec.
+                    const line = lines[i];
+                    controller.enqueue(line.endsWith("\r") ? line.slice(0, -1) : line);
                 }
                 // Keep the last incomplete line in the buffer
                 this.buffer = lines[lines.length - 1] || "";

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

@@ -1 +1 @@
-{"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"}
+{"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,yEAAyE;oBACzE,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAE,CAAC;oBACvB,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACrE,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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-stream",
-  "version": "0.3.9",
+  "version": "0.3.10",
   "description": "Streaming utilities for AI assistants",
   "keywords": [
     "ai",

package/src/core/serialization/assistant-transport/AssistantTransport.test.ts

@@ -161,6 +161,31 @@ describe("AssistantTransportDecoder", () => {
     expect(decodedChunks).toEqual(originalChunks);
   });
 
+  it("should decode SSE with CRLF line endings", async () => {
+    // Simulate a server that uses \r\n line endings (common in HTTP)
+    const sseText =
+      'data: {"type":"text-delta","textDelta":"Hello","path":[]}\r\n\r\n' +
+      "data: [DONE]\r\n\r\n";
+
+    const encoder = new TextEncoder();
+    const stream = new ReadableStream<Uint8Array>({
+      start(controller) {
+        controller.enqueue(encoder.encode(sseText));
+        controller.close();
+      },
+    });
+
+    const decodedStream = stream.pipeThrough(new AssistantTransportDecoder());
+    const decodedChunks = await collectChunks(decodedStream);
+
+    expect(decodedChunks).toHaveLength(1);
+    expect(decodedChunks[0]).toEqual({
+      type: "text-delta",
+      textDelta: "Hello",
+      path: [],
+    });
+  });
+
   it("should throw error when stream ends without [DONE]", async () => {
     // Manually create an SSE stream without [DONE]
     const sseText =

package/src/core/utils/stream/LineDecoderStream.test.ts

@@ -0,0 +1,99 @@
+import { describe, expect, it } from "vitest";
+import { LineDecoderStream } from "./LineDecoderStream";
+
+async function collectLines(stream: ReadableStream<string>): Promise<string[]> {
+  const reader = stream.getReader();
+  const lines: string[] = [];
+  while (true) {
+    const { done, value } = await reader.read();
+    if (done) break;
+    lines.push(value);
+  }
+  return lines;
+}
+
+function createTextStream(chunks: string[]): ReadableStream<string> {
+  return new ReadableStream<string>({
+    start(controller) {
+      for (const chunk of chunks) {
+        controller.enqueue(chunk);
+      }
+      controller.close();
+    },
+  });
+}
+
+describe("LineDecoderStream", () => {
+  it("should split lines on LF (\\n)", async () => {
+    const stream = createTextStream(["line1\nline2\nline3\n"]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(["line1", "line2", "line3"]);
+  });
+
+  it("should split lines on CRLF (\\r\\n)", async () => {
+    const stream = createTextStream(["line1\r\nline2\r\nline3\r\n"]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(["line1", "line2", "line3"]);
+  });
+
+  it("should handle CRLF split across chunks", async () => {
+    // The \r is at the end of one chunk and \n at the start of the next
+    const stream = createTextStream(["line1\r", "\nline2\r\n"]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(["line1", "line2"]);
+  });
+
+  it("should handle empty lines with LF", async () => {
+    const stream = createTextStream(["line1\n\nline2\n"]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(["line1", "", "line2"]);
+  });
+
+  it("should handle empty lines with CRLF", async () => {
+    const stream = createTextStream(["line1\r\n\r\nline2\r\n"]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(["line1", "", "line2"]);
+  });
+
+  it("should handle mixed LF and CRLF", async () => {
+    const stream = createTextStream(["line1\nline2\r\nline3\n"]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(["line1", "line2", "line3"]);
+  });
+
+  it("should handle multiple chunks", async () => {
+    const stream = createTextStream(["li", "ne1\nli", "ne2\n"]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(["line1", "line2"]);
+  });
+
+  it("should throw on incomplete line at end of stream", async () => {
+    const stream = createTextStream(["line1\nincomplete"]);
+    await expect(
+      collectLines(stream.pipeThrough(new LineDecoderStream())),
+    ).rejects.toThrow("Stream ended with an incomplete line");
+  });
+
+  it("should handle SSE-like data with CRLF", async () => {
+    const sseData = 'data: {"type":"text"}\r\n\r\ndata: [DONE]\r\n\r\n';
+    const stream = createTextStream([sseData]);
+    const lines = await collectLines(
+      stream.pipeThrough(new LineDecoderStream()),
+    );
+    expect(lines).toEqual(['data: {"type":"text"}', "", "data: [DONE]", ""]);
+  });
+});

package/src/core/utils/stream/LineDecoderStream.ts

@@ -9,7 +9,9 @@ export class LineDecoderStream extends TransformStream<string, string> {
 
         // Process all complete lines
         for (let i = 0; i < lines.length - 1; i++) {
-          controller.enqueue(lines[i]);
+          // Strip trailing \r to handle \r\n (CRLF) line endings per the SSE spec.
+          const line = lines[i]!;
+          controller.enqueue(line.endsWith("\r") ? line.slice(0, -1) : line);
         }
 
         // Keep the last incomplete line in the buffer