assistant-stream 0.3.15 → 0.3.17

package/dist/core/serialization/ui-message-stream/UIMessageStream.js

@@ -2,234 +2,211 @@ import { AssistantTransformStream } from "../../utils/stream/AssistantTransformS
 import { PipeableTransformStream } from "../../utils/stream/PipeableTransformStream.js";
 import { LineDecoderStream } from "../../utils/stream/LineDecoderStream.js";
 import { generateId } from "../../utils/generateId.js";
-class SSEEventStream extends TransformStream {
-    constructor() {
-        let eventBuffer = {};
-        let dataLines = [];
-        super({
-            start() {
-                eventBuffer = {};
-                dataLines = [];
-            },
-            transform(line, controller) {
-                if (line.startsWith(":"))
-                    return;
-                if (line === "") {
-                    if (dataLines.length > 0) {
-                        controller.enqueue({
-                            event: eventBuffer.event || "message",
-                            data: dataLines.join("\n"),
-                            id: eventBuffer.id,
-                            retry: eventBuffer.retry,
-                        });
-                    }
-                    eventBuffer = {};
-                    dataLines = [];
-                    return;
-                }
-                const [field, ...rest] = line.split(":");
-                const value = rest.join(":").trimStart();
-                switch (field) {
-                    case "event":
-                        eventBuffer.event = value;
-                        break;
-                    case "data":
-                        dataLines.push(value);
-                        break;
-                    case "id":
-                        eventBuffer.id = value;
-                        break;
-                    case "retry":
-                        eventBuffer.retry = Number(value);
-                        break;
-                }
-            },
-            flush(controller) {
-                if (dataLines.length > 0) {
-                    controller.enqueue({
-                        event: eventBuffer.event || "message",
-                        data: dataLines.join("\n"),
-                        id: eventBuffer.id,
-                        retry: eventBuffer.retry,
-                    });
-                }
-            },
-        });
-    }
-}
+//#region src/core/serialization/ui-message-stream/UIMessageStream.ts
+var SSEEventStream = class extends TransformStream {
+	constructor() {
+		let eventBuffer = {};
+		let dataLines = [];
+		super({
+			start() {
+				eventBuffer = {};
+				dataLines = [];
+			},
+			transform(line, controller) {
+				if (line.startsWith(":")) return;
+				if (line === "") {
+					if (dataLines.length > 0) controller.enqueue({
+						event: eventBuffer.event || "message",
+						data: dataLines.join("\n"),
+						id: eventBuffer.id,
+						retry: eventBuffer.retry
+					});
+					eventBuffer = {};
+					dataLines = [];
+					return;
+				}
+				const [field, ...rest] = line.split(":");
+				const value = rest.join(":").trimStart();
+				switch (field) {
+					case "event":
+						eventBuffer.event = value;
+						break;
+					case "data":
+						dataLines.push(value);
+						break;
+					case "id":
+						eventBuffer.id = value;
+						break;
+					case "retry":
+						eventBuffer.retry = Number(value);
+						break;
+				}
+			},
+			flush(controller) {
+				if (dataLines.length > 0) controller.enqueue({
+					event: eventBuffer.event || "message",
+					data: dataLines.join("\n"),
+					id: eventBuffer.id,
+					retry: eventBuffer.retry
+				});
+			}
+		});
+	}
+};
 const isDataChunk = (chunk) => chunk.type.startsWith("data-");
 /**
- * Decodes AI SDK v6 UI Message Stream format into AssistantStreamChunks.
- */
-export class UIMessageStreamDecoder extends PipeableTransformStream {
-    constructor(options = {}) {
-        super((readable) => {
-            const toolCallControllers = new Map();
-            let activeToolCallArgsText;
-            let currentMessageId;
-            let receivedDone = false;
-            const transform = new AssistantTransformStream({
-                transform(chunk, controller) {
-                    const type = chunk.type;
-                    if (isDataChunk(chunk)) {
-                        const name = chunk.type.slice(5);
-                        if (options.onData) {
-                            options.onData({
-                                type: chunk.type,
-                                name,
-                                data: chunk.data,
-                                ...(chunk.transient !== undefined && {
-                                    transient: chunk.transient,
-                                }),
-                            });
-                        }
-                        if (!chunk.transient) {
-                            controller.enqueue({
-                                type: "data",
-                                path: [],
-                                data: [{ name, data: chunk.data }],
-                            });
-                        }
-                        return;
-                    }
-                    switch (type) {
-                        case "start":
-                            currentMessageId = chunk.messageId;
-                            controller.enqueue({
-                                type: "step-start",
-                                path: [],
-                                messageId: chunk.messageId,
-                            });
-                            break;
-                        case "text-start":
-                        case "text-end":
-                        case "reasoning-start":
-                        case "reasoning-end":
-                            break;
-                        case "text-delta":
-                            controller.appendText(chunk.textDelta);
-                            break;
-                        case "reasoning-delta":
-                            controller.appendReasoning(chunk.delta);
-                            break;
-                        case "source":
-                            controller.appendSource({
-                                type: "source",
-                                sourceType: chunk.source.sourceType,
-                                id: chunk.source.id,
-                                url: chunk.source.url,
-                                ...(chunk.source.title && { title: chunk.source.title }),
-                            });
-                            break;
-                        case "file":
-                            controller.appendFile({
-                                type: "file",
-                                mimeType: chunk.file.mimeType,
-                                data: chunk.file.data,
-                            });
-                            break;
-                        case "tool-call-start": {
-                            activeToolCallArgsText?.close();
-                            activeToolCallArgsText = undefined;
-                            if (toolCallControllers.has(chunk.toolCallId)) {
-                                throw new Error(`Encountered duplicate tool call id: ${chunk.toolCallId}`);
-                            }
-                            const toolCallController = controller.addToolCallPart({
-                                toolCallId: chunk.toolCallId,
-                                toolName: chunk.toolName,
-                            });
-                            toolCallControllers.set(chunk.toolCallId, toolCallController);
-                            activeToolCallArgsText = toolCallController.argsText;
-                            break;
-                        }
-                        case "tool-call-delta":
-                            activeToolCallArgsText?.append(chunk.argsText);
-                            break;
-                        case "tool-call-end":
-                            activeToolCallArgsText?.close();
-                            activeToolCallArgsText = undefined;
-                            break;
-                        case "tool-result": {
-                            const toolCallController = toolCallControllers.get(chunk.toolCallId);
-                            if (!toolCallController) {
-                                throw new Error(`Encountered tool result with unknown id: ${chunk.toolCallId}`);
-                            }
-                            toolCallController.setResponse({
-                                result: chunk.result,
-                                isError: chunk.isError ?? false,
-                                ...(chunk.messages !== undefined
-                                    ? { messages: chunk.messages }
-                                    : {}),
-                            });
-                            break;
-                        }
-                        case "start-step":
-                            controller.enqueue({
-                                type: "step-start",
-                                path: [],
-                                messageId: chunk.messageId ?? currentMessageId ?? generateId(),
-                            });
-                            break;
-                        case "finish-step":
-                            controller.enqueue({
-                                type: "step-finish",
-                                path: [],
-                                finishReason: chunk.finishReason,
-                                usage: chunk.usage,
-                                isContinued: chunk.isContinued,
-                            });
-                            break;
-                        case "finish":
-                            controller.enqueue({
-                                type: "message-finish",
-                                path: [],
-                                finishReason: chunk.finishReason,
-                                usage: chunk.usage,
-                            });
-                            break;
-                        case "error":
-                            controller.enqueue({
-                                type: "error",
-                                path: [],
-                                error: chunk.errorText,
-                            });
-                            break;
-                        default:
-                            // ignore unknown types for forward compatibility
-                            break;
-                    }
-                },
-                flush() {
-                    activeToolCallArgsText?.close();
-                    // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return
-                    toolCallControllers.forEach((ctrl) => ctrl.close());
-                    toolCallControllers.clear();
-                },
-            });
-            return readable
-                .pipeThrough(new TextDecoderStream())
-                .pipeThrough(new LineDecoderStream())
-                .pipeThrough(new SSEEventStream())
-                .pipeThrough(new TransformStream({
-                transform(event, controller) {
-                    if (event.event !== "message") {
-                        throw new Error(`Unknown SSE event type: ${event.event}`);
-                    }
-                    if (event.data === "[DONE]") {
-                        receivedDone = true;
-                        controller.terminate();
-                        return;
-                    }
-                    controller.enqueue(JSON.parse(event.data));
-                },
-                flush() {
-                    if (!receivedDone) {
-                        throw new Error("Stream ended abruptly without receiving [DONE] marker");
-                    }
-                },
-            }))
-                .pipeThrough(transform);
-        });
-    }
-}
+* Decodes AI SDK v6 UI Message Stream format into AssistantStreamChunks.
+*/
+var UIMessageStreamDecoder = class extends PipeableTransformStream {
+	constructor(options = {}) {
+		super((readable) => {
+			const toolCallControllers = /* @__PURE__ */ new Map();
+			let activeToolCallArgsText;
+			let currentMessageId;
+			let receivedDone = false;
+			const transform = new AssistantTransformStream({
+				transform(chunk, controller) {
+					const type = chunk.type;
+					if (isDataChunk(chunk)) {
+						const name = chunk.type.slice(5);
+						if (options.onData) options.onData({
+							type: chunk.type,
+							name,
+							data: chunk.data,
+							...chunk.transient !== void 0 && { transient: chunk.transient }
+						});
+						if (!chunk.transient) controller.enqueue({
+							type: "data",
+							path: [],
+							data: [{
+								name,
+								data: chunk.data
+							}]
+						});
+						return;
+					}
+					switch (type) {
+						case "start":
+							currentMessageId = chunk.messageId;
+							controller.enqueue({
+								type: "step-start",
+								path: [],
+								messageId: chunk.messageId
+							});
+							break;
+						case "text-start":
+						case "text-end":
+						case "reasoning-start":
+						case "reasoning-end": break;
+						case "text-delta":
+							controller.appendText(chunk.textDelta);
+							break;
+						case "reasoning-delta":
+							controller.appendReasoning(chunk.delta);
+							break;
+						case "source":
+							controller.appendSource({
+								type: "source",
+								sourceType: chunk.source.sourceType,
+								id: chunk.source.id,
+								url: chunk.source.url,
+								...chunk.source.title && { title: chunk.source.title }
+							});
+							break;
+						case "file":
+							controller.appendFile({
+								type: "file",
+								mimeType: chunk.file.mimeType,
+								data: chunk.file.data
+							});
+							break;
+						case "tool-call-start": {
+							activeToolCallArgsText?.close();
+							activeToolCallArgsText = void 0;
+							if (toolCallControllers.has(chunk.toolCallId)) throw new Error(`Encountered duplicate tool call id: ${chunk.toolCallId}`);
+							const toolCallController = controller.addToolCallPart({
+								toolCallId: chunk.toolCallId,
+								toolName: chunk.toolName
+							});
+							toolCallControllers.set(chunk.toolCallId, toolCallController);
+							activeToolCallArgsText = toolCallController.argsText;
+							break;
+						}
+						case "tool-call-delta":
+							activeToolCallArgsText?.append(chunk.argsText);
+							break;
+						case "tool-call-end":
+							activeToolCallArgsText?.close();
+							activeToolCallArgsText = void 0;
+							break;
+						case "tool-result": {
+							const toolCallController = toolCallControllers.get(chunk.toolCallId);
+							if (!toolCallController) throw new Error(`Encountered tool result with unknown id: ${chunk.toolCallId}`);
+							toolCallController.setResponse({
+								result: chunk.result,
+								isError: chunk.isError ?? false,
+								...chunk.messages !== void 0 ? { messages: chunk.messages } : {}
+							});
+							break;
+						}
+						case "start-step":
+							controller.enqueue({
+								type: "step-start",
+								path: [],
+								messageId: chunk.messageId ?? currentMessageId ?? generateId()
+							});
+							break;
+						case "finish-step":
+							controller.enqueue({
+								type: "step-finish",
+								path: [],
+								finishReason: chunk.finishReason,
+								usage: chunk.usage,
+								isContinued: chunk.isContinued
+							});
+							break;
+						case "finish":
+							controller.enqueue({
+								type: "message-finish",
+								path: [],
+								finishReason: chunk.finishReason,
+								usage: chunk.usage
+							});
+							break;
+						case "error":
+							controller.enqueue({
+								type: "error",
+								path: [],
+								error: chunk.errorText
+							});
+							break;
+						default: break;
+					}
+				},
+				flush() {
+					activeToolCallArgsText?.close();
+					toolCallControllers.forEach((ctrl) => ctrl.close());
+					toolCallControllers.clear();
+				}
+			});
+			return readable.pipeThrough(new TextDecoderStream()).pipeThrough(new LineDecoderStream()).pipeThrough(new SSEEventStream()).pipeThrough(new TransformStream({
+				transform(event, controller) {
+					if (event.event !== "message") throw new Error(`Unknown SSE event type: ${event.event}`);
+					if (event.data === "[DONE]") {
+						receivedDone = true;
+						controller.terminate();
+						return;
+					}
+					controller.enqueue(JSON.parse(event.data));
+				},
+				flush() {
+					if (!receivedDone) throw new Error("Stream ended abruptly without receiving [DONE] marker");
+				}
+			})).pipeThrough(transform);
+		});
+	}
+};
+//#endregion
+export { UIMessageStreamDecoder };
+
 //# sourceMappingURL=UIMessageStream.js.map

package/dist/core/serialization/ui-message-stream/UIMessageStream.js.map

@@ -1 +1 @@
-{"version":3,"file":"UIMessageStream.js","sourceRoot":"","sources":["../../../../src/core/serialization/ui-message-stream/UIMessageStream.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,wBAAwB,EAAE,uDAAoD;AACvF,OAAO,EAAE,uBAAuB,EAAE,sDAAmD;AACrF,OAAO,EAAE,iBAAiB,EAAE,gDAA6C;AAKzE,OAAO,EAAE,UAAU,EAAE,kCAA+B;AAoBpD,MAAM,cAAe,SAAQ,eAAiC;IAC5D;QACE,IAAI,WAAW,GAAsB,EAAE,CAAC;QACxC,IAAI,SAAS,GAAa,EAAE,CAAC;QAE7B,KAAK,CAAC;YACJ,KAAK;gBACH,WAAW,GAAG,EAAE,CAAC;gBACjB,SAAS,GAAG,EAAE,CAAC;YACjB,CAAC;YACD,SAAS,CAAC,IAAI,EAAE,UAAU;gBACxB,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;oBAAE,OAAO;gBAEjC,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;oBAChB,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACzB,UAAU,CAAC,OAAO,CAAC;4BACjB,KAAK,EAAE,WAAW,CAAC,KAAK,IAAI,SAAS;4BACrC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;4BAC1B,EAAE,EAAE,WAAW,CAAC,EAAE;4BAClB,KAAK,EAAE,WAAW,CAAC,KAAK;yBACzB,CAAC,CAAC;oBACL,CAAC;oBACD,WAAW,GAAG,EAAE,CAAC;oBACjB,SAAS,GAAG,EAAE,CAAC;oBACf,OAAO;gBACT,CAAC;gBAED,MAAM,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBACzC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,CAAC;gBAEzC,QAAQ,KAAK,EAAE,CAAC;oBACd,KAAK,OAAO;wBACV,WAAW,CAAC,KAAK,GAAG,KAAK,CAAC;wBAC1B,MAAM;oBACR,KAAK,MAAM;wBACT,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;wBACtB,MAAM;oBACR,KAAK,IAAI;wBACP,WAAW,CAAC,EAAE,GAAG,KAAK,CAAC;wBACvB,MAAM;oBACR,KAAK,OAAO;wBACV,WAAW,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;wBAClC,MAAM;gBACV,CAAC;YACH,CAAC;YACD,KAAK,CAAC,UAAU;gBACd,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,UAAU,CAAC,OAAO,CAAC;wBACjB,KAAK,EAAE,WAAW,CAAC,KAAK,IAAI,SAAS;wBACrC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;wBAC1B,EAAE,EAAE,WAAW,CAAC,EAAE;wBAClB,KAAK,EAAE,WAAW,CAAC,KAAK;qBACzB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,WAAW,GAAG,CAClB,KAA2B,EACQ,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;AAEvE;;GAEG;AACH,MAAM,OAAO,sBAAuB,SAAQ,uBAG3C;IACC,YAAY,UAAyC,EAAE;QACrD,KAAK,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAoC,CAAC;YACxE,IAAI,sBAAwD,CAAC;YAC7D,IAAI,gBAAoC,CAAC;YACzC,IAAI,YAAY,GAAG,KAAK,CAAC;YAEzB,MAAM,SAAS,GAAG,IAAI,wBAAwB,CAAuB;gBACnE,SAAS,CAAC,KAAK,EAAE,UAAU;oBACzB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;oBAExB,IAAI,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;wBACvB,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;wBAEjC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;4BACnB,OAAO,CAAC,MAAM,CAAC;gCACb,IAAI,EAAE,KAAK,CAAC,IAAI;gCAChB,IAAI;gCACJ,IAAI,EAAE,KAAK,CAAC,IAAI;gCAChB,GAAG,CAAC,KAAK,CAAC,SAAS,KAAK,SAAS,IAAI;oCACnC,SAAS,EAAE,KAAK,CAAC,SAAS;iCAC3B,CAAC;6BACH,CAAC,CAAC;wBACL,CAAC;wBAED,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;4BACrB,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,MAAM;gCACZ,IAAI,EAAE,EAAE;gCACR,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;6BACnC,CAAC,CAAC;wBACL,CAAC;wBACD,OAAO;oBACT,CAAC;oBAED,QAAQ,IAAI,EAAE,CAAC;wBACb,KAAK,OAAO;4BACV,gBAAgB,GAAG,KAAK,CAAC,SAAS,CAAC;4BACnC,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,YAAY;gCAClB,IAAI,EAAE,EAAE;gCACR,SAAS,EAAE,KAAK,CAAC,SAAS;6BAC3B,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,YAAY,CAAC;wBAClB,KAAK,UAAU,CAAC;wBAChB,KAAK,iBAAiB,CAAC;wBACvB,KAAK,eAAe;4BAClB,MAAM;wBAER,KAAK,YAAY;4BACf,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;4BACvC,MAAM;wBAER,KAAK,iBAAiB;4BACpB,UAAU,CAAC,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;4BACxC,MAAM;wBAER,KAAK,QAAQ;4BACX,UAAU,CAAC,YAAY,CAAC;gCACtB,IAAI,EAAE,QAAQ;gCACd,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC,UAAU;gCACnC,EAAE,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE;gCACnB,GAAG,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG;gCACrB,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;6BACzD,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,MAAM;4BACT,UAAU,CAAC,UAAU,CAAC;gCACpB,IAAI,EAAE,MAAM;gCACZ,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;gCAC7B,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI;6BACtB,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,iBAAiB,CAAC,CAAC,CAAC;4BACvB,sBAAsB,EAAE,KAAK,EAAE,CAAC;4BAChC,sBAAsB,GAAG,SAAS,CAAC;4BAEnC,IAAI,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;gCAC9C,MAAM,IAAI,KAAK,CACb,uCAAuC,KAAK,CAAC,UAAU,EAAE,CAC1D,CAAC;4BACJ,CAAC;4BAED,MAAM,kBAAkB,GAAG,UAAU,CAAC,eAAe,CAAC;gCACpD,UAAU,EAAE,KAAK,CAAC,UAAU;gCAC5B,QAAQ,EAAE,KAAK,CAAC,QAAQ;6BACzB,CAAC,CAAC;4BACH,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,kBAAkB,CAAC,CAAC;4BAC9D,sBAAsB,GAAG,kBAAkB,CAAC,QAAQ,CAAC;4BACrD,MAAM;wBACR,CAAC;wBAED,KAAK,iBAAiB;4BACpB,sBAAsB,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;4BAC/C,MAAM;wBAER,KAAK,eAAe;4BAClB,sBAAsB,EAAE,KAAK,EAAE,CAAC;4BAChC,sBAAsB,GAAG,SAAS,CAAC;4BACnC,MAAM;wBAER,KAAK,aAAa,CAAC,CAAC,CAAC;4BACnB,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,GAAG,CAChD,KAAK,CAAC,UAAU,CACjB,CAAC;4BACF,IAAI,CAAC,kBAAkB,EAAE,CAAC;gCACxB,MAAM,IAAI,KAAK,CACb,4CAA4C,KAAK,CAAC,UAAU,EAAE,CAC/D,CAAC;4BACJ,CAAC;4BACD,kBAAkB,CAAC,WAAW,CAAC;gCAC7B,MAAM,EAAE,KAAK,CAAC,MAAM;gCACpB,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,KAAK;gCAC/B,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS;oCAC9B,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE;oCAC9B,CAAC,CAAC,EAAE,CAAC;6BACR,CAAC,CAAC;4BACH,MAAM;wBACR,CAAC;wBAED,KAAK,YAAY;4BACf,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,YAAY;gCAClB,IAAI,EAAE,EAAE;gCACR,SAAS,EAAE,KAAK,CAAC,SAAS,IAAI,gBAAgB,IAAI,UAAU,EAAE;6BAC/D,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,aAAa;4BAChB,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,aAAa;gCACnB,IAAI,EAAE,EAAE;gCACR,YAAY,EAAE,KAAK,CAAC,YAAY;gCAChC,KAAK,EAAE,KAAK,CAAC,KAAK;gCAClB,WAAW,EAAE,KAAK,CAAC,WAAW;6BAC/B,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,QAAQ;4BACX,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,gBAAgB;gCACtB,IAAI,EAAE,EAAE;gCACR,YAAY,EAAE,KAAK,CAAC,YAAY;gCAChC,KAAK,EAAE,KAAK,CAAC,KAAK;6BACnB,CAAC,CAAC;4BACH,MAAM;wBAER,KAAK,OAAO;4BACV,UAAU,CAAC,OAAO,CAAC;gCACjB,IAAI,EAAE,OAAO;gCACb,IAAI,EAAE,EAAE;gCACR,KAAK,EAAE,KAAK,CAAC,SAAS;6BACvB,CAAC,CAAC;4BACH,MAAM;wBAER;4BACE,iDAAiD;4BACjD,MAAM;oBACV,CAAC;gBACH,CAAC;gBACD,KAAK;oBACH,sBAAsB,EAAE,KAAK,EAAE,CAAC;oBAChC,uGAAuG;oBACvG,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;oBACpD,mBAAmB,CAAC,KAAK,EAAE,CAAC;gBAC9B,CAAC;aACF,CAAC,CAAC;YAEH,OAAO,QAAQ;iBACZ,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;iBACpC,WAAW,CAAC,IAAI,iBAAiB,EAAE,CAAC;iBACpC,WAAW,CAAC,IAAI,cAAc,EAAE,CAAC;iBACjC,WAAW,CACV,IAAI,eAAe,CAAiC;gBAClD,SAAS,CAAC,KAAK,EAAE,UAAU;oBACzB,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;wBAC9B,MAAM,IAAI,KAAK,CAAC,2BAA2B,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;oBAC5D,CAAC;oBAED,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;wBAC5B,YAAY,GAAG,IAAI,CAAC;wBACpB,UAAU,CAAC,SAAS,EAAE,CAAC;wBACvB,OAAO;oBACT,CAAC;oBAED,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC7C,CAAC;gBACD,KAAK;oBACH,IAAI,CAAC,YAAY,EAAE,CAAC;wBAClB,MAAM,IAAI,KAAK,CACb,uDAAuD,CACxD,CAAC;oBACJ,CAAC;gBACH,CAAC;aACF,CAAC,CACH;iBACA,WAAW,CAAC,SAAS,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC;CACF"}
+{"version":3,"file":"UIMessageStream.js","names":[],"sources":["../../../../src/core/serialization/ui-message-stream/UIMessageStream.ts"],"sourcesContent":["import type { AssistantStreamChunk } from \"../../AssistantStreamChunk\";\nimport type { ToolCallStreamController } from \"../../modules/tool-call\";\nimport type { TextStreamController } from \"../../modules/text\";\nimport { AssistantTransformStream } from \"../../utils/stream/AssistantTransformStream\";\nimport { PipeableTransformStream } from \"../../utils/stream/PipeableTransformStream\";\nimport { LineDecoderStream } from \"../../utils/stream/LineDecoderStream\";\nimport type {\n  UIMessageStreamChunk,\n  UIMessageStreamDataChunk,\n} from \"./chunk-types\";\nimport { generateId } from \"../../utils/generateId\";\n\nexport type { UIMessageStreamChunk, UIMessageStreamDataChunk };\n\nexport type UIMessageStreamDecoderOptions = {\n  onData?: (data: {\n    type: string;\n    name: string;\n    data: unknown;\n    transient?: boolean;\n  }) => void;\n};\n\ntype SSEEvent = {\n  event: string;\n  data: string;\n  id?: string | undefined;\n  retry?: number | undefined;\n};\n\nclass SSEEventStream extends TransformStream<string, SSEEvent> {\n  constructor() {\n    let eventBuffer: Partial<SSEEvent> = {};\n    let dataLines: string[] = [];\n\n    super({\n      start() {\n        eventBuffer = {};\n        dataLines = [];\n      },\n      transform(line, controller) {\n        if (line.startsWith(\":\")) return;\n\n        if (line === \"\") {\n          if (dataLines.length > 0) {\n            controller.enqueue({\n              event: eventBuffer.event || \"message\",\n              data: dataLines.join(\"\\n\"),\n              id: eventBuffer.id,\n              retry: eventBuffer.retry,\n            });\n          }\n          eventBuffer = {};\n          dataLines = [];\n          return;\n        }\n\n        const [field, ...rest] = line.split(\":\");\n        const value = rest.join(\":\").trimStart();\n\n        switch (field) {\n          case \"event\":\n            eventBuffer.event = value;\n            break;\n          case \"data\":\n            dataLines.push(value);\n            break;\n          case \"id\":\n            eventBuffer.id = value;\n            break;\n          case \"retry\":\n            eventBuffer.retry = Number(value);\n            break;\n        }\n      },\n      flush(controller) {\n        if (dataLines.length > 0) {\n          controller.enqueue({\n            event: eventBuffer.event || \"message\",\n            data: dataLines.join(\"\\n\"),\n            id: eventBuffer.id,\n            retry: eventBuffer.retry,\n          });\n        }\n      },\n    });\n  }\n}\n\nconst isDataChunk = (\n  chunk: UIMessageStreamChunk,\n): chunk is UIMessageStreamDataChunk => chunk.type.startsWith(\"data-\");\n\n/**\n * Decodes AI SDK v6 UI Message Stream format into AssistantStreamChunks.\n */\nexport class UIMessageStreamDecoder extends PipeableTransformStream<\n  Uint8Array<ArrayBuffer>,\n  AssistantStreamChunk\n> {\n  constructor(options: UIMessageStreamDecoderOptions = {}) {\n    super((readable) => {\n      const toolCallControllers = new Map<string, ToolCallStreamController>();\n      let activeToolCallArgsText: TextStreamController | undefined;\n      let currentMessageId: string | undefined;\n      let receivedDone = false;\n\n      const transform = new AssistantTransformStream<UIMessageStreamChunk>({\n        transform(chunk, controller) {\n          const type = chunk.type;\n\n          if (isDataChunk(chunk)) {\n            const name = chunk.type.slice(5);\n\n            if (options.onData) {\n              options.onData({\n                type: chunk.type,\n                name,\n                data: chunk.data,\n                ...(chunk.transient !== undefined && {\n                  transient: chunk.transient,\n                }),\n              });\n            }\n\n            if (!chunk.transient) {\n              controller.enqueue({\n                type: \"data\",\n                path: [],\n                data: [{ name, data: chunk.data }],\n              });\n            }\n            return;\n          }\n\n          switch (type) {\n            case \"start\":\n              currentMessageId = chunk.messageId;\n              controller.enqueue({\n                type: \"step-start\",\n                path: [],\n                messageId: chunk.messageId,\n              });\n              break;\n\n            case \"text-start\":\n            case \"text-end\":\n            case \"reasoning-start\":\n            case \"reasoning-end\":\n              break;\n\n            case \"text-delta\":\n              controller.appendText(chunk.textDelta);\n              break;\n\n            case \"reasoning-delta\":\n              controller.appendReasoning(chunk.delta);\n              break;\n\n            case \"source\":\n              controller.appendSource({\n                type: \"source\",\n                sourceType: chunk.source.sourceType,\n                id: chunk.source.id,\n                url: chunk.source.url,\n                ...(chunk.source.title && { title: chunk.source.title }),\n              });\n              break;\n\n            case \"file\":\n              controller.appendFile({\n                type: \"file\",\n                mimeType: chunk.file.mimeType,\n                data: chunk.file.data,\n              });\n              break;\n\n            case \"tool-call-start\": {\n              activeToolCallArgsText?.close();\n              activeToolCallArgsText = undefined;\n\n              if (toolCallControllers.has(chunk.toolCallId)) {\n                throw new Error(\n                  `Encountered duplicate tool call id: ${chunk.toolCallId}`,\n                );\n              }\n\n              const toolCallController = controller.addToolCallPart({\n                toolCallId: chunk.toolCallId,\n                toolName: chunk.toolName,\n              });\n              toolCallControllers.set(chunk.toolCallId, toolCallController);\n              activeToolCallArgsText = toolCallController.argsText;\n              break;\n            }\n\n            case \"tool-call-delta\":\n              activeToolCallArgsText?.append(chunk.argsText);\n              break;\n\n            case \"tool-call-end\":\n              activeToolCallArgsText?.close();\n              activeToolCallArgsText = undefined;\n              break;\n\n            case \"tool-result\": {\n              const toolCallController = toolCallControllers.get(\n                chunk.toolCallId,\n              );\n              if (!toolCallController) {\n                throw new Error(\n                  `Encountered tool result with unknown id: ${chunk.toolCallId}`,\n                );\n              }\n              toolCallController.setResponse({\n                result: chunk.result,\n                isError: chunk.isError ?? false,\n                ...(chunk.messages !== undefined\n                  ? { messages: chunk.messages }\n                  : {}),\n              });\n              break;\n            }\n\n            case \"start-step\":\n              controller.enqueue({\n                type: \"step-start\",\n                path: [],\n                messageId: chunk.messageId ?? currentMessageId ?? generateId(),\n              });\n              break;\n\n            case \"finish-step\":\n              controller.enqueue({\n                type: \"step-finish\",\n                path: [],\n                finishReason: chunk.finishReason,\n                usage: chunk.usage,\n                isContinued: chunk.isContinued,\n              });\n              break;\n\n            case \"finish\":\n              controller.enqueue({\n                type: \"message-finish\",\n                path: [],\n                finishReason: chunk.finishReason,\n                usage: chunk.usage,\n              });\n              break;\n\n            case \"error\":\n              controller.enqueue({\n                type: \"error\",\n                path: [],\n                error: chunk.errorText,\n              });\n              break;\n\n            default:\n              // ignore unknown types for forward compatibility\n              break;\n          }\n        },\n        flush() {\n          activeToolCallArgsText?.close();\n          // biome-ignore lint/suspicious/useIterableCallbackReturn: forEach callback intentionally has no return\n          toolCallControllers.forEach((ctrl) => ctrl.close());\n          toolCallControllers.clear();\n        },\n      });\n\n      return readable\n        .pipeThrough(new TextDecoderStream())\n        .pipeThrough(new LineDecoderStream())\n        .pipeThrough(new SSEEventStream())\n        .pipeThrough(\n          new TransformStream<SSEEvent, UIMessageStreamChunk>({\n            transform(event, controller) {\n              if (event.event !== \"message\") {\n                throw new Error(`Unknown SSE event type: ${event.event}`);\n              }\n\n              if (event.data === \"[DONE]\") {\n                receivedDone = true;\n                controller.terminate();\n                return;\n              }\n\n              controller.enqueue(JSON.parse(event.data));\n            },\n            flush() {\n              if (!receivedDone) {\n                throw new Error(\n                  \"Stream ended abruptly without receiving [DONE] marker\",\n                );\n              }\n            },\n          }),\n        )\n        .pipeThrough(transform);\n    });\n  }\n}\n"],"mappings":";;;;;AA8BA,IAAM,iBAAN,cAA6B,gBAAkC;CAC7D,cAAc;EACZ,IAAI,cAAiC,CAAC;EACtC,IAAI,YAAsB,CAAC;EAE3B,MAAM;GACJ,QAAQ;IACN,cAAc,CAAC;IACf,YAAY,CAAC;GACf;GACA,UAAU,MAAM,YAAY;IAC1B,IAAI,KAAK,WAAW,GAAG,GAAG;IAE1B,IAAI,SAAS,IAAI;KACf,IAAI,UAAU,SAAS,GACrB,WAAW,QAAQ;MACjB,OAAO,YAAY,SAAS;MAC5B,MAAM,UAAU,KAAK,IAAI;MACzB,IAAI,YAAY;MAChB,OAAO,YAAY;KACrB,CAAC;KAEH,cAAc,CAAC;KACf,YAAY,CAAC;KACb;IACF;IAEA,MAAM,CAAC,OAAO,GAAG,QAAQ,KAAK,MAAM,GAAG;IACvC,MAAM,QAAQ,KAAK,KAAK,GAAG,EAAE,UAAU;IAEvC,QAAQ,OAAR;KACE,KAAK;MACH,YAAY,QAAQ;MACpB;KACF,KAAK;MACH,UAAU,KAAK,KAAK;MACpB;KACF,KAAK;MACH,YAAY,KAAK;MACjB;KACF,KAAK;MACH,YAAY,QAAQ,OAAO,KAAK;MAChC;IACJ;GACF;GACA,MAAM,YAAY;IAChB,IAAI,UAAU,SAAS,GACrB,WAAW,QAAQ;KACjB,OAAO,YAAY,SAAS;KAC5B,MAAM,UAAU,KAAK,IAAI;KACzB,IAAI,YAAY;KAChB,OAAO,YAAY;IACrB,CAAC;GAEL;EACF,CAAC;CACH;AACF;AAEA,MAAM,eACJ,UACsC,MAAM,KAAK,WAAW,OAAO;;;;AAKrE,IAAa,yBAAb,cAA4C,wBAG1C;CACA,YAAY,UAAyC,CAAC,GAAG;EACvD,OAAO,aAAa;GAClB,MAAM,sCAAsB,IAAI,IAAsC;GACtE,IAAI;GACJ,IAAI;GACJ,IAAI,eAAe;GAEnB,MAAM,YAAY,IAAI,yBAA+C;IACnE,UAAU,OAAO,YAAY;KAC3B,MAAM,OAAO,MAAM;KAEnB,IAAI,YAAY,KAAK,GAAG;MACtB,MAAM,OAAO,MAAM,KAAK,MAAM,CAAC;MAE/B,IAAI,QAAQ,QACV,QAAQ,OAAO;OACb,MAAM,MAAM;OACZ;OACA,MAAM,MAAM;OACZ,GAAI,MAAM,cAAc,KAAA,KAAa,EACnC,WAAW,MAAM,UACnB;MACF,CAAC;MAGH,IAAI,CAAC,MAAM,WACT,WAAW,QAAQ;OACjB,MAAM;OACN,MAAM,CAAC;OACP,MAAM,CAAC;QAAE;QAAM,MAAM,MAAM;OAAK,CAAC;MACnC,CAAC;MAEH;KACF;KAEA,QAAQ,MAAR;MACE,KAAK;OACH,mBAAmB,MAAM;OACzB,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,WAAW,MAAM;OACnB,CAAC;OACD;MAEF,KAAK;MACL,KAAK;MACL,KAAK;MACL,KAAK,iBACH;MAEF,KAAK;OACH,WAAW,WAAW,MAAM,SAAS;OACrC;MAEF,KAAK;OACH,WAAW,gBAAgB,MAAM,KAAK;OACtC;MAEF,KAAK;OACH,WAAW,aAAa;QACtB,MAAM;QACN,YAAY,MAAM,OAAO;QACzB,IAAI,MAAM,OAAO;QACjB,KAAK,MAAM,OAAO;QAClB,GAAI,MAAM,OAAO,SAAS,EAAE,OAAO,MAAM,OAAO,MAAM;OACxD,CAAC;OACD;MAEF,KAAK;OACH,WAAW,WAAW;QACpB,MAAM;QACN,UAAU,MAAM,KAAK;QACrB,MAAM,MAAM,KAAK;OACnB,CAAC;OACD;MAEF,KAAK,mBAAmB;OACtB,wBAAwB,MAAM;OAC9B,yBAAyB,KAAA;OAEzB,IAAI,oBAAoB,IAAI,MAAM,UAAU,GAC1C,MAAM,IAAI,MACR,uCAAuC,MAAM,YAC/C;OAGF,MAAM,qBAAqB,WAAW,gBAAgB;QACpD,YAAY,MAAM;QAClB,UAAU,MAAM;OAClB,CAAC;OACD,oBAAoB,IAAI,MAAM,YAAY,kBAAkB;OAC5D,yBAAyB,mBAAmB;OAC5C;MACF;MAEA,KAAK;OACH,wBAAwB,OAAO,MAAM,QAAQ;OAC7C;MAEF,KAAK;OACH,wBAAwB,MAAM;OAC9B,yBAAyB,KAAA;OACzB;MAEF,KAAK,eAAe;OAClB,MAAM,qBAAqB,oBAAoB,IAC7C,MAAM,UACR;OACA,IAAI,CAAC,oBACH,MAAM,IAAI,MACR,4CAA4C,MAAM,YACpD;OAEF,mBAAmB,YAAY;QAC7B,QAAQ,MAAM;QACd,SAAS,MAAM,WAAW;QAC1B,GAAI,MAAM,aAAa,KAAA,IACnB,EAAE,UAAU,MAAM,SAAS,IAC3B,CAAC;OACP,CAAC;OACD;MACF;MAEA,KAAK;OACH,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,WAAW,MAAM,aAAa,oBAAoB,WAAW;OAC/D,CAAC;OACD;MAEF,KAAK;OACH,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,cAAc,MAAM;QACpB,OAAO,MAAM;QACb,aAAa,MAAM;OACrB,CAAC;OACD;MAEF,KAAK;OACH,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,cAAc,MAAM;QACpB,OAAO,MAAM;OACf,CAAC;OACD;MAEF,KAAK;OACH,WAAW,QAAQ;QACjB,MAAM;QACN,MAAM,CAAC;QACP,OAAO,MAAM;OACf,CAAC;OACD;MAEF,SAEE;KACJ;IACF;IACA,QAAQ;KACN,wBAAwB,MAAM;KAE9B,oBAAoB,SAAS,SAAS,KAAK,MAAM,CAAC;KAClD,oBAAoB,MAAM;IAC5B;GACF,CAAC;GAED,OAAO,SACJ,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,IAAI,kBAAkB,CAAC,EACnC,YAAY,IAAI,eAAe,CAAC,EAChC,YACC,IAAI,gBAAgD;IAClD,UAAU,OAAO,YAAY;KAC3B,IAAI,MAAM,UAAU,WAClB,MAAM,IAAI,MAAM,2BAA2B,MAAM,OAAO;KAG1D,IAAI,MAAM,SAAS,UAAU;MAC3B,eAAe;MACf,WAAW,UAAU;MACrB;KACF;KAEA,WAAW,QAAQ,KAAK,MAAM,MAAM,IAAI,CAAC;IAC3C;IACA,QAAQ;KACN,IAAI,CAAC,cACH,MAAM,IAAI,MACR,uDACF;IAEJ;GACF,CAAC,CACH,EACC,YAAY,SAAS;EAC1B,CAAC;CACH;AACF"}

package/dist/core/serialization/ui-message-stream/chunk-types.d.ts

@@ -1,79 +1,82 @@
-import type { ReadonlyJSONValue } from "../../../utils/json/json-value.js";
+import { ReadonlyJSONValue } from "../../../utils/json/json-value.js";
+
+//#region src/core/serialization/ui-message-stream/chunk-types.d.ts
 type FinishReason = "stop" | "length" | "content-filter" | "tool-calls" | "error" | "other" | "unknown";
 type Usage = {
-    inputTokens: number;
-    outputTokens: number;
+  inputTokens: number;
+  outputTokens: number;
 };
-export type UIMessageStreamChunk = {
-    type: "start";
-    messageId: string;
+type UIMessageStreamChunk = {
+  type: "start";
+  messageId: string;
 } | {
-    type: "text-start";
-    id: string;
-} | {
-    type: "text-delta";
-    textDelta: string;
-} | {
-    type: "text-end";
+  type: "text-start";
+  id: string;
 } | {
-    type: "reasoning-start";
-    id: string;
+  type: "text-delta";
+  textDelta: string;
 } | {
-    type: "reasoning-delta";
-    delta: string;
+  type: "text-end";
 } | {
-    type: "reasoning-end";
+  type: "reasoning-start";
+  id: string;
 } | {
-    type: "source";
-    source: {
-        sourceType: "url";
-        id: string;
-        url: string;
-        title?: string;
-    };
+  type: "reasoning-delta";
+  delta: string;
 } | {
-    type: "file";
-    file: {
-        mimeType: string;
-        data: string;
-    };
+  type: "reasoning-end";
 } | {
-    type: "tool-call-start";
+  type: "source";
+  source: {
+    sourceType: "url";
     id: string;
-    toolCallId: string;
-    toolName: string;
-} | {
-    type: "tool-call-delta";
-    argsText: string;
-} | {
-    type: "tool-call-end";
-} | {
-    type: "tool-result";
-    toolCallId: string;
-    result: ReadonlyJSONValue;
-    isError?: boolean;
-    messages?: ReadonlyJSONValue;
-} | {
-    type: "start-step";
-    messageId?: string;
-} | {
-    type: "finish-step";
-    finishReason: FinishReason;
-    usage: Usage;
-    isContinued: boolean;
-} | {
-    type: "finish";
-    finishReason: FinishReason;
-    usage: Usage;
-} | {
-    type: "error";
-    errorText: string;
+    url: string;
+    title?: string;
+  };
+} | {
+  type: "file";
+  file: {
+    mimeType: string;
+    data: string;
+  };
+} | {
+  type: "tool-call-start";
+  id: string;
+  toolCallId: string;
+  toolName: string;
+} | {
+  type: "tool-call-delta";
+  argsText: string;
+} | {
+  type: "tool-call-end";
+} | {
+  type: "tool-result";
+  toolCallId: string;
+  result: ReadonlyJSONValue;
+  isError?: boolean;
+  messages?: ReadonlyJSONValue;
+} | {
+  type: "start-step";
+  messageId?: string;
+} | {
+  type: "finish-step";
+  finishReason: FinishReason;
+  usage: Usage;
+  isContinued: boolean;
+} | {
+  type: "finish";
+  finishReason: FinishReason;
+  usage: Usage;
+} | {
+  type: "error";
+  errorText: string;
 } | UIMessageStreamDataChunk;
-export type UIMessageStreamDataChunk = {
-    type: `data-${string}`;
-    id?: string;
-    data: ReadonlyJSONValue;
-    transient?: boolean;
+type UIMessageStreamDataChunk = {
+  type: `data-${string}`;
+  id?: string;
+  data: ReadonlyJSONValue;
+  transient?: boolean;
 };
-export {};
+//#endregion
+export { UIMessageStreamChunk, UIMessageStreamDataChunk };
 //# sourceMappingURL=chunk-types.d.ts.map