pi-credits 0.2.0 → 0.3.1

package/README.md

@@ -12,6 +12,7 @@ A [pi](https://pi.dev/) extension that shows the active model provider's credit
 
 - DeepSeek
 - Fireworks
+- Moonshot
 - OpenAI Codex
 - OpenRouter
 - Vercel AI Gateway

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-credits",
-  "version": "0.2.0",
+  "version": "0.3.1",
   "description": "A pi extension that shows the active provider's credit balance or rate-limit usage",
   "keywords": [
     "pi-coding-agent",

package/src/manager.ts

@@ -9,6 +9,7 @@ const REQUEST_TIMEOUT_MS = 30_000;
 
 export class CreditsManager {
   private inflight: AbortController | undefined = undefined;
+  private currentProvider: string | undefined = undefined;
 
   async refresh(ctx: ExtensionContext): Promise<void> {
     this.inflight?.abort();
@@ -16,10 +17,17 @@ export class CreditsManager {
     const provider = findProvider(ctx.model?.provider);
     if (!provider) {
       this.inflight = undefined;
+      this.currentProvider = undefined;
       ctx.ui.setStatus(STATUS_KEY, undefined);
       return;
     }
 
+    // Clear stale credits from another provider while the new fetch is in flight.
+    if (this.currentProvider !== provider.id) {
+      this.currentProvider = provider.id;
+      ctx.ui.setStatus(STATUS_KEY, undefined);
+    }
+
     const controller = new AbortController();
     this.inflight = controller;
 
@@ -30,7 +38,7 @@ export class CreditsManager {
 
   private async fetch(ctx: ExtensionContext, provider: CreditsProvider, signal: AbortSignal): Promise<void> {
     try {
-      const apiKey = await ctx.modelRegistry.getApiKeyForProvider(provider.provider);
+      const apiKey = await ctx.modelRegistry.getApiKeyForProvider(provider.id);
       if (!apiKey) {
         ctx.ui.setStatus(STATUS_KEY, undefined);
         return;
@@ -42,12 +50,12 @@ export class CreditsManager {
       const credits = await provider.fetch(ctx, apiKey, AbortSignal.any(signals));
 
       // The active model may have changed while the request was in flight.
-      if (ctx.model?.provider !== provider.provider) return;
+      if (ctx.model?.provider !== provider.id) return;
 
       ctx.ui.setStatus(STATUS_KEY, renderCredits(ctx.ui.theme, provider.label, credits));
     } catch (error) {
       if (signal.aborted || ctx.signal?.aborted) return;
-      if (ctx.model?.provider !== provider.provider) return;
+      if (ctx.model?.provider !== provider.id) return;
 
       const message = error instanceof Error ? error.message : String(error);
       ctx.ui.setStatus(STATUS_KEY, renderError(ctx.ui.theme, provider.label, message));

package/src/providers/deepseek.ts

@@ -1,10 +1,9 @@
-import { toNumber } from "../utils";
+import { convertToUSD, toNumber } from "../utils";
 
 import type { Credits, CreditsProvider } from "../types";
 
 const PROVIDER = "deepseek";
 const URL = "https://api.deepseek.com/user/balance";
-const FRANKFURTER_API = "https://api.frankfurter.dev/v2/rate";
 
 interface DeepSeekBalanceResponse {
   balance_infos?: DeepSeekBalanceInfo[] | null;
@@ -15,27 +14,8 @@ interface DeepSeekBalanceInfo {
   total_balance?: string | number;
 }
 
-interface FrankfurterRateResponse {
-  rate?: string | number;
-}
-
-async function convertToUSD(amount: number | undefined, currency: string | undefined, signal: AbortSignal): Promise<number | undefined> {
-  if (amount === undefined) return undefined;
-  if (!currency || currency === "USD") return amount;
-
-  const url = `${FRANKFURTER_API}/${encodeURIComponent(currency)}/USD`;
-  const response = await fetch(url, { headers: { Accept: "application/json" }, signal });
-  if (!response.ok) throw new Error("currency conversion failed");
-
-  const payload = (await response.json()) as FrankfurterRateResponse;
-  const rate = toNumber(payload.rate);
-  if (rate === undefined) throw new Error("currency conversion failed");
-
-  return amount * rate;
-}
-
 export const deepseekProvider: CreditsProvider = {
-  provider: PROVIDER,
+  id: PROVIDER,
   label: "DeepSeek",
 
   async fetch(_ctx, apiKey, signal): Promise<Credits> {

package/src/providers/fireworks.ts

@@ -99,7 +99,7 @@ function moneyToNumber(money: Money | null | undefined): number | undefined {
 }
 
 export const fireworksProvider: CreditsProvider = {
-  provider: PROVIDER,
+  id: PROVIDER,
   label: "Fireworks",
 
   async fetch(_ctx, apiKey, signal): Promise<Credits> {

package/src/providers/index.ts

@@ -1,5 +1,6 @@
 import { deepseekProvider } from "./deepseek";
 import { fireworksProvider } from "./fireworks";
+import { moonshotProvider, moonshotCnProvider } from "./moonshot";
 import { openaiCodexProvider } from "./openai-codex";
 import { openrouterProvider } from "./openrouter";
 import { vercelAiGatewayProvider } from "./vercel-ai-gateway";
@@ -9,11 +10,13 @@ import type { CreditsProvider } from "../types";
 const PROVIDERS: CreditsProvider[] = [
   deepseekProvider,
   fireworksProvider,
+  moonshotProvider,
+  moonshotCnProvider,
   openaiCodexProvider,
   openrouterProvider,
   vercelAiGatewayProvider,
 ];
 
 export function findProvider(provider?: string): CreditsProvider | undefined {
-  return PROVIDERS.find((entry) => entry.provider === provider);
+  return PROVIDERS.find((entry) => entry.id === provider);
 }

package/src/providers/moonshot.ts

@@ -0,0 +1,39 @@
+import { convertToUSD, toNumber } from "../utils";
+
+import type { Credits, CreditsProvider } from "../types";
+
+interface MoonshotBalanceResponse {
+  data?: {
+    available_balance?: string | number;
+  } | null;
+}
+
+/**
+ * For Moonshot, the international and China-mainland accounts live on separate hosts and bill in
+ * different currencies (USD vs CNY), which the endpoint does not report, so each pi provider ID
+ * fixes both host and currency.
+ */
+function createMoonshotProvider(id: string, host: string, currency: string): CreditsProvider {
+  return {
+    id,
+    label: "Moonshot",
+
+    async fetch(_ctx, apiKey, signal): Promise<Credits> {
+      const headers: Record<string, string> = {
+        Accept: "application/json",
+        Authorization: `Bearer ${apiKey}`,
+      };
+
+      const response = await fetch(`https://${host}/v1/users/me/balance`, { headers, signal });
+      if (!response.ok) throw new Error("request failed");
+
+      const payload = (await response.json()) as MoonshotBalanceResponse;
+      const remaining = await convertToUSD(toNumber(payload.data?.available_balance), currency, signal);
+
+      return { type: "balance", remaining };
+    },
+  };
+}
+
+export const moonshotProvider = createMoonshotProvider("moonshotai", "api.moonshot.ai", "USD");
+export const moonshotCnProvider = createMoonshotProvider("moonshotai-cn", "api.moonshot.cn", "CNY");

package/src/providers/openai-codex.ts

@@ -33,7 +33,7 @@ function parseUsedPercent(window?: CodexRateWindow | null): number | undefined {
 }
 
 export const openaiCodexProvider: CreditsProvider = {
-  provider: PROVIDER,
+  id: PROVIDER,
   label: "Codex",
 
   async fetch(ctx, apiKey, signal): Promise<Credits> {

package/src/providers/openrouter.ts

@@ -13,7 +13,7 @@ interface OpenRouterCreditsResponse {
 }
 
 export const openrouterProvider: CreditsProvider = {
-  provider: PROVIDER,
+  id: PROVIDER,
   label: "OpenRouter",
 
   async fetch(_ctx, apiKey, signal): Promise<Credits> {

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

@@ -10,7 +10,7 @@ interface VercelCreditsResponse {
 }
 
 export const vercelAiGatewayProvider: CreditsProvider = {
-  provider: PROVIDER,
+  id: PROVIDER,
   label: "Vercel",
 
   async fetch(_ctx, apiKey, signal): Promise<Credits> {

package/src/types.ts

@@ -18,7 +18,7 @@ export interface CreditsLane {
 
 /** A credits source for a pi provider, shown in the status line while that provider is active. */
 export interface CreditsProvider {
-  readonly provider: Provider;
+  readonly id: Provider;
   readonly label: string;
   fetch(ctx: ExtensionContext, apiKey: string, signal: AbortSignal): Promise<Credits>;
 }

package/src/utils.ts

@@ -29,3 +29,24 @@ export function toNumber(value?: string | number | null): number | undefined {
   const parsed = typeof value === "number" ? value : Number(value);
   return Number.isFinite(parsed) ? parsed : undefined;
 }
+
+const FRANKFURTER_API = "https://api.frankfurter.dev/v2/rate";
+
+interface FrankfurterRateResponse {
+  rate?: string | number;
+}
+
+export async function convertToUSD(amount: number | undefined, currency: string | undefined, signal: AbortSignal): Promise<number | undefined> {
+  if (amount === undefined) return undefined;
+  if (!currency || currency === "USD") return amount;
+
+  const url = `${FRANKFURTER_API}/${encodeURIComponent(currency)}/USD`;
+  const response = await fetch(url, { headers: { Accept: "application/json" }, signal });
+  if (!response.ok) throw new Error("currency conversion failed");
+
+  const payload = (await response.json()) as FrankfurterRateResponse;
+  const rate = toNumber(payload.rate);
+  if (rate === undefined) throw new Error("currency conversion failed");
+
+  return amount * rate;
+}