@oh-my-pi/pi-coding-agent 13.5.8 → 13.6.1

package/src/mcp/types.ts

@@ -244,6 +244,14 @@ export interface MCPServerConnection {
 	tools?: MCPToolDefinition[];
 	/** Source metadata (for display) */
 	_source?: SourceMeta;
+	/** Cached resources (populated on demand) */
+	resources?: MCPResource[];
+	/** Cached resource templates (populated on demand) */
+	resourceTemplates?: MCPResourceTemplate[];
+	/** Server instructions from initialize */
+	instructions?: string;
+	/** Cached prompts (populated on demand) */
+	prompts?: MCPPrompt[];
 }
 
 /** MCP tool with server context */
@@ -251,3 +259,135 @@ export interface MCPToolWithServer {
 	server: MCPServerConnection;
 	tool: MCPToolDefinition;
 }
+
+// =============================================================================
+// MCP Resource Types
+// =============================================================================
+
+/** Annotations for resources, templates, and content blocks */
+export interface MCPAnnotations {
+	audience?: ("user" | "assistant")[];
+	priority?: number;
+	lastModified?: string;
+}
+
+/** A concrete resource exposed by an MCP server */
+export interface MCPResource {
+	uri: string;
+	name: string;
+	title?: string;
+	description?: string;
+	mimeType?: string;
+	size?: number;
+	annotations?: MCPAnnotations;
+}
+
+/** A parameterized resource template (RFC 6570 URI template) */
+export interface MCPResourceTemplate {
+	uriTemplate: string;
+	name: string;
+	title?: string;
+	description?: string;
+	mimeType?: string;
+	annotations?: MCPAnnotations;
+}
+
+/** Result of resources/list */
+export interface MCPResourcesListResult {
+	resources: MCPResource[];
+	nextCursor?: string;
+}
+
+/** Result of resources/templates/list */
+export interface MCPResourceTemplatesListResult {
+	resourceTemplates: MCPResourceTemplate[];
+	nextCursor?: string;
+}
+
+/** A single content item from resources/read */
+export interface MCPResourceContentItem {
+	uri: string;
+	mimeType?: string;
+	text?: string;
+	blob?: string;
+}
+
+/** Result of resources/read */
+export interface MCPResourceReadResult {
+	contents: MCPResourceContentItem[];
+}
+
+/** Params for resources/read */
+export interface MCPResourceReadParams {
+	uri: string;
+}
+
+/** Params for resources/subscribe and resources/unsubscribe */
+export interface MCPResourceSubscribeParams {
+	uri: string;
+}
+
+// =============================================================================
+// MCP Prompt Types
+// =============================================================================
+
+/** An argument definition for an MCP prompt */
+export interface MCPPromptArgument {
+	name: string;
+	description?: string;
+	required?: boolean;
+}
+
+/** A prompt definition exposed by an MCP server */
+export interface MCPPrompt {
+	name: string;
+	title?: string;
+	description?: string;
+	arguments?: MCPPromptArgument[];
+}
+
+/** Result of prompts/list */
+export interface MCPPromptsListResult {
+	prompts: MCPPrompt[];
+	nextCursor?: string;
+}
+
+/** Audio content in prompt messages */
+export interface MCPAudioContent {
+	type: "audio";
+	data: string;
+	mimeType: string;
+}
+
+/** Content type union for prompt messages */
+export type MCPPromptContent = MCPTextContent | MCPImageContent | MCPAudioContent | MCPResourceContent;
+
+/** A single message in a prompt result */
+export interface MCPPromptMessage {
+	role: "user" | "assistant";
+	content: MCPPromptContent | MCPPromptContent[];
+}
+
+/** Params for prompts/get */
+export interface MCPGetPromptParams {
+	name: string;
+	arguments?: Record<string, string>;
+}
+
+/** Result of prompts/get */
+export interface MCPGetPromptResult {
+	description?: string;
+	messages: MCPPromptMessage[];
+}
+
+// =============================================================================
+// MCP Notification Method Names
+// =============================================================================
+
+/** MCP server notification method names */
+export const MCPNotificationMethods = {
+	TOOLS_LIST_CHANGED: "notifications/tools/list_changed",
+	RESOURCES_LIST_CHANGED: "notifications/resources/list_changed",
+	RESOURCES_UPDATED: "notifications/resources/updated",
+	PROMPTS_LIST_CHANGED: "notifications/prompts/list_changed",
+} as const;

package/src/modes/components/footer.ts

@@ -131,6 +131,7 @@ export class FooterComponent implements Component {
 		let totalCacheRead = 0;
 		let totalCacheWrite = 0;
 		let totalCost = 0;
+		let totalPremiumRequests = 0;
 
 		for (const entry of this.session.sessionManager.getEntries()) {
 			if (entry.type === "message" && entry.message.role === "assistant") {
@@ -139,6 +140,7 @@ export class FooterComponent implements Component {
 				totalCacheRead += entry.message.usage.cacheRead;
 				totalCacheWrite += entry.message.usage.cacheWrite;
 				totalCost += entry.message.usage.cost.total;
+				totalPremiumRequests += entry.message.usage.premiumRequests ?? 0;
 			}
 		}
 
@@ -177,11 +179,15 @@ export class FooterComponent implements Component {
 		if (totalCacheRead) statsParts.push(`R${formatNumber(totalCacheRead)}`);
 		if (totalCacheWrite) statsParts.push(`W${formatNumber(totalCacheWrite)}`);
 
-		// Show cost with "(sub)" indicator if using OAuth subscription
+		// Show billing summary with subscription and premium-request indicators
 		const usingSubscription = state.model ? this.session.modelRegistry.isUsingOAuth(state.model) : false;
-		if (totalCost || usingSubscription) {
-			const costStr = `$${totalCost.toFixed(3)}${usingSubscription ? " (sub)" : ""}`;
-			statsParts.push(costStr);
+		const normalizedPremiumRequests = Math.round((totalPremiumRequests + Number.EPSILON) * 100) / 100;
+		if (totalCost || usingSubscription || normalizedPremiumRequests) {
+			const billingParts: string[] = [];
+			if (totalCost) billingParts.push(`$${totalCost.toFixed(3)}`);
+			if (normalizedPremiumRequests) billingParts.push(`★ ${formatNumber(normalizedPremiumRequests)}`);
+			if (usingSubscription) billingParts.push("(sub)");
+			if (billingParts.length > 0) statsParts.push(billingParts.join(" "));
 		}
 
 		// Colorize context percentage based on usage

package/src/modes/components/settings-defs.ts

@@ -97,7 +97,16 @@ const OPTION_PROVIDERS: Partial<Record<SettingPath, OptionProvider>> = {
 	"task.isolation.mode": [
 		{ value: "none", label: "None", description: "No isolation" },
 		{ value: "worktree", label: "Worktree", description: "Git worktree isolation" },
-		{ value: "fuse-overlay", label: "Fuse Overlay", description: "COW overlay via fuse-overlayfs" },
+		{
+			value: "fuse-overlay",
+			label: "Fuse Overlay",
+			description: "COW overlay via fuse-overlayfs (Unix only)",
+		},
+		{
+			value: "fuse-projfs",
+			label: "Fuse ProjFS",
+			description: "COW overlay via ProjFS (Windows only; falls back to worktree if unavailable)",
+		},
 	],
 	// Task isolation merge strategy
 	"task.isolation.merge": [
@@ -206,6 +215,11 @@ const OPTION_PROVIDERS: Partial<Record<SettingPath, OptionProvider>> = {
 		{ value: "openai", label: "OpenAI", description: "api.kimi.com" },
 		{ value: "anthropic", label: "Anthropic", description: "api.moonshot.ai" },
 	],
+	"providers.openaiWebsockets": [
+		{ value: "auto", label: "Auto", description: "Use model/provider default websocket behavior" },
+		{ value: "off", label: "Off", description: "Disable websockets for OpenAI Codex models" },
+		{ value: "on", label: "On", description: "Force websockets for OpenAI Codex models" },
+	],
 	// Default thinking level
 	defaultThinkingLevel: [
 		{ value: "off", label: "off", description: "No reasoning" },

package/src/modes/components/status-line/git-utils.ts

@@ -0,0 +1,42 @@
+/**
+ * Extract "owner/repo" from a GitHub remote URL.
+ * Handles HTTPS, SSH (scp-style), and git:// protocols.
+ *
+ * @returns "owner/repo" or null if the URL isn't a recognized GitHub remote.
+ */
+export function parseGitHubRepo(remoteUrl: string): string | null {
+	const match = remoteUrl.match(/github\.com[:/]([^/]+\/[^/]+)/);
+	if (!match) return null;
+	return match[1].replace(/\.git$/, "");
+}
+
+/**
+ * Extract the branch name from a remote HEAD ref like "origin/main".
+ * Returns the portion after the first "/" or the whole string if no "/" is present.
+ */
+export function parseDefaultBranch(ref: string): string {
+	const slash = ref.indexOf("/");
+	return slash >= 0 ? ref.slice(slash + 1) : ref;
+}
+
+export interface PrCacheContext {
+	branch: string;
+	repoId: string | null;
+}
+
+export function createPrCacheContext(branch: string, repoId: string | null): PrCacheContext {
+	return { branch, repoId };
+}
+
+export function isSamePrCacheContext(a: PrCacheContext | undefined, b: PrCacheContext | undefined): boolean {
+	if (!a || !b) return false;
+	return a.branch === b.branch && a.repoId === b.repoId;
+}
+
+export function canReuseCachedPr(
+	cachedPr: { number: number; url: string } | null | undefined,
+	cachedContext: PrCacheContext | undefined,
+	currentContext: PrCacheContext | null,
+): boolean {
+	return cachedPr !== undefined && currentContext !== null && isSamePrCacheContext(cachedContext, currentContext);
+}

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

@@ -3,7 +3,7 @@ import type { PresetDef, StatusLinePreset } from "./types";
 export const STATUS_LINE_PRESETS: Record<StatusLinePreset, PresetDef> = {
 	default: {
 		// Matches current behavior
-		leftSegments: ["pi", "model", "plan_mode", "path", "git", "context_pct", "token_total", "cost"],
+		leftSegments: ["pi", "model", "plan_mode", "path", "git", "pr", "context_pct", "token_total", "cost"],
 		rightSegments: [],
 		separator: "powerline-thin",
 		segmentOptions: {
@@ -24,7 +24,7 @@ export const STATUS_LINE_PRESETS: Record<StatusLinePreset, PresetDef> = {
 	},
 
 	compact: {
-		leftSegments: ["model", "plan_mode", "git"],
+		leftSegments: ["model", "plan_mode", "git", "pr"],
 		rightSegments: ["cost", "context_pct"],
 		separator: "powerline-thin",
 		segmentOptions: {
@@ -34,7 +34,7 @@ export const STATUS_LINE_PRESETS: Record<StatusLinePreset, PresetDef> = {
 	},
 
 	full: {
-		leftSegments: ["pi", "hostname", "model", "plan_mode", "path", "git", "subagents"],
+		leftSegments: ["pi", "hostname", "model", "plan_mode", "path", "git", "pr", "subagents"],
 		rightSegments: ["token_in", "token_out", "cache_read", "cost", "context_pct", "time_spent", "time"],
 		separator: "powerline",
 		segmentOptions: {
@@ -47,7 +47,7 @@ export const STATUS_LINE_PRESETS: Record<StatusLinePreset, PresetDef> = {
 
 	nerd: {
 		// Full preset with all Nerd Font icons
-		leftSegments: ["pi", "hostname", "model", "plan_mode", "path", "git", "session", "subagents"],
+		leftSegments: ["pi", "hostname", "model", "plan_mode", "path", "git", "pr", "session", "subagents"],
 		rightSegments: [
 			"token_in",
 			"token_out",
@@ -70,7 +70,7 @@ export const STATUS_LINE_PRESETS: Record<StatusLinePreset, PresetDef> = {
 
 	ascii: {
 		// No Nerd Font dependencies
-		leftSegments: ["model", "plan_mode", "path", "git"],
+		leftSegments: ["model", "plan_mode", "path", "git", "pr"],
 		rightSegments: ["token_total", "cost", "context_pct"],
 		separator: "ascii",
 		segmentOptions: {
@@ -82,7 +82,7 @@ export const STATUS_LINE_PRESETS: Record<StatusLinePreset, PresetDef> = {
 
 	custom: {
 		// User-defined - these are just defaults that get overridden
-		leftSegments: ["model", "plan_mode", "path", "git"],
+		leftSegments: ["model", "plan_mode", "path", "git", "pr"],
 		rightSegments: ["token_total", "cost", "context_pct"],
 		separator: "powerline-thin",
 		segmentOptions: {},

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

@@ -1,4 +1,5 @@
 import * as os from "node:os";
+import { TERMINAL } from "@oh-my-pi/pi-tui";
 import { formatDuration, formatNumber, getProjectDir } from "@oh-my-pi/pi-utils";
 import { theme } from "../../../modes/theme/theme";
 import { shortenPath } from "../../../tools/render-utils";
@@ -14,6 +15,10 @@ function withIcon(icon: string, text: string): string {
 	return icon ? `${icon} ${text}` : text;
 }
 
+function normalizePremiumRequests(value: number): number {
+	return Math.round((value + Number.EPSILON) * 100) / 100;
+}
+
 // ═══════════════════════════════════════════════════════════════════════════
 // Segment Implementations
 // ═══════════════════════════════════════════════════════════════════════════
@@ -141,6 +146,18 @@ const gitSegment: StatusLineSegment = {
 	},
 };
 
+const prSegment: StatusLineSegment = {
+	id: "pr",
+	render(ctx) {
+		const { pr } = ctx.git;
+		if (!pr) return { content: "", visible: false };
+
+		const label = withIcon(theme.icon.pr, `#${pr.number}`);
+		const content = TERMINAL.hyperlinks ? `\x1b]8;;${pr.url}\x07${label}\x1b]8;;\x07` : label;
+		return { content: theme.fg("accent", content), visible: true };
+	},
+};
+
 const subagentsSegment: StatusLineSegment = {
 	id: "subagents",
 	render(ctx) {
@@ -189,16 +206,21 @@ const tokenTotalSegment: StatusLineSegment = {
 const costSegment: StatusLineSegment = {
 	id: "cost",
 	render(ctx) {
-		const { cost } = ctx.usageStats;
+		const { cost, premiumRequests } = ctx.usageStats;
+		const normalizedPremiumRequests = normalizePremiumRequests(premiumRequests);
 		const state = ctx.session.state;
 		const usingSubscription = state.model ? ctx.session.modelRegistry.isUsingOAuth(state.model) : false;
 
-		if (!cost && !usingSubscription) {
+		if (!cost && !usingSubscription && !normalizedPremiumRequests) {
 			return { content: "", visible: false };
 		}
 
-		const costDisplay = usingSubscription ? "(sub)" : `$${cost.toFixed(2)}`;
-		return { content: theme.fg("statusLineCost", costDisplay), visible: true };
+		const billingParts: string[] = [];
+		if (cost) billingParts.push(`$${cost.toFixed(2)}`);
+		if (normalizedPremiumRequests) billingParts.push(`★ ${formatNumber(normalizedPremiumRequests)}`);
+		if (usingSubscription) billingParts.push("(sub)");
+
+		return { content: theme.fg("statusLineCost", billingParts.join(" ")), visible: true };
 	},
 };
 
@@ -324,6 +346,7 @@ export const SEGMENTS: Record<StatusLineSegmentId, StatusLineSegment> = {
 	plan_mode: planModeSegment,
 	path: pathSegment,
 	git: gitSegment,
+	pr: prSegment,
 	subagents: subagentsSegment,
 	token_in: tokenInSegment,
 	token_out: tokenOutSegment,

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

@@ -30,6 +30,7 @@ export interface SegmentContext {
 		output: number;
 		cacheRead: number;
 		cacheWrite: number;
+		premiumRequests: number;
 		cost: number;
 	};
 	contextPercent: number;
@@ -40,6 +41,7 @@ export interface SegmentContext {
 	git: {
 		branch: string | null;
 		status: { staged: number; unstaged: number; untracked: number } | null;
+		pr: { number: number; url: string } | null;
 	};
 }
 

package/src/modes/components/status-line-segment-editor.ts

@@ -20,6 +20,7 @@ const SEGMENT_INFO: Record<StatusLineSegmentId, { label: string; short: string }
 	plan_mode: { label: "Plan Mode", short: "plan status" },
 	path: { label: "Path", short: "working dir" },
 	git: { label: "Git", short: "branch/status" },
+	pr: { label: "PR", short: "pull request" },
 	subagents: { label: "Agents", short: "subagent count" },
 	token_in: { label: "Tokens In", short: "input tokens" },
 	token_out: { label: "Tokens Out", short: "output tokens" },

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

@@ -8,6 +8,13 @@ import type { StatusLinePreset, StatusLineSegmentId, StatusLineSeparatorStyle }
 import { theme } from "../../modes/theme/theme";
 import type { AgentSession } from "../../session/agent-session";
 import { findGitHeadPathSync, sanitizeStatusText } from "../shared";
+import {
+	canReuseCachedPr,
+	createPrCacheContext,
+	isSamePrCacheContext,
+	type PrCacheContext,
+	parseDefaultBranch,
+} from "./status-line/git-utils";
 import { getPreset } from "./status-line/presets";
 import { renderSegment, type SegmentContext } from "./status-line/segments";
 import { getSeparator } from "./status-line/separators";
@@ -39,6 +46,7 @@ export interface StatusLineSettings {
 export class StatusLineComponent implements Component {
 	#settings: StatusLineSettings = {};
 	#cachedBranch: string | null | undefined = undefined;
+	#cachedBranchRepoId: string | null | undefined = undefined;
 	#gitWatcher: fs.FSWatcher | null = null;
 	#onBranchChange: (() => void) | null = null;
 	#autoCompactEnabled: boolean = true;
@@ -52,6 +60,12 @@ export class StatusLineComponent implements Component {
 	#gitStatusLastFetch = 0;
 	#gitStatusInFlight = false;
 
+	// PR lookup caching (invalidated on branch/repo context changes)
+	#cachedPr: { number: number; url: string } | null | undefined = undefined;
+	#cachedPrContext: PrCacheContext | undefined = undefined;
+	#prLookupInFlight = false;
+	#defaultBranch?: string;
+
 	constructor(private readonly session: AgentSession) {
 		this.#settings = {
 			preset: settings.get("statusLine.preset"),
@@ -107,13 +121,13 @@ export class StatusLineComponent implements Component {
 
 		try {
 			this.#gitWatcher = fs.watch(gitHeadPath, () => {
-				this.#cachedBranch = undefined;
+				this.#invalidateGitCaches();
 				if (this.#onBranchChange) {
 					this.#onBranchChange();
 				}
 			});
 		} catch {
-			// Silently fail
+			this.#invalidateGitCaches();
 		}
 	}
 
@@ -125,19 +139,27 @@ export class StatusLineComponent implements Component {
 	}
 
 	invalidate(): void {
-		this.#cachedBranch = undefined;
+		this.#invalidateGitCaches();
 	}
 
+	#invalidateGitCaches(): void {
+		this.#cachedBranch = undefined;
+		this.#cachedBranchRepoId = undefined;
+		this.#cachedPr = undefined;
+		this.#cachedPrContext = undefined;
+	}
 	#getCurrentBranch(): string | null {
-		if (this.#cachedBranch !== undefined) {
+		const gitHeadPath = findGitHeadPathSync();
+		if (this.#cachedBranch !== undefined && this.#cachedBranchRepoId === gitHeadPath) {
 			return this.#cachedBranch;
 		}
 
-		const gitHeadPath = findGitHeadPathSync();
+		this.#cachedBranchRepoId = gitHeadPath;
 		if (!gitHeadPath) {
 			this.#cachedBranch = null;
 			return null;
 		}
+
 		try {
 			const content = fs.readFileSync(gitHeadPath, "utf8").trim();
 
@@ -153,6 +175,26 @@ export class StatusLineComponent implements Component {
 		return this.#cachedBranch ?? null;
 	}
 
+	#isDefaultBranch(branch: string): boolean {
+		if (this.#defaultBranch === undefined) {
+			// Kick off async resolution, use hardcoded fallback until it resolves
+			this.#defaultBranch = "main";
+			(async () => {
+				// Try origin/HEAD first, fall back to upstream/HEAD
+				const origin = await $`git rev-parse --abbrev-ref origin/HEAD`.quiet().nothrow();
+				if (origin.exitCode === 0) {
+					this.#defaultBranch = parseDefaultBranch(origin.stdout.toString().trim());
+					return;
+				}
+				const upstream = await $`git rev-parse --abbrev-ref upstream/HEAD`.quiet().nothrow();
+				if (upstream.exitCode === 0) {
+					this.#defaultBranch = parseDefaultBranch(upstream.stdout.toString().trim());
+				}
+			})();
+		}
+		return branch === this.#defaultBranch;
+	}
+
 	#getGitStatus(): { staged: number; unstaged: number; untracked: number } | null {
 		if (this.#gitStatusInFlight || Date.now() - this.#gitStatusLastFetch < 1000) {
 			return this.#cachedGitStatus;
@@ -207,6 +249,66 @@ export class StatusLineComponent implements Component {
 		return this.#cachedGitStatus;
 	}
 
+	#lookupPr(): { number: number; url: string } | null {
+		const branch = this.#getCurrentBranch();
+		const currentContext = branch ? createPrCacheContext(branch, this.#cachedBranchRepoId ?? null) : null;
+
+		if (canReuseCachedPr(this.#cachedPr, this.#cachedPrContext, currentContext)) {
+			return this.#cachedPr ?? null;
+		}
+
+		if (this.#cachedPr !== undefined) {
+			this.#cachedPr = undefined;
+			this.#cachedPrContext = undefined;
+		}
+
+		// Don't look up if no branch, detached HEAD, default branch, or already in flight
+		if (!branch || branch === "detached" || this.#isDefaultBranch(branch) || this.#prLookupInFlight) {
+			return null;
+		}
+
+		this.#prLookupInFlight = true;
+		const lookupContext = currentContext;
+
+		// Fire async lookup, return null until resolved
+		(async () => {
+			// Helper: only write cache if branch/repo context hasn't changed since launch
+			const setCachedPr = (value: { number: number; url: string } | null) => {
+				const latestBranch = this.#getCurrentBranch();
+				const latestContext = latestBranch
+					? createPrCacheContext(latestBranch, this.#cachedBranchRepoId ?? null)
+					: undefined;
+				if (lookupContext && isSamePrCacheContext(latestContext, lookupContext)) {
+					this.#cachedPr = value;
+					this.#cachedPrContext = lookupContext;
+				}
+			};
+			try {
+				// Requires `gh repo set-default` to be configured; fails gracefully if not
+				const result = await $`gh pr view --json number,url`.quiet().nothrow();
+				if (result.exitCode !== 0) {
+					setCachedPr(null);
+					return;
+				}
+				const pr = JSON.parse(result.stdout.toString()) as { number: number; url: string };
+				if (typeof pr.number === "number") {
+					setCachedPr({ number: pr.number, url: pr.url });
+				} else {
+					setCachedPr(null);
+				}
+			} catch {
+				setCachedPr(null);
+			} finally {
+				this.#prLookupInFlight = false;
+				if (this.#cachedPr && this.#onBranchChange) {
+					this.#onBranchChange();
+				}
+			}
+		})();
+
+		return null;
+	}
+
 	#buildSegmentContext(width: number): SegmentContext {
 		const state = this.session.state;
 
@@ -216,6 +318,7 @@ export class StatusLineComponent implements Component {
 			output: 0,
 			cacheRead: 0,
 			cacheWrite: 0,
+			premiumRequests: 0,
 			cost: 0,
 		};
 
@@ -248,6 +351,7 @@ export class StatusLineComponent implements Component {
 			git: {
 				branch: this.#getCurrentBranch(),
 				status: this.#getGitStatus(),
+				pr: this.#lookupPr(),
 			},
 		};
 	}

package/src/modes/controllers/command-controller.ts

@@ -231,6 +231,11 @@ export class CommandController {
 
 	async handleSessionCommand(): Promise<void> {
 		const stats = this.ctx.session.getSessionStats();
+		const premiumRequests =
+			"premiumRequests" in stats && typeof stats.premiumRequests === "number"
+				? stats.premiumRequests
+				: this.ctx.session.sessionManager.getUsageStatistics().premiumRequests;
+		const normalizedPremiumRequests = Math.round((premiumRequests + Number.EPSILON) * 100) / 100;
 
 		let info = `${theme.bold("Session Info")}\n\n`;
 		info += `${theme.fg("dim", "File:")} ${stats.sessionFile ?? "In-memory"}\n`;
@@ -271,9 +276,14 @@ export class CommandController {
 		}
 		info += `${theme.fg("dim", "Total:")} ${stats.tokens.total.toLocaleString()}\n`;
 
-		if (stats.cost > 0) {
+		if (stats.cost > 0 || normalizedPremiumRequests > 0) {
 			info += `\n${theme.bold("Cost")}\n`;
-			info += `${theme.fg("dim", "Total:")} ${stats.cost.toFixed(4)}\n`;
+			if (stats.cost > 0) {
+				info += `${theme.fg("dim", "Total:")} ${stats.cost.toFixed(4)}\n`;
+			}
+			if (normalizedPremiumRequests > 0) {
+				info += `${theme.fg("dim", "Premium Requests:")} ${normalizedPremiumRequests.toLocaleString()}\n`;
+			}
 		}
 
 		const gateway = await getGatewayStatus();