@nmzpy/pi-ember-stack 0.1.6 → 0.2.2

package/plugins/pi-ember-tps/index.ts

@@ -0,0 +1,144 @@
+/**
+ * Ember TPS Meter — minimal tokens-per-second tracker
+ *
+ * Tracks output plus thinking token rate during streaming and exposes the live
+ * value via getLiveTps() for the custom footer to render.
+ */
+
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+
+const STREAM_INTERVAL_MS = 500;
+
+let streamStartMs = 0;
+let firstTokenMs = 0;
+let streamChars = 0;
+let streamThinkingChars = 0;
+let streamTokens = 0;
+let tickTimer: ReturnType<typeof setInterval> | null = null;
+let streaming = false;
+let liveTps = 0;
+let renderTrigger: (() => void) | undefined;
+
+function now(): number {
+	return performance.now();
+}
+
+function tokEst(ch: number): number {
+	return (ch >>> 2) + ((ch & 3) > 0 ? 1 : 0);
+}
+
+function computeTps(): number {
+	const ref = firstTokenMs > 0 ? firstTokenMs : streamStartMs;
+	const elapsed = (now() - ref) / 1000;
+	return elapsed > 0.3 ? streamTokens / elapsed : 0;
+}
+
+function startTick(): void {
+	if (tickTimer) return;
+	tickTimer = setInterval(() => {
+		if (!streaming) {
+			stopTick();
+			return;
+		}
+		liveTps = computeTps();
+		renderTrigger?.();
+	}, STREAM_INTERVAL_MS);
+}
+
+function stopTick(): void {
+	if (tickTimer) {
+		clearInterval(tickTimer);
+		tickTimer = null;
+	}
+}
+
+export function getLiveTps(): number {
+	return liveTps;
+}
+
+export default function piEmberTps(pi: ExtensionAPI): void {
+	pi.on("message_start", async (event, ctx) => {
+		if (event.message.role !== "assistant") return;
+		streamStartMs = now();
+		firstTokenMs = 0;
+		streamChars = 0;
+		streamThinkingChars = 0;
+		streamTokens = 0;
+		liveTps = 0;
+		streaming = true;
+		if (ctx.mode === "tui") {
+			renderTrigger = () => ctx.ui.setStatus("tps", undefined);
+		}
+		startTick();
+	});
+
+	pi.on("message_update", async (event) => {
+		if (event.message.role !== "assistant") return;
+		if (!event.assistantMessageEvent) return;
+		const evt = event.assistantMessageEvent;
+		if (evt.type === "text_delta" || evt.type === "thinking_delta") {
+			const d = evt.delta as string;
+			if (!d) return;
+			if (firstTokenMs === 0) firstTokenMs = now();
+			streamChars += d.length;
+			if (evt.type === "thinking_delta") streamThinkingChars += d.length;
+			streamTokens = tokEst(streamChars);
+		}
+	});
+
+	pi.on("message_end", async (event) => {
+		if (event.message.role !== "assistant") return;
+		streaming = false;
+		stopTick();
+
+		const usage = event.message?.usage;
+		const realOut = usage?.output;
+		// Some providers report output tokens without their thinking tokens.
+		// Keep the streamed combined estimate in that case; providers that
+		// expose `reasoning` already include it in `usage.output`.
+		const tokens = typeof realOut === "number" && realOut > 0
+			? streamThinkingChars > 0 && usage?.reasoning === undefined
+				? Math.max(realOut, streamTokens)
+				: realOut
+			: streamTokens;
+
+		const ref = firstTokenMs > 0 ? firstTokenMs : streamStartMs;
+		const elapsed = (now() - ref) / 1000;
+		if (elapsed < 0.1 || tokens === 0) {
+			liveTps = 0;
+			return;
+		}
+
+		liveTps = tokens / elapsed;
+		renderTrigger?.();
+	});
+
+	pi.on("agent_end", async () => {
+		streaming = false;
+		stopTick();
+	});
+
+	pi.on("session_start", async (_event, ctx) => {
+		streaming = false;
+		stopTick();
+		streamStartMs = 0;
+		firstTokenMs = 0;
+		streamChars = 0;
+		streamThinkingChars = 0;
+		streamTokens = 0;
+		liveTps = 0;
+		renderTrigger = undefined;
+		ctx.ui.setStatus("tps", undefined);
+	});
+
+	// Clear the tick interval and stale render trigger on shutdown so a
+	// subsequent /resume does not keep a setInterval alive against the dead
+	// session's ctx.ui. Without this, the 500ms tick keeps firing and
+	// calls the old renderTrigger long after the session is gone.
+	pi.on("session_shutdown", async () => {
+		streaming = false;
+		stopTick();
+		renderTrigger = undefined;
+		liveTps = 0;
+	});
+}

package/plugins/pi-ember-ui/ember.json

@@ -0,0 +1,99 @@
+{
+	"$schema": "https://raw.githubusercontent.com/earendil-works/pi/main/packages/coding-agent/src/modes/interactive/theme/theme-schema.json",
+	"name": "ember",
+	"vars": {
+		"emberAccent": "#EB6E00",
+		"emberAccent90": "#EB6E00",
+		"emberAccent75": "#D9640F",
+		"emberAccent60": "#C25A17",
+		"emberAccent45": "#A8501F",
+		"emberAccent30": "#8F4627",
+		"emberAccent15": "#783C2F",
+		"emberAccent25": "#4D2E17",
+		"emberGray": "#808080",
+		"emberDarkGray": "#505050",
+		"cyan": "#00d7ff",
+		"blue": "#5f87ff",
+		"green": "#b5bd68",
+		"red": "#cc6666",
+		"yellow": "#ffff00",
+		"text": "#d4d4d4",
+		"gray": "#808080",
+		"dimGray": "#666666",
+		"darkGray": "#505050",
+		"accent": "#EB6E00",
+		"selectedBg": "#3a3a4a",
+		"userMsgBg": "#343541",
+		"subagentBg": "#32333f",
+		"toolPendingBg": "#282832",
+		"toolSuccessBg": "#283228",
+		"toolErrorBg": "#3c2828",
+		"customMsgBg": "#2d2838"
+	},
+	"colors": {
+		"accent": "emberAccent",
+		"border": "emberAccent",
+		"borderAccent": "emberAccent",
+		"borderMuted": "emberDarkGray",
+		"success": "green",
+		"error": "red",
+		"warning": "yellow",
+		"muted": "gray",
+		"dim": "dimGray",
+		"text": "text",
+		"thinkingText": "gray",
+
+		"selectedBg": "selectedBg",
+		"userMessageBg": "userMsgBg",
+		"subagentBg": "subagentBg",
+		"userMessageText": "text",
+		"customMessageBg": "customMsgBg",
+		"customMessageText": "text",
+		"customMessageLabel": "#9575cd",
+		"toolPendingBg": "toolPendingBg",
+		"toolSuccessBg": "toolSuccessBg",
+		"toolErrorBg": "toolErrorBg",
+		"toolTitle": "text",
+		"toolOutput": "gray",
+
+		"mdHeading": "#f0c674",
+		"mdLink": "#81a2be",
+		"mdLinkUrl": "dimGray",
+		"mdCode": "text",
+		"mdCodeBlock": "green",
+		"mdCodeBlockBorder": "gray",
+		"mdQuote": "gray",
+		"mdQuoteBorder": "gray",
+		"mdHr": "gray",
+		"mdListBullet": "emberAccent",
+
+		"toolDiffAdded": "green",
+		"toolDiffRemoved": "red",
+		"toolDiffContext": "gray",
+
+		"syntaxComment": "#6A9955",
+		"syntaxKeyword": "#569CD6",
+		"syntaxFunction": "#DCDCAA",
+		"syntaxVariable": "#9CDCFE",
+		"syntaxString": "#CE9178",
+		"syntaxNumber": "#B5CEA8",
+		"syntaxType": "#4EC9B0",
+		"syntaxOperator": "#D4D4D4",
+		"syntaxPunctuation": "#D4D4D4",
+
+		"thinkingOff": "emberDarkGray",
+		"thinkingMinimal": "emberAccent15",
+		"thinkingLow": "emberAccent30",
+		"thinkingMedium": "emberAccent45",
+		"thinkingHigh": "emberAccent60",
+		"thinkingXhigh": "emberAccent75",
+		"thinkingMax": "emberAccent90",
+
+		"bashMode": "green"
+	},
+	"export": {
+		"pageBg": "#18181e",
+		"cardBg": "#1e1e24",
+		"infoBg": "#3c3728"
+	}
+}