pi-credits 0.1.1 → 0.1.2

package/README.md

@@ -2,16 +2,20 @@
 
 # pi-credits
 
-A [pi](https://pi.dev/) extension that shows the active model provider's credit balance or rate-limit usage as a footer status.
+A [pi](https://pi.dev/) extension that shows the active model provider's credit balance or rate-limit usage in the footer. It appears only for supported providers and uses the provider's stored credential or API key.
 
-- The status appears only while a supported provider is active and uses that provider's stored credential or API key.
-- The status refreshes on session start, model switch, and after each turn that incurs usage.
+![Screenshot](./assets/screenshot.png)
+
+> Example with an OpenAI Codex subscription, paired with my [pi-spark](https://github.com/zlliang/pi-spark) package.
 
 ## Supported providers
 
-- **OpenAI Codex:** rate-limit usage of the 5-hour and weekly windows, as used percentages.
-- **OpenRouter:** remaining credit balance in dollars.
-- **Vercel AI Gateway:** remaining credit balance in dollars.
+- DeepSeek
+- OpenAI Codex
+- OpenRouter
+- Vercel AI Gateway
+
+The provider-specific fetching approaches are strongly inspired by [CodexBar](https://github.com/steipete/codexbar).
 
 ## Install
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-credits",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "A pi extension that shows the active provider's credit balance or rate-limit usage",
   "keywords": [
     "pi-coding-agent",
@@ -15,6 +15,7 @@
   "files": [
     "index.ts",
     "src",
+    "assets",
     "README.md",
     "LICENSE"
   ],

package/src/providers/deepseek.ts

@@ -0,0 +1,56 @@
+import { 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;
+}
+
+interface DeepSeekBalanceInfo {
+  currency?: string;
+  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",
+
+  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 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 };
+  },
+};

package/src/providers/index.ts

@@ -1,3 +1,4 @@
+import { deepseekProvider } from "./deepseek";
 import { openaiCodexProvider } from "./openai-codex";
 import { openrouterProvider } from "./openrouter";
 import { vercelAiGatewayProvider } from "./vercel-ai-gateway";
@@ -5,6 +6,7 @@ import { vercelAiGatewayProvider } from "./vercel-ai-gateway";
 import type { CreditsProvider } from "../types";
 
 const PROVIDERS: CreditsProvider[] = [
+  deepseekProvider,
   openaiCodexProvider,
   openrouterProvider,
   vercelAiGatewayProvider,