pi-spark 0.19.0 → 0.19.2

package/README.md

@@ -155,7 +155,7 @@ All fields are optional. If the recap model configuration is incomplete, pi-spar
 
 | Field | Value | Description |
 | --- | --- | --- |
-| `idle` | number (ms) or duration string | How long the session must stay idle before a recap is generated. Accepts a millisecond number or a [vercel/ms](https://github.com/vercel/ms) string (e.g., `"5m"`); minimum 5000 ms, defaults to 5 minutes. |
+| `idle` | number (ms) or duration string | How long the session must stay idle before a recap is generated. Accepts a millisecond number or a [parse-duration](https://github.com/jkroso/parse-duration) string (e.g., `"5m"`); minimum 5000 ms, defaults to 5 minutes. |
 | `provider` | string | Provider ID for the recap model. |
 | `model` | string | Model ID for the recap model. |
 | `thinkingLevel` | `ModelThinkingLevel` | Thinking level for the recap model. Defaults to `off`. |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-spark",
-  "version": "0.19.0",
+  "version": "0.19.2",
   "description": "Pi package that polishes your daily experience.",
   "keywords": [
     "pi-coding-agent",
@@ -33,21 +33,22 @@
     "@grpc/grpc-js": "^1.14.4",
     "@grpc/proto-loader": "^0.8.1",
     "defu": "^6.1.7",
-    "ms": "4.0.0-nightly.202508271359",
+    "parse-duration": "^2.1.8",
+    "pretty-ms": "^9.3.0",
     "zod": "^4.4.3"
   },
   "peerDependencies": {
-    "@earendil-works/pi-agent-core": ">=0.80.6",
-    "@earendil-works/pi-ai": ">=0.80.6",
-    "@earendil-works/pi-coding-agent": ">=0.80.6",
-    "@earendil-works/pi-tui": ">=0.80.6",
+    "@earendil-works/pi-agent-core": ">=0.80.8",
+    "@earendil-works/pi-ai": ">=0.80.8",
+    "@earendil-works/pi-coding-agent": ">=0.80.8",
+    "@earendil-works/pi-tui": ">=0.80.8",
     "typebox": "*"
   },
   "devDependencies": {
-    "@earendil-works/pi-agent-core": ">=0.80.6",
-    "@earendil-works/pi-ai": ">=0.80.6",
-    "@earendil-works/pi-coding-agent": ">=0.80.6",
-    "@earendil-works/pi-tui": ">=0.80.6",
+    "@earendil-works/pi-agent-core": ">=0.80.8",
+    "@earendil-works/pi-ai": ">=0.80.8",
+    "@earendil-works/pi-coding-agent": ">=0.80.8",
+    "@earendil-works/pi-tui": ">=0.80.8",
     "@types/node": "*",
     "typebox": "*",
     "typescript": "^7.0.2"

package/src/features/credits/manager.ts

@@ -51,7 +51,7 @@ export class CreditsManager {
       }
 
       const signals = AbortSignal.any([AbortSignal.timeout(REQUEST_TIMEOUT_MS), signal]);
-      const credits = await provider.fetch(ctx, apiKey, signals);
+      const credits = await provider.fetch(apiKey, signals);
 
       // The active model may have changed while the request was in flight.
       if (ctx.model?.provider !== provider.id) return;

package/src/features/credits/providers/deepseek.ts

@@ -18,7 +18,7 @@ export const deepseekProvider: CreditsProvider = {
   id: PROVIDER,
   label: "DeepSeek",
 
-  async fetch(_ctx, apiKey, signal): Promise<Credits> {
+  async fetch(apiKey, signal): Promise<Credits> {
     const headers: Record<string, string> = {
       Accept: "application/json",
       Authorization: `Bearer ${apiKey}`,

package/src/features/credits/providers/fireworks.ts

@@ -108,7 +108,7 @@ export const fireworksProvider: CreditsProvider = {
   id: PROVIDER,
   label: "Fireworks",
 
-  async fetch(_ctx, apiKey, signal): Promise<Credits> {
+  async fetch(apiKey, signal): Promise<Credits> {
     const name = await resolveAccount(apiKey, signal);
     const balance = await unary<Balance>("GetBalance", { name }, apiKey, signal);
 

package/src/features/credits/providers/kimi-code.ts

@@ -42,7 +42,7 @@ export const kimiCodeProvider: CreditsProvider = {
   id: PROVIDER,
   label: "Kimi Code",
 
-  async fetch(_ctx, apiKey, signal): Promise<Credits> {
+  async fetch(apiKey, signal): Promise<Credits> {
     const headers: Record<string, string> = {
       Accept: "application/json",
       Authorization: `Bearer ${apiKey}`,

package/src/features/credits/providers/moonshot.ts

@@ -18,7 +18,7 @@ function createMoonshotProvider(id: string, host: string, currency: string): Cre
     id,
     label: "Moonshot",
 
-    async fetch(_ctx, apiKey, signal): Promise<Credits> {
+    async fetch(apiKey, signal): Promise<Credits> {
       const headers: Record<string, string> = {
         Accept: "application/json",
         Authorization: `Bearer ${apiKey}`,

package/src/features/credits/providers/openai-codex.ts

@@ -1,11 +1,13 @@
 import { uuidv7 } from "@earendil-works/pi-agent-core";
+import { readStoredCredential } from "@earendil-works/pi-coding-agent";
+import prettyMilliseconds from "pretty-ms";
 
 import { confirmCodexReset, formatAvailableResets, showCodexResetLoader, showCodexResetSelector } from "./openai-codex-panel";
 import { toNumber } from "../../../utils/format";
 
 import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
 import type { CodexResetPanelData } from "./openai-codex-panel";
-import type { Credits, CreditsProvider, RefreshCredits } from "../types";
+import type { Credits, CreditsLane, CreditsProvider, RefreshCredits } from "../types";
 
 const PROVIDER = "openai-codex";
 const BASE_URL = "https://chatgpt.com/backend-api";
@@ -24,6 +26,7 @@ interface CodexUsageResponse {
 }
 
 interface CodexRateWindow {
+  limit_window_seconds: number;
   used_percent?: number | string;
 }
 
@@ -50,7 +53,7 @@ async function runCodexReset(ctx: ExtensionContext, refresh: RefreshCredits): Pr
   const apiKey = await ctx.modelRegistry.getApiKeyForProvider(PROVIDER);
   if (!apiKey) return;
 
-  const headers = buildHeaders(ctx, apiKey);
+  const headers = buildHeaders(apiKey);
 
   const credit = await showCodexResetSelector(ctx, async (signal) => {
     const requestSignal = AbortSignal.any([signal, AbortSignal.timeout(REQUEST_TIMEOUT_MS)]);
@@ -74,15 +77,15 @@ async function redeemReset(ctx: ExtensionContext, headers: Record<string, string
   }).finally(() => refresh(ctx));
 }
 
-function buildHeaders(ctx: ExtensionContext, apiKey: string): Record<string, string> {
+function buildHeaders(apiKey: string): Record<string, string> {
   const headers: Record<string, string> = {
     Accept: "application/json",
     Authorization: `Bearer ${apiKey}`,
   };
 
-  const credential = ctx.modelRegistry.authStorage.get(PROVIDER);
-  const accountId = credential?.type === "oauth" ? (credential.accountId as string | undefined) : undefined;
-  if (accountId) headers["ChatGPT-Account-ID"] = accountId;
+  const credential = readStoredCredential(PROVIDER);
+  const accountId = credential?.type === "oauth" ? credential.accountId : undefined;
+  if (typeof accountId === "string" && accountId.length > 0) headers["ChatGPT-Account-ID"] = accountId;
 
   return headers;
 }
@@ -126,14 +129,20 @@ function toCredits(usage: CodexUsageResponse, suffix?: string): Credits {
   return {
     type: "windows",
     unlimited: usage.credits?.unlimited === true,
-    lanes: [
-      { label: "5h", percent: parseUsedPercent(usage.rate_limit?.primary_window) },
-      { label: "7d", percent: parseUsedPercent(usage.rate_limit?.secondary_window) },
-    ],
+    lanes: [usage.rate_limit?.primary_window, usage.rate_limit?.secondary_window]
+      .filter((window): window is CodexRateWindow => window != null)
+      .map(toLane),
     suffix,
   };
 }
 
+function toLane(window: CodexRateWindow): CreditsLane {
+  return {
+    label: prettyMilliseconds(window.limit_window_seconds * 1_000, { compact: true }),
+    percent: parseUsedPercent(window),
+  };
+}
+
 function parseUsedPercent(window?: CodexRateWindow | null): number | undefined {
   const value = toNumber(window?.used_percent);
   return value === undefined ? undefined : Math.min(100, Math.max(0, value));
@@ -143,8 +152,8 @@ export const openaiCodexProvider: CreditsProvider = {
   id: PROVIDER,
   label: "Codex",
 
-  async fetch(ctx, apiKey, signal): Promise<Credits> {
-    const headers = buildHeaders(ctx, apiKey);
+  async fetch(apiKey, signal): Promise<Credits> {
+    const headers = buildHeaders(apiKey);
     const usage = await fetchUsage(headers, signal);
     const availableCount = usage.rate_limit_reset_credits?.available_count ?? 0;
     const suffix = availableCount > 0 ? `(${formatAvailableResets(availableCount)})` : undefined;
@@ -152,9 +161,8 @@ export const openaiCodexProvider: CreditsProvider = {
     return toCredits(usage, suffix);
   },
 
-  register(pi, ctx, refresh): void {
-    const credential = ctx.modelRegistry.authStorage.get(PROVIDER);
-    if (credential?.type !== "oauth") return;
+  register(pi, _ctx, refresh): void {
+    if (readStoredCredential(PROVIDER)?.type !== "oauth") return;
 
     pi.registerCommand("codex-reset", {
       description: "Select and redeem a banked OpenAI Codex rate-limit reset",

package/src/features/credits/providers/openrouter.ts

@@ -16,7 +16,7 @@ export const openrouterProvider: CreditsProvider = {
   id: PROVIDER,
   label: "OpenRouter",
 
-  async fetch(_ctx, apiKey, signal): Promise<Credits> {
+  async fetch(apiKey, signal): Promise<Credits> {
     const headers: Record<string, string> = {
       Accept: "application/json",
       Authorization: `Bearer ${apiKey}`,

package/src/features/credits/providers/vercel-ai-gateway.ts

@@ -13,7 +13,7 @@ export const vercelAiGatewayProvider: CreditsProvider = {
   id: PROVIDER,
   label: "Vercel",
 
-  async fetch(_ctx, apiKey, signal): Promise<Credits> {
+  async fetch(apiKey, signal): Promise<Credits> {
     const headers: Record<string, string> = {
       Accept: "application/json",
       Authorization: `Bearer ${apiKey}`,

package/src/features/credits/types.ts

@@ -23,6 +23,6 @@ export type RefreshCredits = (ctx: ExtensionContext) => Promise<void>;
 export interface CreditsProvider {
   readonly id: ProviderId;
   readonly label: string;
-  fetch(ctx: ExtensionContext, apiKey: string, signal: AbortSignal): Promise<Credits>;
+  fetch(apiKey: string, signal: AbortSignal): Promise<Credits>;
   register?(pi: ExtensionAPI, ctx: ExtensionContext, refresh: RefreshCredits): void;
 }

package/src/features/recap/idle.ts

@@ -1,8 +1,6 @@
-import { ms } from "ms";
+import parseDuration from "parse-duration";
 import * as z from "zod";
 
-import type { StringValue } from "ms";
-
 const MIN_IDLE_MS = 5_000;
 
 /** Accept a millisecond number or a human-readable duration (e.g., "3m"), normalized to milliseconds. */
@@ -11,20 +9,11 @@ export const idleTimeoutSchema = z
   .transform((value, ctx) => {
     if (typeof value === "number") return value;
 
-    let parsed: number;
-    try {
-      parsed = ms(value as StringValue);
-    } catch (error) {
-      ctx.addIssue({ code: "custom", message: error instanceof Error ? error.message : String(error) });
-      return z.NEVER;
-    }
-
-    if (Number.isNaN(parsed)) {
-      ctx.addIssue({ code: "custom", message: `Value is not a valid duration. value=${JSON.stringify(value)}` });
-      return z.NEVER;
-    }
+    const parsed = parseDuration(value);
+    if (parsed !== null) return parsed;
 
-    return parsed;
+    ctx.addIssue({ code: "custom", message: `Value is not a valid duration. value=${JSON.stringify(value)}` });
+    return z.NEVER;
   })
   .pipe(z.number().min(MIN_IDLE_MS));