assistant-stream 0.3.26 → 0.3.28

package/dist/core/gorp/GorpStreamResponse.js

@@ -0,0 +1,51 @@
+import { PipeableTransformStream } from "../utils/stream/PipeableTransformStream.js";
+import { GorpStreamAccumulator } from "./GorpStreamAccumulator.js";
+import { SSEDecoder, SSEEncoder } from "../utils/stream/SSE.js";
+//#region src/core/gorp/GorpStreamResponse.ts
+var GorpStreamEncoder = class extends PipeableTransformStream {
+	constructor() {
+		let isFirstChunk = true;
+		super((readable) => readable.pipeThrough(new TransformStream({ transform(chunk, controller) {
+			if (isFirstChunk) controller.enqueue([{
+				type: "set",
+				path: [],
+				value: chunk.snapshot
+			}]);
+			else controller.enqueue(chunk.operations);
+			isFirstChunk = false;
+		} })).pipeThrough(new SSEEncoder()));
+	}
+};
+var GorpStreamDecoder = class extends PipeableTransformStream {
+	constructor() {
+		const accumulator = new GorpStreamAccumulator();
+		super((readable) => readable.pipeThrough(new SSEDecoder()).pipeThrough(new TransformStream({ transform(operations, controller) {
+			accumulator.append(operations);
+			controller.enqueue({
+				snapshot: accumulator.state,
+				operations
+			});
+		} })));
+	}
+};
+var GorpStreamResponse = class extends Response {
+	constructor(body) {
+		super(body.pipeThrough(new GorpStreamEncoder()), { headers: new Headers({
+			"Content-Type": "text/event-stream",
+			"Cache-Control": "no-cache",
+			Connection: "keep-alive",
+			"Assistant-Stream-Format": "object-stream/v0"
+		}) });
+	}
+};
+const fromGorpStreamResponse = (response) => {
+	if (!response.ok) throw new Error(`Response failed, status ${response.status}`);
+	if (!response.body) throw new Error("Response body is null");
+	if (response.headers.get("Content-Type")?.split(";", 1)[0]?.trim().toLowerCase() !== "text/event-stream") throw new Error("Response is not an event stream");
+	if (response.headers.get("Assistant-Stream-Format") !== "object-stream/v0") throw new Error("Unsupported Assistant-Stream-Format header");
+	return response.body.pipeThrough(new GorpStreamDecoder());
+};
+//#endregion
+export { GorpStreamDecoder, GorpStreamEncoder, GorpStreamResponse, fromGorpStreamResponse };
+
+//# sourceMappingURL=GorpStreamResponse.js.map

package/dist/core/gorp/GorpStreamResponse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GorpStreamResponse.js","names":[],"sources":["../../../src/core/gorp/GorpStreamResponse.ts"],"sourcesContent":["import { PipeableTransformStream } from \"../utils/stream/PipeableTransformStream\";\nimport { GorpStreamAccumulator } from \"./GorpStreamAccumulator\";\nimport { SSEDecoder, SSEEncoder } from \"../utils/stream/SSE\";\nimport type { GorpStreamChunk, GorpStreamOperation } from \"./types\";\n\nexport class GorpStreamEncoder extends PipeableTransformStream<\n  GorpStreamChunk,\n  Uint8Array\n> {\n  constructor() {\n    let isFirstChunk = true;\n    super((readable) =>\n      readable\n        .pipeThrough(\n          new TransformStream<GorpStreamChunk, readonly GorpStreamOperation[]>({\n            transform(chunk, controller) {\n              if (isFirstChunk) {\n                controller.enqueue([\n                  { type: \"set\", path: [], value: chunk.snapshot },\n                ]);\n              } else {\n                controller.enqueue(chunk.operations);\n              }\n              isFirstChunk = false;\n            },\n          }),\n        )\n        .pipeThrough(new SSEEncoder()),\n    );\n  }\n}\n\nexport class GorpStreamDecoder extends PipeableTransformStream<\n  Uint8Array<ArrayBuffer>,\n  GorpStreamChunk\n> {\n  constructor() {\n    const accumulator = new GorpStreamAccumulator();\n    super((readable) =>\n      readable\n        .pipeThrough(new SSEDecoder<readonly GorpStreamOperation[]>())\n        .pipeThrough(\n          new TransformStream<readonly GorpStreamOperation[], GorpStreamChunk>({\n            transform(operations, controller) {\n              accumulator.append(operations);\n              controller.enqueue({\n                snapshot: accumulator.state,\n                operations,\n              });\n            },\n          }),\n        ),\n    );\n  }\n}\n\nexport class GorpStreamResponse extends Response {\n  constructor(body: ReadableStream<GorpStreamChunk>) {\n    super(body.pipeThrough(new GorpStreamEncoder()), {\n      headers: new Headers({\n        \"Content-Type\": \"text/event-stream\",\n        \"Cache-Control\": \"no-cache\",\n        Connection: \"keep-alive\",\n        \"Assistant-Stream-Format\": \"object-stream/v0\",\n      }),\n    });\n  }\n}\n\nexport const fromGorpStreamResponse = (\n  response: Response,\n): ReadableStream<GorpStreamChunk> => {\n  if (!response.ok)\n    throw new Error(`Response failed, status ${response.status}`);\n  if (!response.body) throw new Error(\"Response body is null\");\n  const mediaType = response.headers\n    .get(\"Content-Type\")\n    ?.split(\";\", 1)[0]\n    ?.trim()\n    .toLowerCase();\n  if (mediaType !== \"text/event-stream\") {\n    throw new Error(\"Response is not an event stream\");\n  }\n  if (response.headers.get(\"Assistant-Stream-Format\") !== \"object-stream/v0\") {\n    throw new Error(\"Unsupported Assistant-Stream-Format header\");\n  }\n  return response.body.pipeThrough(new GorpStreamDecoder());\n};\n"],"mappings":";;;;AAKA,IAAa,oBAAb,cAAuC,wBAGrC;CACA,cAAc;EACZ,IAAI,eAAe;EACnB,OAAO,aACL,SACG,YACC,IAAI,gBAAiE,EACnE,UAAU,OAAO,YAAY;GAC3B,IAAI,cACF,WAAW,QAAQ,CACjB;IAAE,MAAM;IAAO,MAAM,CAAC;IAAG,OAAO,MAAM;GAAS,CACjD,CAAC;QAED,WAAW,QAAQ,MAAM,UAAU;GAErC,eAAe;EACjB,EACF,CAAC,CACH,CAAC,CACA,YAAY,IAAI,WAAW,CAAC,CACjC;CACF;AACF;AAEA,IAAa,oBAAb,cAAuC,wBAGrC;CACA,cAAc;EACZ,MAAM,cAAc,IAAI,sBAAsB;EAC9C,OAAO,aACL,SACG,YAAY,IAAI,WAA2C,CAAC,CAAC,CAC7D,YACC,IAAI,gBAAiE,EACnE,UAAU,YAAY,YAAY;GAChC,YAAY,OAAO,UAAU;GAC7B,WAAW,QAAQ;IACjB,UAAU,YAAY;IACtB;GACF,CAAC;EACH,EACF,CAAC,CACH,CACJ;CACF;AACF;AAEA,IAAa,qBAAb,cAAwC,SAAS;CAC/C,YAAY,MAAuC;EACjD,MAAM,KAAK,YAAY,IAAI,kBAAkB,CAAC,GAAG,EAC/C,SAAS,IAAI,QAAQ;GACnB,gBAAgB;GAChB,iBAAiB;GACjB,YAAY;GACZ,2BAA2B;EAC7B,CAAC,EACH,CAAC;CACH;AACF;AAEA,MAAa,0BACX,aACoC;CACpC,IAAI,CAAC,SAAS,IACZ,MAAM,IAAI,MAAM,2BAA2B,SAAS,QAAQ;CAC9D,IAAI,CAAC,SAAS,MAAM,MAAM,IAAI,MAAM,uBAAuB;CAM3D,IALkB,SAAS,QACxB,IAAI,cAAc,CAAC,EAClB,MAAM,KAAK,CAAC,CAAC,CAAC,EAAE,EAChB,KAAK,CAAC,CACP,YAAY,MACG,qBAChB,MAAM,IAAI,MAAM,iCAAiC;CAEnD,IAAI,SAAS,QAAQ,IAAI,yBAAyB,MAAM,oBACtD,MAAM,IAAI,MAAM,4CAA4C;CAE9D,OAAO,SAAS,KAAK,YAAY,IAAI,kBAAkB,CAAC;AAC1D"}

package/dist/core/gorp/changeTree.d.ts

@@ -0,0 +1,16 @@
+//#region src/core/gorp/changeTree.d.ts
+type ChangeNode = true | {
+  [key: string]: ChangeNode;
+};
+declare function createChangeNode(): {
+  [key: string]: ChangeNode;
+};
+declare function assertSafePathSegment(segment: string): void;
+declare function markChanged(node: ChangeNode, path: readonly string[]): ChangeNode;
+declare function mergeChanged(target: ChangeNode, source: ChangeNode): ChangeNode;
+declare function lookupChange(node: ChangeNode, path: readonly string[]): ChangeNode | false;
+declare function lookupValue(state: unknown, path: readonly string[]): unknown;
+declare function diffKeys(newNode: unknown, oldNode: unknown): string[];
+//#endregion
+export { ChangeNode, assertSafePathSegment, createChangeNode, diffKeys, lookupChange, lookupValue, markChanged, mergeChanged };
+//# sourceMappingURL=changeTree.d.ts.map

package/dist/core/gorp/changeTree.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"changeTree.d.ts","names":[],"sources":["../../../src/core/gorp/changeTree.ts"],"mappings":";KAAY;GAAuB,cAAc;;iBAEjC;GAAuB,cAAc;;iBAMrC,sBAAsB;iBAMtB,YACd,MAAM,YACN,0BACC;iBAiBa,aACd,QAAQ,YACR,QAAQ,aACP;iBAwBa,aACd,MAAM,YACN,0BACC;iBAUa,YAAY,gBAAgB;iBAY5B,SAAS,kBAAkB"}

package/dist/core/gorp/changeTree.js

@@ -0,0 +1,77 @@
+//#region src/core/gorp/changeTree.ts
+function createChangeNode() {
+	return Object.create(null);
+}
+const UNSAFE_PATH_SEGMENTS = /* @__PURE__ */ new Set([
+	"__proto__",
+	"constructor",
+	"prototype"
+]);
+function assertSafePathSegment(segment) {
+	if (UNSAFE_PATH_SEGMENTS.has(segment)) throw new Error(`Unsafe gorp path segment: ${segment}`);
+}
+function markChanged(node, path) {
+	if (node === true || path.length === 0) return true;
+	let cursor = node;
+	for (let i = 0; i < path.length - 1; i++) {
+		const key = path[i];
+		assertSafePathSegment(key);
+		const next = Object.hasOwn(cursor, key) ? cursor[key] : void 0;
+		if (next === true) return node;
+		if (next === void 0) cursor[key] = createChangeNode();
+		cursor = cursor[key];
+	}
+	const leaf = path[path.length - 1];
+	assertSafePathSegment(leaf);
+	cursor[leaf] = true;
+	return node;
+}
+function mergeChanged(target, source) {
+	if (target === true || source === true) return true;
+	for (const key of Object.keys(source)) {
+		assertSafePathSegment(key);
+		const sChild = source[key];
+		const tChild = Object.hasOwn(target, key) ? target[key] : void 0;
+		target[key] = tChild === void 0 ? cloneChangeNode(sChild) : mergeChanged(tChild, sChild);
+	}
+	return target;
+}
+function cloneChangeNode(source) {
+	if (source === true) return true;
+	const clone = createChangeNode();
+	for (const key of Object.keys(source)) {
+		assertSafePathSegment(key);
+		clone[key] = cloneChangeNode(source[key]);
+	}
+	return clone;
+}
+function lookupChange(node, path) {
+	for (const key of path) {
+		if (node === true) return true;
+		const next = Object.hasOwn(node, key) ? node[key] : void 0;
+		if (next === void 0) return false;
+		node = next;
+	}
+	return node;
+}
+function lookupValue(state, path) {
+	let node = state;
+	for (const key of path) {
+		assertSafePathSegment(key);
+		if (typeof node !== "object" || node === null) return void 0;
+		node = Object.hasOwn(node, key) ? node[key] : void 0;
+	}
+	return node;
+}
+function diffKeys(newNode, oldNode) {
+	const newKeys = typeof newNode === "object" && newNode !== null ? Object.keys(newNode) : [];
+	const oldKeys = typeof oldNode === "object" && oldNode !== null ? Object.keys(oldNode) : [];
+	if (oldKeys.length === 0) return newKeys;
+	const newSet = new Set(newKeys);
+	const deletedKeys = oldKeys.filter((k) => !newSet.has(k)).reverse();
+	return [...newKeys, ...deletedKeys];
+}
+//#endregion
+export { assertSafePathSegment, createChangeNode, diffKeys, lookupChange, lookupValue, markChanged, mergeChanged };
+
+//# sourceMappingURL=changeTree.js.map

package/dist/core/gorp/changeTree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"changeTree.js","names":[],"sources":["../../../src/core/gorp/changeTree.ts"],"sourcesContent":["export type ChangeNode = true | { [key: string]: ChangeNode };\n\nexport function createChangeNode(): { [key: string]: ChangeNode } {\n  return Object.create(null);\n}\n\nconst UNSAFE_PATH_SEGMENTS = new Set([\"__proto__\", \"constructor\", \"prototype\"]);\n\nexport function assertSafePathSegment(segment: string): void {\n  if (UNSAFE_PATH_SEGMENTS.has(segment)) {\n    throw new Error(`Unsafe gorp path segment: ${segment}`);\n  }\n}\n\nexport function markChanged(\n  node: ChangeNode,\n  path: readonly string[],\n): ChangeNode {\n  if (node === true || path.length === 0) return true;\n  let cursor = node;\n  for (let i = 0; i < path.length - 1; i++) {\n    const key = path[i]!;\n    assertSafePathSegment(key);\n    const next = Object.hasOwn(cursor, key) ? cursor[key] : undefined;\n    if (next === true) return node;\n    if (next === undefined) cursor[key] = createChangeNode();\n    cursor = cursor[key] as { [k: string]: ChangeNode };\n  }\n  const leaf = path[path.length - 1]!;\n  assertSafePathSegment(leaf);\n  cursor[leaf] = true;\n  return node;\n}\n\nexport function mergeChanged(\n  target: ChangeNode,\n  source: ChangeNode,\n): ChangeNode {\n  if (target === true || source === true) return true;\n  for (const key of Object.keys(source)) {\n    assertSafePathSegment(key);\n    const sChild = source[key]!;\n    const tChild = Object.hasOwn(target, key) ? target[key] : undefined;\n    target[key] =\n      tChild === undefined\n        ? cloneChangeNode(sChild)\n        : mergeChanged(tChild, sChild);\n  }\n  return target;\n}\n\nfunction cloneChangeNode(source: ChangeNode): ChangeNode {\n  if (source === true) return true;\n  const clone = createChangeNode();\n  for (const key of Object.keys(source)) {\n    assertSafePathSegment(key);\n    clone[key] = cloneChangeNode(source[key]!);\n  }\n  return clone;\n}\n\nexport function lookupChange(\n  node: ChangeNode,\n  path: readonly string[],\n): ChangeNode | false {\n  for (const key of path) {\n    if (node === true) return true;\n    const next = Object.hasOwn(node, key) ? node[key] : undefined;\n    if (next === undefined) return false;\n    node = next;\n  }\n  return node;\n}\n\nexport function lookupValue(state: unknown, path: readonly string[]): unknown {\n  let node = state;\n  for (const key of path) {\n    assertSafePathSegment(key);\n    if (typeof node !== \"object\" || node === null) return undefined;\n    node = Object.hasOwn(node, key)\n      ? (node as Record<string, unknown>)[key]\n      : undefined;\n  }\n  return node;\n}\n\nexport function diffKeys(newNode: unknown, oldNode: unknown): string[] {\n  const newKeys =\n    typeof newNode === \"object\" && newNode !== null ? Object.keys(newNode) : [];\n  const oldKeys =\n    typeof oldNode === \"object\" && oldNode !== null ? Object.keys(oldNode) : [];\n  if (oldKeys.length === 0) return newKeys;\n  const newSet = new Set(newKeys);\n  const deletedKeys = oldKeys.filter((k) => !newSet.has(k)).reverse();\n  return [...newKeys, ...deletedKeys];\n}\n"],"mappings":";AAEA,SAAgB,mBAAkD;CAChE,OAAO,OAAO,OAAO,IAAI;AAC3B;AAEA,MAAM,uCAAuB,IAAI,IAAI;CAAC;CAAa;CAAe;AAAW,CAAC;AAE9E,SAAgB,sBAAsB,SAAuB;CAC3D,IAAI,qBAAqB,IAAI,OAAO,GAClC,MAAM,IAAI,MAAM,6BAA6B,SAAS;AAE1D;AAEA,SAAgB,YACd,MACA,MACY;CACZ,IAAI,SAAS,QAAQ,KAAK,WAAW,GAAG,OAAO;CAC/C,IAAI,SAAS;CACb,KAAK,IAAI,IAAI,GAAG,IAAI,KAAK,SAAS,GAAG,KAAK;EACxC,MAAM,MAAM,KAAK;EACjB,sBAAsB,GAAG;EACzB,MAAM,OAAO,OAAO,OAAO,QAAQ,GAAG,IAAI,OAAO,OAAO,KAAA;EACxD,IAAI,SAAS,MAAM,OAAO;EAC1B,IAAI,SAAS,KAAA,GAAW,OAAO,OAAO,iBAAiB;EACvD,SAAS,OAAO;CAClB;CACA,MAAM,OAAO,KAAK,KAAK,SAAS;CAChC,sBAAsB,IAAI;CAC1B,OAAO,QAAQ;CACf,OAAO;AACT;AAEA,SAAgB,aACd,QACA,QACY;CACZ,IAAI,WAAW,QAAQ,WAAW,MAAM,OAAO;CAC/C,KAAK,MAAM,OAAO,OAAO,KAAK,MAAM,GAAG;EACrC,sBAAsB,GAAG;EACzB,MAAM,SAAS,OAAO;EACtB,MAAM,SAAS,OAAO,OAAO,QAAQ,GAAG,IAAI,OAAO,OAAO,KAAA;EAC1D,OAAO,OACL,WAAW,KAAA,IACP,gBAAgB,MAAM,IACtB,aAAa,QAAQ,MAAM;CACnC;CACA,OAAO;AACT;AAEA,SAAS,gBAAgB,QAAgC;CACvD,IAAI,WAAW,MAAM,OAAO;CAC5B,MAAM,QAAQ,iBAAiB;CAC/B,KAAK,MAAM,OAAO,OAAO,KAAK,MAAM,GAAG;EACrC,sBAAsB,GAAG;EACzB,MAAM,OAAO,gBAAgB,OAAO,IAAK;CAC3C;CACA,OAAO;AACT;AAEA,SAAgB,aACd,MACA,MACoB;CACpB,KAAK,MAAM,OAAO,MAAM;EACtB,IAAI,SAAS,MAAM,OAAO;EAC1B,MAAM,OAAO,OAAO,OAAO,MAAM,GAAG,IAAI,KAAK,OAAO,KAAA;EACpD,IAAI,SAAS,KAAA,GAAW,OAAO;EAC/B,OAAO;CACT;CACA,OAAO;AACT;AAEA,SAAgB,YAAY,OAAgB,MAAkC;CAC5E,IAAI,OAAO;CACX,KAAK,MAAM,OAAO,MAAM;EACtB,sBAAsB,GAAG;EACzB,IAAI,OAAO,SAAS,YAAY,SAAS,MAAM,OAAO,KAAA;EACtD,OAAO,OAAO,OAAO,MAAM,GAAG,IACzB,KAAiC,OAClC,KAAA;CACN;CACA,OAAO;AACT;AAEA,SAAgB,SAAS,SAAkB,SAA4B;CACrE,MAAM,UACJ,OAAO,YAAY,YAAY,YAAY,OAAO,OAAO,KAAK,OAAO,IAAI,CAAC;CAC5E,MAAM,UACJ,OAAO,YAAY,YAAY,YAAY,OAAO,OAAO,KAAK,OAAO,IAAI,CAAC;CAC5E,IAAI,QAAQ,WAAW,GAAG,OAAO;CACjC,MAAM,SAAS,IAAI,IAAI,OAAO;CAC9B,MAAM,cAAc,QAAQ,QAAQ,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ;CAClE,OAAO,CAAC,GAAG,SAAS,GAAG,WAAW;AACpC"}

package/dist/core/gorp/createGorpStream.d.ts

@@ -0,0 +1,16 @@
+import { ReadonlyJSONValue } from "../../utils/json/json-value.js";
+import "../../utils.js";
+import { GorpStreamChunk, GorpStreamOperation } from "./types.js";
+//#region src/core/gorp/createGorpStream.d.ts
+type GorpStreamController = {
+  readonly abortSignal: AbortSignal;
+  enqueue(operations: readonly GorpStreamOperation[]): void;
+};
+type CreateGorpStreamOptions = {
+  execute: (controller: GorpStreamController) => void | PromiseLike<void>;
+  defaultValue?: ReadonlyJSONValue;
+};
+declare const createGorpStream: ({ execute, defaultValue }: CreateGorpStreamOptions) => ReadableStream<GorpStreamChunk>;
+//#endregion
+export { createGorpStream };
+//# sourceMappingURL=createGorpStream.d.ts.map

package/dist/core/gorp/createGorpStream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createGorpStream.d.ts","names":[],"sources":["../../../src/core/gorp/createGorpStream.ts"],"mappings":";;;;KAKK;WACM,aAAa;EAEtB,QAAQ,qBAAqB;;KA8D1B;EACH,UAAU,YAAY,gCAAgC;EACtD,eAAe;;cAGJ,qBAAoB,SAAA,gBAG9B,4BAAuB,eAAA"}

package/dist/core/{object/createObjectStream.js → gorp/createGorpStream.js}

@@ -1,18 +1,20 @@
-import { ObjectStreamAccumulator } from "./ObjectStreamAccumulator.js";
+import { GorpStreamAccumulator } from "./GorpStreamAccumulator.js";
 import { withPromiseOrValue } from "../utils/withPromiseOrValue.js";
-//#region src/core/object/createObjectStream.ts
-var ObjectStreamControllerImpl = class {
+//#region src/core/gorp/createGorpStream.ts
+var GorpStreamControllerImpl = class {
 	_controller;
 	_abortController = new AbortController();
 	_accumulator;
+	_cancelled = false;
 	get abortSignal() {
 		return this._abortController.signal;
 	}
 	constructor(controller, defaultValue) {
 		this._controller = controller;
-		this._accumulator = new ObjectStreamAccumulator(defaultValue);
+		this._accumulator = new GorpStreamAccumulator(defaultValue);
 	}
 	enqueue(operations) {
+		if (this._cancelled) return;
 		this._accumulator.append(operations);
 		this._controller.enqueue({
 			snapshot: this._accumulator.state,
@@ -20,12 +22,15 @@ var ObjectStreamControllerImpl = class {
 		});
 	}
 	__internalError(error) {
+		if (this._cancelled) return;
 		this._controller.error(error);
 	}
 	__internalClose() {
+		if (this._cancelled) return;
 		this._controller.close();
 	}
 	__internalCancel(reason) {
+		this._cancelled = true;
 		this._abortController.abort(reason);
 	}
 };
@@ -33,14 +38,14 @@ const getStreamControllerPair = (defaultValue) => {
 	let controller;
 	return [new ReadableStream({
 		start(c) {
-			controller = new ObjectStreamControllerImpl(c, defaultValue);
+			controller = new GorpStreamControllerImpl(c, defaultValue);
 		},
 		cancel(reason) {
 			controller.__internalCancel(reason);
 		}
 	}), controller];
 };
-const createObjectStream = ({ execute, defaultValue = {} }) => {
+const createGorpStream = ({ execute, defaultValue = {} }) => {
 	const [stream, controller] = getStreamControllerPair(defaultValue);
 	withPromiseOrValue(() => execute(controller), () => {
 		controller.__internalClose();
@@ -50,6 +55,6 @@ const createObjectStream = ({ execute, defaultValue = {} }) => {
 	return stream;
 };
 //#endregion
-export { createObjectStream };
+export { createGorpStream };
 
-//# sourceMappingURL=createObjectStream.js.map
+//# sourceMappingURL=createGorpStream.js.map

package/dist/core/gorp/createGorpStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"createGorpStream.js","names":[],"sources":["../../../src/core/gorp/createGorpStream.ts"],"sourcesContent":["import type { ReadonlyJSONValue } from \"../../utils\";\nimport { withPromiseOrValue } from \"../utils/withPromiseOrValue\";\nimport { GorpStreamAccumulator } from \"./GorpStreamAccumulator\";\nimport type { GorpStreamOperation, GorpStreamChunk } from \"./types\";\n\ntype GorpStreamController = {\n  readonly abortSignal: AbortSignal;\n\n  enqueue(operations: readonly GorpStreamOperation[]): void;\n};\n\nclass GorpStreamControllerImpl implements GorpStreamController {\n  private _controller: ReadableStreamDefaultController<GorpStreamChunk>;\n  private _abortController = new AbortController();\n  private _accumulator: GorpStreamAccumulator;\n  private _cancelled = false;\n\n  get abortSignal() {\n    return this._abortController.signal;\n  }\n\n  constructor(\n    controller: ReadableStreamDefaultController<GorpStreamChunk>,\n    defaultValue: ReadonlyJSONValue,\n  ) {\n    this._controller = controller;\n    this._accumulator = new GorpStreamAccumulator(defaultValue);\n  }\n\n  enqueue(operations: readonly GorpStreamOperation[]) {\n    if (this._cancelled) return;\n\n    this._accumulator.append(operations);\n\n    this._controller.enqueue({\n      snapshot: this._accumulator.state,\n      operations,\n    });\n  }\n\n  __internalError(error: unknown) {\n    if (this._cancelled) return;\n    this._controller.error(error);\n  }\n\n  __internalClose() {\n    if (this._cancelled) return;\n    this._controller.close();\n  }\n\n  __internalCancel(reason?: unknown) {\n    this._cancelled = true;\n    this._abortController.abort(reason);\n  }\n}\n\nconst getStreamControllerPair = (defaultValue: ReadonlyJSONValue) => {\n  let controller!: GorpStreamControllerImpl;\n  const stream = new ReadableStream<GorpStreamChunk>({\n    start(c) {\n      controller = new GorpStreamControllerImpl(c, defaultValue);\n    },\n    cancel(reason: unknown) {\n      controller.__internalCancel(reason);\n    },\n  });\n\n  return [stream, controller] as const;\n};\n\ntype CreateGorpStreamOptions = {\n  execute: (controller: GorpStreamController) => void | PromiseLike<void>;\n  defaultValue?: ReadonlyJSONValue;\n};\n\nexport const createGorpStream = ({\n  execute,\n  defaultValue = {},\n}: CreateGorpStreamOptions) => {\n  const [stream, controller] = getStreamControllerPair(defaultValue);\n\n  withPromiseOrValue(\n    () => execute(controller),\n    () => {\n      controller.__internalClose();\n    },\n    (e: unknown) => {\n      controller.__internalError(e);\n    },\n  );\n\n  return stream;\n};\n"],"mappings":";;;AAWA,IAAM,2BAAN,MAA+D;CAC7D;CACA,mBAA2B,IAAI,gBAAgB;CAC/C;CACA,aAAqB;CAErB,IAAI,cAAc;EAChB,OAAO,KAAK,iBAAiB;CAC/B;CAEA,YACE,YACA,cACA;EACA,KAAK,cAAc;EACnB,KAAK,eAAe,IAAI,sBAAsB,YAAY;CAC5D;CAEA,QAAQ,YAA4C;EAClD,IAAI,KAAK,YAAY;EAErB,KAAK,aAAa,OAAO,UAAU;EAEnC,KAAK,YAAY,QAAQ;GACvB,UAAU,KAAK,aAAa;GAC5B;EACF,CAAC;CACH;CAEA,gBAAgB,OAAgB;EAC9B,IAAI,KAAK,YAAY;EACrB,KAAK,YAAY,MAAM,KAAK;CAC9B;CAEA,kBAAkB;EAChB,IAAI,KAAK,YAAY;EACrB,KAAK,YAAY,MAAM;CACzB;CAEA,iBAAiB,QAAkB;EACjC,KAAK,aAAa;EAClB,KAAK,iBAAiB,MAAM,MAAM;CACpC;AACF;AAEA,MAAM,2BAA2B,iBAAoC;CACnE,IAAI;CAUJ,OAAO,CAAC,IATW,eAAgC;EACjD,MAAM,GAAG;GACP,aAAa,IAAI,yBAAyB,GAAG,YAAY;EAC3D;EACA,OAAO,QAAiB;GACtB,WAAW,iBAAiB,MAAM;EACpC;CACF,CAEa,GAAG,UAAU;AAC5B;AAOA,MAAa,oBAAoB,EAC/B,SACA,eAAe,CAAC,QACa;CAC7B,MAAM,CAAC,QAAQ,cAAc,wBAAwB,YAAY;CAEjE,yBACQ,QAAQ,UAAU,SAClB;EACJ,WAAW,gBAAgB;CAC7B,IACC,MAAe;EACd,WAAW,gBAAgB,CAAC;CAC9B,CACF;CAEA,OAAO;AACT"}

package/dist/core/{object → gorp}/types.d.ts

@@ -1,7 +1,7 @@
 import { ReadonlyJSONValue } from "../../utils/json/json-value.js";
 import "../../utils.js";
-//#region src/core/object/types.d.ts
-type ObjectStreamOperation = {
+//#region src/core/gorp/types.d.ts
+type AssistantTransportStateOperation = {
   readonly type: "set";
   readonly path: readonly string[];
   readonly value: ReadonlyJSONValue;
@@ -10,10 +10,11 @@ type ObjectStreamOperation = {
   readonly path: readonly string[];
   readonly value: string;
 };
-type ObjectStreamChunk = {
+type GorpStreamOperation = AssistantTransportStateOperation;
+type GorpStreamChunk = {
   readonly snapshot: ReadonlyJSONValue;
-  readonly operations: readonly ObjectStreamOperation[];
+  readonly operations: readonly GorpStreamOperation[];
 };
 //#endregion
-export { ObjectStreamChunk, ObjectStreamOperation };
+export { AssistantTransportStateOperation, GorpStreamChunk, GorpStreamOperation };
 //# sourceMappingURL=types.d.ts.map

package/dist/core/gorp/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/core/gorp/types.ts"],"mappings":";;;KAEY;WAEG;WACA;WACA,OAAO;;WAGP;WACA;WACA;;KAGH,sBAAsB;KAEtB;WACD,UAAU;WACV,qBAAqB"}

package/dist/core/modules/assistant-stream.d.ts

@@ -76,10 +76,11 @@ type AssistantStreamController = {
  * Creates an {@link AssistantStream} and writes to it with an
  * {@link AssistantStreamController}.
  *
- * The callback may write synchronously or asynchronously. If it throws, an
- * `error` chunk is emitted before the error is rethrown; when the callback
- * settles, the stream is closed automatically unless the controller was
- * already closed.
+ * The callback may write synchronously or asynchronously. If it throws while
+ * the stream is open, an `error` chunk is emitted. Failures after an explicit
+ * close are logged, while failures after consumer cancellation are discarded.
+ * When the callback settles, the stream is closed automatically unless the
+ * controller was already closed.
  */
 declare function createAssistantStream(callback: (controller: AssistantStreamController) => PromiseLike<void> | void): AssistantStream;
 /**

package/dist/core/modules/assistant-stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"assistant-stream.d.ts","names":[],"sources":["../../../src/core/modules/assistant-stream.ts"],"mappings":";;;;;;;;KAuBK;EACH;EACA;EACA;EACA,OAAO;EACP,WAAW,iBAAiB;;;;;;;;;KAUlB;;EAEV,WAAW;;EAEX,gBAAgB;;EAEhB,aAAa,SAAS;;EAEtB,WAAW,SAAS;;EAEpB,WAAW,SAAS;;;;;;;;EAQpB,eAAe;;;;;;;EAOf,gBAAgB,kBAAkB;;;;;;;EAOlC,gBAAgB,SAAS,mBAAmB;;EAE5C,QAAQ,OAAO;;;;;;;EAOf,MAAM,QAAQ;;EAEd;;;;;;;EAOA,aAAa,mBAAmB;;;;;;;;;;;iBA2MlB,sBACd,WAAW,YAAY,8BAA8B,2BACpD;;;;;;;;iBAiCa,6CAA+B,iBAAA;;;;;;;;iBAsB/B,8BACd,WAAW,YAAY,8BAA8B,2BAAwB"}
+{"version":3,"file":"assistant-stream.d.ts","names":[],"sources":["../../../src/core/modules/assistant-stream.ts"],"mappings":";;;;;;;;KAuBK;EACH;EACA;EACA;EACA,OAAO;EACP,WAAW,iBAAiB;;;;;;;;;KAUlB;;EAEV,WAAW;;EAEX,gBAAgB;;EAEhB,aAAa,SAAS;;EAEtB,WAAW,SAAS;;EAEpB,WAAW,SAAS;;;;;;;;EAQpB,eAAe;;;;;;;EAOf,gBAAgB,kBAAkB;;;;;;;EAOlC,gBAAgB,SAAS,mBAAmB;;EAE5C,QAAQ,OAAO;;;;;;;EAOf,MAAM,QAAQ;;EAEd;;;;;;;EAOA,aAAa,mBAAmB;;;;;;;;;;;;iBAoNlB,sBACd,WAAW,YAAY,8BAA8B,2BACpD;;;;;;;;iBAkCa,6CAA+B,iBAAA;;;;;;;;iBAsB/B,8BACd,WAAW,YAAY,8BAA8B,2BAAwB"}

package/dist/core/modules/assistant-stream.js

@@ -18,7 +18,10 @@ var AssistantStreamControllerImpl = class AssistantStreamControllerImpl {
 		};
 	}
 	get __internal_isClosed() {
-		return this._state.merger.isSealed();
+		return this._state.merger.isSealed() || this._state.merger.isCancelled() || this._state.merger.isErrored();
+	}
+	get __internal_isCancelled() {
+		return this._state.merger.isCancelled();
 	}
 	__internal_getReadable() {
 		return this._state.merger.readable;
@@ -133,10 +136,11 @@ var AssistantStreamControllerImpl = class AssistantStreamControllerImpl {
 * Creates an {@link AssistantStream} and writes to it with an
 * {@link AssistantStreamController}.
 *
-* The callback may write synchronously or asynchronously. If it throws, an
-* `error` chunk is emitted before the error is rethrown; when the callback
-* settles, the stream is closed automatically unless the controller was
-* already closed.
+* The callback may write synchronously or asynchronously. If it throws while
+* the stream is open, an `error` chunk is emitted. Failures after an explicit
+* close are logged, while failures after consumer cancellation are discarded.
+* When the callback settles, the stream is closed automatically unless the
+* controller was already closed.
 */
 function createAssistantStream(callback) {
 	const controller = new AssistantStreamControllerImpl();
@@ -149,7 +153,7 @@ function createAssistantStream(callback) {
 				path: [],
 				error: String(e)
 			});
-			throw e;
+			else if (!controller.__internal_isCancelled) console.error(e);
 		} finally {
 			if (!controller.__internal_isClosed) controller.close();
 		}

package/dist/core/modules/assistant-stream.js.map

@@ -1 +1 @@
-{"version":3,"file":"assistant-stream.js","names":[],"sources":["../../../src/core/modules/assistant-stream.ts"],"sourcesContent":["import { AssistantStream } from \"../AssistantStream\";\nimport type { AssistantStreamChunk, PartInit } from \"../AssistantStreamChunk\";\nimport { createMergeStream } from \"../utils/stream/merge\";\nimport { createTextStreamController, type TextStreamController } from \"./text\";\nimport {\n  createToolCallStreamController,\n  type ToolCallStreamController,\n} from \"./tool-call\";\nimport { Counter } from \"../utils/Counter\";\nimport {\n  PathAppendEncoder,\n  PathMergeEncoder,\n} from \"../utils/stream/path-utils\";\nimport { DataStreamEncoder } from \"../serialization/data-stream/DataStream\";\nimport type { DataPart, FilePart, SourcePart } from \"../utils/types\";\nimport { generateId } from \"../utils/generateId\";\nimport type {\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"../../utils/json/json-value\";\nimport type { ToolResponseLike } from \"../tool/ToolResponse\";\nimport { promiseWithResolvers } from \"../../utils/promiseWithResolvers\";\n\ntype ToolCallPartInit = {\n  toolCallId?: string;\n  toolName: string;\n  argsText?: string;\n  args?: ReadonlyJSONObject;\n  response?: ToolResponseLike<ReadonlyJSONValue>;\n};\n\n/**\n * Imperative writer for constructing an {@link AssistantStream}.\n *\n * The controller handles part boundaries for common streaming operations. Use\n * `appendText` and `appendReasoning` for simple token streams, or open explicit\n * parts with `addTextPart` and `addToolCallPart` when you need direct control.\n */\nexport type AssistantStreamController = {\n  /** Appends text to the current text part, opening one if needed. */\n  appendText(textDelta: string): void;\n  /** Appends reasoning text to the current reasoning part, opening one if needed. */\n  appendReasoning(reasoningDelta: string): void;\n  /** Appends a source citation part to the stream. */\n  appendSource(options: SourcePart): void;\n  /** Appends a file part to the stream. */\n  appendFile(options: FilePart): void;\n  /** Appends a named data part to the stream. */\n  appendData(options: DataPart): void;\n  /**\n   * Opens a text part and returns its writer.\n   *\n   * Close the returned controller when the text part is complete. Opening a new\n   * part through this controller closes any implicit text or reasoning append\n   * part first.\n   */\n  addTextPart(): TextStreamController;\n  /**\n   * Opens a tool-call part by tool name and returns its writer.\n   *\n   * A tool call id is generated automatically. Use the object overload when the\n   * caller already has an id, initial args, args text, or response.\n   */\n  addToolCallPart(options: string): ToolCallStreamController;\n  /**\n   * Opens a tool-call part and returns its writer.\n   *\n   * Use this overload to provide a stable `toolCallId`, initial arguments,\n   * streamed argument text, or an immediate {@link ToolResponseLike}.\n   */\n  addToolCallPart(options: ToolCallPartInit): ToolCallStreamController;\n  /** Enqueues a raw protocol chunk. Prefer higher-level helpers when possible. */\n  enqueue(chunk: AssistantStreamChunk): void;\n  /**\n   * Merges another assistant stream into this stream.\n   *\n   * Paths from the merged stream are remapped so its parts appear at the next\n   * available position in this controller's output.\n   */\n  merge(stream: AssistantStream): void;\n  /** Closes any active part and finishes the stream. */\n  close(): void;\n  /**\n   * Returns a controller that writes child parts with `parentId` attached.\n   *\n   * Use this for nested or related parts that should be associated with an\n   * existing message or part in downstream renderers.\n   */\n  withParentId(parentId: string): AssistantStreamController;\n};\n\n// Shared state between controller instances\ntype AssistantStreamControllerState = {\n  merger: ReturnType<typeof createMergeStream>;\n  append?:\n    | {\n        controller: TextStreamController;\n        kind: \"text\" | \"reasoning\";\n        parentId: string | undefined;\n      }\n    | undefined;\n  contentCounter: Counter;\n  closeSubscriber?: () => void;\n};\n\nclass AssistantStreamControllerImpl implements AssistantStreamController {\n  private readonly _state: AssistantStreamControllerState;\n  private _parentId?: string;\n\n  constructor(state?: AssistantStreamControllerState) {\n    this._state = state || {\n      merger: createMergeStream(),\n      contentCounter: new Counter(),\n    };\n  }\n\n  get __internal_isClosed() {\n    return this._state.merger.isSealed();\n  }\n\n  __internal_getReadable() {\n    return this._state.merger.readable;\n  }\n\n  __internal_subscribeToClose(callback: () => void) {\n    this._state.closeSubscriber = callback;\n  }\n\n  private _addPart(part: PartInit, stream: AssistantStream) {\n    if (this._state.append) {\n      this._state.append.controller.close();\n      this._state.append = undefined;\n    }\n\n    this.enqueue({\n      type: \"part-start\",\n      part,\n      path: [],\n    });\n    this._state.merger.addStream(\n      stream.pipeThrough(\n        new PathAppendEncoder(this._state.contentCounter.value),\n      ),\n    );\n  }\n\n  merge(stream: AssistantStream) {\n    this._state.merger.addStream(\n      stream.pipeThrough(new PathMergeEncoder(this._state.contentCounter)),\n    );\n  }\n\n  appendText(textDelta: string) {\n    if (\n      this._state.append?.kind !== \"text\" ||\n      this._state.append.parentId !== this._parentId\n    ) {\n      this._state.append = {\n        kind: \"text\",\n        parentId: this._parentId,\n        controller: this.addTextPart(),\n      };\n    }\n    this._state.append.controller.append(textDelta);\n  }\n\n  appendReasoning(textDelta: string) {\n    if (\n      this._state.append?.kind !== \"reasoning\" ||\n      this._state.append.parentId !== this._parentId\n    ) {\n      this._state.append = {\n        kind: \"reasoning\",\n        parentId: this._parentId,\n        controller: this.addReasoningPart(),\n      };\n    }\n    this._state.append.controller.append(textDelta);\n  }\n\n  addTextPart() {\n    const [stream, controller] = createTextStreamController();\n    this._addPart(this._withParentIdOption({ type: \"text\" }), stream);\n    return controller;\n  }\n\n  addReasoningPart() {\n    const [stream, controller] = createTextStreamController();\n    this._addPart(this._withParentIdOption({ type: \"reasoning\" }), stream);\n    return controller;\n  }\n\n  addToolCallPart(\n    options: string | ToolCallPartInit,\n  ): ToolCallStreamController {\n    const opt = typeof options === \"string\" ? { toolName: options } : options;\n    const toolName = opt.toolName;\n    const toolCallId = opt.toolCallId ?? generateId();\n\n    const [stream, controller] = createToolCallStreamController();\n    this._addPart(\n      {\n        type: \"tool-call\",\n        toolName,\n        toolCallId,\n        ...(this._parentId && { parentId: this._parentId }),\n      },\n      stream,\n    );\n\n    if (opt.argsText !== undefined) {\n      controller.argsText.append(opt.argsText);\n      controller.argsText.close();\n    }\n    if (opt.args !== undefined) {\n      controller.argsText.append(JSON.stringify(opt.args));\n      controller.argsText.close();\n    }\n    if (opt.response !== undefined) {\n      controller.setResponse(opt.response);\n    }\n\n    return controller;\n  }\n\n  private _finishedPartStream(): AssistantStream {\n    return new ReadableStream({\n      start(controller) {\n        controller.enqueue({ type: \"part-finish\", path: [] });\n        controller.close();\n      },\n    });\n  }\n\n  private _withParentIdOption<T>(options: T): T {\n    if (!this._parentId) return options;\n    return { ...options, parentId: this._parentId };\n  }\n\n  appendSource(options: SourcePart) {\n    this._addPart(\n      this._withParentIdOption(options),\n      this._finishedPartStream(),\n    );\n  }\n\n  appendFile(options: FilePart) {\n    this._addPart(\n      this._withParentIdOption(options),\n      this._finishedPartStream(),\n    );\n  }\n\n  appendData(options: DataPart) {\n    this._addPart(\n      this._withParentIdOption(options),\n      this._finishedPartStream(),\n    );\n  }\n\n  enqueue(chunk: AssistantStreamChunk) {\n    this._state.merger.enqueue(chunk);\n\n    if (chunk.type === \"part-start\" && chunk.path.length === 0) {\n      this._state.contentCounter.up();\n    }\n  }\n\n  withParentId(parentId: string): AssistantStreamController {\n    const controller = new AssistantStreamControllerImpl(this._state);\n    controller._parentId = parentId;\n    return controller;\n  }\n\n  close() {\n    this._state.append?.controller?.close();\n    this._state.merger.seal();\n\n    this._state.closeSubscriber?.();\n  }\n}\n\n/**\n * Creates an {@link AssistantStream} and writes to it with an\n * {@link AssistantStreamController}.\n *\n * The callback may write synchronously or asynchronously. If it throws, an\n * `error` chunk is emitted before the error is rethrown; when the callback\n * settles, the stream is closed automatically unless the controller was\n * already closed.\n */\nexport function createAssistantStream(\n  callback: (controller: AssistantStreamController) => PromiseLike<void> | void,\n): AssistantStream {\n  const controller = new AssistantStreamControllerImpl();\n\n  const runTask = async () => {\n    try {\n      await callback(controller);\n    } catch (e) {\n      if (!controller.__internal_isClosed) {\n        controller.enqueue({\n          type: \"error\",\n          path: [],\n          error: String(e),\n        });\n      }\n      throw e;\n    } finally {\n      if (!controller.__internal_isClosed) {\n        controller.close();\n      }\n    }\n  };\n  runTask();\n\n  return controller.__internal_getReadable();\n}\n\n/**\n * Creates an {@link AssistantStream} together with the controller used to\n * write into it.\n *\n * Use this when the stream needs to be returned before all writers are known.\n * Closing the returned controller finishes the paired stream.\n */\nexport function createAssistantStreamController() {\n  const { resolve, promise } = promiseWithResolvers<void>();\n  let controller!: AssistantStreamController;\n  const stream = createAssistantStream((c) => {\n    controller = c;\n\n    (controller as AssistantStreamControllerImpl).__internal_subscribeToClose(\n      resolve,\n    );\n\n    return promise;\n  });\n  return [stream, controller] as const;\n}\n\n/**\n * Creates a `Response` whose body is an encoded {@link AssistantStream}.\n *\n * This is the HTTP-route convenience form of {@link createAssistantStream}; it\n * uses {@link DataStreamEncoder} so the response can be consumed by matching\n * assistant-ui data stream decoders.\n */\nexport function createAssistantStreamResponse(\n  callback: (controller: AssistantStreamController) => PromiseLike<void> | void,\n) {\n  return AssistantStream.toResponse(\n    createAssistantStream(callback),\n    new DataStreamEncoder(),\n  );\n}\n"],"mappings":";;;;;;;;;;AAyGA,IAAM,gCAAN,MAAM,8BAAmE;CACvE;CACA;CAEA,YAAY,OAAwC;EAClD,KAAK,SAAS,SAAS;GACrB,QAAQ,kBAAkB;GAC1B,gBAAgB,IAAI,QAAQ;EAC9B;CACF;CAEA,IAAI,sBAAsB;EACxB,OAAO,KAAK,OAAO,OAAO,SAAS;CACrC;CAEA,yBAAyB;EACvB,OAAO,KAAK,OAAO,OAAO;CAC5B;CAEA,4BAA4B,UAAsB;EAChD,KAAK,OAAO,kBAAkB;CAChC;CAEA,SAAiB,MAAgB,QAAyB;EACxD,IAAI,KAAK,OAAO,QAAQ;GACtB,KAAK,OAAO,OAAO,WAAW,MAAM;GACpC,KAAK,OAAO,SAAS,KAAA;EACvB;EAEA,KAAK,QAAQ;GACX,MAAM;GACN;GACA,MAAM,CAAC;EACT,CAAC;EACD,KAAK,OAAO,OAAO,UACjB,OAAO,YACL,IAAI,kBAAkB,KAAK,OAAO,eAAe,KAAK,CACxD,CACF;CACF;CAEA,MAAM,QAAyB;EAC7B,KAAK,OAAO,OAAO,UACjB,OAAO,YAAY,IAAI,iBAAiB,KAAK,OAAO,cAAc,CAAC,CACrE;CACF;CAEA,WAAW,WAAmB;EAC5B,IACE,KAAK,OAAO,QAAQ,SAAS,UAC7B,KAAK,OAAO,OAAO,aAAa,KAAK,WAErC,KAAK,OAAO,SAAS;GACnB,MAAM;GACN,UAAU,KAAK;GACf,YAAY,KAAK,YAAY;EAC/B;EAEF,KAAK,OAAO,OAAO,WAAW,OAAO,SAAS;CAChD;CAEA,gBAAgB,WAAmB;EACjC,IACE,KAAK,OAAO,QAAQ,SAAS,eAC7B,KAAK,OAAO,OAAO,aAAa,KAAK,WAErC,KAAK,OAAO,SAAS;GACnB,MAAM;GACN,UAAU,KAAK;GACf,YAAY,KAAK,iBAAiB;EACpC;EAEF,KAAK,OAAO,OAAO,WAAW,OAAO,SAAS;CAChD;CAEA,cAAc;EACZ,MAAM,CAAC,QAAQ,cAAc,2BAA2B;EACxD,KAAK,SAAS,KAAK,oBAAoB,EAAE,MAAM,OAAO,CAAC,GAAG,MAAM;EAChE,OAAO;CACT;CAEA,mBAAmB;EACjB,MAAM,CAAC,QAAQ,cAAc,2BAA2B;EACxD,KAAK,SAAS,KAAK,oBAAoB,EAAE,MAAM,YAAY,CAAC,GAAG,MAAM;EACrE,OAAO;CACT;CAEA,gBACE,SAC0B;EAC1B,MAAM,MAAM,OAAO,YAAY,WAAW,EAAE,UAAU,QAAQ,IAAI;EAClE,MAAM,WAAW,IAAI;EACrB,MAAM,aAAa,IAAI,cAAc,WAAW;EAEhD,MAAM,CAAC,QAAQ,cAAc,+BAA+B;EAC5D,KAAK,SACH;GACE,MAAM;GACN;GACA;GACA,GAAI,KAAK,aAAa,EAAE,UAAU,KAAK,UAAU;EACnD,GACA,MACF;EAEA,IAAI,IAAI,aAAa,KAAA,GAAW;GAC9B,WAAW,SAAS,OAAO,IAAI,QAAQ;GACvC,WAAW,SAAS,MAAM;EAC5B;EACA,IAAI,IAAI,SAAS,KAAA,GAAW;GAC1B,WAAW,SAAS,OAAO,KAAK,UAAU,IAAI,IAAI,CAAC;GACnD,WAAW,SAAS,MAAM;EAC5B;EACA,IAAI,IAAI,aAAa,KAAA,GACnB,WAAW,YAAY,IAAI,QAAQ;EAGrC,OAAO;CACT;CAEA,sBAA+C;EAC7C,OAAO,IAAI,eAAe,EACxB,MAAM,YAAY;GAChB,WAAW,QAAQ;IAAE,MAAM;IAAe,MAAM,CAAC;GAAE,CAAC;GACpD,WAAW,MAAM;EACnB,EACF,CAAC;CACH;CAEA,oBAA+B,SAAe;EAC5C,IAAI,CAAC,KAAK,WAAW,OAAO;EAC5B,OAAO;GAAE,GAAG;GAAS,UAAU,KAAK;EAAU;CAChD;CAEA,aAAa,SAAqB;EAChC,KAAK,SACH,KAAK,oBAAoB,OAAO,GAChC,KAAK,oBAAoB,CAC3B;CACF;CAEA,WAAW,SAAmB;EAC5B,KAAK,SACH,KAAK,oBAAoB,OAAO,GAChC,KAAK,oBAAoB,CAC3B;CACF;CAEA,WAAW,SAAmB;EAC5B,KAAK,SACH,KAAK,oBAAoB,OAAO,GAChC,KAAK,oBAAoB,CAC3B;CACF;CAEA,QAAQ,OAA6B;EACnC,KAAK,OAAO,OAAO,QAAQ,KAAK;EAEhC,IAAI,MAAM,SAAS,gBAAgB,MAAM,KAAK,WAAW,GACvD,KAAK,OAAO,eAAe,GAAG;CAElC;CAEA,aAAa,UAA6C;EACxD,MAAM,aAAa,IAAI,8BAA8B,KAAK,MAAM;EAChE,WAAW,YAAY;EACvB,OAAO;CACT;CAEA,QAAQ;EACN,KAAK,OAAO,QAAQ,YAAY,MAAM;EACtC,KAAK,OAAO,OAAO,KAAK;EAExB,KAAK,OAAO,kBAAkB;CAChC;AACF;;;;;;;;;;AAWA,SAAgB,sBACd,UACiB;CACjB,MAAM,aAAa,IAAI,8BAA8B;CAErD,MAAM,UAAU,YAAY;EAC1B,IAAI;GACF,MAAM,SAAS,UAAU;EAC3B,SAAS,GAAG;GACV,IAAI,CAAC,WAAW,qBACd,WAAW,QAAQ;IACjB,MAAM;IACN,MAAM,CAAC;IACP,OAAO,OAAO,CAAC;GACjB,CAAC;GAEH,MAAM;EACR,UAAU;GACR,IAAI,CAAC,WAAW,qBACd,WAAW,MAAM;EAErB;CACF;CACA,QAAQ;CAER,OAAO,WAAW,uBAAuB;AAC3C;;;;;;;;AASA,SAAgB,kCAAkC;CAChD,MAAM,EAAE,SAAS,YAAY,qBAA2B;CACxD,IAAI;CAUJ,OAAO,CATQ,uBAAuB,MAAM;EAC1C,aAAa;EAEb,WAA8C,4BAC5C,OACF;EAEA,OAAO;CACT,CACa,GAAG,UAAU;AAC5B;;;;;;;;AASA,SAAgB,8BACd,UACA;CACA,OAAO,gBAAgB,WACrB,sBAAsB,QAAQ,GAC9B,IAAI,kBAAkB,CACxB;AACF"}
+{"version":3,"file":"assistant-stream.js","names":[],"sources":["../../../src/core/modules/assistant-stream.ts"],"sourcesContent":["import { AssistantStream } from \"../AssistantStream\";\nimport type { AssistantStreamChunk, PartInit } from \"../AssistantStreamChunk\";\nimport { createMergeStream } from \"../utils/stream/merge\";\nimport { createTextStreamController, type TextStreamController } from \"./text\";\nimport {\n  createToolCallStreamController,\n  type ToolCallStreamController,\n} from \"./tool-call\";\nimport { Counter } from \"../utils/Counter\";\nimport {\n  PathAppendEncoder,\n  PathMergeEncoder,\n} from \"../utils/stream/path-utils\";\nimport { DataStreamEncoder } from \"../serialization/data-stream/DataStream\";\nimport type { DataPart, FilePart, SourcePart } from \"../utils/types\";\nimport { generateId } from \"../utils/generateId\";\nimport type {\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"../../utils/json/json-value\";\nimport type { ToolResponseLike } from \"../tool/ToolResponse\";\nimport { promiseWithResolvers } from \"../../utils/promiseWithResolvers\";\n\ntype ToolCallPartInit = {\n  toolCallId?: string;\n  toolName: string;\n  argsText?: string;\n  args?: ReadonlyJSONObject;\n  response?: ToolResponseLike<ReadonlyJSONValue>;\n};\n\n/**\n * Imperative writer for constructing an {@link AssistantStream}.\n *\n * The controller handles part boundaries for common streaming operations. Use\n * `appendText` and `appendReasoning` for simple token streams, or open explicit\n * parts with `addTextPart` and `addToolCallPart` when you need direct control.\n */\nexport type AssistantStreamController = {\n  /** Appends text to the current text part, opening one if needed. */\n  appendText(textDelta: string): void;\n  /** Appends reasoning text to the current reasoning part, opening one if needed. */\n  appendReasoning(reasoningDelta: string): void;\n  /** Appends a source citation part to the stream. */\n  appendSource(options: SourcePart): void;\n  /** Appends a file part to the stream. */\n  appendFile(options: FilePart): void;\n  /** Appends a named data part to the stream. */\n  appendData(options: DataPart): void;\n  /**\n   * Opens a text part and returns its writer.\n   *\n   * Close the returned controller when the text part is complete. Opening a new\n   * part through this controller closes any implicit text or reasoning append\n   * part first.\n   */\n  addTextPart(): TextStreamController;\n  /**\n   * Opens a tool-call part by tool name and returns its writer.\n   *\n   * A tool call id is generated automatically. Use the object overload when the\n   * caller already has an id, initial args, args text, or response.\n   */\n  addToolCallPart(options: string): ToolCallStreamController;\n  /**\n   * Opens a tool-call part and returns its writer.\n   *\n   * Use this overload to provide a stable `toolCallId`, initial arguments,\n   * streamed argument text, or an immediate {@link ToolResponseLike}.\n   */\n  addToolCallPart(options: ToolCallPartInit): ToolCallStreamController;\n  /** Enqueues a raw protocol chunk. Prefer higher-level helpers when possible. */\n  enqueue(chunk: AssistantStreamChunk): void;\n  /**\n   * Merges another assistant stream into this stream.\n   *\n   * Paths from the merged stream are remapped so its parts appear at the next\n   * available position in this controller's output.\n   */\n  merge(stream: AssistantStream): void;\n  /** Closes any active part and finishes the stream. */\n  close(): void;\n  /**\n   * Returns a controller that writes child parts with `parentId` attached.\n   *\n   * Use this for nested or related parts that should be associated with an\n   * existing message or part in downstream renderers.\n   */\n  withParentId(parentId: string): AssistantStreamController;\n};\n\n// Shared state between controller instances\ntype AssistantStreamControllerState = {\n  merger: ReturnType<typeof createMergeStream>;\n  append?:\n    | {\n        controller: TextStreamController;\n        kind: \"text\" | \"reasoning\";\n        parentId: string | undefined;\n      }\n    | undefined;\n  contentCounter: Counter;\n  closeSubscriber?: () => void;\n};\n\nclass AssistantStreamControllerImpl implements AssistantStreamController {\n  private readonly _state: AssistantStreamControllerState;\n  private _parentId?: string;\n\n  constructor(state?: AssistantStreamControllerState) {\n    this._state = state || {\n      merger: createMergeStream(),\n      contentCounter: new Counter(),\n    };\n  }\n\n  get __internal_isClosed() {\n    return (\n      this._state.merger.isSealed() ||\n      this._state.merger.isCancelled() ||\n      this._state.merger.isErrored()\n    );\n  }\n\n  get __internal_isCancelled() {\n    return this._state.merger.isCancelled();\n  }\n\n  __internal_getReadable() {\n    return this._state.merger.readable;\n  }\n\n  __internal_subscribeToClose(callback: () => void) {\n    this._state.closeSubscriber = callback;\n  }\n\n  private _addPart(part: PartInit, stream: AssistantStream) {\n    if (this._state.append) {\n      this._state.append.controller.close();\n      this._state.append = undefined;\n    }\n\n    this.enqueue({\n      type: \"part-start\",\n      part,\n      path: [],\n    });\n    this._state.merger.addStream(\n      stream.pipeThrough(\n        new PathAppendEncoder(this._state.contentCounter.value),\n      ),\n    );\n  }\n\n  merge(stream: AssistantStream) {\n    this._state.merger.addStream(\n      stream.pipeThrough(new PathMergeEncoder(this._state.contentCounter)),\n    );\n  }\n\n  appendText(textDelta: string) {\n    if (\n      this._state.append?.kind !== \"text\" ||\n      this._state.append.parentId !== this._parentId\n    ) {\n      this._state.append = {\n        kind: \"text\",\n        parentId: this._parentId,\n        controller: this.addTextPart(),\n      };\n    }\n    this._state.append.controller.append(textDelta);\n  }\n\n  appendReasoning(textDelta: string) {\n    if (\n      this._state.append?.kind !== \"reasoning\" ||\n      this._state.append.parentId !== this._parentId\n    ) {\n      this._state.append = {\n        kind: \"reasoning\",\n        parentId: this._parentId,\n        controller: this.addReasoningPart(),\n      };\n    }\n    this._state.append.controller.append(textDelta);\n  }\n\n  addTextPart() {\n    const [stream, controller] = createTextStreamController();\n    this._addPart(this._withParentIdOption({ type: \"text\" }), stream);\n    return controller;\n  }\n\n  addReasoningPart() {\n    const [stream, controller] = createTextStreamController();\n    this._addPart(this._withParentIdOption({ type: \"reasoning\" }), stream);\n    return controller;\n  }\n\n  addToolCallPart(\n    options: string | ToolCallPartInit,\n  ): ToolCallStreamController {\n    const opt = typeof options === \"string\" ? { toolName: options } : options;\n    const toolName = opt.toolName;\n    const toolCallId = opt.toolCallId ?? generateId();\n\n    const [stream, controller] = createToolCallStreamController();\n    this._addPart(\n      {\n        type: \"tool-call\",\n        toolName,\n        toolCallId,\n        ...(this._parentId && { parentId: this._parentId }),\n      },\n      stream,\n    );\n\n    if (opt.argsText !== undefined) {\n      controller.argsText.append(opt.argsText);\n      controller.argsText.close();\n    }\n    if (opt.args !== undefined) {\n      controller.argsText.append(JSON.stringify(opt.args));\n      controller.argsText.close();\n    }\n    if (opt.response !== undefined) {\n      controller.setResponse(opt.response);\n    }\n\n    return controller;\n  }\n\n  private _finishedPartStream(): AssistantStream {\n    return new ReadableStream({\n      start(controller) {\n        controller.enqueue({ type: \"part-finish\", path: [] });\n        controller.close();\n      },\n    });\n  }\n\n  private _withParentIdOption<T>(options: T): T {\n    if (!this._parentId) return options;\n    return { ...options, parentId: this._parentId };\n  }\n\n  appendSource(options: SourcePart) {\n    this._addPart(\n      this._withParentIdOption(options),\n      this._finishedPartStream(),\n    );\n  }\n\n  appendFile(options: FilePart) {\n    this._addPart(\n      this._withParentIdOption(options),\n      this._finishedPartStream(),\n    );\n  }\n\n  appendData(options: DataPart) {\n    this._addPart(\n      this._withParentIdOption(options),\n      this._finishedPartStream(),\n    );\n  }\n\n  enqueue(chunk: AssistantStreamChunk) {\n    this._state.merger.enqueue(chunk);\n\n    if (chunk.type === \"part-start\" && chunk.path.length === 0) {\n      this._state.contentCounter.up();\n    }\n  }\n\n  withParentId(parentId: string): AssistantStreamController {\n    const controller = new AssistantStreamControllerImpl(this._state);\n    controller._parentId = parentId;\n    return controller;\n  }\n\n  close() {\n    this._state.append?.controller?.close();\n    this._state.merger.seal();\n\n    this._state.closeSubscriber?.();\n  }\n}\n\n/**\n * Creates an {@link AssistantStream} and writes to it with an\n * {@link AssistantStreamController}.\n *\n * The callback may write synchronously or asynchronously. If it throws while\n * the stream is open, an `error` chunk is emitted. Failures after an explicit\n * close are logged, while failures after consumer cancellation are discarded.\n * When the callback settles, the stream is closed automatically unless the\n * controller was already closed.\n */\nexport function createAssistantStream(\n  callback: (controller: AssistantStreamController) => PromiseLike<void> | void,\n): AssistantStream {\n  const controller = new AssistantStreamControllerImpl();\n\n  const runTask = async () => {\n    try {\n      await callback(controller);\n    } catch (e) {\n      if (!controller.__internal_isClosed) {\n        controller.enqueue({\n          type: \"error\",\n          path: [],\n          error: String(e),\n        });\n      } else if (!controller.__internal_isCancelled) {\n        console.error(e);\n      }\n    } finally {\n      if (!controller.__internal_isClosed) {\n        controller.close();\n      }\n    }\n  };\n  runTask();\n\n  return controller.__internal_getReadable();\n}\n\n/**\n * Creates an {@link AssistantStream} together with the controller used to\n * write into it.\n *\n * Use this when the stream needs to be returned before all writers are known.\n * Closing the returned controller finishes the paired stream.\n */\nexport function createAssistantStreamController() {\n  const { resolve, promise } = promiseWithResolvers<void>();\n  let controller!: AssistantStreamController;\n  const stream = createAssistantStream((c) => {\n    controller = c;\n\n    (controller as AssistantStreamControllerImpl).__internal_subscribeToClose(\n      resolve,\n    );\n\n    return promise;\n  });\n  return [stream, controller] as const;\n}\n\n/**\n * Creates a `Response` whose body is an encoded {@link AssistantStream}.\n *\n * This is the HTTP-route convenience form of {@link createAssistantStream}; it\n * uses {@link DataStreamEncoder} so the response can be consumed by matching\n * assistant-ui data stream decoders.\n */\nexport function createAssistantStreamResponse(\n  callback: (controller: AssistantStreamController) => PromiseLike<void> | void,\n) {\n  return AssistantStream.toResponse(\n    createAssistantStream(callback),\n    new DataStreamEncoder(),\n  );\n}\n"],"mappings":";;;;;;;;;;AAyGA,IAAM,gCAAN,MAAM,8BAAmE;CACvE;CACA;CAEA,YAAY,OAAwC;EAClD,KAAK,SAAS,SAAS;GACrB,QAAQ,kBAAkB;GAC1B,gBAAgB,IAAI,QAAQ;EAC9B;CACF;CAEA,IAAI,sBAAsB;EACxB,OACE,KAAK,OAAO,OAAO,SAAS,KAC5B,KAAK,OAAO,OAAO,YAAY,KAC/B,KAAK,OAAO,OAAO,UAAU;CAEjC;CAEA,IAAI,yBAAyB;EAC3B,OAAO,KAAK,OAAO,OAAO,YAAY;CACxC;CAEA,yBAAyB;EACvB,OAAO,KAAK,OAAO,OAAO;CAC5B;CAEA,4BAA4B,UAAsB;EAChD,KAAK,OAAO,kBAAkB;CAChC;CAEA,SAAiB,MAAgB,QAAyB;EACxD,IAAI,KAAK,OAAO,QAAQ;GACtB,KAAK,OAAO,OAAO,WAAW,MAAM;GACpC,KAAK,OAAO,SAAS,KAAA;EACvB;EAEA,KAAK,QAAQ;GACX,MAAM;GACN;GACA,MAAM,CAAC;EACT,CAAC;EACD,KAAK,OAAO,OAAO,UACjB,OAAO,YACL,IAAI,kBAAkB,KAAK,OAAO,eAAe,KAAK,CACxD,CACF;CACF;CAEA,MAAM,QAAyB;EAC7B,KAAK,OAAO,OAAO,UACjB,OAAO,YAAY,IAAI,iBAAiB,KAAK,OAAO,cAAc,CAAC,CACrE;CACF;CAEA,WAAW,WAAmB;EAC5B,IACE,KAAK,OAAO,QAAQ,SAAS,UAC7B,KAAK,OAAO,OAAO,aAAa,KAAK,WAErC,KAAK,OAAO,SAAS;GACnB,MAAM;GACN,UAAU,KAAK;GACf,YAAY,KAAK,YAAY;EAC/B;EAEF,KAAK,OAAO,OAAO,WAAW,OAAO,SAAS;CAChD;CAEA,gBAAgB,WAAmB;EACjC,IACE,KAAK,OAAO,QAAQ,SAAS,eAC7B,KAAK,OAAO,OAAO,aAAa,KAAK,WAErC,KAAK,OAAO,SAAS;GACnB,MAAM;GACN,UAAU,KAAK;GACf,YAAY,KAAK,iBAAiB;EACpC;EAEF,KAAK,OAAO,OAAO,WAAW,OAAO,SAAS;CAChD;CAEA,cAAc;EACZ,MAAM,CAAC,QAAQ,cAAc,2BAA2B;EACxD,KAAK,SAAS,KAAK,oBAAoB,EAAE,MAAM,OAAO,CAAC,GAAG,MAAM;EAChE,OAAO;CACT;CAEA,mBAAmB;EACjB,MAAM,CAAC,QAAQ,cAAc,2BAA2B;EACxD,KAAK,SAAS,KAAK,oBAAoB,EAAE,MAAM,YAAY,CAAC,GAAG,MAAM;EACrE,OAAO;CACT;CAEA,gBACE,SAC0B;EAC1B,MAAM,MAAM,OAAO,YAAY,WAAW,EAAE,UAAU,QAAQ,IAAI;EAClE,MAAM,WAAW,IAAI;EACrB,MAAM,aAAa,IAAI,cAAc,WAAW;EAEhD,MAAM,CAAC,QAAQ,cAAc,+BAA+B;EAC5D,KAAK,SACH;GACE,MAAM;GACN;GACA;GACA,GAAI,KAAK,aAAa,EAAE,UAAU,KAAK,UAAU;EACnD,GACA,MACF;EAEA,IAAI,IAAI,aAAa,KAAA,GAAW;GAC9B,WAAW,SAAS,OAAO,IAAI,QAAQ;GACvC,WAAW,SAAS,MAAM;EAC5B;EACA,IAAI,IAAI,SAAS,KAAA,GAAW;GAC1B,WAAW,SAAS,OAAO,KAAK,UAAU,IAAI,IAAI,CAAC;GACnD,WAAW,SAAS,MAAM;EAC5B;EACA,IAAI,IAAI,aAAa,KAAA,GACnB,WAAW,YAAY,IAAI,QAAQ;EAGrC,OAAO;CACT;CAEA,sBAA+C;EAC7C,OAAO,IAAI,eAAe,EACxB,MAAM,YAAY;GAChB,WAAW,QAAQ;IAAE,MAAM;IAAe,MAAM,CAAC;GAAE,CAAC;GACpD,WAAW,MAAM;EACnB,EACF,CAAC;CACH;CAEA,oBAA+B,SAAe;EAC5C,IAAI,CAAC,KAAK,WAAW,OAAO;EAC5B,OAAO;GAAE,GAAG;GAAS,UAAU,KAAK;EAAU;CAChD;CAEA,aAAa,SAAqB;EAChC,KAAK,SACH,KAAK,oBAAoB,OAAO,GAChC,KAAK,oBAAoB,CAC3B;CACF;CAEA,WAAW,SAAmB;EAC5B,KAAK,SACH,KAAK,oBAAoB,OAAO,GAChC,KAAK,oBAAoB,CAC3B;CACF;CAEA,WAAW,SAAmB;EAC5B,KAAK,SACH,KAAK,oBAAoB,OAAO,GAChC,KAAK,oBAAoB,CAC3B;CACF;CAEA,QAAQ,OAA6B;EACnC,KAAK,OAAO,OAAO,QAAQ,KAAK;EAEhC,IAAI,MAAM,SAAS,gBAAgB,MAAM,KAAK,WAAW,GACvD,KAAK,OAAO,eAAe,GAAG;CAElC;CAEA,aAAa,UAA6C;EACxD,MAAM,aAAa,IAAI,8BAA8B,KAAK,MAAM;EAChE,WAAW,YAAY;EACvB,OAAO;CACT;CAEA,QAAQ;EACN,KAAK,OAAO,QAAQ,YAAY,MAAM;EACtC,KAAK,OAAO,OAAO,KAAK;EAExB,KAAK,OAAO,kBAAkB;CAChC;AACF;;;;;;;;;;;AAYA,SAAgB,sBACd,UACiB;CACjB,MAAM,aAAa,IAAI,8BAA8B;CAErD,MAAM,UAAU,YAAY;EAC1B,IAAI;GACF,MAAM,SAAS,UAAU;EAC3B,SAAS,GAAG;GACV,IAAI,CAAC,WAAW,qBACd,WAAW,QAAQ;IACjB,MAAM;IACN,MAAM,CAAC;IACP,OAAO,OAAO,CAAC;GACjB,CAAC;QACI,IAAI,CAAC,WAAW,wBACrB,QAAQ,MAAM,CAAC;EAEnB,UAAU;GACR,IAAI,CAAC,WAAW,qBACd,WAAW,MAAM;EAErB;CACF;CACA,QAAQ;CAER,OAAO,WAAW,uBAAuB;AAC3C;;;;;;;;AASA,SAAgB,kCAAkC;CAChD,MAAM,EAAE,SAAS,YAAY,qBAA2B;CACxD,IAAI;CAUJ,OAAO,CATQ,uBAAuB,MAAM;EAC1C,aAAa;EAEb,WAA8C,4BAC5C,OACF;EAEA,OAAO;CACT,CACa,GAAG,UAAU;AAC5B;;;;;;;;AASA,SAAgB,8BACd,UACA;CACA,OAAO,gBAAgB,WACrB,sBAAsB,QAAQ,GAC9B,IAAI,kBAAkB,CACxB;AACF"}

package/dist/core/modules/tool-call.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool-call.d.ts","names":[],"sources":["../../../src/core/modules/tool-call.ts"],"mappings":";;;;;;KAOY;EACV,UAAU;EAEV,YAAY,UAAU,iBAAiB;EACvC;;cA2FW,uBACX,UAAU,mBAAmB,8BAC5B;cAcU,gDAA8B,iBAAA"}
+{"version":3,"file":"tool-call.d.ts","names":[],"sources":["../../../src/core/modules/tool-call.ts"],"mappings":";;;;;;KAOY;EACV,UAAU;EAEV,YAAY,UAAU,iBAAiB;EACvC;;cA8FW,uBACX,UAAU,mBAAmB,8BAC5B;cAcU,gDAA8B,iBAAA"}

package/dist/core/modules/tool-call.js

@@ -1,9 +1,9 @@
 import { createTextStream } from "./text.js";
 //#region src/core/modules/tool-call.ts
 var ToolCallStreamControllerImpl = class {
-	_controller;
 	_isClosed = false;
 	_mergeTask;
+	_controller;
 	constructor(_controller) {
 		this._controller = _controller;
 		const stream = createTextStream({ start: (c) => {
@@ -36,8 +36,6 @@ var ToolCallStreamControllerImpl = class {
 	}
 	_argsTextController;
 	async setResponse(response) {
-		this._argsTextController.close();
-		await Promise.resolve();
 		this._controller.enqueue({
 			type: "result",
 			path: [],
@@ -47,6 +45,8 @@ var ToolCallStreamControllerImpl = class {
 			...response.modelContent !== void 0 ? { modelContent: response.modelContent } : {},
 			...response.messages !== void 0 ? { messages: response.messages } : {}
 		});
+		this._argsTextController.close();
+		await Promise.resolve();
 	}
 	async close() {
 		if (this._isClosed) return;

package/dist/core/modules/tool-call.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-call.js","names":[],"sources":["../../../src/core/modules/tool-call.ts"],"sourcesContent":["import type { AssistantStream } from \"../AssistantStream\";\nimport type { AssistantStreamChunk } from \"../AssistantStreamChunk\";\nimport type { ToolResponseLike } from \"../tool/ToolResponse\";\nimport type { ReadonlyJSONValue } from \"../../utils/json/json-value\";\nimport type { UnderlyingReadable } from \"../utils/stream/UnderlyingReadable\";\nimport { createTextStream, type TextStreamController } from \"./text\";\n\nexport type ToolCallStreamController = {\n  argsText: TextStreamController;\n\n  setResponse(response: ToolResponseLike<ReadonlyJSONValue>): void;\n  close(): void;\n};\n\nclass ToolCallStreamControllerImpl implements ToolCallStreamController {\n  private _isClosed = false;\n\n  private _mergeTask: Promise<void>;\n  constructor(\n    private _controller: ReadableStreamDefaultController<AssistantStreamChunk>,\n  ) {\n    const stream = createTextStream({\n      start: (c) => {\n        this._argsTextController = c;\n      },\n    });\n\n    let hasArgsText = false;\n    this._mergeTask = stream.pipeTo(\n      new WritableStream({\n        write: (chunk) => {\n          switch (chunk.type) {\n            case \"text-delta\":\n              hasArgsText = true;\n              this._controller.enqueue(chunk);\n              break;\n\n            case \"part-finish\":\n              if (!hasArgsText) {\n                // if no argsText was provided, assume empty object\n                this._controller.enqueue({\n                  type: \"text-delta\",\n                  textDelta: \"{}\",\n                  path: [],\n                });\n              }\n              this._controller.enqueue({\n                type: \"tool-call-args-text-finish\",\n                path: [],\n              });\n              break;\n\n            default:\n              throw new Error(`Unexpected chunk type: ${chunk.type}`);\n          }\n        },\n      }),\n    );\n  }\n\n  get argsText() {\n    return this._argsTextController;\n  }\n\n  private _argsTextController!: TextStreamController;\n\n  async setResponse(response: ToolResponseLike<ReadonlyJSONValue>) {\n    this._argsTextController.close();\n    await Promise.resolve(); // flush microtask queue\n    // TODO switch argsTextController to be something that doesn'#t require this\n    this._controller.enqueue({\n      type: \"result\",\n      path: [],\n      ...(response.artifact !== undefined\n        ? { artifact: response.artifact }\n        : {}),\n      result: response.result,\n      isError: response.isError ?? false,\n      ...(response.modelContent !== undefined\n        ? { modelContent: response.modelContent }\n        : {}),\n      ...(response.messages !== undefined\n        ? { messages: response.messages }\n        : {}),\n    });\n  }\n\n  async close() {\n    if (this._isClosed) return;\n\n    this._isClosed = true;\n    this._argsTextController.close();\n    await this._mergeTask;\n\n    this._controller.enqueue({\n      type: \"part-finish\",\n      path: [],\n    });\n    this._controller.close();\n  }\n}\n\nexport const createToolCallStream = (\n  readable: UnderlyingReadable<ToolCallStreamController>,\n): AssistantStream => {\n  return new ReadableStream({\n    start(c) {\n      return readable.start?.(new ToolCallStreamControllerImpl(c));\n    },\n    pull(c) {\n      return readable.pull?.(new ToolCallStreamControllerImpl(c));\n    },\n    cancel(c) {\n      return readable.cancel?.(c);\n    },\n  });\n};\n\nexport const createToolCallStreamController = () => {\n  let controller!: ToolCallStreamController;\n  const stream = createToolCallStream({\n    start(c) {\n      controller = c;\n    },\n  });\n  return [stream, controller] as const;\n};\n"],"mappings":";;AAcA,IAAM,+BAAN,MAAuE;CAK3D;CAJV,YAAoB;CAEpB;CACA,YACE,aACA;EADQ,KAAA,cAAA;EAER,MAAM,SAAS,iBAAiB,EAC9B,QAAQ,MAAM;GACZ,KAAK,sBAAsB;EAC7B,EACF,CAAC;EAED,IAAI,cAAc;EAClB,KAAK,aAAa,OAAO,OACvB,IAAI,eAAe,EACjB,QAAQ,UAAU;GAChB,QAAQ,MAAM,MAAd;IACE,KAAK;KACH,cAAc;KACd,KAAK,YAAY,QAAQ,KAAK;KAC9B;IAEF,KAAK;KACH,IAAI,CAAC,aAEH,KAAK,YAAY,QAAQ;MACvB,MAAM;MACN,WAAW;MACX,MAAM,CAAC;KACT,CAAC;KAEH,KAAK,YAAY,QAAQ;MACvB,MAAM;MACN,MAAM,CAAC;KACT,CAAC;KACD;IAEF,SACE,MAAM,IAAI,MAAM,0BAA0B,MAAM,MAAM;GAC1D;EACF,EACF,CAAC,CACH;CACF;CAEA,IAAI,WAAW;EACb,OAAO,KAAK;CACd;CAEA;CAEA,MAAM,YAAY,UAA+C;EAC/D,KAAK,oBAAoB,MAAM;EAC/B,MAAM,QAAQ,QAAQ;EAEtB,KAAK,YAAY,QAAQ;GACvB,MAAM;GACN,MAAM,CAAC;GACP,GAAI,SAAS,aAAa,KAAA,IACtB,EAAE,UAAU,SAAS,SAAS,IAC9B,CAAC;GACL,QAAQ,SAAS;GACjB,SAAS,SAAS,WAAW;GAC7B,GAAI,SAAS,iBAAiB,KAAA,IAC1B,EAAE,cAAc,SAAS,aAAa,IACtC,CAAC;GACL,GAAI,SAAS,aAAa,KAAA,IACtB,EAAE,UAAU,SAAS,SAAS,IAC9B,CAAC;EACP,CAAC;CACH;CAEA,MAAM,QAAQ;EACZ,IAAI,KAAK,WAAW;EAEpB,KAAK,YAAY;EACjB,KAAK,oBAAoB,MAAM;EAC/B,MAAM,KAAK;EAEX,KAAK,YAAY,QAAQ;GACvB,MAAM;GACN,MAAM,CAAC;EACT,CAAC;EACD,KAAK,YAAY,MAAM;CACzB;AACF;AAEA,MAAa,wBACX,aACoB;CACpB,OAAO,IAAI,eAAe;EACxB,MAAM,GAAG;GACP,OAAO,SAAS,QAAQ,IAAI,6BAA6B,CAAC,CAAC;EAC7D;EACA,KAAK,GAAG;GACN,OAAO,SAAS,OAAO,IAAI,6BAA6B,CAAC,CAAC;EAC5D;EACA,OAAO,GAAG;GACR,OAAO,SAAS,SAAS,CAAC;EAC5B;CACF,CAAC;AACH;AAEA,MAAa,uCAAuC;CAClD,IAAI;CAMJ,OAAO,CALQ,qBAAqB,EAClC,MAAM,GAAG;EACP,aAAa;CACf,EACF,CACa,GAAG,UAAU;AAC5B"}
+{"version":3,"file":"tool-call.js","names":[],"sources":["../../../src/core/modules/tool-call.ts"],"sourcesContent":["import type { AssistantStream } from \"../AssistantStream\";\nimport type { AssistantStreamChunk } from \"../AssistantStreamChunk\";\nimport type { ToolResponseLike } from \"../tool/ToolResponse\";\nimport type { ReadonlyJSONValue } from \"../../utils/json/json-value\";\nimport type { UnderlyingReadable } from \"../utils/stream/UnderlyingReadable\";\nimport { createTextStream, type TextStreamController } from \"./text\";\n\nexport type ToolCallStreamController = {\n  argsText: TextStreamController;\n\n  setResponse(response: ToolResponseLike<ReadonlyJSONValue>): void;\n  close(): void;\n};\n\nclass ToolCallStreamControllerImpl implements ToolCallStreamController {\n  private _isClosed = false;\n\n  private _mergeTask: Promise<void>;\n  private _controller: ReadableStreamDefaultController<AssistantStreamChunk>;\n\n  constructor(\n    _controller: ReadableStreamDefaultController<AssistantStreamChunk>,\n  ) {\n    this._controller = _controller;\n    const stream = createTextStream({\n      start: (c) => {\n        this._argsTextController = c;\n      },\n    });\n\n    let hasArgsText = false;\n    this._mergeTask = stream.pipeTo(\n      new WritableStream({\n        write: (chunk) => {\n          switch (chunk.type) {\n            case \"text-delta\":\n              hasArgsText = true;\n              this._controller.enqueue(chunk);\n              break;\n\n            case \"part-finish\":\n              if (!hasArgsText) {\n                // if no argsText was provided, assume empty object\n                this._controller.enqueue({\n                  type: \"text-delta\",\n                  textDelta: \"{}\",\n                  path: [],\n                });\n              }\n              this._controller.enqueue({\n                type: \"tool-call-args-text-finish\",\n                path: [],\n              });\n              break;\n\n            default:\n              throw new Error(`Unexpected chunk type: ${chunk.type}`);\n          }\n        },\n      }),\n    );\n  }\n\n  get argsText() {\n    return this._argsTextController;\n  }\n\n  private _argsTextController!: TextStreamController;\n\n  async setResponse(response: ToolResponseLike<ReadonlyJSONValue>) {\n    this._controller.enqueue({\n      type: \"result\",\n      path: [],\n      ...(response.artifact !== undefined\n        ? { artifact: response.artifact }\n        : {}),\n      result: response.result,\n      isError: response.isError ?? false,\n      ...(response.modelContent !== undefined\n        ? { modelContent: response.modelContent }\n        : {}),\n      ...(response.messages !== undefined\n        ? { messages: response.messages }\n        : {}),\n    });\n    this._argsTextController.close();\n    await Promise.resolve(); // flush microtask queue\n    // TODO switch argsTextController to be something that doesn'#t require this\n  }\n\n  async close() {\n    if (this._isClosed) return;\n\n    this._isClosed = true;\n    this._argsTextController.close();\n    await this._mergeTask;\n\n    this._controller.enqueue({\n      type: \"part-finish\",\n      path: [],\n    });\n    this._controller.close();\n  }\n}\n\nexport const createToolCallStream = (\n  readable: UnderlyingReadable<ToolCallStreamController>,\n): AssistantStream => {\n  return new ReadableStream({\n    start(c) {\n      return readable.start?.(new ToolCallStreamControllerImpl(c));\n    },\n    pull(c) {\n      return readable.pull?.(new ToolCallStreamControllerImpl(c));\n    },\n    cancel(c) {\n      return readable.cancel?.(c);\n    },\n  });\n};\n\nexport const createToolCallStreamController = () => {\n  let controller!: ToolCallStreamController;\n  const stream = createToolCallStream({\n    start(c) {\n      controller = c;\n    },\n  });\n  return [stream, controller] as const;\n};\n"],"mappings":";;AAcA,IAAM,+BAAN,MAAuE;CACrE,YAAoB;CAEpB;CACA;CAEA,YACE,aACA;EACA,KAAK,cAAc;EACnB,MAAM,SAAS,iBAAiB,EAC9B,QAAQ,MAAM;GACZ,KAAK,sBAAsB;EAC7B,EACF,CAAC;EAED,IAAI,cAAc;EAClB,KAAK,aAAa,OAAO,OACvB,IAAI,eAAe,EACjB,QAAQ,UAAU;GAChB,QAAQ,MAAM,MAAd;IACE,KAAK;KACH,cAAc;KACd,KAAK,YAAY,QAAQ,KAAK;KAC9B;IAEF,KAAK;KACH,IAAI,CAAC,aAEH,KAAK,YAAY,QAAQ;MACvB,MAAM;MACN,WAAW;MACX,MAAM,CAAC;KACT,CAAC;KAEH,KAAK,YAAY,QAAQ;MACvB,MAAM;MACN,MAAM,CAAC;KACT,CAAC;KACD;IAEF,SACE,MAAM,IAAI,MAAM,0BAA0B,MAAM,MAAM;GAC1D;EACF,EACF,CAAC,CACH;CACF;CAEA,IAAI,WAAW;EACb,OAAO,KAAK;CACd;CAEA;CAEA,MAAM,YAAY,UAA+C;EAC/D,KAAK,YAAY,QAAQ;GACvB,MAAM;GACN,MAAM,CAAC;GACP,GAAI,SAAS,aAAa,KAAA,IACtB,EAAE,UAAU,SAAS,SAAS,IAC9B,CAAC;GACL,QAAQ,SAAS;GACjB,SAAS,SAAS,WAAW;GAC7B,GAAI,SAAS,iBAAiB,KAAA,IAC1B,EAAE,cAAc,SAAS,aAAa,IACtC,CAAC;GACL,GAAI,SAAS,aAAa,KAAA,IACtB,EAAE,UAAU,SAAS,SAAS,IAC9B,CAAC;EACP,CAAC;EACD,KAAK,oBAAoB,MAAM;EAC/B,MAAM,QAAQ,QAAQ;CAExB;CAEA,MAAM,QAAQ;EACZ,IAAI,KAAK,WAAW;EAEpB,KAAK,YAAY;EACjB,KAAK,oBAAoB,MAAM;EAC/B,MAAM,KAAK;EAEX,KAAK,YAAY,QAAQ;GACvB,MAAM;GACN,MAAM,CAAC;EACT,CAAC;EACD,KAAK,YAAY,MAAM;CACzB;AACF;AAEA,MAAa,wBACX,aACoB;CACpB,OAAO,IAAI,eAAe;EACxB,MAAM,GAAG;GACP,OAAO,SAAS,QAAQ,IAAI,6BAA6B,CAAC,CAAC;EAC7D;EACA,KAAK,GAAG;GACN,OAAO,SAAS,OAAO,IAAI,6BAA6B,CAAC,CAAC;EAC5D;EACA,OAAO,GAAG;GACR,OAAO,SAAS,SAAS,CAAC;EAC5B;CACF,CAAC;AACH;AAEA,MAAa,uCAAuC;CAClD,IAAI;CAMJ,OAAO,CALQ,qBAAqB,EAClC,MAAM,GAAG;EACP,aAAa;CACf,EACF,CACa,GAAG,UAAU;AAC5B"}

package/dist/core/serialization/assistant-transport/AssistantTransport.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantTransport.d.ts","names":[],"sources":["../../../../src/core/serialization/assistant-transport/AssistantTransport.ts"],"mappings":";;;;;;;;cAYa,kCACH,wBAAwB,sBAAsB,WAAW,yBACtD;EAEX,SAAO;;;;;;;cA4BI,kCAAkC,wBAC7C,WAAW,cACX"}
+{"version":3,"file":"AssistantTransport.d.ts","names":[],"sources":["../../../../src/core/serialization/assistant-transport/AssistantTransport.ts"],"mappings":";;;;;;;;cA6Ea,kCACH,wBAAwB,sBAAsB,WAAW,yBACtD;EAEX,SAAO;;;;;;;cA4BI,kCAAkC,wBAC7C,WAAW,cACX"}

package/dist/core/serialization/assistant-transport/AssistantTransport.js

@@ -1,6 +1,87 @@
 import { PipeableTransformStream } from "../../utils/stream/PipeableTransformStream.js";
 import { SSEEventDecoderStream } from "../../utils/stream/SSEEventDecoderStream.js";
+import sjson from "secure-json-parse";
 //#region src/core/serialization/assistant-transport/AssistantTransport.ts
+const noFields = () => true;
+const requiredObject = (key) => (c) => {
+	const value = c[key];
+	return typeof value === "object" && value !== null && !Array.isArray(value);
+};
+const requiredString = (key) => (c) => typeof c[key] === "string";
+const requiredArray = (key) => (c) => Array.isArray(c[key]);
+const optionalBoolean = (key) => (c) => c[key] === void 0 || typeof c[key] === "boolean";
+const KNOWN_CHUNK_TYPES = {
+	"part-start": {
+		kind: "message",
+		valid: requiredObject("part")
+	},
+	"part-finish": {
+		kind: "part-addressed",
+		valid: noFields
+	},
+	"tool-call-args-text-finish": {
+		kind: "part-addressed",
+		valid: noFields
+	},
+	"text-delta": {
+		kind: "part-addressed",
+		valid: requiredString("textDelta")
+	},
+	annotations: {
+		kind: "message",
+		valid: requiredArray("annotations")
+	},
+	data: {
+		kind: "message",
+		valid: requiredArray("data")
+	},
+	"step-start": {
+		kind: "message",
+		valid: noFields
+	},
+	"step-finish": {
+		kind: "message",
+		valid: requiredString("finishReason")
+	},
+	"message-finish": {
+		kind: "message",
+		valid: requiredString("finishReason")
+	},
+	result: {
+		kind: "part-addressed",
+		valid: optionalBoolean("isError")
+	},
+	error: {
+		kind: "message",
+		valid: noFields
+	},
+	"update-state": {
+		kind: "message",
+		valid: requiredArray("operations")
+	}
+};
+const parseChunk = (data) => {
+	let value;
+	try {
+		value = sjson.parse(data);
+	} catch {
+		return "unparseable";
+	}
+	if (typeof value !== "object" || value === null || Array.isArray(value)) return "not-an-object";
+	const { type, path } = value;
+	if (typeof type !== "string" || !Object.prototype.hasOwnProperty.call(KNOWN_CHUNK_TYPES, type)) return "unknown-type";
+	const rule = KNOWN_CHUNK_TYPES[type];
+	if (!rule.valid(value)) return `invalid-fields:${type}`;
+	if (path === void 0) {
+		if (rule.kind !== "message") return `missing-path:${type}`;
+		return {
+			...value,
+			path: []
+		};
+	}
+	if (!Array.isArray(path) || !path.every((entry) => Number.isInteger(entry) && entry >= 0)) return "invalid-path";
+	return value;
+};
 /**
 * AssistantTransportEncoder encodes AssistantStreamChunks into SSE format
 * and emits [DONE] when the stream completes.
@@ -32,6 +113,7 @@ var AssistantTransportDecoder = class extends PipeableTransformStream {
 	constructor() {
 		super((readable) => {
 			let receivedDone = false;
+			const warnedReasons = /* @__PURE__ */ new Set();
 			return readable.pipeThrough(new TextDecoderStream()).pipeThrough(new SSEEventDecoderStream()).pipeThrough(new TransformStream({
 				transform(event, controller) {
 					switch (event.event) {
@@ -39,7 +121,15 @@ var AssistantTransportDecoder = class extends PipeableTransformStream {
 							if (event.data === "[DONE]") {
 								receivedDone = true;
 								controller.terminate();
-							} else controller.enqueue(JSON.parse(event.data));
+							} else {
+								const chunk = parseChunk(event.data);
+								if (typeof chunk === "string") {
+									if (!warnedReasons.has(chunk)) {
+										warnedReasons.add(chunk);
+										console.warn(`Dropped invalid assistant-transport chunk (${chunk}): ${event.data.slice(0, 200)}`);
+									}
+								} else controller.enqueue(chunk);
+							}
 							break;
 						default: throw new Error(`Unknown SSE event type: ${event.event}`);
 					}