pi-credits 0.2.0 → 0.3.0

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.0",
   "description": "A pi extension that shows the active provider's credit balance or rate-limit usage",
   "keywords": [
     "pi-coding-agent",

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,25 +14,6 @@ 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,
   label: "DeepSeek",

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,6 +10,8 @@ import type { CreditsProvider } from "../types";
 const PROVIDERS: CreditsProvider[] = [
   deepseekProvider,
   fireworksProvider,
+  moonshotProvider,
+  moonshotCnProvider,
   openaiCodexProvider,
   openrouterProvider,
   vercelAiGatewayProvider,

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(provider: string, host: string, currency: string): CreditsProvider {
+  return {
+    provider,
+    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/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;
+}