@duke-dsh-plugins/dsh-token-stats 1.3.3 → 1.4.0

package/typert.host.js

@@ -80,6 +80,84 @@ const tokenStatsResultSchema = z.union([
     .readonly(),
 ]);
 
+// ---- quota / balance (套餐余额) ---------------------------------------------
+
+const quotaDisplaySchema = z
+  .object({
+    fiveHrPct: z.union([z.number(), z.null()]),
+    weeklyPct: z.union([z.number(), z.null()]),
+    fiveHrResetsIn: z.union([z.string(), z.null()]),
+    weeklyResetsIn: z.union([z.string(), z.null()]),
+    balanceText: z.union([z.string(), z.null()]),
+    balanceUsd: z.union([z.number(), z.null()]),
+    currency: z.union([z.string(), z.null()]),
+  })
+  .readonly();
+
+const quotaValueSchema = z.union([
+  z
+    .object({
+      ok: z.literal(true).readonly(),
+      provider: z.string(),
+      display: quotaDisplaySchema,
+    })
+    .readonly(),
+  z
+    .object({
+      ok: z.literal(false).readonly(),
+      provider: z.string(),
+      kind: z.string(),
+      message: z.string(),
+    })
+    .readonly(),
+]);
+
+const tokenStatsQuotaResultSchema = z.union([
+  z
+    .object({
+      ok: z.literal(true).readonly(),
+      value: quotaValueSchema,
+    })
+    .readonly(),
+  z
+    .object({
+      ok: z.literal(false).readonly(),
+      error: z
+        .object({
+          code: z.string().readonly(),
+          message: z.string().readonly().optional(),
+        })
+        .readonly(),
+    })
+    .readonly(),
+]);
+
+const allQuotasValueSchema = z
+  .object({
+    quotas: z.record(z.string(), quotaValueSchema).readonly(),
+  })
+  .readonly();
+
+const tokenStatsAllQuotasResultSchema = z.union([
+  z
+    .object({
+      ok: z.literal(true).readonly(),
+      value: allQuotasValueSchema,
+    })
+    .readonly(),
+  z
+    .object({
+      ok: z.literal(false).readonly(),
+      error: z
+        .object({
+          code: z.string().readonly(),
+          message: z.string().readonly().optional(),
+        })
+        .readonly(),
+    })
+    .readonly(),
+]);
+
 export const TYPERT = {
   package: "@duke-dsh-plugins/dsh-token-stats",
   face: "host",
@@ -99,6 +177,66 @@ export const TYPERT = {
       },
       sourceLocation: { file: "index.js", line: 1, column: 1 },
     },
+    {
+      id: "dsh-token-stats#tokenStats/getQuota",
+      service: "tokenStats",
+      namespace: "tokenStats",
+      method: "getQuota",
+      invocation: { kind: "direct" },
+      parameters: [
+        {
+          name: "provider",
+          wire: "provider",
+          source: "json",
+          codec: {
+            mode: "strict",
+            typeSymbol: "dsh-token-stats#tokenStats/getQuota:provider",
+            schema: z.string(),
+          },
+        },
+        {
+          name: "force",
+          wire: "force",
+          source: "json",
+          codec: {
+            mode: "strict",
+            typeSymbol: "dsh-token-stats#tokenStats/getQuota:force",
+            schema: z.boolean(),
+          },
+        },
+      ],
+      result: {
+        mode: "strict",
+        typeSymbol: "dsh-token-stats#TokenStatsQuotaResult",
+        schema: tokenStatsQuotaResultSchema,
+      },
+      sourceLocation: { file: "index.js", line: 1, column: 1 },
+    },
+    {
+      id: "dsh-token-stats#tokenStats/getAllQuotas",
+      service: "tokenStats",
+      namespace: "tokenStats",
+      method: "getAllQuotas",
+      invocation: { kind: "direct" },
+      parameters: [
+        {
+          name: "force",
+          wire: "force",
+          source: "json",
+          codec: {
+            mode: "strict",
+            typeSymbol: "dsh-token-stats#tokenStats/getAllQuotas:force",
+            schema: z.boolean(),
+          },
+        },
+      ],
+      result: {
+        mode: "strict",
+        typeSymbol: "dsh-token-stats#TokenStatsAllQuotasResult",
+        schema: tokenStatsAllQuotasResultSchema,
+      },
+      sourceLocation: { file: "index.js", line: 1, column: 1 },
+    },
   ],
   model: {
     services: [
@@ -120,6 +258,26 @@ export const TYPERT = {
             jsDoc:
               "/**\n * Return the whole-dataset snapshot: ready/collecting state, backfill progress,\n * per-day per-model aggregates and the ordered model list.\n * @returns success or a business failure.\n */",
           },
+          {
+            kind: "method",
+            name: "getQuota",
+            signature:
+              "@Remote('getQuota') async getQuota(provider: string, force: boolean): Promise<TokenStatsQuotaResult>",
+            summary:
+              "Read one provider's plan quota / balance (minimax | deepseek | kimi | openrouter | zhipu), cached per provider.",
+            jsDoc:
+              "/**\n * Read the plan quota / balance for one provider through the credentials seam\n * and the provider's own endpoint. Success caches 30 s; failures back off\n * exponentially (5 s → 30 min). `force` drops the cached entry first.\n * @param provider - internal provider key.\n * @param force - bypass the cache.\n * @returns success with the display payload or a business failure.\n */",
+          },
+          {
+            kind: "method",
+            name: "getAllQuotas",
+            signature:
+              "@Remote('getAllQuotas') async getAllQuotas(force: boolean): Promise<TokenStatsAllQuotasResult>",
+            summary:
+              "Read every supported provider's quota / balance in parallel (settings page balance section).",
+            jsDoc:
+              "/**\n * Snapshot all supported providers at once; each entry is the same wire\n * value getQuota returns. `force` bypasses every provider's cache.\n * @param force - bypass the caches.\n * @returns success with the per-provider quota map.\n */",
+          },
         ],
         types: [
           {
@@ -142,6 +300,31 @@ export const TYPERT = {
             declaration:
               "export type TokenStatsResult = { ok: true; value: TokenStatsValue } | { ok: false; error: { code: string; message?: string } };",
           },
+          {
+            name: "TokenStatsQuotaDisplay",
+            declaration:
+              "export interface TokenStatsQuotaDisplay {\n    readonly fiveHrPct: number | null;\n    readonly weeklyPct: number | null;\n    readonly fiveHrResetsIn: string | null;\n    readonly weeklyResetsIn: string | null;\n    readonly balanceText: string | null;\n    readonly balanceUsd: number | null;\n    readonly currency: string | null;\n}",
+          },
+          {
+            name: "TokenStatsQuotaValue",
+            declaration:
+              "export type TokenStatsQuotaValue =\n    | { readonly ok: true; readonly provider: string; readonly display: TokenStatsQuotaDisplay }\n    | { readonly ok: false; readonly provider: string; readonly kind: string; readonly message: string };",
+          },
+          {
+            name: "TokenStatsQuotaResult",
+            declaration:
+              "export type TokenStatsQuotaResult = { ok: true; value: TokenStatsQuotaValue } | { ok: false; error: { code: string; message?: string } };",
+          },
+          {
+            name: "TokenStatsAllQuotasValue",
+            declaration:
+              "export interface TokenStatsAllQuotasValue {\n    readonly quotas: Record<string, TokenStatsQuotaValue>;\n}",
+          },
+          {
+            name: "TokenStatsAllQuotasResult",
+            declaration:
+              "export type TokenStatsAllQuotasResult = { ok: true; value: TokenStatsAllQuotasValue } | { ok: false; error: { code: string; message?: string } };",
+          },
         ],
       },
     ],