gsd-pi 2.76.0 → 2.77.0

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-discovery.test.js

@@ -1,11 +1,12 @@
 import assert from "node:assert/strict";
-import { mkdirSync, rmSync } from "node:fs";
+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 { ModelDiscoveryCache } from "./discovery-cache.js";
 import { getDefaultTTL, getDiscoverableProviders, getDiscoveryAdapter } from "./model-discovery.js";
+import { ModelRegistry } from "./model-registry.js";
 let testDir;
 beforeEach(() => {
     testDir = join(tmpdir(), `model-registry-discovery-test-${Date.now()}-${Math.random().toString(36).slice(2)}`);
@@ -110,4 +111,63 @@ describe("Discovery TTL configuration", () => {
         assert.equal(customTTL, defaultTTL);
     });
 });
+describe("ModelRegistry discovery — OpenAI-compatible custom providers", () => {
+    it("discovers custom OpenAI-compatible providers and maps capability metadata", async () => {
+        const providerName = `minimax-openai-${Date.now()}-${Math.random().toString(36).slice(2)}`;
+        const modelsPath = join(testDir, "models.json");
+        writeFileSync(modelsPath, JSON.stringify({
+            providers: {
+                [providerName]: {
+                    baseUrl: "https://api.minimax.example",
+                    apiKey: "minimax-test-key",
+                    api: "openai-completions",
+                    models: [{ id: "bootstrap-model" }],
+                },
+            },
+        }, null, 2), "utf-8");
+        const prevFetch = globalThis.fetch;
+        let requestedUrl = "";
+        globalThis.fetch = (async (input) => {
+            requestedUrl = typeof input === "string" ? input : input instanceof URL ? input.toString() : input.url;
+            return new Response(JSON.stringify({
+                data: [
+                    {
+                        id: "MiniMax-M2.7-highspeed",
+                        name: "MiniMax M2.7 Highspeed",
+                        context_window: 165000,
+                        max_output_tokens: 32768,
+                        supports_reasoning: true,
+                        input_modalities: ["text", "image"],
+                    },
+                ],
+            }), {
+                status: 200,
+                headers: { "content-type": "application/json" },
+            });
+        });
+        try {
+            const registry = new ModelRegistry(AuthStorage.inMemory({}), modelsPath);
+            // Guard against global cache leakage from prior test runs.
+            registry.getDiscoveryCache().clear(providerName);
+            const results = await registry.discoverModels([providerName]);
+            const discovery = results.find((r) => r.provider === providerName);
+            assert.ok(discovery, "discovery result should include custom provider");
+            assert.equal(discovery?.error, undefined, "custom provider discovery should succeed");
+            assert.equal(requestedUrl, "https://api.minimax.example/v1/models");
+            const discovered = registry
+                .getAllWithDiscovered()
+                .find((m) => m.provider === providerName && m.id === "MiniMax-M2.7-highspeed");
+            assert.ok(discovered, "discovered model should be merged into model list");
+            assert.equal(discovered?.api, "openai-completions");
+            assert.equal(discovered?.baseUrl, "https://api.minimax.example");
+            assert.equal(discovered?.contextWindow, 165000);
+            assert.equal(discovered?.maxTokens, 32768);
+            assert.equal(discovered?.reasoning, true);
+            assert.deepEqual(discovered?.input, ["text", "image"]);
+        }
+        finally {
+            globalThis.fetch = prevFetch;
+        }
+    });
+});
 //# sourceMappingURL=model-registry-discovery.test.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"model-registry-discovery.test.js","sourceRoot":"","sources":["../../src/core/model-registry-discovery.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAiB,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,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,wBAAwB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAEpG,IAAI,OAAe,CAAC;AAEpB,UAAU,CAAC,GAAG,EAAE;IACf,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,iCAAiC,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC/G,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,gFAAgF;AAEhF,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IACjE,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAEjD,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACxC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAEjD,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,+EAA+E;AAE/E,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACnD,MAAM,SAAS,GAAG,wBAAwB,EAAE,CAAC;QAC7C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,EAAE,GAAG,QAAQ,2BAA2B,CAAC,CAAC;QACvF,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,eAAe,GAAG,CAAC,WAAW,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACrF,KAAK,MAAM,QAAQ,IAAI,eAAe,EAAE,CAAC;YACxC,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,EAAE,GAAG,QAAQ,+BAA+B,CAAC,CAAC;YAC3F,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YACtD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,EAAE,GAAG,QAAQ,6BAA6B,CAAC,CAAC;QACxE,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,+CAA+C,EAAE,GAAG,EAAE;IAC9D,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACnD,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACtD,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC;YACpC,MAAM,EAAE,EAAE,IAAI,EAAE,SAAkB,EAAE,GAAG,EAAE,SAAS,EAAE;SACpD,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QAEhD,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE;YACpB,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,EAAE;YACvD,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE;SAC1C,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAE9C,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAEvB,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC9D,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,EAAE,CAAC,SAAS,GAAG,SAAS,EAAE,0CAA0C,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC5C,MAAM,SAAS,GAAG,aAAa,CAAC,oBAAoB,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;QAC5C,0EAA0E;QAC1E,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import 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 { ModelDiscoveryCache } from \"./discovery-cache.js\";\nimport { getDefaultTTL, getDiscoverableProviders, getDiscoveryAdapter } from \"./model-discovery.js\";\n\nlet testDir: string;\n\nbeforeEach(() => {\n\ttestDir = join(tmpdir(), `model-registry-discovery-test-${Date.now()}-${Math.random().toString(36).slice(2)}`);\n\tmkdirSync(testDir, { recursive: true });\n});\n\nafterEach(() => {\n\ttry {\n\t\trmSync(testDir, { recursive: true, force: true });\n\t} catch {\n\t\t// Cleanup best-effort\n\t}\n});\n\n// ─── discovery cache integration ─────────────────────────────────────────────\n\ndescribe(\"ModelDiscoveryCache — integration with discovery\", () => {\n\tit(\"cache respects provider-specific TTLs\", () => {\n\t\tconst cachePath = join(testDir, \"cache.json\");\n\t\tconst cache = new ModelDiscoveryCache(cachePath);\n\n\t\tcache.set(\"ollama\", [{ id: \"llama2\" }]);\n\t\tconst entry = cache.get(\"ollama\");\n\t\tassert.ok(entry);\n\t\tassert.equal(entry.ttlMs, getDefaultTTL(\"ollama\"));\n\t});\n\n\tit(\"cache uses custom TTL when provided\", () => {\n\t\tconst cachePath = join(testDir, \"cache.json\");\n\t\tconst cache = new ModelDiscoveryCache(cachePath);\n\n\t\tcache.set(\"openai\", [{ id: \"gpt-4o\" }], 999);\n\t\tconst entry = cache.get(\"openai\");\n\t\tassert.ok(entry);\n\t\tassert.equal(entry.ttlMs, 999);\n\t});\n});\n\n// ─── adapter resolution ─────────────────────────────────────────────────────\n\ndescribe(\"Discovery adapter resolution\", () => {\n\tit(\"all discoverable providers have adapters\", () => {\n\t\tconst providers = getDiscoverableProviders();\n\t\tfor (const provider of providers) {\n\t\t\tconst adapter = getDiscoveryAdapter(provider);\n\t\t\tassert.equal(adapter.supportsDiscovery, true, `${provider} should support discovery`);\n\t\t}\n\t});\n\n\tit(\"static adapters return empty model lists\", async () => {\n\t\tconst staticProviders = [\"anthropic\", \"bedrock\", \"azure-openai\", \"groq\", \"cerebras\"];\n\t\tfor (const provider of staticProviders) {\n\t\t\tconst adapter = getDiscoveryAdapter(provider);\n\t\t\tassert.equal(adapter.supportsDiscovery, false, `${provider} should not support discovery`);\n\t\t\tconst models = await adapter.fetchModels(\"dummy-key\");\n\t\t\tassert.deepEqual(models, [], `${provider} should return empty models`);\n\t\t}\n\t});\n});\n\n// ─── AuthStorage hasAuth for discovery ───────────────────────────────────────\n\ndescribe(\"AuthStorage — hasAuth for discovery providers\", () => {\n\tit(\"returns false for providers without auth\", () => {\n\t\tconst storage = AuthStorage.inMemory({});\n\t\tassert.equal(storage.hasAuth(\"openai\"), false);\n\t\tassert.equal(storage.hasAuth(\"ollama\"), false);\n\t});\n\n\tit(\"returns true for providers with stored keys\", () => {\n\t\tconst storage = AuthStorage.inMemory({\n\t\t\topenai: { type: \"api_key\" as const, key: \"sk-test\" },\n\t\t});\n\t\tassert.equal(storage.hasAuth(\"openai\"), true);\n\t\tassert.equal(storage.hasAuth(\"ollama\"), false);\n\t});\n});\n\n// ─── cache persistence across instances ──────────────────────────────────────\n\ndescribe(\"ModelDiscoveryCache — persistence\", () => {\n\tit(\"data survives across cache instances\", () => {\n\t\tconst cachePath = join(testDir, \"persist.json\");\n\n\t\tconst cache1 = new ModelDiscoveryCache(cachePath);\n\t\tcache1.set(\"openai\", [\n\t\t\t{ id: \"gpt-4o\", name: \"GPT-4o\", contextWindow: 128000 },\n\t\t\t{ id: \"gpt-4o-mini\", name: \"GPT-4o Mini\" },\n\t\t]);\n\n\t\tconst cache2 = new ModelDiscoveryCache(cachePath);\n\t\tconst entry = cache2.get(\"openai\");\n\t\tassert.ok(entry);\n\t\tassert.equal(entry.models.length, 2);\n\t\tassert.equal(entry.models[0].contextWindow, 128000);\n\t});\n\n\tit(\"clear persists across instances\", () => {\n\t\tconst cachePath = join(testDir, \"clear.json\");\n\n\t\tconst cache1 = new ModelDiscoveryCache(cachePath);\n\t\tcache1.set(\"openai\", [{ id: \"gpt-4o\" }]);\n\t\tcache1.clear(\"openai\");\n\n\t\tconst cache2 = new ModelDiscoveryCache(cachePath);\n\t\tassert.equal(cache2.get(\"openai\"), undefined);\n\t});\n});\n\n// ─── discovery TTL values ────────────────────────────────────────────────────\n\ndescribe(\"Discovery TTL configuration\", () => {\n\tit(\"ollama has shortest TTL (local models change often)\", () => {\n\t\tconst ollamaTTL = getDefaultTTL(\"ollama\");\n\t\tconst openaiTTL = getDefaultTTL(\"openai\");\n\t\tassert.ok(ollamaTTL < openaiTTL, \"ollama TTL should be shorter than openai\");\n\t});\n\n\tit(\"unknown providers get default TTL\", () => {\n\t\tconst customTTL = getDefaultTTL(\"my-custom-provider\");\n\t\tconst defaultTTL = getDefaultTTL(\"default\");\n\t\t// Unknown providers should get the same TTL as the explicit \"default\" key\n\t\tassert.equal(customTTL, defaultTTL);\n\t});\n});\n"]}
+{"version":3,"file":"model-registry-discovery.test.js","sourceRoot":"","sources":["../../src/core/model-registry-discovery.test.ts"],"names":[],"mappings":"AAAA,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,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,wBAAwB,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AACpG,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,IAAI,OAAe,CAAC;AAEpB,UAAU,CAAC,GAAG,EAAE;IACf,OAAO,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,iCAAiC,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC/G,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,gFAAgF;AAEhF,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IACjE,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAEjD,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACxC,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAEjD,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,+EAA+E;AAE/E,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACnD,MAAM,SAAS,GAAG,wBAAwB,EAAE,CAAC;QAC7C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,EAAE,GAAG,QAAQ,2BAA2B,CAAC,CAAC;QACvF,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,eAAe,GAAG,CAAC,WAAW,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACrF,KAAK,MAAM,QAAQ,IAAI,eAAe,EAAE,CAAC;YACxC,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,EAAE,GAAG,QAAQ,+BAA+B,CAAC,CAAC;YAC3F,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YACtD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,EAAE,GAAG,QAAQ,6BAA6B,CAAC,CAAC;QACxE,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,+CAA+C,EAAE,GAAG,EAAE;IAC9D,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QACnD,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACtD,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,CAAC;YACpC,MAAM,EAAE,EAAE,IAAI,EAAE,SAAkB,EAAE,GAAG,EAAE,SAAS,EAAE;SACpD,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;QAEhD,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE;YACpB,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,EAAE;YACvD,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE;SAC1C,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;QAE9C,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAEvB,MAAM,MAAM,GAAG,IAAI,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC9D,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,EAAE,CAAC,SAAS,GAAG,SAAS,EAAE,0CAA0C,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC5C,MAAM,SAAS,GAAG,aAAa,CAAC,oBAAoB,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC;QAC5C,0EAA0E;QAC1E,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,8DAA8D,EAAE,GAAG,EAAE;IAC7E,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,YAAY,GAAG,kBAAkB,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3F,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAChD,aAAa,CACZ,UAAU,EACV,IAAI,CAAC,SAAS,CACb;YACC,SAAS,EAAE;gBACV,CAAC,YAAY,CAAC,EAAE;oBACf,OAAO,EAAE,6BAA6B;oBACtC,MAAM,EAAE,kBAAkB;oBAC1B,GAAG,EAAE,oBAAoB;oBACzB,MAAM,EAAE,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnC;aACD;SACD,EACD,IAAI,EACJ,CAAC,CACD,EACD,OAAO,CACP,CAAC;QAEF,MAAM,SAAS,GAAG,UAAU,CAAC,KAAK,CAAC;QACnC,IAAI,YAAY,GAAG,EAAE,CAAC;QACtB,UAAU,CAAC,KAAK,GAAG,CAAC,KAAK,EAAE,KAA6B,EAAE,EAAE;YAC3D,YAAY,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,YAAY,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC;YACvG,OAAO,IAAI,QAAQ,CAClB,IAAI,CAAC,SAAS,CAAC;gBACd,IAAI,EAAE;oBACL;wBACC,EAAE,EAAE,wBAAwB;wBAC5B,IAAI,EAAE,wBAAwB;wBAC9B,cAAc,EAAE,MAAM;wBACtB,iBAAiB,EAAE,KAAK;wBACxB,kBAAkB,EAAE,IAAI;wBACxB,gBAAgB,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;qBACnC;iBACD;aACD,CAAC,EACF;gBACC,MAAM,EAAE,GAAG;gBACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;aAC/C,CACD,CAAC;QACH,CAAC,CAA4B,CAAC;QAE9B,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;YACzE,2DAA2D;YAC3D,QAAQ,CAAC,iBAAiB,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACjD,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;YAE9D,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC;YACnE,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,iDAAiD,CAAC,CAAC;YACxE,MAAM,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,EAAE,SAAS,EAAE,0CAA0C,CAAC,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,uCAAuC,CAAC,CAAC;YAEpE,MAAM,UAAU,GAAG,QAAQ;iBACzB,oBAAoB,EAAE;iBACtB,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,IAAI,CAAC,CAAC,EAAE,KAAK,wBAAwB,CAAC,CAAC;YAChF,MAAM,CAAC,EAAE,CAAC,UAAU,EAAE,mDAAmD,CAAC,CAAC;YAC3E,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,GAAG,EAAE,oBAAoB,CAAC,CAAC;YACpD,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,EAAE,6BAA6B,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;YAC1C,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;QACxD,CAAC;gBAAS,CAAC;YACV,UAAU,CAAC,KAAK,GAAG,SAAS,CAAC;QAC9B,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import 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 { ModelDiscoveryCache } from \"./discovery-cache.js\";\nimport { getDefaultTTL, getDiscoverableProviders, getDiscoveryAdapter } from \"./model-discovery.js\";\nimport { ModelRegistry } from \"./model-registry.js\";\n\nlet testDir: string;\n\nbeforeEach(() => {\n\ttestDir = join(tmpdir(), `model-registry-discovery-test-${Date.now()}-${Math.random().toString(36).slice(2)}`);\n\tmkdirSync(testDir, { recursive: true });\n});\n\nafterEach(() => {\n\ttry {\n\t\trmSync(testDir, { recursive: true, force: true });\n\t} catch {\n\t\t// Cleanup best-effort\n\t}\n});\n\n// ─── discovery cache integration ─────────────────────────────────────────────\n\ndescribe(\"ModelDiscoveryCache — integration with discovery\", () => {\n\tit(\"cache respects provider-specific TTLs\", () => {\n\t\tconst cachePath = join(testDir, \"cache.json\");\n\t\tconst cache = new ModelDiscoveryCache(cachePath);\n\n\t\tcache.set(\"ollama\", [{ id: \"llama2\" }]);\n\t\tconst entry = cache.get(\"ollama\");\n\t\tassert.ok(entry);\n\t\tassert.equal(entry.ttlMs, getDefaultTTL(\"ollama\"));\n\t});\n\n\tit(\"cache uses custom TTL when provided\", () => {\n\t\tconst cachePath = join(testDir, \"cache.json\");\n\t\tconst cache = new ModelDiscoveryCache(cachePath);\n\n\t\tcache.set(\"openai\", [{ id: \"gpt-4o\" }], 999);\n\t\tconst entry = cache.get(\"openai\");\n\t\tassert.ok(entry);\n\t\tassert.equal(entry.ttlMs, 999);\n\t});\n});\n\n// ─── adapter resolution ─────────────────────────────────────────────────────\n\ndescribe(\"Discovery adapter resolution\", () => {\n\tit(\"all discoverable providers have adapters\", () => {\n\t\tconst providers = getDiscoverableProviders();\n\t\tfor (const provider of providers) {\n\t\t\tconst adapter = getDiscoveryAdapter(provider);\n\t\t\tassert.equal(adapter.supportsDiscovery, true, `${provider} should support discovery`);\n\t\t}\n\t});\n\n\tit(\"static adapters return empty model lists\", async () => {\n\t\tconst staticProviders = [\"anthropic\", \"bedrock\", \"azure-openai\", \"groq\", \"cerebras\"];\n\t\tfor (const provider of staticProviders) {\n\t\t\tconst adapter = getDiscoveryAdapter(provider);\n\t\t\tassert.equal(adapter.supportsDiscovery, false, `${provider} should not support discovery`);\n\t\t\tconst models = await adapter.fetchModels(\"dummy-key\");\n\t\t\tassert.deepEqual(models, [], `${provider} should return empty models`);\n\t\t}\n\t});\n});\n\n// ─── AuthStorage hasAuth for discovery ───────────────────────────────────────\n\ndescribe(\"AuthStorage — hasAuth for discovery providers\", () => {\n\tit(\"returns false for providers without auth\", () => {\n\t\tconst storage = AuthStorage.inMemory({});\n\t\tassert.equal(storage.hasAuth(\"openai\"), false);\n\t\tassert.equal(storage.hasAuth(\"ollama\"), false);\n\t});\n\n\tit(\"returns true for providers with stored keys\", () => {\n\t\tconst storage = AuthStorage.inMemory({\n\t\t\topenai: { type: \"api_key\" as const, key: \"sk-test\" },\n\t\t});\n\t\tassert.equal(storage.hasAuth(\"openai\"), true);\n\t\tassert.equal(storage.hasAuth(\"ollama\"), false);\n\t});\n});\n\n// ─── cache persistence across instances ──────────────────────────────────────\n\ndescribe(\"ModelDiscoveryCache — persistence\", () => {\n\tit(\"data survives across cache instances\", () => {\n\t\tconst cachePath = join(testDir, \"persist.json\");\n\n\t\tconst cache1 = new ModelDiscoveryCache(cachePath);\n\t\tcache1.set(\"openai\", [\n\t\t\t{ id: \"gpt-4o\", name: \"GPT-4o\", contextWindow: 128000 },\n\t\t\t{ id: \"gpt-4o-mini\", name: \"GPT-4o Mini\" },\n\t\t]);\n\n\t\tconst cache2 = new ModelDiscoveryCache(cachePath);\n\t\tconst entry = cache2.get(\"openai\");\n\t\tassert.ok(entry);\n\t\tassert.equal(entry.models.length, 2);\n\t\tassert.equal(entry.models[0].contextWindow, 128000);\n\t});\n\n\tit(\"clear persists across instances\", () => {\n\t\tconst cachePath = join(testDir, \"clear.json\");\n\n\t\tconst cache1 = new ModelDiscoveryCache(cachePath);\n\t\tcache1.set(\"openai\", [{ id: \"gpt-4o\" }]);\n\t\tcache1.clear(\"openai\");\n\n\t\tconst cache2 = new ModelDiscoveryCache(cachePath);\n\t\tassert.equal(cache2.get(\"openai\"), undefined);\n\t});\n});\n\n// ─── discovery TTL values ────────────────────────────────────────────────────\n\ndescribe(\"Discovery TTL configuration\", () => {\n\tit(\"ollama has shortest TTL (local models change often)\", () => {\n\t\tconst ollamaTTL = getDefaultTTL(\"ollama\");\n\t\tconst openaiTTL = getDefaultTTL(\"openai\");\n\t\tassert.ok(ollamaTTL < openaiTTL, \"ollama TTL should be shorter than openai\");\n\t});\n\n\tit(\"unknown providers get default TTL\", () => {\n\t\tconst customTTL = getDefaultTTL(\"my-custom-provider\");\n\t\tconst defaultTTL = getDefaultTTL(\"default\");\n\t\t// Unknown providers should get the same TTL as the explicit \"default\" key\n\t\tassert.equal(customTTL, defaultTTL);\n\t});\n});\n\ndescribe(\"ModelRegistry discovery — OpenAI-compatible custom providers\", () => {\n\tit(\"discovers custom OpenAI-compatible providers and maps capability metadata\", async () => {\n\t\tconst providerName = `minimax-openai-${Date.now()}-${Math.random().toString(36).slice(2)}`;\n\t\tconst modelsPath = join(testDir, \"models.json\");\n\t\twriteFileSync(\n\t\t\tmodelsPath,\n\t\t\tJSON.stringify(\n\t\t\t\t{\n\t\t\t\t\tproviders: {\n\t\t\t\t\t\t[providerName]: {\n\t\t\t\t\t\t\tbaseUrl: \"https://api.minimax.example\",\n\t\t\t\t\t\t\tapiKey: \"minimax-test-key\",\n\t\t\t\t\t\t\tapi: \"openai-completions\",\n\t\t\t\t\t\t\tmodels: [{ id: \"bootstrap-model\" }],\n\t\t\t\t\t\t},\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t\tnull,\n\t\t\t\t2,\n\t\t\t),\n\t\t\t\"utf-8\",\n\t\t);\n\n\t\tconst prevFetch = globalThis.fetch;\n\t\tlet requestedUrl = \"\";\n\t\tglobalThis.fetch = (async (input: string | URL | Request) => {\n\t\t\trequestedUrl = typeof input === \"string\" ? input : input instanceof URL ? input.toString() : input.url;\n\t\t\treturn new Response(\n\t\t\t\tJSON.stringify({\n\t\t\t\t\tdata: [\n\t\t\t\t\t\t{\n\t\t\t\t\t\t\tid: \"MiniMax-M2.7-highspeed\",\n\t\t\t\t\t\t\tname: \"MiniMax M2.7 Highspeed\",\n\t\t\t\t\t\t\tcontext_window: 165000,\n\t\t\t\t\t\t\tmax_output_tokens: 32768,\n\t\t\t\t\t\t\tsupports_reasoning: true,\n\t\t\t\t\t\t\tinput_modalities: [\"text\", \"image\"],\n\t\t\t\t\t\t},\n\t\t\t\t\t],\n\t\t\t\t}),\n\t\t\t\t{\n\t\t\t\t\tstatus: 200,\n\t\t\t\t\theaders: { \"content-type\": \"application/json\" },\n\t\t\t\t},\n\t\t\t);\n\t\t}) as typeof globalThis.fetch;\n\n\t\ttry {\n\t\t\tconst registry = new ModelRegistry(AuthStorage.inMemory({}), modelsPath);\n\t\t\t// Guard against global cache leakage from prior test runs.\n\t\t\tregistry.getDiscoveryCache().clear(providerName);\n\t\t\tconst results = await registry.discoverModels([providerName]);\n\n\t\t\tconst discovery = results.find((r) => r.provider === providerName);\n\t\t\tassert.ok(discovery, \"discovery result should include custom provider\");\n\t\t\tassert.equal(discovery?.error, undefined, \"custom provider discovery should succeed\");\n\t\t\tassert.equal(requestedUrl, \"https://api.minimax.example/v1/models\");\n\n\t\t\tconst discovered = registry\n\t\t\t\t.getAllWithDiscovered()\n\t\t\t\t.find((m) => m.provider === providerName && m.id === \"MiniMax-M2.7-highspeed\");\n\t\t\tassert.ok(discovered, \"discovered model should be merged into model list\");\n\t\t\tassert.equal(discovered?.api, \"openai-completions\");\n\t\t\tassert.equal(discovered?.baseUrl, \"https://api.minimax.example\");\n\t\t\tassert.equal(discovered?.contextWindow, 165000);\n\t\t\tassert.equal(discovered?.maxTokens, 32768);\n\t\t\tassert.equal(discovered?.reasoning, true);\n\t\t\tassert.deepEqual(discovered?.input, [\"text\", \"image\"]);\n\t\t} finally {\n\t\t\tglobalThis.fetch = prevFetch;\n\t\t}\n\t});\n});\n"]}

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

@@ -115,6 +115,11 @@ export declare class ModelRegistry {
      * Convert DiscoveryResult[] into Model<Api>[] with default values.
      */
     private convertDiscoveredModels;
+    private getProviderApis;
+    private getAutoDiscoverableProviders;
+    private getProviderBaseUrl;
+    private getDiscoveryProviderDefaults;
+    private getDiscoveryTtl;
     /**
      * Check if a model's baseUrl points to a local endpoint.
      * Delegates to standalone isLocalModel() function.

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;IA+CtE;;;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;IAsB/B;;;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

@@ -9,7 +9,7 @@ import { existsSync, readFileSync } from "fs";
 import { join } from "path";
 import { getAgentDir } from "../config.js";
 import { ModelDiscoveryCache } from "./discovery-cache.js";
-import { getDiscoverableProviders, getDiscoveryAdapter } from "./model-discovery.js";
+import { getDefaultTTL, getDiscoverableProviders, getDiscoveryAdapter, supportsDiscoveryForApi } from "./model-discovery.js";
 import { resolveConfigValue, resolveHeaders } from "./resolve-config-value.js";
 import { isLocalModel } from "./local-model-check.js";
 const Ajv = AjvModule.default || AjvModule;
@@ -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,
                 });
             }
         }
@@ -640,10 +654,11 @@ export class ModelRegistry {
      * Results are cached and merged into the registry (never overrides existing models).
      */
     async discoverModels(providers) {
-        const targetProviders = providers ?? getDiscoverableProviders();
+        const targetProviders = providers ?? this.getAutoDiscoverableProviders();
         const results = [];
         for (const providerName of targetProviders) {
-            const adapter = getDiscoveryAdapter(providerName);
+            const providerApis = this.getProviderApis(providerName);
+            const adapter = getDiscoveryAdapter(providerName, providerApis);
             if (!adapter.supportsDiscovery)
                 continue;
             // Skip if cache is still fresh
@@ -662,8 +677,10 @@ export class ModelRegistry {
                 const apiKey = await this.authStorage.getApiKey(providerName);
                 if (!apiKey && !this.isProviderRequestReady(providerName))
                     continue;
-                const models = await adapter.fetchModels(apiKey ?? "", undefined);
-                this.discoveryCache.set(providerName, models);
+                const baseUrl = this.getProviderBaseUrl(providerName);
+                const models = await adapter.fetchModels(apiKey ?? "", baseUrl);
+                const ttlMs = this.getDiscoveryTtl(providerName, providerApis);
+                this.discoveryCache.set(providerName, models, ttlMs);
                 results.push({
                     provider: providerName,
                     models,
@@ -712,23 +729,86 @@ export class ModelRegistry {
         for (const result of results) {
             if (result.error)
                 continue;
+            const providerDefaults = this.getDiscoveryProviderDefaults(result.provider);
             for (const dm of result.models) {
                 converted.push({
                     id: dm.id,
                     name: dm.name ?? dm.id,
-                    api: "openai",
+                    api: providerDefaults.api,
                     provider: result.provider,
-                    baseUrl: "",
+                    baseUrl: providerDefaults.baseUrl,
                     reasoning: dm.reasoning ?? false,
-                    input: dm.input ?? ["text"],
+                    input: dm.input ?? providerDefaults.input,
                     cost: dm.cost ?? { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-                    contextWindow: dm.contextWindow ?? 128000,
-                    maxTokens: dm.maxTokens ?? 16384,
+                    contextWindow: dm.contextWindow ?? providerDefaults.contextWindow,
+                    maxTokens: dm.maxTokens ?? providerDefaults.maxTokens,
                 });
             }
         }
         return converted;
     }
+    getProviderApis(provider) {
+        const apis = new Set();
+        for (const model of this.models) {
+            if (model.provider === provider && typeof model.api === "string" && model.api.length > 0) {
+                apis.add(model.api);
+            }
+        }
+        const providerConfig = this.registeredProviders.get(provider);
+        if (providerConfig?.api)
+            apis.add(providerConfig.api);
+        for (const modelDef of providerConfig?.models ?? []) {
+            if (modelDef.api)
+                apis.add(modelDef.api);
+        }
+        return apis;
+    }
+    getAutoDiscoverableProviders() {
+        const discoverable = new Set(getDiscoverableProviders());
+        for (const provider of new Set(this.models.map((m) => m.provider))) {
+            const apis = this.getProviderApis(provider);
+            for (const api of apis) {
+                if (supportsDiscoveryForApi(api)) {
+                    discoverable.add(provider);
+                    break;
+                }
+            }
+        }
+        return [...discoverable];
+    }
+    getProviderBaseUrl(provider) {
+        const fromModels = this.models.find((m) => m.provider === provider && typeof m.baseUrl === "string" && m.baseUrl.length > 0);
+        if (fromModels?.baseUrl)
+            return fromModels.baseUrl;
+        return this.registeredProviders.get(provider)?.baseUrl;
+    }
+    getDiscoveryProviderDefaults(provider) {
+        const first = this.models.find((m) => m.provider === provider);
+        if (first) {
+            return {
+                api: first.api,
+                baseUrl: first.baseUrl,
+                input: first.input,
+                contextWindow: first.contextWindow,
+                maxTokens: first.maxTokens,
+            };
+        }
+        return {
+            api: "openai-completions",
+            baseUrl: this.registeredProviders.get(provider)?.baseUrl ?? "",
+            input: ["text"],
+            contextWindow: 128000,
+            maxTokens: 16384,
+        };
+    }
+    getDiscoveryTtl(provider, providerApis) {
+        for (const api of providerApis) {
+            if (supportsDiscoveryForApi(api)) {
+                return getDefaultTTL("openai");
+            }
+        }
+        return getDefaultTTL(provider);
+    }
     /**
      * Check if a model's baseUrl points to a local endpoint.
      * Delegates to standalone isLocalModel() function.