pi-spark 0.21.0 → 0.21.1

package/README.md

@@ -80,7 +80,7 @@ pi-spark names the session automatically after the first completed turn, so it's
 
 Long, streaming `write` tool calls keep their latest lines visible instead of pinning the preview to the beginning. The omitted-line count appears above the preview, while `ctrl+o` still expands the full content.
 
-https://raw.githubusercontent.com/zlliang/pi-spark/refs/heads/main/assets/screencast-write.mp4
+https://github.com/user-attachments/assets/b7bb77c1-c188-46a2-aba9-6487bdf58fa0
 
 ## Configuration
 

package/docs/reverse-engineering-fireworks-credits.md

@@ -89,7 +89,7 @@ The amount is `units + nanos / 1e9` (the standard `google.type.Money` encoding),
 The provider lives in [`src/features/credits/providers/fireworks.ts`](../src/features/credits/providers/fireworks.ts) and follows the same `CreditsProvider` contract as the HTTP-based providers; only the transport differs.
 
 - **Transport.** A single, reused `@grpc/grpc-js` client dials `gateway.fireworks.ai:443` over TLS. The service is declared in [`fireworks.proto`](../src/features/credits/providers/fireworks.proto) — the minimal slice needed, loaded at runtime with `@grpc/proto-loader`. The `.proto` is resolved relative to the module via `import.meta.url`, so it ships and loads from the package directory.
-- **Auth.** Each call sends `x-api-key: <key>` in gRPC metadata. The key comes from Pi's `modelRegistry.getApiKeyForProvider("fireworks")` (env `FIREWORKS_API_KEY`), handled by the manager.
+- **Auth.** Each call sends `x-api-key: <key>` in gRPC metadata. The manager resolves provider authentication through Pi's `modelRegistry.getProviderAuth("fireworks")`. For Fireworks, Pi returns the API key sourced from `FIREWORKS_API_KEY`.
 - **Account resolution.** `ListAccounts` is called once and the resolved `accounts/<id>` is cached per API key, so steady-state refreshes are a single `GetBalance` round trip.
 - **Cancellation.** The shared `AbortSignal` cancels the in-flight gRPC call, and a defensive deadline bounds each request.
 - **Result.** `Money` is converted to a dollar amount and returned as `{ type: "balance", remaining }`, which the status line renders like any other balance provider.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-spark",
-  "version": "0.21.0",
+  "version": "0.21.1",
   "description": "Pi package that polishes your daily experience.",
   "keywords": [
     "pi-coding-agent",

package/src/features/credits/manager.ts

@@ -1,3 +1,4 @@
+import { getAuthToken } from "../../utils/auth";
 import { findProvider } from "./providers";
 import { renderCredits, renderError } from "./status";
 
@@ -47,14 +48,14 @@ export class CreditsManager {
 
   private async fetch(ctx: ExtensionContext, provider: CreditsProvider, signal: AbortSignal): Promise<void> {
     try {
-      const apiKey = await ctx.modelRegistry.getApiKeyForProvider(provider.id);
+      const token = await getAuthToken(ctx.modelRegistry, provider.id);
       if (signal.aborted) return;
-      if (!apiKey) {
+      if (!token) {
         ctx.ui.setStatus(STATUS_KEY, undefined);
         return;
       }
 
-      const credits = await provider.fetch(apiKey, signal);
+      const credits = await provider.fetch(token, signal);
 
       // The active model may have changed while the request was in flight.
       if (ctx.model?.provider !== provider.id) return;

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

@@ -3,6 +3,7 @@ import { readStoredCredential } from "@earendil-works/pi-coding-agent";
 import prettyMilliseconds from "pretty-ms";
 
 import { confirmCodexReset, formatAvailableResets, showCodexResetLoader, showCodexResetSelector } from "./openai-codex-panel";
+import { getAuthToken } from "../../../utils/auth";
 import { toNumber } from "../../../utils/format";
 import { http, withAuth } from "../../../utils/http";
 
@@ -51,10 +52,10 @@ interface ConsumeResetResponse {
 }
 
 async function runCodexReset(ctx: ExtensionContext, refresh: RefreshCredits): Promise<void> {
-  const apiKey = await ctx.modelRegistry.getApiKeyForProvider(PROVIDER);
-  if (!apiKey) return;
+  const token = await getAuthToken(ctx.modelRegistry, PROVIDER);
+  if (!token) return;
 
-  const client = createClient(apiKey);
+  const client = createClient(token);
 
   const credit = await showCodexResetSelector(ctx, async (signal) => {
     const [usage, details] = await Promise.all([fetchUsage(client, signal), fetchResetCredits(client, signal)]);

package/src/features/presets/manager.ts

@@ -80,10 +80,14 @@ export class PresetManager {
     const currentIndex = currentKey ? this.keys.indexOf(currentKey) : -1;
     const step = direction === "forward" ? 1 : -1;
     const nextIndex = currentIndex === -1 ? direction === "forward" ? 0 : this.keys.length - 1 : (currentIndex + step + this.keys.length) % this.keys.length;
-    const nextKey = this.keys[nextIndex];
-    if (!nextKey) return;
 
-    await this.apply(nextKey, ctx);
+    for (let offset = 0; offset < this.keys.length; offset++) {
+      const candidateIndex = (nextIndex + step * offset + this.keys.length) % this.keys.length;
+      const candidateKey = this.keys[candidateIndex];
+      if (!candidateKey) continue;
+
+      if (await this.apply(candidateKey, ctx)) return;
+    }
   }
 
   describe(key: string): string {

package/src/utils/auth.ts

@@ -0,0 +1,11 @@
+import type { ProviderId } from "@earendil-works/pi-ai";
+import type { ModelRegistry } from "@earendil-works/pi-coding-agent";
+
+export async function getAuthToken(modelRegistry: ModelRegistry, provider: ProviderId): Promise<string | undefined> {
+  const providerAuth = await modelRegistry.getProviderAuth(provider);
+
+  const authorization = Object.entries(providerAuth?.auth.headers ?? {}).find(([name]) => name.toLowerCase() === "authorization")?.[1];
+  const bearerToken = authorization?.match(/^Bearer\s+(.+)$/i)?.[1];
+
+  return providerAuth?.auth.apiKey ?? bearerToken;
+}