gsd-pi 2.76.0-dev.76f9a2dc5 → 2.76.0-dev.97807402

package/packages/pi-coding-agent/dist/core/model-registry-custom-caps.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=model-registry-custom-caps.test.d.ts.map

package/packages/pi-coding-agent/dist/core/model-registry-custom-caps.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-custom-caps.test.d.ts","sourceRoot":"","sources":["../../src/core/model-registry-custom-caps.test.ts"],"names":[],"mappings":""}

package/packages/pi-coding-agent/dist/core/model-registry-custom-caps.test.js

@@ -0,0 +1,203 @@
+/**
+ * Regression tests for #4563:
+ *   Bug 1 — custom/Anthropic-compatible models were hard-capped to 32 k output tokens
+ *   Bug 2 — custom models in models.json could not declare capabilities.supportsXhigh
+ */
+import assert from "node:assert/strict";
+import { mkdirSync, rmSync, writeFileSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { afterEach, beforeEach, describe, it } from "node:test";
+import { AuthStorage } from "./auth-storage.js";
+import { ModelRegistry } from "./model-registry.js";
+// ─── Helpers ─────────────────────────────────────────────────────────────────
+let testDir;
+beforeEach(() => {
+    testDir = join(tmpdir(), `model-registry-custom-caps-${Date.now()}-${Math.random().toString(36).slice(2)}`);
+    mkdirSync(testDir, { recursive: true });
+});
+afterEach(() => {
+    try {
+        rmSync(testDir, { recursive: true, force: true });
+    }
+    catch {
+        // best-effort cleanup
+    }
+});
+function createRegistry(modelsJson) {
+    const path = join(testDir, "models.json");
+    writeFileSync(path, JSON.stringify(modelsJson));
+    return new ModelRegistry(AuthStorage.inMemory(), path);
+}
+function writeModelsJson(obj) {
+    const path = join(testDir, "models.json");
+    writeFileSync(path, JSON.stringify(obj));
+    return path;
+}
+// ─── Bug 1: 32 k cap must not apply to custom/OpenAI-compatible models ────────
+describe("Bug 1 — maxTokens cap (#4563)", () => {
+    it("custom openai-completions model with maxTokens > 32 k is not capped", () => {
+        const registry = createRegistry({
+            providers: {
+                "kimi-custom": {
+                    baseUrl: "https://api.example.com/v1",
+                    apiKey: "sk-test",
+                    api: "openai-completions",
+                    models: [
+                        {
+                            id: "kimi-k2.6-code-preview",
+                            name: "Kimi K2.6 Code Preview",
+                            maxTokens: 131072,
+                            contextWindow: 262144,
+                        },
+                    ],
+                },
+            },
+        });
+        const model = registry.getAll().find((m) => m.id === "kimi-k2.6-code-preview");
+        assert.ok(model, "model should be registered");
+        assert.equal(model.maxTokens, 131072, "maxTokens must be preserved as declared — not capped to 32 000");
+    });
+    it("custom model with maxTokens exactly 32 k is not affected", () => {
+        const registry = createRegistry({
+            providers: {
+                "custom-provider": {
+                    baseUrl: "https://api.example.com/v1",
+                    apiKey: "sk-test",
+                    api: "openai-completions",
+                    models: [{ id: "model-32k", maxTokens: 32000, contextWindow: 128000 }],
+                },
+            },
+        });
+        const model = registry.getAll().find((m) => m.id === "model-32k");
+        assert.ok(model);
+        assert.equal(model.maxTokens, 32000);
+    });
+    it("custom model with maxTokens 65 k is stored at full value", () => {
+        const registry = createRegistry({
+            providers: {
+                "dashscope-custom": {
+                    baseUrl: "https://dashscope-intl.aliyuncs.com/compatible-mode/v1",
+                    apiKey: "sk-test",
+                    api: "openai-completions",
+                    models: [
+                        {
+                            id: "qwen3.5-plus",
+                            name: "Qwen3.5 Plus",
+                            maxTokens: 65536,
+                            contextWindow: 1000000,
+                        },
+                    ],
+                },
+            },
+        });
+        const model = registry.getAll().find((m) => m.id === "qwen3.5-plus" && m.provider === "dashscope-custom");
+        assert.ok(model);
+        assert.equal(model.maxTokens, 65536);
+    });
+});
+// ─── Bug 2: capabilities.supportsXhigh must be declarable in models.json ──────
+describe("Bug 2 — capabilities.supportsXhigh in models.json (#4563)", () => {
+    it("model with capabilities.supportsXhigh: true surfaces the flag", () => {
+        const registry = createRegistry({
+            providers: {
+                "kimi-custom": {
+                    baseUrl: "https://api.example.com/v1",
+                    apiKey: "sk-test",
+                    api: "anthropic-messages",
+                    models: [
+                        {
+                            id: "kimi-k2.6-code-preview",
+                            name: "Kimi K2.6 Code Preview",
+                            maxTokens: 131072,
+                            contextWindow: 262144,
+                            capabilities: { supportsXhigh: true },
+                        },
+                    ],
+                },
+            },
+        });
+        const model = registry.getAll().find((m) => m.id === "kimi-k2.6-code-preview");
+        assert.ok(model, "model should be registered");
+        assert.equal(model.capabilities?.supportsXhigh, true, "supportsXhigh must be true as declared in models.json");
+    });
+    it("model without capabilities declaration has no supportsXhigh", () => {
+        const registry = createRegistry({
+            providers: {
+                "plain-provider": {
+                    baseUrl: "https://api.example.com/v1",
+                    apiKey: "sk-test",
+                    api: "openai-completions",
+                    models: [{ id: "plain-model", maxTokens: 16384, contextWindow: 128000 }],
+                },
+            },
+        });
+        const model = registry.getAll().find((m) => m.id === "plain-model");
+        assert.ok(model);
+        // supportsXhigh should be absent or explicitly false — never implicitly true
+        assert.ok(!model.capabilities?.supportsXhigh, "supportsXhigh must not be set for models that don't declare it");
+    });
+    it("capabilities.supportsXhigh: false is respected", () => {
+        const registry = createRegistry({
+            providers: {
+                "explicit-provider": {
+                    baseUrl: "https://api.example.com/v1",
+                    apiKey: "sk-test",
+                    api: "openai-completions",
+                    models: [
+                        {
+                            id: "no-xhigh-model",
+                            capabilities: { supportsXhigh: false },
+                        },
+                    ],
+                },
+            },
+        });
+        const model = registry.getAll().find((m) => m.id === "no-xhigh-model");
+        assert.ok(model);
+        assert.equal(model.capabilities?.supportsXhigh, false);
+    });
+    it("supportsXhigh declared in models.json is not overwritten by capability patches", () => {
+        // The capability-patches system must not overwrite an explicit declaration in models.json.
+        // applyCapabilityPatches uses spread: { ...patch.caps, ...model.capabilities }
+        // so model.capabilities wins. This test verifies the precedence end-to-end.
+        const registry = createRegistry({
+            providers: {
+                "compat-provider": {
+                    baseUrl: "https://api.example.com/v1",
+                    apiKey: "sk-test",
+                    api: "openai-completions",
+                    models: [
+                        {
+                            id: "custom-xhigh-model",
+                            capabilities: { supportsXhigh: true },
+                        },
+                    ],
+                },
+            },
+        });
+        const model = registry.getAll().find((m) => m.id === "custom-xhigh-model");
+        assert.ok(model);
+        assert.equal(model.capabilities?.supportsXhigh, true);
+    });
+    it("modelOverrides can set capabilities.supportsXhigh on built-in models", () => {
+        // A user-facing override in models.json should be able to add supportsXhigh
+        // to a built-in model that doesn't declare it.
+        const path = writeModelsJson({
+            providers: {
+                anthropic: {
+                    modelOverrides: {
+                        "claude-3-5-haiku-20241022": {
+                            capabilities: { supportsXhigh: true },
+                        },
+                    },
+                },
+            },
+        });
+        const registry = new ModelRegistry(AuthStorage.inMemory(), path);
+        const model = registry.getAll().find((m) => m.provider === "anthropic" && m.id === "claude-3-5-haiku-20241022");
+        assert.ok(model, "built-in model must still be present");
+        assert.equal(model.capabilities?.supportsXhigh, true, "modelOverrides must be able to set capabilities.supportsXhigh");
+    });
+});
+//# sourceMappingURL=model-registry-custom-caps.test.js.map

package/packages/pi-coding-agent/dist/core/model-registry-custom-caps.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-custom-caps.test.js","sourceRoot":"","sources":["../../src/core/model-registry-custom-caps.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC3D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AAChE,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,gFAAgF;AAEhF,IAAI,OAAe,CAAC;AAEpB,UAAU,CAAC,GAAG,EAAE;IACf,OAAO,GAAG,IAAI,CACb,MAAM,EAAE,EACR,8BAA8B,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CACjF,CAAC;IACF,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;AACzC,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,GAAG,EAAE;IACd,IAAI,CAAC;QACJ,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACR,sBAAsB;IACvB,CAAC;AACF,CAAC,CAAC,CAAC;AAEH,SAAS,cAAc,CAAC,UAAkB;IACzC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;IAC1C,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC;IAChD,OAAO,IAAI,aAAa,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,IAAI,CAAC,CAAC;AACxD,CAAC;AAED,SAAS,eAAe,CAAC,GAAW;IACnC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;IAC1C,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IACzC,OAAO,IAAI,CAAC;AACb,CAAC;AAED,iFAAiF;AAEjF,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC9E,MAAM,QAAQ,GAAG,cAAc,CAAC;YAC/B,SAAS,EAAE;gBACV,aAAa,EAAE;oBACd,OAAO,EAAE,4BAA4B;oBACrC,MAAM,EAAE,SAAS;oBACjB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE;wBACP;4BACC,EAAE,EAAE,wBAAwB;4BAC5B,IAAI,EAAE,wBAAwB;4BAC9B,SAAS,EAAE,MAAM;4BACjB,aAAa,EAAE,MAAM;yBACrB;qBACD;iBACD;aACD;SACD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,wBAAwB,CAAC,CAAC;QAC/E,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,4BAA4B,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CACX,KAAK,CAAC,SAAS,EACf,MAAM,EACN,gEAAgE,CAChE,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QACnE,MAAM,QAAQ,GAAG,cAAc,CAAC;YAC/B,SAAS,EAAE;gBACV,iBAAiB,EAAE;oBAClB,OAAO,EAAE,4BAA4B;oBACrC,MAAM,EAAE,SAAS;oBACjB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;iBACtE;aACD;SACD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,WAAW,CAAC,CAAC;QAClE,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QACnE,MAAM,QAAQ,GAAG,cAAc,CAAC;YAC/B,SAAS,EAAE;gBACV,kBAAkB,EAAE;oBACnB,OAAO,EAAE,wDAAwD;oBACjE,MAAM,EAAE,SAAS;oBACjB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE;wBACP;4BACC,EAAE,EAAE,cAAc;4BAClB,IAAI,EAAE,cAAc;4BACpB,SAAS,EAAE,KAAK;4BAChB,aAAa,EAAE,OAAO;yBACtB;qBACD;iBACD;aACD;SACD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,cAAc,IAAI,CAAC,CAAC,QAAQ,KAAK,kBAAkB,CAAC,CAAC;QAC1G,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,iFAAiF;AAEjF,QAAQ,CAAC,2DAA2D,EAAE,GAAG,EAAE;IAC1E,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACxE,MAAM,QAAQ,GAAG,cAAc,CAAC;YAC/B,SAAS,EAAE;gBACV,aAAa,EAAE;oBACd,OAAO,EAAE,4BAA4B;oBACrC,MAAM,EAAE,SAAS;oBACjB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE;wBACP;4BACC,EAAE,EAAE,wBAAwB;4BAC5B,IAAI,EAAE,wBAAwB;4BAC9B,SAAS,EAAE,MAAM;4BACjB,aAAa,EAAE,MAAM;4BACrB,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;yBACrC;qBACD;iBACD;aACD;SACD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,wBAAwB,CAAC,CAAC;QAC/E,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,4BAA4B,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CACX,KAAK,CAAC,YAAY,EAAE,aAAa,EACjC,IAAI,EACJ,uDAAuD,CACvD,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACtE,MAAM,QAAQ,GAAG,cAAc,CAAC;YAC/B,SAAS,EAAE;gBACV,gBAAgB,EAAE;oBACjB,OAAO,EAAE,4BAA4B;oBACrC,MAAM,EAAE,SAAS;oBACjB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;iBACxE;aACD;SACD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,aAAa,CAAC,CAAC;QACpE,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,6EAA6E;QAC7E,MAAM,CAAC,EAAE,CACR,CAAC,KAAK,CAAC,YAAY,EAAE,aAAa,EAClC,gEAAgE,CAChE,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACzD,MAAM,QAAQ,GAAG,cAAc,CAAC;YAC/B,SAAS,EAAE;gBACV,mBAAmB,EAAE;oBACpB,OAAO,EAAE,4BAA4B;oBACrC,MAAM,EAAE,SAAS;oBACjB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE;wBACP;4BACC,EAAE,EAAE,gBAAgB;4BACpB,YAAY,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE;yBACtC;qBACD;iBACD;aACD;SACD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,gBAAgB,CAAC,CAAC;QACvE,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACzF,2FAA2F;QAC3F,+EAA+E;QAC/E,4EAA4E;QAC5E,MAAM,QAAQ,GAAG,cAAc,CAAC;YAC/B,SAAS,EAAE;gBACV,iBAAiB,EAAE;oBAClB,OAAO,EAAE,4BAA4B;oBACrC,MAAM,EAAE,SAAS;oBACjB,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE;wBACP;4BACC,EAAE,EAAE,oBAAoB;4BACxB,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;yBACrC;qBACD;iBACD;aACD;SACD,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,oBAAoB,CAAC,CAAC;QAC3E,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,aAAa,EAAE,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC/E,4EAA4E;QAC5E,+CAA+C;QAC/C,MAAM,IAAI,GAAG,eAAe,CAAC;YAC5B,SAAS,EAAE;gBACV,SAAS,EAAE;oBACV,cAAc,EAAE;wBACf,2BAA2B,EAAE;4BAC5B,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE;yBACrC;qBACD;iBACD;aACD;SACD,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,IAAI,CAAC,CAAC;QACjE,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,IAAI,CACnC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,WAAW,IAAI,CAAC,CAAC,EAAE,KAAK,2BAA2B,CACzE,CAAC;QACF,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,sCAAsC,CAAC,CAAC;QACzD,MAAM,CAAC,KAAK,CACX,KAAK,CAAC,YAAY,EAAE,aAAa,EACjC,IAAI,EACJ,+DAA+D,CAC/D,CAAC;IACH,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["/**\n * Regression tests for #4563:\n *   Bug 1 — custom/Anthropic-compatible models were hard-capped to 32 k output tokens\n *   Bug 2 — custom models in models.json could not declare capabilities.supportsXhigh\n */\nimport assert from \"node:assert/strict\";\nimport { mkdirSync, rmSync, writeFileSync } from \"node:fs\";\nimport { tmpdir } from \"node:os\";\nimport { join } from \"node:path\";\nimport { afterEach, beforeEach, describe, it } from \"node:test\";\nimport { AuthStorage } from \"./auth-storage.js\";\nimport { ModelRegistry } from \"./model-registry.js\";\n\n// ─── Helpers ─────────────────────────────────────────────────────────────────\n\nlet testDir: string;\n\nbeforeEach(() => {\n\ttestDir = join(\n\t\ttmpdir(),\n\t\t`model-registry-custom-caps-${Date.now()}-${Math.random().toString(36).slice(2)}`,\n\t);\n\tmkdirSync(testDir, { recursive: true });\n});\n\nafterEach(() => {\n\ttry {\n\t\trmSync(testDir, { recursive: true, force: true });\n\t} catch {\n\t\t// best-effort cleanup\n\t}\n});\n\nfunction createRegistry(modelsJson: object): ModelRegistry {\n\tconst path = join(testDir, \"models.json\");\n\twriteFileSync(path, JSON.stringify(modelsJson));\n\treturn new ModelRegistry(AuthStorage.inMemory(), path);\n}\n\nfunction writeModelsJson(obj: object): string {\n\tconst path = join(testDir, \"models.json\");\n\twriteFileSync(path, JSON.stringify(obj));\n\treturn path;\n}\n\n// ─── Bug 1: 32 k cap must not apply to custom/OpenAI-compatible models ────────\n\ndescribe(\"Bug 1 — maxTokens cap (#4563)\", () => {\n\tit(\"custom openai-completions model with maxTokens > 32 k is not capped\", () => {\n\t\tconst registry = createRegistry({\n\t\t\tproviders: {\n\t\t\t\t\"kimi-custom\": {\n\t\t\t\t\tbaseUrl: \"https://api.example.com/v1\",\n\t\t\t\t\tapiKey: \"sk-test\",\n\t\t\t\t\tapi: \"openai-completions\",\n\t\t\t\t\tmodels: [\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tid: \"kimi-k2.6-code-preview\",\n\t\t\t\t\t\t\tname: \"Kimi K2.6 Code Preview\",\n\t\t\t\t\t\t\tmaxTokens: 131072,\n\t\t\t\t\t\t\tcontextWindow: 262144,\n\t\t\t\t\t\t},\n\t\t\t\t\t],\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst model = registry.getAll().find((m) => m.id === \"kimi-k2.6-code-preview\");\n\t\tassert.ok(model, \"model should be registered\");\n\t\tassert.equal(\n\t\t\tmodel.maxTokens,\n\t\t\t131072,\n\t\t\t\"maxTokens must be preserved as declared — not capped to 32 000\",\n\t\t);\n\t});\n\n\tit(\"custom model with maxTokens exactly 32 k is not affected\", () => {\n\t\tconst registry = createRegistry({\n\t\t\tproviders: {\n\t\t\t\t\"custom-provider\": {\n\t\t\t\t\tbaseUrl: \"https://api.example.com/v1\",\n\t\t\t\t\tapiKey: \"sk-test\",\n\t\t\t\t\tapi: \"openai-completions\",\n\t\t\t\t\tmodels: [{ id: \"model-32k\", maxTokens: 32000, contextWindow: 128000 }],\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst model = registry.getAll().find((m) => m.id === \"model-32k\");\n\t\tassert.ok(model);\n\t\tassert.equal(model.maxTokens, 32000);\n\t});\n\n\tit(\"custom model with maxTokens 65 k is stored at full value\", () => {\n\t\tconst registry = createRegistry({\n\t\t\tproviders: {\n\t\t\t\t\"dashscope-custom\": {\n\t\t\t\t\tbaseUrl: \"https://dashscope-intl.aliyuncs.com/compatible-mode/v1\",\n\t\t\t\t\tapiKey: \"sk-test\",\n\t\t\t\t\tapi: \"openai-completions\",\n\t\t\t\t\tmodels: [\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tid: \"qwen3.5-plus\",\n\t\t\t\t\t\t\tname: \"Qwen3.5 Plus\",\n\t\t\t\t\t\t\tmaxTokens: 65536,\n\t\t\t\t\t\t\tcontextWindow: 1000000,\n\t\t\t\t\t\t},\n\t\t\t\t\t],\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst model = registry.getAll().find((m) => m.id === \"qwen3.5-plus\" && m.provider === \"dashscope-custom\");\n\t\tassert.ok(model);\n\t\tassert.equal(model.maxTokens, 65536);\n\t});\n});\n\n// ─── Bug 2: capabilities.supportsXhigh must be declarable in models.json ──────\n\ndescribe(\"Bug 2 — capabilities.supportsXhigh in models.json (#4563)\", () => {\n\tit(\"model with capabilities.supportsXhigh: true surfaces the flag\", () => {\n\t\tconst registry = createRegistry({\n\t\t\tproviders: {\n\t\t\t\t\"kimi-custom\": {\n\t\t\t\t\tbaseUrl: \"https://api.example.com/v1\",\n\t\t\t\t\tapiKey: \"sk-test\",\n\t\t\t\t\tapi: \"anthropic-messages\",\n\t\t\t\t\tmodels: [\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tid: \"kimi-k2.6-code-preview\",\n\t\t\t\t\t\t\tname: \"Kimi K2.6 Code Preview\",\n\t\t\t\t\t\t\tmaxTokens: 131072,\n\t\t\t\t\t\t\tcontextWindow: 262144,\n\t\t\t\t\t\t\tcapabilities: { supportsXhigh: true },\n\t\t\t\t\t\t},\n\t\t\t\t\t],\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst model = registry.getAll().find((m) => m.id === \"kimi-k2.6-code-preview\");\n\t\tassert.ok(model, \"model should be registered\");\n\t\tassert.equal(\n\t\t\tmodel.capabilities?.supportsXhigh,\n\t\t\ttrue,\n\t\t\t\"supportsXhigh must be true as declared in models.json\",\n\t\t);\n\t});\n\n\tit(\"model without capabilities declaration has no supportsXhigh\", () => {\n\t\tconst registry = createRegistry({\n\t\t\tproviders: {\n\t\t\t\t\"plain-provider\": {\n\t\t\t\t\tbaseUrl: \"https://api.example.com/v1\",\n\t\t\t\t\tapiKey: \"sk-test\",\n\t\t\t\t\tapi: \"openai-completions\",\n\t\t\t\t\tmodels: [{ id: \"plain-model\", maxTokens: 16384, contextWindow: 128000 }],\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst model = registry.getAll().find((m) => m.id === \"plain-model\");\n\t\tassert.ok(model);\n\t\t// supportsXhigh should be absent or explicitly false — never implicitly true\n\t\tassert.ok(\n\t\t\t!model.capabilities?.supportsXhigh,\n\t\t\t\"supportsXhigh must not be set for models that don't declare it\",\n\t\t);\n\t});\n\n\tit(\"capabilities.supportsXhigh: false is respected\", () => {\n\t\tconst registry = createRegistry({\n\t\t\tproviders: {\n\t\t\t\t\"explicit-provider\": {\n\t\t\t\t\tbaseUrl: \"https://api.example.com/v1\",\n\t\t\t\t\tapiKey: \"sk-test\",\n\t\t\t\t\tapi: \"openai-completions\",\n\t\t\t\t\tmodels: [\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tid: \"no-xhigh-model\",\n\t\t\t\t\t\t\tcapabilities: { supportsXhigh: false },\n\t\t\t\t\t\t},\n\t\t\t\t\t],\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst model = registry.getAll().find((m) => m.id === \"no-xhigh-model\");\n\t\tassert.ok(model);\n\t\tassert.equal(model.capabilities?.supportsXhigh, false);\n\t});\n\n\tit(\"supportsXhigh declared in models.json is not overwritten by capability patches\", () => {\n\t\t// The capability-patches system must not overwrite an explicit declaration in models.json.\n\t\t// applyCapabilityPatches uses spread: { ...patch.caps, ...model.capabilities }\n\t\t// so model.capabilities wins. This test verifies the precedence end-to-end.\n\t\tconst registry = createRegistry({\n\t\t\tproviders: {\n\t\t\t\t\"compat-provider\": {\n\t\t\t\t\tbaseUrl: \"https://api.example.com/v1\",\n\t\t\t\t\tapiKey: \"sk-test\",\n\t\t\t\t\tapi: \"openai-completions\",\n\t\t\t\t\tmodels: [\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tid: \"custom-xhigh-model\",\n\t\t\t\t\t\t\tcapabilities: { supportsXhigh: true },\n\t\t\t\t\t\t},\n\t\t\t\t\t],\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst model = registry.getAll().find((m) => m.id === \"custom-xhigh-model\");\n\t\tassert.ok(model);\n\t\tassert.equal(model.capabilities?.supportsXhigh, true);\n\t});\n\n\tit(\"modelOverrides can set capabilities.supportsXhigh on built-in models\", () => {\n\t\t// A user-facing override in models.json should be able to add supportsXhigh\n\t\t// to a built-in model that doesn't declare it.\n\t\tconst path = writeModelsJson({\n\t\t\tproviders: {\n\t\t\t\tanthropic: {\n\t\t\t\t\tmodelOverrides: {\n\t\t\t\t\t\t\"claude-3-5-haiku-20241022\": {\n\t\t\t\t\t\t\tcapabilities: { supportsXhigh: true },\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t});\n\n\t\tconst registry = new ModelRegistry(AuthStorage.inMemory(), path);\n\t\tconst model = registry.getAll().find(\n\t\t\t(m) => m.provider === \"anthropic\" && m.id === \"claude-3-5-haiku-20241022\",\n\t\t);\n\t\tassert.ok(model, \"built-in model must still be present\");\n\t\tassert.equal(\n\t\t\tmodel.capabilities?.supportsXhigh,\n\t\t\ttrue,\n\t\t\t\"modelOverrides must be able to set capabilities.supportsXhigh\",\n\t\t);\n\t});\n});\n"]}

package/packages/pi-coding-agent/dist/core/model-registry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model-registry.d.ts","sourceRoot":"","sources":["../../src/core/model-registry.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACN,KAAK,GAAG,EAER,KAAK,2BAA2B,EAChC,KAAK,OAAO,EAMZ,KAAK,KAAK,EACV,KAAK,sBAAsB,EAK3B,KAAK,mBAAmB,EACxB,MAAM,YAAY,CAAC;AAOpB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,KAAK,EAAmB,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAwG7E,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG,OAAO,GAAG,aAAa,GAAG,MAAM,CAAC;AA2F3E;;GAEG;AACH,qBAAa,aAAa;IASxB,QAAQ,CAAC,WAAW,EAAE,WAAW;IACjC,QAAQ,CAAC,cAAc,EAAE,MAAM,GAAG,SAAS;IAT5C,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,cAAc,CAAsB;IAC5C,OAAO,CAAC,qBAAqB,CAAkC;IAC/D,OAAO,CAAC,mBAAmB,CAA+C;IAC1E,OAAO,CAAC,SAAS,CAAiC;gBAGxC,WAAW,EAAE,WAAW,EACxB,cAAc,GAAE,MAAM,GAAG,SAA8C;IAoBjF;;OAEG;IACH,OAAO,IAAI,IAAI;IAef;;OAEG;IACH,QAAQ,IAAI,MAAM,GAAG,SAAS;IAI9B,OAAO,CAAC,UAAU;IA+BlB,8DAA8D;IAC9D,OAAO,CAAC,iBAAiB;IAiCzB,wFAAwF;IACxF,OAAO,CAAC,iBAAiB;IAazB,OAAO,CAAC,gBAAgB;IAuDxB,OAAO,CAAC,cAAc;IAyCtB,OAAO,CAAC,WAAW;IAiEnB;;;OAGG;IACH,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAItB;;;OAGG;IACH,YAAY,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAI5B;;;OAGG;IACH,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,gBAAgB;IASvD;;OAEG;IACH,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAQjD;;OAEG;IACH,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS;IAI/D;;;;OAIG;IACG,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAMnF;;;;OAIG;IACG,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAM7F;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO;IAKxC;;;;;;OAMG;IACH,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,GAAG,IAAI;IAKzE;;;;;;;;OAQG;IACH,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI;IAO9C,OAAO,CAAC,mBAAmB;IA6J3B;;;OAGG;IACG,cAAc,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAkDtE;;;OAGG;IACH,oBAAoB,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAMpC;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO;IAIxC;;OAEG;IACH,iBAAiB,IAAI,mBAAmB;IAIxC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAuB/B,OAAO,CAAC,eAAe;IAgBvB,OAAO,CAAC,4BAA4B;IAcpC,OAAO,CAAC,kBAAkB;IAM1B,OAAO,CAAC,4BAA4B;IA2BpC,OAAO,CAAC,eAAe;IASvB;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO;IAI/C;;;;OAIG;IACH,eAAe,IAAI,OAAO;CAK1B;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IACnC,QAAQ,CAAC,EAAE,gBAAgB,CAAC;IAC5B;yGACqG;IACrG,OAAO,CAAC,EAAE,MAAM,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,mBAAmB,KAAK,2BAA2B,CAAC;IACnH,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,wCAAwC;IACxC,KAAK,CAAC,EAAE,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;IAC3C,MAAM,CAAC,EAAE,KAAK,CAAC;QACd,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,GAAG,CAAC,EAAE,GAAG,CAAC;QACV,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,OAAO,CAAC;QACnB,KAAK,EAAE,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;QAC5B,IAAI,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,CAAC;QAC/E,aAAa,EAAE,MAAM,CAAC;QACtB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC9B,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KAC1C,CAAC,CAAC;CACH"}
+{"version":3,"file":"model-registry.d.ts","sourceRoot":"","sources":["../../src/core/model-registry.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EACN,KAAK,GAAG,EAER,KAAK,2BAA2B,EAChC,KAAK,OAAO,EAMZ,KAAK,KAAK,EACV,KAAK,sBAAsB,EAK3B,KAAK,mBAAmB,EACxB,MAAM,YAAY,CAAC;AAOpB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,KAAK,EAAmB,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAkH7E,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG,OAAO,GAAG,aAAa,GAAG,MAAM,CAAC;AAgG3E;;GAEG;AACH,qBAAa,aAAa;IASxB,QAAQ,CAAC,WAAW,EAAE,WAAW;IACjC,QAAQ,CAAC,cAAc,EAAE,MAAM,GAAG,SAAS;IAT5C,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,cAAc,CAAsB;IAC5C,OAAO,CAAC,qBAAqB,CAAkC;IAC/D,OAAO,CAAC,mBAAmB,CAA+C;IAC1E,OAAO,CAAC,SAAS,CAAiC;gBAGxC,WAAW,EAAE,WAAW,EACxB,cAAc,GAAE,MAAM,GAAG,SAA8C;IAoBjF;;OAEG;IACH,OAAO,IAAI,IAAI;IAef;;OAEG;IACH,QAAQ,IAAI,MAAM,GAAG,SAAS;IAI9B,OAAO,CAAC,UAAU;IA+BlB,8DAA8D;IAC9D,OAAO,CAAC,iBAAiB;IAiCzB,wFAAwF;IACxF,OAAO,CAAC,iBAAiB;IAazB,OAAO,CAAC,gBAAgB;IAuDxB,OAAO,CAAC,cAAc;IAyCtB,OAAO,CAAC,WAAW;IAkEnB;;;OAGG;IACH,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAItB;;;OAGG;IACH,YAAY,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAI5B;;;OAGG;IACH,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,gBAAgB;IASvD;;OAEG;IACH,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO;IAQjD;;OAEG;IACH,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS;IAI/D;;;;OAIG;IACG,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAMnF;;;;OAIG;IACG,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAM7F;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO;IAKxC;;;;;;OAMG;IACH,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,GAAG,IAAI;IAKzE;;;;;;;;OAQG;IACH,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI;IAO9C,OAAO,CAAC,mBAAmB;IA6J3B;;;OAGG;IACG,cAAc,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAkDtE;;;OAGG;IACH,oBAAoB,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAMpC;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO;IAIxC;;OAEG;IACH,iBAAiB,IAAI,mBAAmB;IAIxC;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAuB/B,OAAO,CAAC,eAAe;IAgBvB,OAAO,CAAC,4BAA4B;IAcpC,OAAO,CAAC,kBAAkB;IAM1B,OAAO,CAAC,4BAA4B;IA2BpC,OAAO,CAAC,eAAe;IASvB;;;OAGG;IACH,MAAM,CAAC,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO;IAI/C;;;;OAIG;IACH,eAAe,IAAI,OAAO;CAK1B;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IACnC,QAAQ,CAAC,EAAE,gBAAgB,CAAC;IAC5B;yGACqG;IACrG,OAAO,CAAC,EAAE,MAAM,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,GAAG,CAAC,EAAE,GAAG,CAAC;IACV,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,mBAAmB,KAAK,2BAA2B,CAAC;IACnH,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,wCAAwC;IACxC,KAAK,CAAC,EAAE,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,CAAC;IAC3C,MAAM,CAAC,EAAE,KAAK,CAAC;QACd,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,GAAG,CAAC,EAAE,GAAG,CAAC;QACV,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,SAAS,EAAE,OAAO,CAAC;QACnB,KAAK,EAAE,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;QAC5B,IAAI,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAC;YAAC,UAAU,EAAE,MAAM,CAAA;SAAE,CAAC;QAC/E,aAAa,EAAE,MAAM,CAAC;QACtB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC9B,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KAC1C,CAAC,CAAC;CACH"}

package/packages/pi-coding-agent/dist/core/model-registry.js

@@ -24,6 +24,13 @@ const VercelGatewayRoutingSchema = Type.Object({
     only: Type.Optional(Type.Array(Type.String())),
     order: Type.Optional(Type.Array(Type.String())),
 });
+// Schema for model capability declarations (mirrors ModelCapabilities in pi-ai types)
+const ModelCapabilitiesSchema = Type.Object({
+    supportsXhigh: Type.Optional(Type.Boolean()),
+    requiresToolCallId: Type.Optional(Type.Boolean()),
+    supportsServiceTier: Type.Optional(Type.Boolean()),
+    charsPerToken: Type.Optional(Type.Number()),
+});
 // Schema for OpenAI compatibility settings
 const OpenAICompletionsCompatSchema = Type.Object({
     supportsStore: Type.Optional(Type.Boolean()),
@@ -62,6 +69,7 @@ const ModelDefinitionSchema = Type.Object({
     maxTokens: Type.Optional(Type.Number()),
     headers: Type.Optional(Type.Record(Type.String(), Type.String())),
     compat: Type.Optional(OpenAICompatSchema),
+    capabilities: Type.Optional(ModelCapabilitiesSchema),
 });
 // Schema for per-model overrides (all fields optional, merged with built-in model)
 const ModelOverrideSchema = Type.Object({
@@ -78,6 +86,7 @@ const ModelOverrideSchema = Type.Object({
     maxTokens: Type.Optional(Type.Number()),
     headers: Type.Optional(Type.Record(Type.String(), Type.String())),
     compat: Type.Optional(OpenAICompatSchema),
+    capabilities: Type.Optional(ModelCapabilitiesSchema),
 });
 const ProviderConfigSchema = Type.Object({
     baseUrl: Type.Optional(Type.String({ minLength: 1 })),
@@ -151,6 +160,10 @@ function applyModelOverride(model, override) {
     }
     // Deep merge compat
     result.compat = mergeCompat(model.compat, override.compat);
+    // Merge capabilities (override wins per-field)
+    if (override.capabilities) {
+        result.capabilities = { ...model.capabilities, ...override.capabilities };
+    }
     return result;
 }
 /**
@@ -392,6 +405,7 @@ export class ModelRegistry {
                     maxTokens: modelDef.maxTokens ?? 16384,
                     headers,
                     compat: modelDef.compat,
+                    capabilities: modelDef.capabilities,
                 });
             }
         }