@bugabinga/pi-ext-cost 0.1.0

package/CHANGELOG.md

@@ -0,0 +1,12 @@
+# Changelog
+
+## 0.1.0 - 2026-05-21
+
+- a40a427 prepare extensions for npm release
+- 6a97080 add asciinema demo workflow
+- 517ba54 pi: add extension footer fallbacks
+- 133cb7d chore(pi): migrate extensions to earendil packages
+- 5ca1296 Rework Pi agent extensions
+- b87a61a feat(pi): monorepo workspace — all extensions are proper packages
+- b408bba pi(ext): footer segment producers (context, timing, cost, runtime, git-status, pr-status, thinking)
+

package/README.md

@@ -0,0 +1,11 @@
+# cost
+
+Pi model, token, and cost telemetry.
+
+Shows model id, last-turn tokens, session cost, and monthly cost. Uses `footer:segment` with `@bugabinga/pi-ext-footer`; otherwise falls back to Pi status text.
+
+## Demo
+
+<!-- demo:footer_suite:start -->
+![Footer suite](assets/footer_suite.gif)
+<!-- demo:footer_suite:end -->

package/index.ts

@@ -0,0 +1,390 @@
+/**
+ * Cost, token, and model footer segments.
+ *
+ * Emits into the shared footer "llm" zone:
+ * - model: current model id
+ * - tokens: last assistant turn token usage
+ * - cost: turn/session/monthly spend when non-zero
+ */
+
+import type { ExtensionAPI, ExtensionContext, ThemeColor } from "@earendil-works/pi-coding-agent";
+import { parseSessionEntries } from "@earendil-works/pi-coding-agent";
+import { readdirSync, readFileSync, type Dirent } from "node:fs";
+import { dirname, join } from "node:path";
+
+// ── Constants ─────────────────────────────────────────────────────────────
+
+const FOOTER_KEY = Symbol.for("@bugabinga/pi-ext-footer");
+const STATUS_ID = "cost";
+const PRICE_UNIT_TOKENS = 1_000_000;
+const COUNT_DECIMAL_THRESHOLD = 1_000;
+const COUNT_ROUND_THRESHOLD = 10_000;
+const LOW_COST_THRESHOLD = 0.01;
+const MEDIUM_COST_THRESHOLD = 10;
+const SESSION_EXT = ".jsonl";
+const SESSION_MONTH_RE = /^(\d{4})-(\d{2})/;
+
+const FOOTER = {
+	model: { id: "model", color: "syntaxKeyword", order: 0 },
+	tokens: { id: "tokens", color: "muted", order: 2 },
+	cost: { id: "cost", color: "success", order: 3 },
+	zone: "llm",
+} as const satisfies Record<string, unknown>;
+
+const UPDATE_EVENTS = ["agent_start", "agent_end", "tool_execution_end"] as const;
+const RESET_SIGNATURE_EVENTS = ["model_select", "session_compact"] as const;
+
+// ── Types ─────────────────────────────────────────────────────────────────
+
+type PriceRow = {
+	input: number;
+	cacheRead: number;
+	output: number;
+};
+
+type TokenUsage = {
+	input: number;
+	output: number;
+	cacheRead: number;
+	cacheWrite: number;
+};
+
+type UsageWithCost = Partial<TokenUsage> & {
+	cost?: { total?: number };
+};
+
+type AssistantLikeMessage = {
+	role?: unknown;
+	usage?: UsageWithCost;
+	provider?: string;
+	model?: string;
+};
+
+type Branch = ReturnType<ExtensionContext["sessionManager"]["getBranch"]>;
+
+type Snapshot = {
+	modelLabel: string;
+	usage: TokenUsage;
+	turnCost: number;
+	sessionCost: number;
+	monthlyCost: number;
+};
+
+type SegmentId = "model" | "tokens" | "cost";
+
+// ── Footer capability ─────────────────────────────────────────────────────
+
+function hasFooter(): boolean {
+	const footer = (globalThis as any)[FOOTER_KEY];
+	return footer?.loaded === true && footer.version >= 1;
+}
+
+// ── Pricing ───────────────────────────────────────────────────────────────
+
+// Prices per 1M tokens. Used only when providers report $0 cost.
+const PRICING: Record<string, Record<string, PriceRow>> = {
+	zai: {
+		"glm-5.1": { input: 1.4, cacheRead: 0.26, output: 4.4 },
+		"glm-5": { input: 1.0, cacheRead: 0.2, output: 4.0 },
+		"glm-5-turbo": { input: 0.5, cacheRead: 0.1, output: 2.0 },
+		"glm-4.7": { input: 1.0, cacheRead: 0.2, output: 4.0 },
+		"glm-4.7-flash": { input: 0.1, cacheRead: 0.02, output: 0.4 },
+		"glm-4.7-flashx": { input: 0.1, cacheRead: 0.02, output: 0.4 },
+		"glm-4.6": { input: 1.0, cacheRead: 0.2, output: 4.0 },
+		"glm-4.5": { input: 1.0, cacheRead: 0.2, output: 4.0 },
+		"glm-4.5-air": { input: 0.1, cacheRead: 0.02, output: 0.4 },
+		"glm-4.5-flash": { input: 0.1, cacheRead: 0.02, output: 0.4 },
+	},
+	minimax: {
+		"MiniMax-M2.7": { input: 1.0, cacheRead: 0.06, output: 4.0 },
+		"MiniMax-M2.7-highspeed": { input: 0.6, cacheRead: 0.06, output: 2.4 },
+		"MiniMax-M1": { input: 1.0, cacheRead: 0.06, output: 4.0 },
+	},
+};
+
+function priceFor(provider: string, model: string): PriceRow | undefined {
+	const providerPrices = PRICING[provider];
+	if (!providerPrices) return undefined;
+
+	return providerPrices[model]
+		?? Object.entries(providerPrices).find(([prefix]) => model.startsWith(prefix))?.[1];
+}
+
+function manualTokenCost(provider: string, model: string, usage: TokenUsage): number {
+	const price = priceFor(provider, model);
+	if (!price) return 0;
+
+	return (
+		usage.input * price.input
+		+ usage.cacheRead * price.cacheRead
+		+ usage.output * price.output
+	) / PRICE_UNIT_TOKENS;
+}
+
+// ── Message extraction ────────────────────────────────────────────────────
+
+function isAssistantMessageEntry(entry: unknown): entry is { type: "message"; message: AssistantLikeMessage } {
+	const candidate = entry as { type?: unknown; message?: AssistantLikeMessage } | undefined;
+	return candidate?.type === "message" && candidate.message?.role === "assistant";
+}
+
+function normalizeUsage(usage: UsageWithCost | undefined): TokenUsage {
+	return {
+		input: usage?.input ?? 0,
+		output: usage?.output ?? 0,
+		cacheRead: usage?.cacheRead ?? 0,
+		cacheWrite: usage?.cacheWrite ?? 0,
+	};
+}
+
+function messageCost(message: AssistantLikeMessage): number {
+	const usage = normalizeUsage(message.usage);
+	const reported = message.usage?.cost?.total ?? 0;
+	if (reported > 0) return reported;
+
+	return manualTokenCost(message.provider ?? "", message.model ?? "", usage);
+}
+
+function sumMessageCosts(entries: readonly unknown[]): number {
+	let total = 0;
+	for (const entry of entries) {
+		if (isAssistantMessageEntry(entry)) total += messageCost(entry.message);
+	}
+	return total;
+}
+
+function lastAssistantUsage(branch: Branch): TokenUsage {
+	for (let i = branch.length - 1; i >= 0; i--) {
+		const entry = branch[i];
+		if (!isAssistantMessageEntry(entry)) continue;
+		if (!entry.message.usage) continue;
+		return normalizeUsage(entry.message.usage);
+	}
+	return { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
+}
+
+// ── Session file scan ─────────────────────────────────────────────────────
+
+function safeReadDir(path: string): Dirent[] {
+	try {
+		return readdirSync(path, { withFileTypes: true });
+	} catch {
+		return [];
+	}
+}
+
+function currentMonthKey(date = new Date()): string {
+	return `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, "0")}`;
+}
+
+function sessionFileMonth(name: string): string | undefined {
+	const match = name.match(SESSION_MONTH_RE);
+	return match ? `${match[1]}-${match[2]}` : undefined;
+}
+
+function isSessionFileForMonth(file: Dirent, month: string): boolean {
+	return file.isFile() && file.name.endsWith(SESSION_EXT) && sessionFileMonth(file.name) === month;
+}
+
+function sessionFilesForMonth(sessionDir: string, month: string): string[] {
+	const root = dirname(sessionDir);
+	const files: string[] = [];
+
+	for (const entry of safeReadDir(root)) {
+		if (entry.isFile() && isSessionFileForMonth(entry, month)) {
+			files.push(join(root, entry.name));
+			continue;
+		}
+
+		if (!entry.isDirectory()) continue;
+		const childDir = join(root, entry.name);
+		for (const file of safeReadDir(childDir)) {
+			if (isSessionFileForMonth(file, month)) files.push(join(childDir, file.name));
+		}
+	}
+
+	return files;
+}
+
+function sumFileCosts(filePath: string): number {
+	try {
+		return sumMessageCosts(parseSessionEntries(readFileSync(filePath, "utf-8")));
+	} catch {
+		return 0;
+	}
+}
+
+async function calculateMonthlyCost(sessionDir: string): Promise<number> {
+	let total = 0;
+	for (const file of sessionFilesForMonth(sessionDir, currentMonthKey())) {
+		total += sumFileCosts(file);
+	}
+	return total;
+}
+
+// ── Formatting ────────────────────────────────────────────────────────────
+
+function formatCount(value: number): string {
+	if (value < COUNT_DECIMAL_THRESHOLD) return `${value}`;
+	if (value < COUNT_ROUND_THRESHOLD) return `${(value / COUNT_DECIMAL_THRESHOLD).toFixed(1)}k`;
+	return `${Math.round(value / COUNT_DECIMAL_THRESHOLD)}k`;
+}
+
+function formatCost(value: number): string {
+	if (value < LOW_COST_THRESHOLD) return `$${value.toFixed(4)}`;
+	if (value < MEDIUM_COST_THRESHOLD) return `$${value.toFixed(2)}`;
+	return `$${value.toFixed(1)}`;
+}
+
+function modelLabel(ctx: ExtensionContext): string {
+	const modelId = ctx.model?.id ?? "no-model";
+	return modelId.split("/").pop() ?? modelId;
+}
+
+function tokenLabel(usage: TokenUsage): string {
+	const totalPrompt = usage.input + usage.cacheRead;
+	if (totalPrompt === 0 && usage.output === 0) return "↑0";
+	if (usage.input === 0 && usage.cacheRead > 0) return `◐ ${formatCount(usage.cacheRead)}`;
+	if (usage.cacheRead === 0) return `↑${formatCount(usage.input)} ↓${formatCount(usage.output)}`;
+
+	const cachePercent = Math.round((usage.cacheRead / totalPrompt) * 100);
+	return `◐${formatCount(usage.input)} ↓${formatCount(usage.output)} (${cachePercent}% cached)`;
+}
+
+function costLabel(snapshot: Snapshot): string | undefined {
+	const parts: string[] = [];
+	if (snapshot.turnCost > 0) parts.push(formatCost(snapshot.turnCost));
+	if (snapshot.sessionCost > 0) parts.push(`Σ${formatCost(snapshot.sessionCost)}`);
+	if (snapshot.monthlyCost > snapshot.sessionCost) parts.push(` ${formatCost(snapshot.monthlyCost)}`);
+	return parts.length > 0 ? parts.join(" · ") : undefined;
+}
+
+// ── Footer emission ───────────────────────────────────────────────────────
+
+function emitSegment(pi: ExtensionAPI, id: SegmentId, text: string | undefined) {
+	if (text === undefined) {
+		pi.events.emit("footer:segment", { id: FOOTER[id].id, text: undefined });
+		return;
+	}
+
+	pi.events.emit("footer:segment", {
+		id: FOOTER[id].id,
+		text,
+		color: FOOTER[id].color as ThemeColor,
+		zone: FOOTER.zone,
+		order: FOOTER[id].order,
+	});
+}
+
+function statusLabel(snapshot: Snapshot): string {
+	return [snapshot.modelLabel, tokenLabel(snapshot.usage), costLabel(snapshot)]
+		.filter((part): part is string => !!part)
+		.join(" · ");
+}
+
+function publishSnapshot(pi: ExtensionAPI, ctx: ExtensionContext, snapshot: Snapshot) {
+	if (hasFooter()) {
+		ctx.ui.setStatus(STATUS_ID, undefined);
+		emitSegment(pi, "model", snapshot.modelLabel);
+		emitSegment(pi, "tokens", tokenLabel(snapshot.usage));
+		emitSegment(pi, "cost", costLabel(snapshot));
+		return;
+	}
+
+	clearSegments(pi);
+	ctx.ui.setStatus(STATUS_ID, statusLabel(snapshot));
+}
+
+function clearSegments(pi: ExtensionAPI) {
+	emitSegment(pi, "model", undefined);
+	emitSegment(pi, "tokens", undefined);
+	emitSegment(pi, "cost", undefined);
+}
+
+function clearPublished(pi: ExtensionAPI, ctx?: ExtensionContext) {
+	clearSegments(pi);
+	if (ctx?.hasUI) ctx.ui.setStatus(STATUS_ID, undefined);
+}
+
+// ── Snapshot ──────────────────────────────────────────────────────────────
+
+function buildSnapshot(ctx: ExtensionContext, previousSessionCost: number, monthlyCost: number): Snapshot {
+	const branch = ctx.sessionManager.getBranch();
+	const sessionCost = sumMessageCosts(branch);
+
+	return {
+		modelLabel: modelLabel(ctx),
+		usage: lastAssistantUsage(branch),
+		turnCost: Math.max(0, sessionCost - previousSessionCost),
+		sessionCost,
+		monthlyCost,
+	};
+}
+
+function signature(snapshot: Snapshot): string {
+	const usage = snapshot.usage;
+	return [
+		snapshot.modelLabel,
+		usage.input,
+		usage.output,
+		usage.cacheRead,
+		usage.cacheWrite,
+		snapshot.sessionCost,
+		snapshot.monthlyCost,
+	].join("|");
+}
+
+// ── Extension ─────────────────────────────────────────────────────────────
+
+export default function costExtension(pi: ExtensionAPI) {
+	let previousSessionCost = 0;
+	let monthlyCost = 0;
+	let lastSignature = "";
+	let hasUI = false;
+
+	async function sync(ctx: ExtensionContext) {
+		if (!hasUI) return;
+
+		const snapshot = buildSnapshot(ctx, previousSessionCost, monthlyCost);
+		previousSessionCost = snapshot.sessionCost;
+
+		const nextSignature = signature(snapshot);
+		if (nextSignature === lastSignature) return;
+		lastSignature = nextSignature;
+
+		publishSnapshot(pi, ctx, snapshot);
+	}
+
+	async function syncBestEffort(ctx: ExtensionContext) {
+		try {
+			await sync(ctx);
+		} catch {
+			// Footer telemetry must never disturb the agent loop.
+		}
+	}
+
+	pi.on("session_start", async (_event, ctx) => {
+		hasUI = ctx.hasUI;
+		if (!hasUI) return;
+
+		previousSessionCost = 0;
+		lastSignature = "";
+		monthlyCost = await calculateMonthlyCost(ctx.sessionManager.getSessionDir());
+		await syncBestEffort(ctx);
+	});
+
+	for (const eventName of UPDATE_EVENTS) {
+		pi.on(eventName, async (_event, ctx) => syncBestEffort(ctx));
+	}
+
+	for (const eventName of RESET_SIGNATURE_EVENTS) {
+		pi.on(eventName, async (_event, ctx) => {
+			lastSignature = "";
+			await syncBestEffort(ctx);
+		});
+	}
+
+	pi.on("session_shutdown", async (_event, ctx) => {
+		clearPublished(pi, ctx);
+	});
+}

package/package.json

@@ -0,0 +1,15 @@
+{
+  "name": "@bugabinga/pi-ext-cost",
+  "version": "0.1.0",
+  "type": "module",
+  "main": "index.ts",
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "license": "MIT",
+  "description": "Model, token, and spend telemetry for Pi.",
+  "keywords": [
+    "pi",
+    "pi-extension"
+  ]
+}