pi-spark 0.19.1 → 0.19.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-spark",
-  "version": "0.19.1",
+  "version": "0.19.2",
   "description": "Pi package that polishes your daily experience.",
   "keywords": [
     "pi-coding-agent",
@@ -38,17 +38,17 @@
     "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,4 +1,5 @@
 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";
@@ -52,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)]);
@@ -76,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;
 }
@@ -151,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;
@@ -160,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;
 }