ai-stream-utils 1.5.0 → 2.0.0

package/dist/convert-ui-message-stream-to-sse-stream-CcQRXKju.mjs

@@ -0,0 +1,165 @@
+import { JsonToSseTransformStream, parseJsonEventStream, uiMessageChunkSchema } from "ai";
+
+//#region src/utils/convert-async-iterable-to-stream.ts
+/**
+* Converts an AsyncIterable to a ReadableStream.
+* Copied from https://github.com/vercel/ai/blob/main/packages/provider-utils/src/convert-async-iterator-to-readable-stream.ts
+*/
+function convertAsyncIterableToStream(iterable) {
+	const iterator = iterable[Symbol.asyncIterator]();
+	let cancelled = false;
+	return new ReadableStream({
+		async pull(controller) {
+			if (cancelled) return;
+			try {
+				const { value, done } = await iterator.next();
+				if (done) controller.close();
+				else controller.enqueue(value);
+			} catch (error) {
+				controller.error(error);
+			}
+		},
+		async cancel(reason) {
+			cancelled = true;
+			try {
+				await iterator.return?.(reason);
+			} catch {}
+		}
+	});
+}
+
+//#endregion
+//#region src/utils/create-async-iterable-stream.ts
+/**
+* Converts a ReadableStream to an AsyncIterableStream.
+* Copied from https://github.com/vercel/ai/blob/main/packages/ai/src/util/async-iterable-stream.ts
+*/
+function createAsyncIterableStream(source) {
+	/** Pipe through a TransformStream to ensure a fresh, unlocked stream. */
+	const stream = source.pipeThrough(new TransformStream());
+	/** Implements the async iterator protocol for the stream. */
+	return Object.assign(stream, { [Symbol.asyncIterator]() {
+		const reader = stream.getReader();
+		let finished = false;
+		/** Cleans up the reader by cancelling and releasing the lock. */
+		async function cleanup(cancelStream) {
+			finished = true;
+			try {
+				if (cancelStream) await reader.cancel?.();
+			} finally {
+				try {
+					reader.releaseLock();
+				} catch {}
+			}
+		}
+		return {
+			async next() {
+				if (finished) return {
+					done: true,
+					value: void 0
+				};
+				const { done, value } = await reader.read();
+				if (done) {
+					await cleanup(true);
+					return {
+						done: true,
+						value: void 0
+					};
+				}
+				return {
+					done: false,
+					value
+				};
+			},
+			async return() {
+				await cleanup(true);
+				return {
+					done: true,
+					value: void 0
+				};
+			},
+			async throw(err) {
+				await cleanup(true);
+				throw err;
+			}
+		};
+	} });
+}
+
+//#endregion
+//#region src/utils/convert-array-to-async-iterable.ts
+/**
+* Converts an array to an AsyncIterable.
+*/
+async function* convertArrayToAsyncIterable(array) {
+	for (const item of array) yield item;
+}
+
+//#endregion
+//#region src/utils/convert-array-to-stream.ts
+/**
+* Converts an array to a ReadableStream.
+*/
+function convertArrayToStream(array) {
+	return new ReadableStream({ start(controller) {
+		for (const item of array) controller.enqueue(item);
+		controller.close();
+	} });
+}
+
+//#endregion
+//#region src/utils/convert-async-iterable-to-array.ts
+/**
+* Converts an AsyncIterable to an array.
+*/
+async function convertAsyncIterableToArray(iterable) {
+	const result = [];
+	for await (const item of iterable) result.push(item);
+	return result;
+}
+
+//#endregion
+//#region src/utils/convert-sse-stream-to-ui-message-stream.ts
+/**
+* Converts an SSE stream to a UI message stream.
+*/
+function convertSSEToUIMessageStream(stream) {
+	return parseJsonEventStream({
+		stream: stream.pipeThrough(new TextEncoderStream()),
+		schema: uiMessageChunkSchema
+	}).pipeThrough(new TransformStream({ transform(result, controller) {
+		if (result.success) controller.enqueue(result.value);
+	} }));
+}
+
+//#endregion
+//#region src/utils/convert-stream-to-array.ts
+/**
+* Converts a ReadableStream to an array.
+*/
+async function convertStreamToArray(stream) {
+	const reader = stream.getReader();
+	const result = [];
+	try {
+		while (true) {
+			const { done, value } = await reader.read();
+			if (done) break;
+			result.push(value);
+		}
+	} finally {
+		reader.releaseLock();
+	}
+	return result;
+}
+
+//#endregion
+//#region src/utils/convert-ui-message-stream-to-sse-stream.ts
+/**
+* Converts a UI message stream to an SSE stream.
+*/
+function convertUIMessageToSSEStream(stream) {
+	return stream.pipeThrough(new JsonToSseTransformStream());
+}
+
+//#endregion
+export { convertArrayToStream as a, convertAsyncIterableToStream as c, convertAsyncIterableToArray as i, convertStreamToArray as n, convertArrayToAsyncIterable as o, convertSSEToUIMessageStream as r, createAsyncIterableStream as s, convertUIMessageToSSEStream as t };