@varo-ui/ai 1.2.0 → 2.1.0

package/dist/index.mjs

@@ -10,13 +10,51 @@ function splitGraphemes(value) {
 function clamp(value, min, max) {
 	return Math.min(max, Math.max(min, value));
 }
+function hasUnclosedFence(value) {
+	let marker;
+	let markerLength = 0;
+	for (const line of value.split(/\r?\n/)) {
+		const match = /^ {0,3}(`{3,}|~{3,})/.exec(line);
+		if (!match) continue;
+		const nextMarker = match[1][0];
+		const nextLength = match[1].length;
+		if (!marker) {
+			marker = nextMarker;
+			markerLength = nextLength;
+		} else if (marker === nextMarker && nextLength >= markerLength) {
+			marker = void 0;
+			markerLength = 0;
+		}
+	}
+	return marker !== void 0;
+}
+function fenceSafeCommitCount(visible, source, segments) {
+	if (segments.length === 0) return 0;
+	const candidate = visible + segments.join("");
+	const lineStart = candidate.lastIndexOf("\n") + 1;
+	const candidateLine = candidate.slice(lineStart);
+	const sourceLineEnd = source.indexOf("\n", lineStart);
+	const sourceLine = source.slice(lineStart, sourceLineEnd < 0 ? source.length : sourceLineEnd);
+	const match = /^ {0,3}(`+|~+)/.exec(sourceLine);
+	if (!match) return segments.length;
+	const marker = match[1];
+	const unresolvedCandidate = marker.length < 3 && sourceLine.trim() === marker;
+	const completeFence = marker.length >= 3;
+	if ((unresolvedCandidate || completeFence) && /^ {0,3}(?:`+|~+)/.test(candidateLine)) return 0;
+	return segments.length;
+}
 function createTimerTextStream(options = {}) {
 	const minCharsPerSecond = Math.max(1, options.minCharsPerSecond ?? 40);
-	const maxCharsPerSecond = Math.max(minCharsPerSecond, options.maxCharsPerSecond ?? 800);
-	const targetLatencyMs = Math.max(50, options.targetLatencyMs ?? 700);
-	const maxCommitFps = Math.max(1, Math.min(30, options.maxCommitFps ?? 20));
-	const maxCharsPerCommit = Math.max(1, options.maxCharsPerCommit ?? 64);
-	const catchUpThreshold = Math.max(1, options.catchUpThreshold ?? 320);
+	const maxCharsPerSecond = Math.max(minCharsPerSecond, options.maxCharsPerSecond ?? 1e3);
+	const targetLatencyMs = Math.max(1, options.targetLatencyMs ?? 900);
+	const catchUpLatencyMs = Math.max(1, options.catchUpLatencyMs ?? 350);
+	const maxCommitFps = Math.max(1, Math.min(30, options.maxCommitFps ?? 30));
+	const maxCharsPerCommit = Math.max(1, options.maxCharsPerCommit ?? 80);
+	const catchUpThreshold = Math.max(0, options.catchUpThreshold ?? 600);
+	const startDelayMs = Math.max(0, options.startDelayMs ?? 80);
+	const flushOnFinish = options.flushOnFinish ?? false;
+	const burstInitialContent = options.burstInitialContent ?? false;
+	const burstRevealThresholdChars = Math.max(1, options.burstRevealThresholdChars ?? 2048);
 	const listeners = /* @__PURE__ */ new Set();
 	const pending = [];
 	let pendingOffset = 0;
@@ -27,16 +65,17 @@ function createTimerTextStream(options = {}) {
 	let destroyed = false;
 	let timer;
 	let lastTick = Date.now();
+	let firstPendingAt = 0;
 	let budget = 0;
 	function pendingCount() {
 		return pending.length - pendingOffset;
 	}
 	function snapshot() {
-		const pendingChars = pendingCount();
+		const pendingChars = Math.max(0, source.length - visible.length);
 		return {
-			caughtUp: pendingChars === 0,
+			caughtUp: pendingCount() === 0,
 			done,
-			final: done && pendingChars === 0,
+			final: done && pendingCount() === 0,
 			paused,
 			pendingChars,
 			source,
@@ -59,6 +98,11 @@ function createTimerTextStream(options = {}) {
 		timer = void 0;
 		if (destroyed || paused) return;
 		const now = Date.now();
+		if (firstPendingAt && now - firstPendingAt < startDelayMs) {
+			schedule();
+			return;
+		}
+		firstPendingAt = 0;
 		const deltaMs = Math.max(1, now - lastTick);
 		lastTick = now;
 		const backlog = pendingCount();
@@ -66,12 +110,27 @@ function createTimerTextStream(options = {}) {
 			emit();
 			return;
 		}
-		const cps = clamp(backlog * 1e3 / targetLatencyMs * (backlog >= catchUpThreshold ? 1.8 : 1), minCharsPerSecond, maxCharsPerSecond);
+		const latencyMs = backlog >= catchUpThreshold ? catchUpLatencyMs : targetLatencyMs;
+		const cps = clamp(backlog * 1e3 / latencyMs, minCharsPerSecond, maxCharsPerSecond);
 		budget += cps * deltaMs / 1e3;
 		const commitCount = Math.min(backlog, maxCharsPerCommit, Math.max(1, Math.floor(budget)));
-		budget = Math.max(0, budget - commitCount);
-		visible += pending.slice(pendingOffset, pendingOffset + commitCount).join("");
-		pendingOffset += commitCount;
+		let proposedSegments = pending.slice(pendingOffset, pendingOffset + commitCount);
+		let safeCommitCount = fenceSafeCommitCount(visible, source, proposedSegments);
+		if (safeCommitCount === 0) {
+			const newlineOffset = pending.slice(pendingOffset).findIndex((segment) => segment.includes("\n"));
+			if (newlineOffset >= 0) {
+				proposedSegments = pending.slice(pendingOffset, pendingOffset + newlineOffset + 1);
+				safeCommitCount = fenceSafeCommitCount(visible, source, proposedSegments);
+			}
+		}
+		if (safeCommitCount === 0) {
+			emit();
+			schedule();
+			return;
+		}
+		budget = Math.max(0, budget - safeCommitCount);
+		visible += proposedSegments.slice(0, safeCommitCount).join("");
+		pendingOffset += safeCommitCount;
 		compactPending();
 		emit();
 		schedule();
@@ -80,27 +139,37 @@ function createTimerTextStream(options = {}) {
 		clearTimeout(timer);
 		timer = void 0;
 	}
+	function destroy() {
+		if (destroyed) return;
+		destroyed = true;
+		cancelTimer();
+		pending.length = 0;
+		listeners.clear();
+	}
 	return {
-		destroy() {
-			if (destroyed) return;
-			destroyed = true;
-			cancelTimer();
-			pending.length = 0;
-			listeners.clear();
-		},
+		destroy,
+		dispose: destroy,
 		enqueue(chunk) {
 			if (destroyed || !chunk) return;
 			if (done) done = false;
+			const initialChunk = source.length === 0;
 			source += chunk;
 			pending.push(...splitGraphemes(chunk));
+			if (initialChunk) firstPendingAt = Date.now();
 			lastTick = Date.now();
+			if (initialChunk && burstInitialContent && pendingCount() >= burstRevealThresholdChars && !hasUnclosedFence(source)) {
+				visible = source;
+				pending.length = 0;
+				pendingOffset = 0;
+				firstPendingAt = 0;
+			}
 			emit();
 			schedule();
 		},
 		finish(finishOptions = {}) {
 			if (destroyed) return;
 			done = true;
-			if (finishOptions.flush) {
+			if (finishOptions.flush ?? flushOnFinish) {
 				visible = source;
 				pending.length = 0;
 				pendingOffset = 0;
@@ -125,7 +194,7 @@ function createTimerTextStream(options = {}) {
 			cancelTimer();
 			emit();
 		},
-		reset(content = "") {
+		reset(content = "", _resetOptions = {}) {
 			cancelTimer();
 			source = content;
 			visible = content;
@@ -134,6 +203,7 @@ function createTimerTextStream(options = {}) {
 			done = false;
 			paused = false;
 			budget = 0;
+			firstPendingAt = 0;
 			lastTick = Date.now();
 			emit();
 		},
@@ -156,6 +226,7 @@ function createTextStream(options = {}) {
 	const controller = createSmoothMarkdownStream(options);
 	return {
 		destroy: controller.destroy,
+		dispose: controller.dispose,
 		enqueue: controller.enqueue,
 		finish: controller.finish,
 		flush: controller.flush,
@@ -168,6 +239,12 @@ function createTextStream(options = {}) {
 }
 //#endregion
 //#region src/agent-stream.ts
+function requestIteratorCleanup(iterator) {
+	try {
+		const cleanup = iterator.return?.();
+		if (cleanup) cleanup.catch(() => void 0);
+	} catch {}
+}
 function initialSnapshot() {
 	return {
 		data: [],
@@ -395,6 +472,7 @@ function createAgentStreamController(options = {}) {
 				resolveCancellation = resolve;
 			});
 			cancelActiveConnect = resolveCancellation;
+			let protocolTerminal = false;
 			try {
 				try {
 					while (!cancelled && !destroyed) {
@@ -403,7 +481,12 @@ function createAgentStreamController(options = {}) {
 							result
 						})), cancellation.then(() => ({ kind: "cancelled" }))]);
 						if (outcome.kind === "cancelled" || outcome.result.done) break;
-						push(outcome.result.value);
+						const event = outcome.result.value;
+						push(event);
+						if (event.type === "done" || event.type === "error") {
+							protocolTerminal = true;
+							break;
+						}
 					}
 					if (!transportDone && !cancelled && !destroyed) push({ type: "done" });
 				} catch (error) {
@@ -413,7 +496,7 @@ function createAgentStreamController(options = {}) {
 						type: "error"
 					});
 				} finally {
-					if (cancelled || destroyed) iterator.return?.();
+					if (protocolTerminal || cancelled || destroyed) requestIteratorCleanup(iterator);
 				}
 				return await waitForTerminal();
 			} finally {
@@ -444,20 +527,41 @@ function createAgentEventChannel() {
 	let ended = false;
 	let failure;
 	let wake;
+	function stopConsuming() {
+		ended = true;
+		queue.length = 0;
+		notify();
+	}
 	async function* iterate() {
-		while (!ended || queue.length > 0) {
-			if (failure) throw failure;
-			const event = queue.shift();
-			if (event) {
-				yield event;
-				continue;
+		try {
+			while (!ended || queue.length > 0) {
+				if (failure) throw failure;
+				const event = queue.shift();
+				if (event) {
+					yield event;
+					continue;
+				}
+				await new Promise((resolve) => {
+					wake = resolve;
+				});
+				wake = void 0;
 			}
-			await new Promise((resolve) => {
-				wake = resolve;
-			});
-			wake = void 0;
+			if (failure) throw failure;
+		} finally {
+			stopConsuming();
 		}
-		if (failure) throw failure;
+	}
+	function createIterator() {
+		const iterator = iterate();
+		return {
+			next() {
+				return iterator.next();
+			},
+			return() {
+				stopConsuming();
+				return iterator.return(void 0);
+			}
+		};
 	}
 	function notify() {
 		wake?.();
@@ -477,7 +581,7 @@ function createAgentEventChannel() {
 			queue.push(event);
 			notify();
 		},
-		source: { [Symbol.asyncIterator]: iterate }
+		source: { [Symbol.asyncIterator]: createIterator }
 	};
 }
 //#endregion
@@ -656,6 +760,68 @@ function createStreamingMarkdownParser(options = {}) {
 }
 //#endregion
 //#region src/markdown-view.ts
+function richTextChildren(children) {
+	const result = [];
+	for (const child of children ?? []) {
+		const normalized = richTextNode(child);
+		if (!normalized) return;
+		result.push(normalized);
+	}
+	return result;
+}
+function richTextNode(node) {
+	if (node.kind === "text") return {
+		text: node.text ?? "",
+		type: "text"
+	};
+	if (node.kind === "hardbreak") return { name: "br" };
+	if (node.kind === "checkbox") return {
+		text: node.checked ? "☑ " : "☐ ",
+		type: "text"
+	};
+	const name = {
+		emphasis: "em",
+		strikethrough: "del",
+		strong: "strong",
+		subscript: "sub",
+		superscript: "sup"
+	}[node.kind];
+	if (name) {
+		const children = richTextChildren(node.children);
+		return children ? {
+			children,
+			name
+		} : void 0;
+	}
+	if (node.kind === "highlight" || node.kind === "insert") {
+		const children = richTextChildren(node.children);
+		if (!children) return;
+		return {
+			attrs: { style: node.kind === "highlight" ? "background:rgba(250,146,0,.18);border-radius:2px;" : "text-decoration:underline;" },
+			children,
+			name: "span"
+		};
+	}
+	if (node.kind === "inline-code") return {
+		attrs: { style: "font-family:monospace;padding:1px 4px;border-radius:4px;background:rgba(127,127,127,.12);" },
+		children: [{
+			text: node.text ?? "",
+			type: "text"
+		}],
+		name: "code"
+	};
+	if (node.kind === "math-inline") return {
+		attrs: { style: "font-family:monospace;" },
+		children: [{
+			text: node.text ?? "",
+			type: "text"
+		}],
+		name: "span"
+	};
+}
+function toAgentRichTextNodes(nodes) {
+	return richTextChildren(nodes);
+}
 function stringValue(value, fallback = "") {
 	return typeof value === "string" ? value : fallback;
 }
@@ -672,7 +838,7 @@ function nodeField(node, key) {
 	return key in node ? Reflect.get(node, key) : void 0;
 }
 function normalizeUnknownNode(value) {
-	if (!isUnknownRecord(value) || typeof value.type !== "string" || typeof value.raw !== "string") return void 0;
+	if (!isUnknownRecord(value) || typeof value.type !== "string" || typeof value.raw !== "string") return;
 	return normalizeMarkdownNode(value);
 }
 function normalizeChildren(value) {
@@ -917,19 +1083,40 @@ function createAgentSseEventSource() {
 	const dataLines = [];
 	let textBuffer = "";
 	let ended = false;
+	function clearBufferedInput() {
+		dataLines.length = 0;
+		textBuffer = "";
+	}
+	function closeSource() {
+		if (ended) return;
+		ended = true;
+		clearBufferedInput();
+		channel.end();
+	}
+	function failSource(error) {
+		if (ended) return;
+		ended = true;
+		clearBufferedInput();
+		channel.fail(error);
+	}
+	function finishWith(event) {
+		channel.push(event);
+		closeSource();
+	}
 	function dispatch() {
 		if (dataLines.length === 0 || ended) return;
 		const payload = dataLines.join("\n");
 		dataLines.length = 0;
 		if (payload === "[DONE]") {
-			channel.push({ type: "done" });
+			finishWith({ type: "done" });
 			return;
 		}
 		try {
-			channel.push(parseAgentStreamEvent(JSON.parse(payload)));
+			const event = parseAgentStreamEvent(JSON.parse(payload));
+			if (event.type === "done" || event.type === "error") finishWith(event);
+			else channel.push(event);
 		} catch (error) {
-			ended = true;
-			channel.fail(error);
+			failSource(error);
 		}
 	}
 	function processLine(line) {
@@ -949,10 +1136,12 @@ function createAgentSseEventSource() {
 		}
 	}
 	function appendText(value) {
+		if (ended) return;
 		textBuffer += value;
 		let lineEnd = textBuffer.indexOf("\n");
 		while (lineEnd >= 0) {
 			processLine(textBuffer.slice(0, lineEnd));
+			if (ended) return;
 			textBuffer = textBuffer.slice(lineEnd + 1);
 			lineEnd = textBuffer.indexOf("\n");
 		}
@@ -961,15 +1150,13 @@ function createAgentSseEventSource() {
 		end() {
 			if (ended) return;
 			appendText(decoder.decode(/* @__PURE__ */ new Uint8Array(0), true));
+			if (ended) return;
 			if (textBuffer.length > 0) processLine(textBuffer);
 			dispatch();
-			ended = true;
-			channel.end();
+			closeSource();
 		},
 		fail(error) {
-			if (ended) return;
-			ended = true;
-			channel.fail(error);
+			failSource(error);
 		},
 		feed(chunk) {
 			if (ended) return;
@@ -979,6 +1166,113 @@ function createAgentSseEventSource() {
 	};
 }
 //#endregion
-export { createAgentEventChannel, createAgentSseEventSource, createAgentStreamController, createStreamingMarkdownParser, createTextStream, normalizeMarkdownNode, normalizeMarkdownNodes, parseAgentStreamEvent };
+//#region src/thread.ts
+const DESTROYED_ERROR = "Agent thread controller is destroyed";
+function assertVersionId(id) {
+	if (id.length === 0) throw new Error("Agent thread version id must not be empty");
+}
+function freezeVersion(version) {
+	return Object.freeze({ ...version });
+}
+function freezeSnapshot(versions, activeVersionId) {
+	if (activeVersionId === void 0) return Object.freeze({ versions });
+	return Object.freeze({
+		activeVersionId,
+		versions
+	});
+}
+function validateSnapshot(initial) {
+	const sourceVersions = initial?.versions ?? [];
+	const versionsById = /* @__PURE__ */ new Map();
+	for (const version of sourceVersions) {
+		assertVersionId(version.id);
+		if (versionsById.has(version.id)) throw new Error(`Agent thread version "${version.id}" is duplicated`);
+		versionsById.set(version.id, version);
+	}
+	for (const version of sourceVersions) if (version.parentId !== void 0 && !versionsById.has(version.parentId)) throw new Error(`Agent thread parent version "${version.parentId}" does not exist`);
+	const resolvedIds = /* @__PURE__ */ new Set();
+	for (const version of sourceVersions) {
+		if (resolvedIds.has(version.id)) continue;
+		const path = [];
+		const pathIds = /* @__PURE__ */ new Set();
+		let current = version;
+		while (current !== void 0 && !resolvedIds.has(current.id)) {
+			if (pathIds.has(current.id)) throw new Error(`Agent thread version "${current.id}" has a cyclic parent chain`);
+			path.push(current.id);
+			pathIds.add(current.id);
+			current = current.parentId === void 0 ? void 0 : versionsById.get(current.parentId);
+		}
+		path.forEach((id) => resolvedIds.add(id));
+	}
+	const activeVersionId = initial?.activeVersionId;
+	if (activeVersionId !== void 0 && !versionsById.has(activeVersionId)) throw new Error(`Agent thread active version "${activeVersionId}" does not exist`);
+	return freezeSnapshot(Object.freeze(sourceVersions.map(freezeVersion)), activeVersionId);
+}
+function createAgentThreadController(initial) {
+	const listeners = /* @__PURE__ */ new Set();
+	let snapshot = validateSnapshot(initial);
+	let destroyed = false;
+	function assertActive() {
+		if (destroyed) throw new Error(DESTROYED_ERROR);
+	}
+	function findVersion(id) {
+		return snapshot.versions.find((version) => version.id === id);
+	}
+	function replace(nextSnapshot) {
+		snapshot = nextSnapshot;
+		listeners.forEach((listener) => listener());
+	}
+	function append(version) {
+		assertActive();
+		assertVersionId(version.id);
+		if (findVersion(version.id)) throw new Error(`Agent thread version "${version.id}" already exists`);
+		if (version.parentId !== void 0 && !findVersion(version.parentId)) throw new Error(`Agent thread parent version "${version.parentId}" does not exist`);
+		replace(freezeSnapshot(Object.freeze([...snapshot.versions, freezeVersion(version)]), version.id));
+	}
+	function fork(parentId, version) {
+		assertActive();
+		assertVersionId(version.id);
+		if (findVersion(version.id)) throw new Error(`Agent thread version "${version.id}" already exists`);
+		if (!findVersion(parentId)) throw new Error(`Agent thread parent version "${parentId}" does not exist`);
+		const forkedVersion = freezeVersion({
+			...version,
+			parentId
+		});
+		replace(freezeSnapshot(Object.freeze([...snapshot.versions, forkedVersion]), forkedVersion.id));
+	}
+	function select(id) {
+		assertActive();
+		if (!findVersion(id)) throw new Error(`Agent thread version "${id}" does not exist`);
+		replace(freezeSnapshot(snapshot.versions, id));
+	}
+	function reset(nextInitial) {
+		assertActive();
+		replace(validateSnapshot(nextInitial));
+	}
+	function subscribe(listener) {
+		assertActive();
+		listeners.add(listener);
+		return () => {
+			listeners.delete(listener);
+		};
+	}
+	return {
+		append,
+		destroy() {
+			if (destroyed) return;
+			destroyed = true;
+			listeners.clear();
+		},
+		fork,
+		getSnapshot() {
+			return snapshot;
+		},
+		reset,
+		select,
+		subscribe
+	};
+}
+//#endregion
+export { createAgentEventChannel, createAgentSseEventSource, createAgentStreamController, createAgentThreadController, createStreamingMarkdownParser, createTextStream, normalizeMarkdownNode, normalizeMarkdownNodes, parseAgentStreamEvent, toAgentRichTextNodes };
 
 //# sourceMappingURL=index.mjs.map