pi-bailian 1.0.0

package/README.md

@@ -0,0 +1,50 @@
+# pi-bailian
+
+Alibaba Bailian (ModelStudio) provider for [pi.dev](https://pi.dev) — routes Qwen3 / GLM-4 / Kimi models through the Anthropic-compatible coding endpoint.
+
+## Models
+
+| ID | Name | Context | Output | Thinking |
+|---|---|---|---|---|
+| `qwen3.5-plus` | Qwen 3.5 Plus | 960K | 64K | ✅ |
+| `qwen3-max-2026-01-23` | Qwen3 Max (2026-01-23) | 252K | 32K | ✅ |
+| `qwen3-coder-plus` | Qwen3 Coder Plus | 975K | 64K | ❌ |
+| `qwen3-coder-next` | Qwen3 Coder Next | 200K | 64K | ❌ |
+| `glm-4.7` | GLM-4.7 | 166K | 16K | ✅ |
+| `kimi-k2.5` | Kimi K2.5 | 252K | 32K | ✅ |
+
+## Install
+
+```bash
+# Install from local path
+pi install ./pi-bailian
+
+# Or test without installing
+pi -e ./pi-bailian
+```
+
+## Configuration
+
+### Interactive setup (recommended)
+
+After installing, run the configure command inside pi:
+
+```
+/bailian-configure
+```
+
+It will prompt you for your API key and save it to `~/.pi/agent/bailian.json`. The key is applied immediately — no restart needed.
+
+### Environment variable
+
+Set `BAILIAN_API_KEY` in your environment (takes precedence over the saved config):
+
+```bash
+export BAILIAN_API_KEY=sk-sp-...
+```
+
+Get your API key from the [Model Studio console](https://modelstudio.console.alibabacloud.com).
+
+## Usage
+
+After installing, use `/model` or `Ctrl+L` in pi to select a `bailian/...` model. Use `Ctrl+T` to toggle thinking for models that support it.

package/index.ts

@@ -0,0 +1,170 @@
+/**
+ * Alibaba Bailian (ModelStudio) provider for pi.dev
+ *
+ * Registers the Bailian Anthropic-compatible coding endpoint with all
+ * configured Qwen3 / GLM-4 / Kimi models.
+ *
+ * Usage:
+ *   # Install from local path
+ *   pi install ./pi-bailian
+ *
+ *   # Or test in-place without installing
+ *   pi -e ./pi-bailian
+ *
+ *   # Configure your API key interactively
+ *   /bailian-configure
+ *
+ *   # Or set via environment variable (takes precedence over saved config)
+ *   BAILIAN_API_KEY=sk-sp-... pi
+ *
+ *   # Then switch models with /model or Ctrl+L
+ *   # Look for "bailian/..." entries
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { readFileSync, writeFileSync, mkdirSync, existsSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+
+const CONFIG_FILE = join(homedir(), ".pi", "agent", "bailian.json");
+
+function readStoredKey(): string | undefined {
+    try {
+        if (existsSync(CONFIG_FILE)) {
+            const data = JSON.parse(readFileSync(CONFIG_FILE, "utf-8"));
+            return typeof data.apiKey === "string" ? data.apiKey : undefined;
+        }
+    } catch {
+        // ignore malformed config
+    }
+    return undefined;
+}
+
+function saveKey(apiKey: string): void {
+    const dir = join(homedir(), ".pi", "agent");
+    mkdirSync(dir, { recursive: true });
+    writeFileSync(CONFIG_FILE, JSON.stringify({ apiKey }, null, 2), "utf-8");
+}
+
+const BAILIAN_MODELS = [
+    // ── Qwen 3.5 Plus ──────────────────────────────────────────────────────
+    // Extended thinking enabled | 960K context | 64K output
+    {
+        id: "qwen3.5-plus",
+        name: "Qwen 3.5 Plus",
+        reasoning: true,
+        input: ["text"] as ["text"],
+        contextWindow: 983616,
+        maxTokens: 65536,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    },
+    // ── Qwen3 Max 2026-01-23 ───────────────────────────────────────────────
+    // Extended thinking enabled | 252K context | 32K output
+    {
+        id: "qwen3-max-2026-01-23",
+        name: "Qwen3 Max (2026-01-23)",
+        reasoning: true,
+        input: ["text"] as ["text"],
+        contextWindow: 258048,
+        maxTokens: 32768,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    },
+    // ── Qwen3 Coder Plus ───────────────────────────────────────────────────
+    // No thinking | 975K context | 64K output
+    {
+        id: "qwen3-coder-plus",
+        name: "Qwen3 Coder Plus",
+        reasoning: false,
+        input: ["text"] as ["text"],
+        contextWindow: 997952,
+        maxTokens: 65536,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    },
+    // ── Qwen3 Coder Next ───────────────────────────────────────────────────
+    // No thinking | 200K context | 64K output
+    {
+        id: "qwen3-coder-next",
+        name: "Qwen3 Coder Next",
+        reasoning: false,
+        input: ["text"] as ["text"],
+        contextWindow: 204800,
+        maxTokens: 65536,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    },
+    // ── GLM-4.7 ────────────────────────────────────────────────────────────
+    // Extended thinking enabled | 166K context | 16K output
+    {
+        id: "glm-4.7",
+        name: "GLM-4.7",
+        reasoning: true,
+        input: ["text"] as ["text"],
+        contextWindow: 169984,
+        maxTokens: 16384,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    },
+    // ── Kimi K2.5 ─────────────────────────────────────────────────────────
+    // Extended thinking enabled | 252K context | 32K output
+    {
+        id: "kimi-k2.5",
+        name: "Kimi K2.5",
+        reasoning: true,
+        input: ["text"] as ["text"],
+        contextWindow: 258048,
+        maxTokens: 32768,
+        cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    },
+];
+
+const BAILIAN_BASE_URL = "https://coding-intl.dashscope.aliyuncs.com/apps/anthropic/v1";
+
+export default function (pi: ExtensionAPI) {
+    // Resolve API key: env var takes precedence, then saved config, then env var name
+    // (pi resolves env var name strings automatically, so "BAILIAN_API_KEY" is a
+    // valid apiKey value that tells pi to look up process.env.BAILIAN_API_KEY).
+    const resolvedKey = process.env.BAILIAN_API_KEY ?? readStoredKey() ?? "BAILIAN_API_KEY";
+
+    pi.registerProvider("bailian", {
+        baseUrl: BAILIAN_BASE_URL,
+        apiKey: resolvedKey,
+        api: "anthropic-messages",
+        models: BAILIAN_MODELS,
+    });
+
+    pi.on("session_start", async (_event, ctx) => {
+        const hasKey = process.env.BAILIAN_API_KEY !== undefined || readStoredKey() !== undefined;
+        if (!hasKey) {
+            ctx.ui.notify(
+                "Bailian: No API key configured. Run /bailian-configure to set up.",
+                "warning",
+            );
+        }
+    });
+
+    pi.registerCommand("bailian-configure", {
+        description: "Configure your Alibaba Bailian API key",
+        handler: async (_args, ctx) => {
+            const apiKey = await ctx.ui.input(
+                "Bailian API key (from https://bailian.console.aliyun.com):",
+                "sk-sp-...",
+            );
+
+            if (!apiKey || apiKey.trim() === "") {
+                ctx.ui.notify("No API key provided. Configuration cancelled.", "warning");
+                return;
+            }
+
+            const trimmedKey = apiKey.trim();
+            saveKey(trimmedKey);
+
+            // Re-register provider immediately with the new key — no reload needed.
+            pi.registerProvider("bailian", {
+                baseUrl: BAILIAN_BASE_URL,
+                apiKey: trimmedKey,
+                api: "anthropic-messages",
+                models: BAILIAN_MODELS,
+            });
+
+            ctx.ui.notify("Bailian API key saved and applied successfully!", "success");
+        },
+    });
+}

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "pi-bailian",
+  "version": "1.0.0",
+  "description": "Alibaba Bailian (ModelStudio) provider for pi.dev — Qwen3, GLM-4, Kimi via Anthropic-compatible API",
+  "type": "module",
+  "license": "MIT",
+  "author": "mkurman",
+  "files": [
+    "index.ts",
+    "README.md"
+  ],
+  "keywords": [
+    "pi-package"
+  ],
+  "pi": {
+    "extensions": [
+      "./index.ts"
+    ]
+  },
+  "scripts": {
+    "clean": "echo 'nothing to clean'",
+    "build": "echo 'nothing to build'",
+    "check": "echo 'nothing to check'"
+  }
+}