assistant-stream 0.3.28 → 0.3.29

package/dist/core/serialization/PlainText.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"PlainText.d.ts","names":[],"sources":["../../../src/core/serialization/PlainText.ts"],"mappings":";;;;cAKa,yBACH,wBAAwB,sBAAsB,WAAW,yBACtD;EAEX,SAAO;;;cA6CI,yBAAyB,wBACpC,WAAW,cACX"}
+{"version":3,"file":"PlainText.d.ts","names":[],"sources":["../../../src/core/serialization/PlainText.ts"],"mappings":";;;;cASa,yBACH,wBAAwB,sBAAsB,WAAW,yBACtD;EAEX,SAAO;;;cAsBI,yBAAyB,wBACpC,WAAW,cACX"}

package/dist/core/serialization/PlainText.js

@@ -1,29 +1,15 @@
 import { AssistantTransformStream } from "../utils/stream/AssistantTransformStream.js";
 import { PipeableTransformStream } from "../utils/stream/PipeableTransformStream.js";
+import { AssistantMetaTransformStream } from "../utils/stream/AssistantMetaTransformStream.js";
 //#region src/core/serialization/PlainText.ts
 var PlainTextEncoder = class extends PipeableTransformStream {
-	headers = new Headers({
-		"Content-Type": "text/plain; charset=utf-8",
-		"x-vercel-ai-data-stream": "v1"
-	});
+	headers = new Headers({ "Content-Type": "text/plain; charset=utf-8" });
 	constructor() {
 		super((readable) => {
 			const transform = new TransformStream({ transform(chunk, controller) {
-				const type = chunk.type;
-				switch (type) {
-					case "text-delta":
-						controller.enqueue(chunk.textDelta);
-						break;
-					case "part-start":
-					case "part-finish":
-					case "step-start":
-					case "step-finish":
-					case "message-finish":
-					case "error": break;
-					default: throw new Error(`unsupported chunk type: ${type}`);
-				}
+				if (chunk.type === "text-delta" && chunk.meta.type === "text") controller.enqueue(chunk.textDelta);
 			} });
-			return readable.pipeThrough(transform).pipeThrough(new TextEncoderStream());
+			return readable.pipeThrough(new AssistantMetaTransformStream()).pipeThrough(transform).pipeThrough(new TextEncoderStream());
 		});
 	}
 };

package/dist/core/serialization/PlainText.js.map

@@ -1 +1 @@
-{"version":3,"file":"PlainText.js","names":["unsupportedType"],"sources":["../../../src/core/serialization/PlainText.ts"],"sourcesContent":["import type { AssistantStreamEncoder } from \"../AssistantStream\";\nimport type { AssistantStreamChunk } from \"../AssistantStreamChunk\";\nimport { AssistantTransformStream } from \"../utils/stream/AssistantTransformStream\";\nimport { PipeableTransformStream } from \"../utils/stream/PipeableTransformStream\";\n\nexport class PlainTextEncoder\n  extends PipeableTransformStream<AssistantStreamChunk, Uint8Array<ArrayBuffer>>\n  implements AssistantStreamEncoder\n{\n  headers = new Headers({\n    \"Content-Type\": \"text/plain; charset=utf-8\",\n    \"x-vercel-ai-data-stream\": \"v1\",\n  });\n\n  constructor() {\n    super((readable) => {\n      const transform = new TransformStream<AssistantStreamChunk, string>({\n        transform(chunk, controller) {\n          const type = chunk.type;\n          switch (type) {\n            case \"text-delta\":\n              controller.enqueue(chunk.textDelta);\n              break;\n\n            case \"part-start\":\n            case \"part-finish\":\n            case \"step-start\":\n            case \"step-finish\":\n            case \"message-finish\":\n            case \"error\":\n              break;\n\n            default: {\n              const unsupportedType:\n                | \"tool-call-args-text-finish\"\n                | \"data\"\n                | \"annotations\"\n                | \"tool-call-begin\"\n                | \"tool-call-delta\"\n                | \"result\"\n                | \"update-state\" = type;\n              throw new Error(`unsupported chunk type: ${unsupportedType}`);\n            }\n          }\n        },\n      });\n\n      return readable\n        .pipeThrough(transform)\n        .pipeThrough(new TextEncoderStream());\n    });\n  }\n}\n\nexport class PlainTextDecoder extends PipeableTransformStream<\n  Uint8Array<ArrayBuffer>,\n  AssistantStreamChunk\n> {\n  constructor() {\n    super((readable) => {\n      const transform = new AssistantTransformStream<string>({\n        transform(chunk, controller) {\n          controller.appendText(chunk);\n        },\n      });\n\n      return readable\n        .pipeThrough(new TextDecoderStream())\n        .pipeThrough(transform);\n    });\n  }\n}\n"],"mappings":";;;AAKA,IAAa,mBAAb,cACU,wBAEV;CACE,UAAU,IAAI,QAAQ;EACpB,gBAAgB;EAChB,2BAA2B;CAC7B,CAAC;CAED,cAAc;EACZ,OAAO,aAAa;GAClB,MAAM,YAAY,IAAI,gBAA8C,EAClE,UAAU,OAAO,YAAY;IAC3B,MAAM,OAAO,MAAM;IACnB,QAAQ,MAAR;KACE,KAAK;MACH,WAAW,QAAQ,MAAM,SAAS;MAClC;KAEF,KAAK;KACL,KAAK;KACL,KAAK;KACL,KAAK;KACL,KAAK;KACL,KAAK,SACH;KAEF,SASE,MAAM,IAAI,MAAM,2BAA2BA,MAAiB;IAEhE;GACF,EACF,CAAC;GAED,OAAO,SACJ,YAAY,SAAS,CAAC,CACtB,YAAY,IAAI,kBAAkB,CAAC;EACxC,CAAC;CACH;AACF;AAEA,IAAa,mBAAb,cAAsC,wBAGpC;CACA,cAAc;EACZ,OAAO,aAAa;GAClB,MAAM,YAAY,IAAI,yBAAiC,EACrD,UAAU,OAAO,YAAY;IAC3B,WAAW,WAAW,KAAK;GAC7B,EACF,CAAC;GAED,OAAO,SACJ,YAAY,IAAI,kBAAkB,CAAC,CAAC,CACpC,YAAY,SAAS;EAC1B,CAAC;CACH;AACF"}
+{"version":3,"file":"PlainText.js","names":[],"sources":["../../../src/core/serialization/PlainText.ts"],"sourcesContent":["import type { AssistantStreamEncoder } from \"../AssistantStream\";\nimport type { AssistantStreamChunk } from \"../AssistantStreamChunk\";\nimport {\n  AssistantMetaTransformStream,\n  type AssistantMetaStreamChunk,\n} from \"../utils/stream/AssistantMetaTransformStream\";\nimport { AssistantTransformStream } from \"../utils/stream/AssistantTransformStream\";\nimport { PipeableTransformStream } from \"../utils/stream/PipeableTransformStream\";\n\nexport class PlainTextEncoder\n  extends PipeableTransformStream<AssistantStreamChunk, Uint8Array<ArrayBuffer>>\n  implements AssistantStreamEncoder\n{\n  headers = new Headers({\n    \"Content-Type\": \"text/plain; charset=utf-8\",\n  });\n\n  constructor() {\n    super((readable) => {\n      const transform = new TransformStream<AssistantMetaStreamChunk, string>({\n        transform(chunk, controller) {\n          if (chunk.type === \"text-delta\" && chunk.meta.type === \"text\") {\n            controller.enqueue(chunk.textDelta);\n          }\n        },\n      });\n\n      return readable\n        .pipeThrough(new AssistantMetaTransformStream())\n        .pipeThrough(transform)\n        .pipeThrough(new TextEncoderStream());\n    });\n  }\n}\n\nexport class PlainTextDecoder extends PipeableTransformStream<\n  Uint8Array<ArrayBuffer>,\n  AssistantStreamChunk\n> {\n  constructor() {\n    super((readable) => {\n      const transform = new AssistantTransformStream<string>({\n        transform(chunk, controller) {\n          controller.appendText(chunk);\n        },\n      });\n\n      return readable\n        .pipeThrough(new TextDecoderStream())\n        .pipeThrough(transform);\n    });\n  }\n}\n"],"mappings":";;;;AASA,IAAa,mBAAb,cACU,wBAEV;CACE,UAAU,IAAI,QAAQ,EACpB,gBAAgB,4BAClB,CAAC;CAED,cAAc;EACZ,OAAO,aAAa;GAClB,MAAM,YAAY,IAAI,gBAAkD,EACtE,UAAU,OAAO,YAAY;IAC3B,IAAI,MAAM,SAAS,gBAAgB,MAAM,KAAK,SAAS,QACrD,WAAW,QAAQ,MAAM,SAAS;GAEtC,EACF,CAAC;GAED,OAAO,SACJ,YAAY,IAAI,6BAA6B,CAAC,CAAC,CAC/C,YAAY,SAAS,CAAC,CACtB,YAAY,IAAI,kBAAkB,CAAC;EACxC,CAAC;CACH;AACF;AAEA,IAAa,mBAAb,cAAsC,wBAGpC;CACA,cAAc;EACZ,OAAO,aAAa;GAClB,MAAM,YAAY,IAAI,yBAAiC,EACrD,UAAU,OAAO,YAAY;IAC3B,WAAW,WAAW,KAAK;GAC7B,EACF,CAAC;GAED,OAAO,SACJ,YAAY,IAAI,kBAAkB,CAAC,CAAC,CACpC,YAAY,SAAS;EAC1B,CAAC;CACH;AACF"}

package/package.json

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

package/src/core/serialization/PlainText.test.ts

@@ -0,0 +1,54 @@
+import { describe, expect, it } from "vitest";
+import { AssistantStream } from "../AssistantStream";
+import type { AssistantStreamChunk } from "../AssistantStreamChunk";
+import { createAssistantStream } from "../modules/assistant-stream";
+import { PlainTextEncoder } from "./PlainText";
+
+describe("PlainTextEncoder", () => {
+  it("emits only the assistant text for a reasoning + tool-call + text stream", async () => {
+    const stream = createAssistantStream(async (controller) => {
+      controller.appendReasoning("internal chain of thought. ");
+      const tool = controller.addToolCallPart({
+        toolCallId: "call_1",
+        toolName: "search",
+      });
+      tool.argsText.append('{"query":"secret"}');
+      tool.setResponse({ result: "ok" });
+      tool.close();
+      controller.appendText("Hello!");
+    });
+
+    const response = AssistantStream.toResponse(stream, new PlainTextEncoder());
+    await expect(response.text()).resolves.toBe("Hello!");
+  });
+
+  it("skips annotations, data, and update-state chunks without throwing", async () => {
+    const chunks: AssistantStreamChunk[] = [
+      { type: "part-start", path: [], part: { type: "text" } },
+      { type: "text-delta", path: [0], textDelta: "Hi" },
+      { type: "annotations", path: [], annotations: [{ note: "cited" }] },
+      { type: "data", path: [], data: [{ progress: 1 }] },
+      {
+        type: "update-state",
+        path: [],
+        operations: [{ type: "set", path: ["k"], value: "v" }],
+      },
+      { type: "part-finish", path: [0] },
+    ];
+    const stream = new ReadableStream<AssistantStreamChunk>({
+      start(controller) {
+        for (const chunk of chunks) controller.enqueue(chunk);
+        controller.close();
+      },
+    });
+
+    const response = AssistantStream.toResponse(stream, new PlainTextEncoder());
+    await expect(response.text()).resolves.toBe("Hi");
+  });
+
+  it("sends plain text headers without the data-stream protocol marker", () => {
+    const headers = new PlainTextEncoder().headers;
+    expect(headers.get("Content-Type")).toBe("text/plain; charset=utf-8");
+    expect(headers.has("x-vercel-ai-data-stream")).toBe(false);
+  });
+});

package/src/core/serialization/PlainText.ts

@@ -1,5 +1,9 @@
 import type { AssistantStreamEncoder } from "../AssistantStream";
 import type { AssistantStreamChunk } from "../AssistantStreamChunk";
+import {
+  AssistantMetaTransformStream,
+  type AssistantMetaStreamChunk,
+} from "../utils/stream/AssistantMetaTransformStream";
 import { AssistantTransformStream } from "../utils/stream/AssistantTransformStream";
 import { PipeableTransformStream } from "../utils/stream/PipeableTransformStream";
 
@@ -9,43 +13,20 @@ export class PlainTextEncoder
 {
   headers = new Headers({
     "Content-Type": "text/plain; charset=utf-8",
-    "x-vercel-ai-data-stream": "v1",
   });
 
   constructor() {
     super((readable) => {
-      const transform = new TransformStream<AssistantStreamChunk, string>({
+      const transform = new TransformStream<AssistantMetaStreamChunk, string>({
         transform(chunk, controller) {
-          const type = chunk.type;
-          switch (type) {
-            case "text-delta":
-              controller.enqueue(chunk.textDelta);
-              break;
-
-            case "part-start":
-            case "part-finish":
-            case "step-start":
-            case "step-finish":
-            case "message-finish":
-            case "error":
-              break;
-
-            default: {
-              const unsupportedType:
-                | "tool-call-args-text-finish"
-                | "data"
-                | "annotations"
-                | "tool-call-begin"
-                | "tool-call-delta"
-                | "result"
-                | "update-state" = type;
-              throw new Error(`unsupported chunk type: ${unsupportedType}`);
-            }
+          if (chunk.type === "text-delta" && chunk.meta.type === "text") {
+            controller.enqueue(chunk.textDelta);
           }
         },
       });
 
       return readable
+        .pipeThrough(new AssistantMetaTransformStream())
         .pipeThrough(transform)
         .pipeThrough(new TextEncoderStream());
     });