@narumitw/pi-usage 0.60.7 → 0.60.9

package/docs/providers.md

@@ -54,7 +54,8 @@ The extension queries the fixed usage endpoint only when both the selected model
 Custom and proxy origins fail before network access, redirects are rejected, and the credential is never sent to an override from Kimi Code's environment-specific development path.
 
 Plan buckets remain integer request counts and are rendered with their source-defined windows.
-Unknown units, duplicate windows, missing counts, invalid timestamps, and malformed rows remain unavailable rather than receiving guessed semantics.
+Live responses omit both `used` and `remaining` on untouched rows and may report only `remaining` once usage starts, so a row with a positive `limit` and no counters means zero usage, and when only one counter is present the other is derived from `limit`.
+Counters that are present but malformed, unknown units, duplicate windows, invalid timestamps, and malformed rows remain unavailable rather than receiving guessed semantics.
 Booster-wallet `amount` and `amountLeft` values use Kimi's first-party conversion of 1,000,000 fixed-point units per cent, while monthly values already arrive in cents.
 Wallet values retain their currency and stay separate from plan requests and percentages in reports and the statusline.
 Wallet fields remain unavailable unless the response supplies one consistent currency; missing monthly values are omitted, and an enabled zero cap is shown as zero.

package/package.json

@@ -1,76 +1,75 @@
 {
-	"name": "@narumitw/pi-usage",
-	"version": "0.60.7",
-	"description": "Pi extension that shows current-account usage and DeepSeek API balance for supported providers.",
-	"type": "module",
-	"license": "MIT",
-	"private": false,
-	"keywords": [
-		"pi-package",
-		"pi-extension",
-		"pi",
-		"usage",
-		"quota",
-		"balance",
-		"deepseek",
-		"fireworks",
-		"baseten",
-		"vercel-ai-gateway",
-		"codex",
-		"kimi",
-		"kimi-coding",
-		"moonshot",
-		"moonshotai",
-		"minimax",
-		"token-plan",
-		"copilot",
-		"openrouter",
-		"opencode",
-		"zen",
-		"zai",
-		"glm",
-		"xai",
-		"grok"
-	],
-	"files": [
-		"src",
-		"dist",
-		"docs",
-		"README.md",
-		"LICENSE"
-	],
-	"pi": {
-		"extensions": [
-			"./dist/index.ts"
-		]
-	},
-	"scripts": {
-		"build": "node scripts/build-runtime.mjs",
-		"check": "npm run build && biome check --vcs-use-ignore-file=false src test scripts package.json tsconfig.json README.md && npm run typecheck",
-		"format": "biome check --write --vcs-use-ignore-file=false src test scripts package.json tsconfig.json README.md",
-		"typecheck": "tsc --noEmit",
-		"prepack": "npm run build"
-	},
-	"peerDependencies": {
-		"@earendil-works/pi-ai": "*",
-		"@earendil-works/pi-coding-agent": "*",
-		"@earendil-works/pi-tui": "*"
-	},
-	"devDependencies": {
-		"@biomejs/biome": "2.5.11",
-		"@earendil-works/pi-ai": "0.85.0",
-		"@earendil-works/pi-coding-agent": "0.85.0",
-		"@earendil-works/pi-tui": "0.85.0",
-		"@types/node": "26.4.1",
-		"esbuild": "0.28.2",
-		"typescript": "7.0.2"
-	},
-	"repository": {
-		"type": "git",
-		"url": "https://github.com/narumiruna/pi-extensions",
-		"directory": "packages/pi-usage"
-	},
-	"dependencies": {
-		"@narumitw/pi-tui-kit": "^0.61.0"
-	}
+  "name": "@narumitw/pi-usage",
+  "version": "0.60.9",
+  "description": "Pi extension that shows current-account usage and DeepSeek API balance for supported providers.",
+  "type": "module",
+  "license": "MIT",
+  "private": false,
+  "keywords": [
+    "pi-package",
+    "pi-extension",
+    "pi",
+    "usage",
+    "quota",
+    "balance",
+    "deepseek",
+    "fireworks",
+    "baseten",
+    "vercel-ai-gateway",
+    "codex",
+    "kimi",
+    "kimi-coding",
+    "moonshot",
+    "moonshotai",
+    "minimax",
+    "token-plan",
+    "copilot",
+    "openrouter",
+    "opencode",
+    "zen",
+    "zai",
+    "glm",
+    "xai",
+    "grok"
+  ],
+  "files": [
+    "src",
+    "dist",
+    "docs",
+    "README.md",
+    "LICENSE"
+  ],
+  "pi": {
+    "extensions": [
+      "./dist/index.ts"
+    ]
+  },
+  "scripts": {
+    "build": "node scripts/build-runtime.mjs",
+    "check": "npm run build && biome check --vcs-use-ignore-file=false src test scripts package.json tsconfig.json README.md && npm run typecheck",
+    "format": "biome check --write --vcs-use-ignore-file=false src test scripts package.json tsconfig.json README.md",
+    "typecheck": "tsc --noEmit",
+    "prepack": "npm run build"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-ai": "*",
+    "@earendil-works/pi-coding-agent": "*",
+    "@earendil-works/pi-tui": "*"
+  },
+  "devDependencies": {
+    "@earendil-works/pi-ai": "0.86.0",
+    "@earendil-works/pi-coding-agent": "0.86.0",
+    "@earendil-works/pi-tui": "0.86.0",
+    "@types/node": "26.6.2",
+    "esbuild": "0.28.2",
+    "typescript": "7.0.2"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/narumiruna/pi-extensions",
+    "directory": "packages/pi-usage"
+  },
+  "dependencies": {
+    "@narumitw/pi-tui-kit": "^0.61.0"
+  }
 }

package/src/codex-fast-runtime.ts

@@ -1,14 +1,10 @@
-import type {
-	ExtensionAPI,
-	ExtensionCommandContext,
-	ExtensionContext,
-} from "@earendil-works/pi-coding-agent";
+import type { ExtensionAPI, ExtensionCommandContext, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import {
-	codexFastAvailability,
-	codexFastIsEffective,
-	codexFastStatusLabel,
-	correctCodexFastMessageCost,
-	rewriteCodexFastPayload,
+  codexFastAvailability,
+  codexFastIsEffective,
+  codexFastStatusLabel,
+  correctCodexFastMessageCost,
+  rewriteCodexFastPayload,
 } from "./codex-fast.js";
 import { errorMessage } from "./core.js";
 import { isStaleExtensionContextError } from "./query.js";
@@ -21,200 +17,168 @@ type PendingFastRequest = { fastRequested: boolean; model: PiModel };
 export const FAST_USAGE_WARNING = "Fast is about 1.5× faster and uses more of your plan allowance.";
 
 export function registerCodexFastMode(
-	pi: ExtensionAPI,
-	settingsRuntime: UsageSettingsRuntime,
-	refreshStatus: (ctx: ExtensionContext) => void,
-	options: { registerSessionStart?: boolean } = {},
+  pi: ExtensionAPI,
+  settingsRuntime: UsageSettingsRuntime,
+  refreshStatus: (ctx: ExtensionContext) => void,
+  options: { registerSessionStart?: boolean } = {},
 ) {
-	let sessionController = new AbortController();
-	let generation = 0;
-	const pendingFastRequests = new Map<string, PendingFastRequest>();
+  let sessionController = new AbortController();
+  let generation = 0;
+  const pendingFastRequests = new Map<string, PendingFastRequest>();
 
-	const toggle = async (
-		ctx: ExtensionCommandContext,
-		enabled: boolean,
-		callerSignal?: AbortSignal,
-	): Promise<boolean> => {
-		const ownerGeneration = generation;
-		const sessionId = ctx.sessionManager.getSessionId();
-		const signal = callerSignal
-			? AbortSignal.any([callerSignal, sessionController.signal])
-			: sessionController.signal;
-		try {
-			await settingsRuntime.update({ codexFastMode: enabled }, signal);
-		} catch (error) {
-			if (isAbortError(error) || isStaleExtensionContextError(error)) return false;
-			ctx.ui.notify(`Could not save pi-usage.json: ${errorMessage(error)}`, "error");
-			return false;
-		}
-		if (
-			signal.aborted ||
-			ownerGeneration !== generation ||
-			ctx.sessionManager.getSessionId() !== sessionId
-		) {
-			return false;
-		}
-		refreshStatus(ctx);
-		ctx.ui.notify(
-			enabled
-				? `Codex Fast mode enabled. ${FAST_USAGE_WARNING}`
-				: "Codex Fast mode disabled; standard routing will be used.",
-			"info",
-		);
-		return true;
-	};
+  const toggle = async (
+    ctx: ExtensionCommandContext,
+    enabled: boolean,
+    callerSignal?: AbortSignal,
+  ): Promise<boolean> => {
+    const ownerGeneration = generation;
+    const sessionId = ctx.sessionManager.getSessionId();
+    const signal = callerSignal ? AbortSignal.any([callerSignal, sessionController.signal]) : sessionController.signal;
+    try {
+      await settingsRuntime.update({ codexFastMode: enabled }, signal);
+    } catch (error) {
+      if (isAbortError(error) || isStaleExtensionContextError(error)) return false;
+      ctx.ui.notify(`Could not save pi-usage.json: ${errorMessage(error)}`, "error");
+      return false;
+    }
+    if (signal.aborted || ownerGeneration !== generation || ctx.sessionManager.getSessionId() !== sessionId) {
+      return false;
+    }
+    refreshStatus(ctx);
+    ctx.ui.notify(
+      enabled
+        ? `Codex Fast mode enabled. ${FAST_USAGE_WARNING}`
+        : "Codex Fast mode disabled; standard routing will be used.",
+      "info",
+    );
+    return true;
+  };
 
-	pi.registerCommand("fast", {
-		description: "Toggle Codex Fast mode",
-		handler: async (args, ctx) => {
-			if (args.trim()) {
-				if (!ctx.hasUI) throw new Error("/fast does not accept arguments.");
-				ctx.ui.notify("/fast does not accept arguments.", "warning");
-				return;
-			}
-			if (!ctx.hasUI) throw new Error("/fast requires TUI or RPC mode.");
-			const availability = codexFastAvailability(
-				ctx.model,
-				settingsRuntime.get().settings.codexFastMode,
-			);
-			if (availability.kind === "not-codex") {
-				ctx.ui.notify("/fast is available only for the active OpenAI Codex model.", "warning");
-				return;
-			}
-			if (availability.kind === "unavailable") {
-				ctx.ui.notify(availability.reason, "warning");
-				return;
-			}
-			if (settingsRuntime.get().kind === "invalid") {
-				ctx.ui.notify(
-					"pi-usage.json is invalid; repair it and run /reload before changing Fast mode.",
-					"error",
-				);
-				return;
-			}
-			await toggle(ctx, !availability.enabled);
-		},
-	});
+  pi.registerCommand("fast", {
+    description: "Toggle Codex Fast mode",
+    handler: async (args, ctx) => {
+      if (args.trim()) {
+        if (!ctx.hasUI) throw new Error("/fast does not accept arguments.");
+        ctx.ui.notify("/fast does not accept arguments.", "warning");
+        return;
+      }
+      if (!ctx.hasUI) throw new Error("/fast requires TUI or RPC mode.");
+      const availability = codexFastAvailability(ctx.model, settingsRuntime.get().settings.codexFastMode);
+      if (availability.kind === "not-codex") {
+        ctx.ui.notify("/fast is available only for the active OpenAI Codex model.", "warning");
+        return;
+      }
+      if (availability.kind === "unavailable") {
+        ctx.ui.notify(availability.reason, "warning");
+        return;
+      }
+      if (settingsRuntime.get().kind === "invalid") {
+        ctx.ui.notify("pi-usage.json is invalid; repair it and run /reload before changing Fast mode.", "error");
+        return;
+      }
+      await toggle(ctx, !availability.enabled);
+    },
+  });
 
-	const prepareSession = (ctx: ExtensionContext): Promise<void> => {
-		const sessionId = ctx.sessionManager.getSessionId();
-		generation += 1;
-		sessionController.abort();
-		pendingFastRequests.clear();
-		sessionController = new AbortController();
-		const ownerGeneration = generation;
-		return (async () => {
-			let state: Readonly<UsageSettingsState>;
-			try {
-				state = await settingsRuntime.reload(sessionController.signal);
-			} catch (error) {
-				if (sessionController.signal.aborted || ownerGeneration !== generation) return;
-				if (ctx.hasUI) {
-					ctx.ui.notify(
-						`Could not load pi-usage.json; using defaults. ${errorMessage(error)}`,
-						"warning",
-					);
-				}
-				return;
-			}
-			if (
-				sessionController.signal.aborted ||
-				ownerGeneration !== generation ||
-				ctx.sessionManager.getSessionId() !== sessionId
-			) {
-				return;
-			}
-			if (ctx.hasUI && state.kind === "invalid") {
-				ctx.ui.notify(
-					`Invalid pi-usage.json; using defaults without overwriting it. ${state.issue}`,
-					"warning",
-				);
-			}
-			refreshStatus(ctx);
-		})();
-	};
+  const prepareSession = (ctx: ExtensionContext): Promise<void> => {
+    const sessionId = ctx.sessionManager.getSessionId();
+    generation += 1;
+    sessionController.abort();
+    pendingFastRequests.clear();
+    sessionController = new AbortController();
+    const ownerGeneration = generation;
+    return (async () => {
+      let state: Readonly<UsageSettingsState>;
+      try {
+        state = await settingsRuntime.reload(sessionController.signal);
+      } catch (error) {
+        if (sessionController.signal.aborted || ownerGeneration !== generation) return;
+        if (ctx.hasUI) {
+          ctx.ui.notify(`Could not load pi-usage.json; using defaults. ${errorMessage(error)}`, "warning");
+        }
+        return;
+      }
+      if (
+        sessionController.signal.aborted ||
+        ownerGeneration !== generation ||
+        ctx.sessionManager.getSessionId() !== sessionId
+      ) {
+        return;
+      }
+      if (ctx.hasUI && state.kind === "invalid") {
+        ctx.ui.notify(`Invalid pi-usage.json; using defaults without overwriting it. ${state.issue}`, "warning");
+      }
+      refreshStatus(ctx);
+    })();
+  };
 
-	if (options.registerSessionStart !== false) {
-		pi.on("session_start", async (_event, ctx) => prepareSession(ctx));
-	}
+  if (options.registerSessionStart !== false) {
+    pi.on("session_start", async (_event, ctx) => prepareSession(ctx));
+  }
 
-	pi.on("before_provider_request", (event, ctx) => {
-		const rewritten = rewriteCodexFastPayload(
-			event.payload,
-			ctx.model,
-			settingsRuntime.get().settings.codexFastMode,
-		);
-		const key = activeRequestKey(ctx);
-		if (key && ctx.model) {
-			pendingFastRequests.set(key, {
-				fastRequested: isRecord(rewritten) && rewritten.service_tier === "priority",
-				model: ctx.model,
-			});
-		}
-		return rewritten;
-	});
-	pi.on("message_end", (event, ctx) => {
-		const request = consumeFastRequest(ctx, event.message, pendingFastRequests);
-		if (request === NO_FAST_REQUEST) return undefined;
-		const message = correctCodexFastMessageCost(
-			event.message,
-			request.model,
-			request.fastRequested,
-		);
-		return message ? { message: message as never } : undefined;
-	});
-	pi.on("session_shutdown", async () => {
-		generation += 1;
-		sessionController.abort();
-		pendingFastRequests.clear();
-		await settingsRuntime.flush();
-	});
+  pi.on("before_provider_request", (event, ctx) => {
+    const rewritten = rewriteCodexFastPayload(event.payload, ctx.model, settingsRuntime.get().settings.codexFastMode);
+    const key = activeRequestKey(ctx);
+    if (key && ctx.model) {
+      pendingFastRequests.set(key, {
+        fastRequested: isRecord(rewritten) && rewritten.service_tier === "priority",
+        model: ctx.model,
+      });
+    }
+    return rewritten;
+  });
+  pi.on("message_end", (event, ctx) => {
+    const request = consumeFastRequest(ctx, event.message, pendingFastRequests);
+    if (request === NO_FAST_REQUEST) return undefined;
+    const message = correctCodexFastMessageCost(event.message, request.model, request.fastRequested);
+    return message ? { message: message as never } : undefined;
+  });
+  pi.on("session_shutdown", async () => {
+    generation += 1;
+    sessionController.abort();
+    pendingFastRequests.clear();
+    await settingsRuntime.flush();
+  });
 
-	return {
-		prepareSession,
-		availability(model: PiModel | undefined) {
-			return codexFastAvailability(model, settingsRuntime.get().settings.codexFastMode);
-		},
-		decorateStatus(model: PiModel | undefined, status: string) {
-			return codexFastStatusLabel(
-				status,
-				codexFastIsEffective(model, settingsRuntime.get().settings.codexFastMode),
-			);
-		},
-		toggle,
-	};
+  return {
+    prepareSession,
+    availability(model: PiModel | undefined) {
+      return codexFastAvailability(model, settingsRuntime.get().settings.codexFastMode);
+    },
+    decorateStatus(model: PiModel | undefined, status: string) {
+      return codexFastStatusLabel(status, codexFastIsEffective(model, settingsRuntime.get().settings.codexFastMode));
+    },
+    toggle,
+  };
 }
 
 function activeRequestKey(ctx: ExtensionContext): string | undefined {
-	const model = ctx.model;
-	return model ? `${ctx.sessionManager.getSessionId()}:${model.provider}/${model.id}` : undefined;
+  const model = ctx.model;
+  return model ? `${ctx.sessionManager.getSessionId()}:${model.provider}/${model.id}` : undefined;
 }
 
 function consumeFastRequest(
-	ctx: ExtensionContext,
-	message: unknown,
-	pending: Map<string, PendingFastRequest>,
+  ctx: ExtensionContext,
+  message: unknown,
+  pending: Map<string, PendingFastRequest>,
 ): PendingFastRequest | typeof NO_FAST_REQUEST {
-	if (!isRecord(message) || message.role !== "assistant") return NO_FAST_REQUEST;
-	const key = messageRequestKey(ctx, message);
-	if (!key) return NO_FAST_REQUEST;
-	const request = pending.get(key);
-	pending.delete(key);
-	return request ?? NO_FAST_REQUEST;
+  if (!isRecord(message) || message.role !== "assistant") return NO_FAST_REQUEST;
+  const key = messageRequestKey(ctx, message);
+  if (!key) return NO_FAST_REQUEST;
+  const request = pending.get(key);
+  pending.delete(key);
+  return request ?? NO_FAST_REQUEST;
 }
 
-function messageRequestKey(
-	ctx: ExtensionContext,
-	message: Record<string, unknown>,
-): string | undefined {
-	if (typeof message.provider !== "string" || typeof message.model !== "string") return undefined;
-	return `${ctx.sessionManager.getSessionId()}:${message.provider}/${message.model}`;
+function messageRequestKey(ctx: ExtensionContext, message: Record<string, unknown>): string | undefined {
+  if (typeof message.provider !== "string" || typeof message.model !== "string") return undefined;
+  return `${ctx.sessionManager.getSessionId()}:${message.provider}/${message.model}`;
 }
 
 function isRecord(value: unknown): value is Record<string, unknown> {
-	return typeof value === "object" && value !== null && !Array.isArray(value);
+  return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 
 function isAbortError(error: unknown): boolean {
-	return error instanceof Error && error.name === "AbortError";
+  return error instanceof Error && error.name === "AbortError";
 }