gsd-pi 2.76.0 → 2.77.0-dev.1d17f366c

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.