pi-spark 0.22.3 → 0.23.1

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # Changelog
 
+## [0.23.1](https://github.com/zlliang/pi-spark/compare/v0.23.0...v0.23.1) (2026-09-12)
+
+
+### Bug Fixes
+
+* **credits:** refresh pricing on user input ([8183402](https://github.com/zlliang/pi-spark/commit/8183402da978b41696685e244a698214fd519ce4))
+
+## [0.23.0](https://github.com/zlliang/pi-spark/compare/v0.22.3...v0.23.0) (2026-09-11)
+
+
+### Features
+
+* **credits:** support provider peak/off-peak pricing ([f0dbf19](https://github.com/zlliang/pi-spark/commit/f0dbf19dca6f7a3eb290490052489eb0649680d1))
+
 ## [0.22.3](https://github.com/zlliang/pi-spark/compare/v0.22.2...v0.22.3) (2026-09-10)
 
 

package/README.md

@@ -56,6 +56,12 @@ Banked rate-limit resets are saved benefits that can reset eligible Codex usage
 
 ![Codex resets](./assets/screenshot-codex-resets.png)
 
+#### Peak/off-peak pricing
+
+Pi does not currently support time-based pricing, so its cost estimates can be inaccurate for providers with peak/off-peak rates, such as [DeepSeek](https://api-docs.deepseek.com/quick_start/pricing). pi-spark updates model cost metadata so subsequent usage calculations reflect the current rate.
+
+![Peak/off-peak pricing](./assets/screenshot-peak-off-peak-pricing.png)
+
 ### Presets
 
 pi-spark lets you define named model presets in `spark.json` (see [Configuration](#configuration)), so you can switch between models and thinking levels without retyping provider details. The active preset is shown on the editor's top border.
@@ -145,7 +151,7 @@ All fields are optional. Each top-level feature runs with the defaults below unl
 
 #### `CreditsConfig`
 
-All supported providers are enabled by default. Set a provider to `false` to disable its credits status, or to `true` to override a global `false` in project config.
+All supported providers are enabled by default. Set a provider to `false` to disable its credits status and pricing updates, or to `true` to override a global `false` in project config.
 
 ```json
 {

package/package.json

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

package/src/features/credits/index.ts

@@ -1,24 +1,15 @@
 import { CreditsManager } from "./manager";
 import { getEnabledProviders, registerProviderExtensions } from "./providers";
 import { loadConfig } from "../../config";
-import { isUsage } from "../../utils/usage";
 
-import type { AgentMessage } from "@earendil-works/pi-agent-core";
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 
-function hasCost(message: AgentMessage): boolean {
-  const usage = (message as { usage?: unknown }).usage;
-  if (!isUsage(usage)) return false;
-
-  return usage.cost.total > 0 || usage.input > 0 || usage.output > 0;
-}
-
 export function registerCredits(pi: ExtensionAPI): void {
   let creditsManager: CreditsManager | undefined = undefined;
 
   pi.on("session_start", (_event, ctx) => {
     const config = loadConfig(ctx).credits;
-    if (!["tui", "rpc"].includes(ctx.mode) || !config) return;
+    if (!config) return;
 
     const providers = getEnabledProviders(config);
     if (providers.length === 0) return;
@@ -29,13 +20,29 @@ export function registerCredits(pi: ExtensionAPI): void {
     creditsManager.refresh(ctx);
   });
 
+  // Background calls run at `before_agent_start`, before the first `context` event, and the model
+  // registry may have been refreshed since the last apply.
+  pi.on("input", (_event, ctx) => {
+    creditsManager?.refresh(ctx);
+  });
+
   pi.on("model_select", (_event, ctx) => {
     creditsManager?.refresh(ctx);
   });
 
-  pi.on("turn_end", (event, ctx) => {
-    if (!hasCost(event.message)) return;
+  pi.on("context", (_event, ctx) => {
+    creditsManager?.refresh(ctx);
+  });
+
+  pi.on("turn_end", (_event, ctx) => {
+    creditsManager?.refresh(ctx);
+  });
+
+  pi.on("agent_settled", (_event, ctx) => {
+    creditsManager?.refresh(ctx);
+  });
 
+  pi.on("session_before_compact", (_event, ctx) => {
     creditsManager?.refresh(ctx);
   });
 
@@ -43,6 +50,12 @@ export function registerCredits(pi: ExtensionAPI): void {
     creditsManager?.refresh(ctx);
   });
 
+  pi.on("session_before_tree", (event, ctx) => {
+    if (!event.preparation.userWantsSummary) return;
+
+    creditsManager?.refresh(ctx);
+  });
+
   pi.on("session_tree", (event, ctx) => {
     if (!event.summaryEntry) return;
 

package/src/features/credits/manager.ts

@@ -1,5 +1,6 @@
 import { getAuthToken } from "../../utils/auth";
 import { findProvider } from "./providers";
+import { applyPricingRules } from "./pricing";
 import { renderCredits, renderError } from "./status";
 
 import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
@@ -17,6 +18,7 @@ export class CreditsManager {
   }
 
   async refresh(ctx: ExtensionContext): Promise<void> {
+    this.providers.forEach((provider) => applyPricingRules(ctx, provider, Date.now()));
     this.inflight?.abort();
 
     const provider = findProvider(this.providers, ctx.model?.provider);
@@ -55,10 +57,8 @@ export class CreditsManager {
         return;
       }
 
-      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;
+      const credits = await provider.fetch(token, signal, ctx.model?.cost);
+      if (signal.aborted || ctx.model?.provider !== provider.id) return;
 
       ctx.ui.setStatus(STATUS_KEY, renderCredits(ctx.ui.theme, provider.label, credits, provider.link));
     } catch (error) {

package/src/features/credits/pricing.ts

@@ -0,0 +1,57 @@
+import type { Api, Model } from "@earendil-works/pi-ai";
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+import type { CreditsProvider } from "./types";
+
+export type Cost = Model<Api>["cost"] & { label?: string };
+
+/**
+ * A time-of-day price window. `start`/`end` are `HH:MM` in UTC. Bounds are [start, end); equal
+ * bounds mean all day. When `start > end` the window wraps past midnight (e.g., 16:30 → 00:30).
+ */
+interface PricingWindow {
+  start: string;
+  end: string;
+  days?: number[];
+  cost: Cost;
+}
+
+export interface PricingRule {
+  model: string;
+  defaultCost: Cost;
+  windows: PricingWindow[];
+}
+
+/** Updates registry and active model prices without changing the provider's rule data. */
+export function applyPricingRules(ctx: ExtensionContext, provider: CreditsProvider, timestamp: number): void {
+  (provider.pricingRules ?? []).forEach((rule) => {
+    const cost = resolveCost(rule, timestamp);
+    const model = ctx.modelRegistry.find(provider.id, rule.model);
+
+    if (model) model.cost = { ...cost };
+    if (ctx.model?.provider === provider.id && ctx.model.id === rule.model) ctx.model.cost = { ...cost };
+  });
+}
+
+function resolveCost(rule: PricingRule, timestamp: number): Cost {
+  const date = new Date(timestamp);
+  const minutes = date.getUTCHours() * 60 + date.getUTCMinutes();
+
+  const cost = rule.windows.reduce((result, window) => {
+    if (window.days && !window.days.includes(date.getUTCDay())) return result;
+    if (inDailyRange(minutes, parseHHMM(window.start), parseHHMM(window.end))) return window.cost;
+    return result;
+  }, rule.defaultCost);
+
+  return cost;
+}
+
+function parseHHMM(hhmm: string): number {
+  const [hours, minutes] = hhmm.split(":");
+  return Number(hours) * 60 + Number(minutes);
+}
+
+function inDailyRange(minutes: number, start: number, end: number): boolean {
+  if (start === end) return true;
+  if (start < end) return minutes >= start && minutes < end;
+  return minutes >= start || minutes < end;
+}

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

@@ -1,6 +1,7 @@
 import { convertToUSD, toNumber } from "../../../utils/format";
 import { http, withAuth } from "../../../utils/http";
 
+import type { Cost } from "../pricing";
 import type { Credits, CreditsProvider } from "../types";
 
 const PROVIDER = "deepseek";
@@ -20,11 +21,47 @@ export const deepseekProvider: CreditsProvider = {
   label: "DeepSeek",
   link: "https://platform.deepseek.com/usage",
 
-  async fetch(apiKey, signal): Promise<Credits> {
+  // See https://api-docs.deepseek.com/quick_start/pricing/.
+  pricingRules: [
+    {
+      model: "deepseek-flash",
+      defaultCost: { input: 0.15, output: 0.6, cacheRead: 0.003, cacheWrite: 0, label: "off-peak" },
+      windows: [
+        { start: "01:00", end: "04:00", days: [1, 2, 3, 4, 5], cost: { input: 0.3, output: 1.2, cacheRead: 0.006, cacheWrite: 0, label: "peak" } },
+        { start: "06:00", end: "10:00", days: [1, 2, 3, 4, 5], cost: { input: 0.3, output: 1.2, cacheRead: 0.006, cacheWrite: 0, label: "peak" } },
+      ],
+    },
+    {
+      model: "deepseek-v4-flash",
+      defaultCost: { input: 0.15, output: 0.6, cacheRead: 0.003, cacheWrite: 0, label: "off-peak" },
+      windows: [
+        { start: "01:00", end: "04:00", days: [1, 2, 3, 4, 5], cost: { input: 0.3, output: 1.2, cacheRead: 0.006, cacheWrite: 0, label: "peak" } },
+        { start: "06:00", end: "10:00", days: [1, 2, 3, 4, 5], cost: { input: 0.3, output: 1.2, cacheRead: 0.006, cacheWrite: 0, label: "peak" } },
+      ],
+    },
+    {
+      model: "deepseek-v4-flash-vision-exp",
+      defaultCost: { input: 0.15, output: 0.6, cacheRead: 0.003, cacheWrite: 0, label: "off-peak" },
+      windows: [
+        { start: "01:00", end: "04:00", days: [1, 2, 3, 4, 5], cost: { input: 0.3, output: 1.2, cacheRead: 0.006, cacheWrite: 0, label: "peak" } },
+        { start: "06:00", end: "10:00", days: [1, 2, 3, 4, 5], cost: { input: 0.3, output: 1.2, cacheRead: 0.006, cacheWrite: 0, label: "peak" } },
+      ],
+    },
+    {
+      model: "deepseek-v4-pro",
+      defaultCost: { input: 0.66, output: 1.98, cacheRead: 0.022, cacheWrite: 0, label: "off-peak" },
+      windows: [
+        { start: "01:00", end: "04:00", days: [1, 2, 3, 4, 5], cost: { input: 1.32, output: 3.96, cacheRead: 0.044, cacheWrite: 0, label: "peak" } },
+        { start: "06:00", end: "10:00", days: [1, 2, 3, 4, 5], cost: { input: 1.32, output: 3.96, cacheRead: 0.044, cacheWrite: 0, label: "peak" } },
+      ],
+    },
+  ],
+
+  async fetch(apiKey, signal, cost?: Cost): Promise<Credits> {
     const payload = await withAuth(http, apiKey).get(URL, { signal }).json<DeepSeekBalanceResponse>();
     const balance = payload.balance_infos?.find((entry) => entry.currency === "USD") ?? payload.balance_infos?.[0];
     const remaining = await convertToUSD(toNumber(balance?.total_balance), balance?.currency, signal);
 
-    return { type: "balance", remaining };
+    return { type: "balance", remaining, suffix: cost?.label ? `(${cost.label})` : undefined };
   },
 };

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

@@ -31,5 +31,5 @@ export function findProvider(providers: CreditsProvider[], provider?: string): C
 }
 
 export function registerProviderExtensions(pi: ExtensionAPI, ctx: ExtensionContext, providers: CreditsProvider[], refresh: RefreshCredits): void {
-  for (const provider of providers) provider.register?.(pi, ctx, refresh);
+  providers.forEach((provider) => provider.register?.(pi, ctx, refresh));
 }

package/src/features/credits/types.ts

@@ -1,5 +1,6 @@
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import type { CreditsProviderId } from "./config";
+import type { Cost, PricingRule } from "./pricing";
 
 /**
  * Normalized credits/usage for a provider.
@@ -26,6 +27,7 @@ export interface CreditsProvider {
   readonly id: CreditsProviderId;
   readonly label: string;
   readonly link?: string;
-  fetch(apiKey: string, signal: AbortSignal): Promise<Credits>;
+  readonly pricingRules?: PricingRule[];
+  fetch(apiKey: string, signal: AbortSignal, cost?: Cost): Promise<Credits>;
   register?(pi: ExtensionAPI, ctx: ExtensionContext, refresh: RefreshCredits): void;
 }