@oh-my-pi/pi-coding-agent 3.13.1337 → 3.14.0

package/CHANGELOG.md

@@ -2,6 +2,15 @@
 
 ## [Unreleased]
 
+## [3.14.0] - 2026-01-04
+### Added
+
+- Added `getUsageStatistics()` method to SessionManager for tracking cumulative token usage and costs across session messages
+
+### Changed
+
+- Changed status line to display usage statistics more efficiently by using centralized session statistics instead of recalculating from entries
+
 ## [3.13.1337] - 2026-01-04
 
 ## [3.9.1337] - 2026-01-04

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-coding-agent",
-	"version": "3.13.1337",
+	"version": "3.14.0",
 	"description": "Coding agent CLI with read, bash, edit, write tools and session management",
 	"type": "module",
 	"ompConfig": {
@@ -39,9 +39,9 @@
 		"prepublishOnly": "bun run generate-template && bun run clean && bun run build"
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-agent-core": "3.13.1337",
-		"@oh-my-pi/pi-ai": "3.13.1337",
-		"@oh-my-pi/pi-tui": "3.13.1337",
+		"@oh-my-pi/pi-agent-core": "3.14.0",
+		"@oh-my-pi/pi-ai": "3.14.0",
+		"@oh-my-pi/pi-tui": "3.14.0",
 		"@sinclair/typebox": "^0.34.46",
 		"ajv": "^8.17.1",
 		"chalk": "^5.5.0",

package/src/core/hooks/index.ts

@@ -13,4 +13,4 @@ export {
 export { execCommand, HookRunner, type HookErrorListener } from "./runner";
 export { wrapToolsWithHooks, wrapToolWithHooks } from "./tool-wrapper";
 export * from "./types";
-export type { ReadonlySessionManager } from "../session-manager";
+export type { UsageStatistics, ReadonlySessionManager } from "../session-manager";

package/src/core/session-manager.ts

@@ -192,6 +192,7 @@ export type ReadonlySessionManager = Pick<
 	| "getHeader"
 	| "getEntries"
 	| "getTree"
+	| "getUsageStatistics"
 >;
 
 /** Generate a unique short ID (8 hex chars, collision-checked) */
@@ -586,6 +587,14 @@ export function getRecentSessions(sessionDir: string, limit = 3): RecentSessionI
  * Use buildSessionContext() to get the resolved message list for the LLM, which
  * handles compaction summaries and follows the path from root to current leaf.
  */
+export interface UsageStatistics {
+	input: number;
+	output: number;
+	cacheRead: number;
+	cacheWrite: number;
+	cost: number;
+}
+
 export class SessionManager {
 	private sessionId: string = "";
 	private sessionTitle: string | undefined;
@@ -598,6 +607,7 @@ export class SessionManager {
 	private byId: Map<string, SessionEntry> = new Map();
 	private labelsById: Map<string, string> = new Map();
 	private leafId: string | null = null;
+	private usageStatistics: UsageStatistics = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0 };
 
 	private constructor(cwd: string, sessionDir: string, sessionFile: string | undefined, persist: boolean) {
 		this.cwd = cwd;
@@ -649,6 +659,7 @@ export class SessionManager {
 		this.byId.clear();
 		this.leafId = null;
 		this.flushed = false;
+		this.usageStatistics = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0 };
 
 		// Only generate filename if persisting and not already set (e.g., via --session flag)
 		if (this.persist && !this.sessionFile) {
@@ -662,6 +673,7 @@ export class SessionManager {
 		this.byId.clear();
 		this.labelsById.clear();
 		this.leafId = null;
+		this.usageStatistics = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, cost: 0 };
 		for (const entry of this.fileEntries) {
 			if (entry.type === "session") continue;
 			this.byId.set(entry.id, entry);
@@ -673,6 +685,14 @@ export class SessionManager {
 					this.labelsById.delete(entry.targetId);
 				}
 			}
+			if (entry.type === "message" && entry.message.role === "assistant") {
+				const usage = entry.message.usage;
+				this.usageStatistics.input += usage.input;
+				this.usageStatistics.output += usage.output;
+				this.usageStatistics.cacheRead += usage.cacheRead;
+				this.usageStatistics.cacheWrite += usage.cacheWrite;
+				this.usageStatistics.cost += usage.cost.total;
+			}
 		}
 	}
 
@@ -690,6 +710,11 @@ export class SessionManager {
 		return this.cwd;
 	}
 
+	/** Get usage statistics across all assistant messages in the session. */
+	getUsageStatistics(): UsageStatistics {
+		return this.usageStatistics;
+	}
+
 	getSessionDir(): string {
 		return this.sessionDir;
 	}
@@ -755,6 +780,14 @@ export class SessionManager {
 		this.byId.set(entry.id, entry);
 		this.leafId = entry.id;
 		this._persist(entry);
+		if (entry.type === "message" && entry.message.role === "assistant") {
+			const usage = entry.message.usage;
+			this.usageStatistics.input += usage.input;
+			this.usageStatistics.output += usage.output;
+			this.usageStatistics.cacheRead += usage.cacheRead;
+			this.usageStatistics.cacheWrite += usage.cacheWrite;
+			this.usageStatistics.cost += usage.cost.total;
+		}
 	}
 
 	/** Append a message as child of current leaf, then advance leaf. Returns entry id.

package/src/modes/interactive/components/status-line.ts

@@ -17,13 +17,14 @@ const THINKING_ICONS: Record<string, string> = {
 
 // Nerd Font icons
 const ICONS = {
-	model: "\uf4bc", //  robot/model
-	folder: "\uf115", //  folder
+	model: "\uec19", //  robot/model
+	folder: "\uf115 ", //  folder
 	branch: "\ue725", //  git branch
 	sep: "\ue0b1", //  powerline thin chevron
 	tokens: "\ue26b", //  coins
 	context: "\ue70f", //  window
 	auto: "\udb80\udc68", //  auto
+	pi: "\ue22c", //  pi
 } as const;
 
 /** Create a colored text segment with background */
@@ -251,23 +252,6 @@ export class StatusLineComponent implements Component {
 	private buildStatusLine(): string {
 		const state = this.session.state;
 
-		// Calculate cumulative usage from ALL session entries
-		let totalInput = 0;
-		let totalOutput = 0;
-		let totalCacheRead = 0;
-		let totalCacheWrite = 0;
-		let totalCost = 0;
-
-		for (const entry of this.session.sessionManager.getEntries()) {
-			if (entry.type === "message" && entry.message.role === "assistant") {
-				totalInput += entry.message.usage.input;
-				totalOutput += entry.message.usage.output;
-				totalCacheRead += entry.message.usage.cacheRead;
-				totalCacheWrite += entry.message.usage.cacheWrite;
-				totalCost += entry.message.usage.cost.total;
-			}
-		}
-
 		// Get context percentage from last assistant message
 		const lastAssistantMessage = state.messages
 			.slice()
@@ -370,14 +354,15 @@ export class StatusLineComponent implements Component {
 		// ═══════════════════════════════════════════════════════════════════════
 		const spendParts: string[] = [];
 
-		const totalTokens = totalInput + totalOutput + totalCacheRead + totalCacheWrite;
+		const { input, output, cacheRead, cacheWrite, cost } = this.session.sessionManager.getUsageStatistics();
+		const totalTokens = input + output + cacheRead + cacheWrite;
 		if (totalTokens) {
 			spendParts.push(`${ICONS.tokens} ${formatTokens(totalTokens)}`);
 		}
 
 		const usingSubscription = state.model ? this.session.modelRegistry.isUsingOAuth(state.model) : false;
-		if (totalCost || usingSubscription) {
-			const costDisplay = `$${totalCost.toFixed(2)}${usingSubscription ? " (sub)" : ""}`;
+		if (cost || usingSubscription) {
+			const costDisplay = `$${cost.toFixed(2)}${usingSubscription ? " (sub)" : ""}`;
 			spendParts.push(costDisplay);
 		}
 
@@ -391,6 +376,10 @@ export class StatusLineComponent implements Component {
 
 		let statusLine = "";
 
+		// Pi segment
+		statusLine += plSegment(`${ICONS.pi} `, theme.getFgAnsi("statusLineContext"), bgAnsi);
+		statusLine += plSep(sepAnsi, bgAnsi);
+
 		// Model segment
 		statusLine += plSegment(modelContent, theme.getFgAnsi("statusLineModel"), bgAnsi);
 		statusLine += plSep(sepAnsi, bgAnsi);