@d3ara1n/pi-provider-zhipu-coding-plan 1.0.0

package/index.ts

@@ -0,0 +1,205 @@
+/**
+ * pi-provider-zhipu-coding-plan
+ *
+ * Registers "zhipu-coding" provider with dynamic model discovery
+ * and usage quota reporting via the shared UsageRegistry.
+ *
+ * Auth: API key stored in ~/.pi/agent/auth.json under "zhipu-coding"
+ */
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { usageRegistry } from "@d3ara1n/pi-usage-block-core";
+import type { UsageProvider, UsageWindow } from "@d3ara1n/pi-usage-block-core";
+import fs from "node:fs";
+import path from "node:path";
+import os from "node:os";
+
+// ── Config ────────────────────────────────────────────────────────────────
+
+const PROVIDER_ID = "zhipu-coding";
+const PROVIDER_NAME = "Zhipu AI (Coding Plan)";
+const CODING_BASE_URL = "https://open.bigmodel.cn/api/coding/paas/v4";
+const QUOTA_API_URL = "https://bigmodel.cn/api/monitor/usage/quota/limit";
+const API_KEY_ENV = "ZHIPUAI_API_KEY";
+
+// ── Types ─────────────────────────────────────────────────────────────────
+
+interface ModelMeta {
+  contextWindow: number;
+  maxTokens: number;
+  reasoning: boolean;
+  input: ("text" | "image")[];
+  compat: Record<string, unknown>;
+}
+
+interface QuotaLimitItem {
+  type: "TOKENS_LIMIT" | "TIME_LIMIT";
+  percentage?: number;
+  usage?: number;
+  currentValue?: number;
+  remaining?: number;
+  nextResetTime?: number;
+  /** Window duration count (e.g. 5) paired with unit */
+  number?: number;
+  /** Window duration unit: 3 = hours */
+  unit?: number;
+}
+
+interface QuotaResponse {
+  code: number;
+  success: boolean;
+  data?: { limits?: QuotaLimitItem[]; level?: string };
+}
+
+interface ApiModel { id: string }
+
+// ── Known model metadata (fallback) ───────────────────────────────────────
+
+const DEFAULT_META: ModelMeta = {
+  contextWindow: 128_000,
+  maxTokens: 16_384,
+  reasoning: true,
+  input: ["text"],
+  compat: { supportsDeveloperRole: false, thinkingFormat: "zai" },
+};
+
+const ZAI_STREAM = {
+  supportsDeveloperRole: false,
+  thinkingFormat: "zai",
+  zaiToolStream: true,
+} as const;
+
+const KNOWN_MODELS: Record<string, Partial<ModelMeta>> = {
+  "glm-4.5":        { contextWindow: 131_072, maxTokens: 98_304 },
+  "glm-4.5-air":    { contextWindow: 131_072, maxTokens: 98_304 },
+  "glm-4.5-flash":  { contextWindow: 131_072, maxTokens: 98_304 },
+  "glm-4.5v":       { contextWindow: 65_536,  maxTokens: 4_096, input: ["text", "image"] },
+  "glm-4.6":        { contextWindow: 205_000, maxTokens: 131_072 },
+  "glm-4.6v":       { contextWindow: 128_000, maxTokens: 4_096, input: ["text", "image"] },
+  "glm-4.6v-flash": { contextWindow: 128_000, maxTokens: 4_096, input: ["text", "image"] },
+  "glm-4.7":        { contextWindow: 205_000, maxTokens: 131_072, compat: ZAI_STREAM },
+  "glm-4.7-flash":  { contextWindow: 200_000, maxTokens: 131_072, compat: ZAI_STREAM },
+  "glm-4.7-flashx": { contextWindow: 200_000, maxTokens: 131_072, compat: ZAI_STREAM },
+  "glm-5":          { contextWindow: 205_000, maxTokens: 131_072, compat: ZAI_STREAM },
+  "glm-5-turbo":    { contextWindow: 200_000, maxTokens: 131_072, compat: ZAI_STREAM },
+  "glm-5.1":        { contextWindow: 200_000, maxTokens: 131_072, compat: ZAI_STREAM },
+  "glm-5v-turbo":   { contextWindow: 200_000, maxTokens: 131_072, input: ["text", "image"], compat: ZAI_STREAM },
+};
+
+// ── API helpers ───────────────────────────────────────────────────────────
+
+function resolveApiKey(): string | undefined {
+  try {
+    const piHome = process.env.PI_HOME ?? path.join(os.homedir(), ".pi");
+    const authPath = path.join(piHome, "agent", "auth.json");
+    if (!fs.existsSync(authPath)) return undefined;
+    const auth = JSON.parse(fs.readFileSync(authPath, "utf8"));
+    return auth[PROVIDER_ID]?.apiKey ?? auth[PROVIDER_ID]?.key;
+  } catch { return undefined; }
+}
+
+async function fetchModelList(apiKey: string): Promise<ApiModel[]> {
+  const res = await fetch(`${CODING_BASE_URL}/models`, {
+    headers: { Authorization: `Bearer ${apiKey}` },
+    signal: AbortSignal.timeout(10_000),
+  });
+  if (!res.ok) throw new Error(`GET /models ${res.status}`);
+  return (await res.json() as { data?: ApiModel[] }).data ?? [];
+}
+
+async function fetchQuota(apiKey: string): Promise<QuotaResponse | null> {
+  try {
+    const res = await fetch(QUOTA_API_URL, {
+      headers: { Authorization: apiKey, "Content-Type": "application/json" },
+      signal: AbortSignal.timeout(10_000),
+    });
+    if (!res.ok) return null;
+    const body = await res.json() as QuotaResponse;
+    return body.success ? body : null;
+  } catch { return null; }
+}
+
+// ── Model config builder ──────────────────────────────────────────────────
+
+function buildModelConfig(id: string) {
+  const k = KNOWN_MODELS[id];
+  return {
+    id,
+    name: id
+      .replace(/^glm-/, "GLM-")
+      .replace(/-/g, " ")
+      .replace(/\b(\w)/g, (_, c: string) => c.toUpperCase()),
+    api: "openai-completions" as const,
+    reasoning: k?.reasoning ?? DEFAULT_META.reasoning,
+    input: k?.input ?? DEFAULT_META.input,
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 }, // subscription plan
+    contextWindow: k?.contextWindow ?? DEFAULT_META.contextWindow,
+    maxTokens: k?.maxTokens ?? DEFAULT_META.maxTokens,
+    compat: { ...DEFAULT_META.compat, ...(k?.compat ?? {}) },
+  };
+}
+
+const FALLBACK_MODELS = Object.keys(KNOWN_MODELS).map(buildModelConfig);
+
+// ── Usage provider ────────────────────────────────────────────────────────
+
+function createUsageProvider(): UsageProvider {
+  return {
+    id: PROVIDER_ID,
+    name: "Zhipu Coding",
+    icon: "⚡",
+    async fetchUsage(): Promise<UsageWindow[]> {
+      const apiKey = resolveApiKey();
+      if (!apiKey) return [];
+
+      const quota = await fetchQuota(apiKey);
+      if (!quota?.data?.limits?.length) return [];
+
+      const limits = quota.data.limits;
+
+      // Each TOKENS_LIMIT is a quota window (5h, weekly, etc.)
+      // It carries its own percentage and nextResetTime.
+      // Lite plan has one; Pro plan may have two (5h + weekly).
+      const windows = limits
+        .filter(l => l.type === "TOKENS_LIMIT" && typeof l.percentage === "number")
+        .map(l => {
+          const period = l.number && l.unit ? `${l.number}${l.unit === 3 ? "h" : "w"}` : "";
+          return {
+            period,
+            used: l.percentage!,
+            limit: 100,
+            unit: "tokens" as const,
+            resetAt: l.nextResetTime ? new Date(l.nextResetTime) : undefined,
+          };
+        });
+
+      return windows;
+    },
+  };
+}
+
+// ── Entry point ───────────────────────────────────────────────────────────
+
+export default async function (pi: ExtensionAPI) {
+  const apiKey = resolveApiKey();
+
+  let models = FALLBACK_MODELS;
+  if (apiKey) {
+    try {
+      const apiModels = await fetchModelList(apiKey);
+      if (apiModels.length) models = apiModels.map(m => buildModelConfig(m.id));
+    } catch (e) {
+      console.error("[zhipu-coding] model discovery failed, using fallback:", e);
+    }
+  }
+
+  pi.registerProvider(PROVIDER_ID, {
+    name: PROVIDER_NAME,
+    baseUrl: CODING_BASE_URL,
+    apiKey: `$${API_KEY_ENV}`,
+    api: "openai-completions",
+    authHeader: true,
+    models,
+  });
+
+  usageRegistry.register(createUsageProvider());
+}

package/package.json

@@ -0,0 +1,22 @@
+{
+  "name": "@d3ara1n/pi-provider-zhipu-coding-plan",
+  "version": "1.0.0",
+  "description": "Zhipu AI Coding Plan provider for pi — auto-discovers models, reports usage quota",
+  "keywords": ["pi-package"],
+  "main": "index.ts",
+  "dependencies": {
+    "@d3ara1n/pi-usage-block-core": "^1.0.0"
+  },
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": "*"
+  },
+  "pi": {
+    "extensions": ["./index.ts"]
+  },
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/d3ara1n/pi-extensions",
+    "directory": "packages/pi-provider-zhipu-coding-plan"
+  }
+}