pi-spark 0.16.0 → 0.17.1

package/README.md

@@ -36,7 +36,7 @@ pi-spark ships with custom editor, footer, and fullscreen rendering, replacing t
 
 pi-spark shows the active provider's credit balance or rate-limit usage in the status line, so you can keep an eye on what's left without leaving the terminal.
 
-- Supported providers: DeepSeek, Fireworks, Moonshot, OpenAI Codex, OpenRouter, and Vercel AI Gateway.
+- Supported providers: DeepSeek, Fireworks, Kimi Code, Moonshot, OpenAI Codex, OpenRouter, and Vercel AI Gateway.
 - Most provider fetching follows [CodexBar](https://github.com/steipete/codexbar). Fireworks is the exception: its balance sits behind an internal gRPC API, reverse-engineered from the `firectl` binary (see [docs/fireworks.md](./docs/fireworks.md)).
 
 ![Credits](./assets/screenshot-credits.png)

package/package.json

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

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

@@ -1,5 +1,6 @@
 import { deepseekProvider } from "./deepseek";
 import { fireworksProvider } from "./fireworks";
+import { kimiCodeProvider } from "./kimi-code";
 import { moonshotProvider, moonshotCnProvider } from "./moonshot";
 import { openaiCodexProvider } from "./openai-codex";
 import { openrouterProvider } from "./openrouter";
@@ -10,6 +11,7 @@ import type { CreditsProvider } from "../types";
 const PROVIDERS: CreditsProvider[] = [
   deepseekProvider,
   fireworksProvider,
+  kimiCodeProvider,
   moonshotProvider,
   moonshotCnProvider,
   openaiCodexProvider,

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

@@ -0,0 +1,69 @@
+import { toNumber } from "../../../utils/format";
+
+import type { Credits, CreditsLane, CreditsProvider } from "../types";
+
+const PROVIDER = "kimi-coding";
+const URL = "https://api.kimi.com/coding/v1/usages";
+
+interface KimiCodeUsageResponse {
+  usage?: { used?: string | number } | null;
+  limits?: {
+    window?: { duration?: number; timeUnit?: string } | null;
+    detail?: { used?: string | number } | null;
+  }[] | null;
+}
+
+function toPercent(used: number | undefined): number | undefined {
+  if (used === undefined) return undefined;
+  return Math.min(100, Math.max(0, used));
+}
+
+function formatWindowLabel(window?: { duration?: number; timeUnit?: string } | null): string {
+  if (!window?.duration) return "Limit";
+
+  const duration = window.duration;
+  const unit = window.timeUnit?.toUpperCase() ?? "";
+
+  if (unit === "TIME_UNIT_MINUTE") {
+    if (duration >= 60 && duration % 60 === 0) return `${duration / 60}h`;
+    return `${duration}m`;
+  }
+  if (unit === "TIME_UNIT_HOUR") return `${duration}h`;
+  if (unit === "TIME_UNIT_DAY") return `${duration}d`;
+  if (unit === "TIME_UNIT_MONTH") return `${duration}mo`;
+  return `${duration}${unit.replace("TIME_UNIT_", "").toLowerCase()}`;
+}
+
+function buildLane(label: string, detail?: KimiCodeUsageResponse["usage"]): CreditsLane {
+  return { label, percent: toPercent(toNumber(detail?.used)) };
+}
+
+export const kimiCodeProvider: CreditsProvider = {
+  id: PROVIDER,
+  label: "Kimi Code",
+
+  async fetch(_ctx, apiKey, signal): Promise<Credits> {
+    const headers: Record<string, string> = {
+      Accept: "application/json",
+      Authorization: `Bearer ${apiKey}`,
+    };
+
+    const response = await fetch(URL, { headers, signal });
+    if (!response.ok) throw new Error("request failed");
+
+    const payload = (await response.json()) as KimiCodeUsageResponse;
+    const lanes: CreditsLane[] = [];
+
+    payload.limits?.forEach((item) => {
+      if (!item) return;
+
+      const label = formatWindowLabel(item.window);
+      lanes.push(buildLane(label, item.detail));
+    });
+
+    const weeklyLane = buildLane("7d", payload.usage);
+    lanes.push(weeklyLane);
+
+    return { type: "windows", lanes };
+  },
+};

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

@@ -11,16 +11,14 @@ interface CodexUsageResponse {
     primary_window?: CodexRateWindow | null;
     secondary_window?: CodexRateWindow | null;
   } | null;
-  credits?: {
-    unlimited?: boolean;
-  } | null;
+  credits?: { unlimited?: boolean } | null;
 }
 
 interface CodexRateWindow {
   used_percent?: number | string;
 }
 
-export function getAccountId(ctx: ExtensionContext): string | undefined {
+function getAccountId(ctx: ExtensionContext): string | undefined {
   const credential = ctx.modelRegistry.authStorage.get(PROVIDER) as { accountId?: string } | undefined;
   const accountId = credential?.accountId;
 

package/src/features/recap/manager.ts

@@ -85,7 +85,13 @@ export class RecapManager {
     const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
     if (!auth.ok) throw new Error(auth.error);
 
-    const options: SimpleStreamOptions = { maxTokens: MAX_TOKENS, signal };
+    const options: SimpleStreamOptions = {
+      maxTokens: MAX_TOKENS,
+      // Codex uses the session ID for request routing; without it, background calls may be
+      // routed to an unavailable model.
+      sessionId: ctx.sessionManager.getSessionId(),
+      signal,
+    };
     if (auth.apiKey) options.apiKey = auth.apiKey;
     if (auth.headers) options.headers = auth.headers;
     if (thinkingLevel !== "off") options.reasoning = thinkingLevel;
@@ -99,6 +105,10 @@ export class RecapManager {
       }],
     }, options);
 
+    if (response.stopReason === "error") {
+      throw new Error(response.errorMessage ?? "Recap generation failed");
+    }
+
     const content = response.content
       .filter((block) => block.type === "text")
       .map((block) => block.text)

package/src/features/title/manager.ts

@@ -68,7 +68,13 @@ export class TitleManager {
     const auth = await ctx.modelRegistry.getApiKeyAndHeaders(model);
     if (!auth.ok) throw new Error(auth.error);
 
-    const options: SimpleStreamOptions = { maxTokens: MAX_TOKENS, signal };
+    const options: SimpleStreamOptions = {
+      maxTokens: MAX_TOKENS,
+      // Codex uses the session ID for request routing; without it, background calls may be
+      // routed to an unavailable model.
+      sessionId: ctx.sessionManager.getSessionId(),
+      signal,
+    };
     if (auth.apiKey) options.apiKey = auth.apiKey;
     if (auth.headers) options.headers = auth.headers;
     if (thinkingLevel !== "off") options.reasoning = thinkingLevel;
@@ -82,6 +88,10 @@ export class TitleManager {
       }],
     }, options);
 
+    if (response.stopReason === "error") {
+      throw new Error(response.errorMessage ?? "Title generation failed");
+    }
+
     const content = response.content
       .filter((block) => block.type === "text")
       .map((block) => block.text)