gsd-pi 2.17.0 → 2.18.0

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

@@ -0,0 +1,35 @@
+/**
+ * Provider discovery adapters for runtime model enumeration.
+ * Each adapter implements ProviderDiscoveryAdapter to fetch models from provider APIs.
+ */
+export interface DiscoveredModel {
+    id: string;
+    name?: string;
+    contextWindow?: number;
+    maxTokens?: number;
+    reasoning?: boolean;
+    input?: ("text" | "image")[];
+    cost?: {
+        input: number;
+        output: number;
+        cacheRead: number;
+        cacheWrite: number;
+    };
+}
+export interface DiscoveryResult {
+    provider: string;
+    models: DiscoveredModel[];
+    fetchedAt: number;
+    error?: string;
+}
+export interface ProviderDiscoveryAdapter {
+    provider: string;
+    supportsDiscovery: boolean;
+    fetchModels(apiKey: string, baseUrl?: string): Promise<DiscoveredModel[]>;
+}
+/** Per-provider TTLs in milliseconds */
+export declare const DISCOVERY_TTLS: Record<string, number>;
+export declare function getDefaultTTL(provider: string): number;
+export declare function getDiscoveryAdapter(provider: string): ProviderDiscoveryAdapter;
+export declare function getDiscoverableProviders(): string[];
+//# sourceMappingURL=model-discovery.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"model-discovery.d.ts","sourceRoot":"","sources":["../../src/core/model-discovery.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,MAAM,WAAW,eAAe;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,KAAK,CAAC,EAAE,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;IAC7B,IAAI,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE,CAAC;CAChF;AAED,MAAM,WAAW,eAAe;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,eAAe,EAAE,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,wBAAwB;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;CAC1E;AAED,wCAAwC;AACxC,eAAO,MAAM,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAMjD,CAAC;AAEF,wBAAgB,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAEtD;AAuLD,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,wBAAwB,CAE9E;AAED,wBAAgB,wBAAwB,IAAI,MAAM,EAAE,CAInD"}

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

@@ -0,0 +1,162 @@
+/**
+ * Provider discovery adapters for runtime model enumeration.
+ * Each adapter implements ProviderDiscoveryAdapter to fetch models from provider APIs.
+ */
+/** Per-provider TTLs in milliseconds */
+export const DISCOVERY_TTLS = {
+    ollama: 5 * 60 * 1000, // 5 minutes (local, models change often)
+    openai: 60 * 60 * 1000, // 1 hour
+    google: 60 * 60 * 1000, // 1 hour
+    openrouter: 60 * 60 * 1000, // 1 hour
+    default: 24 * 60 * 60 * 1000, // 24 hours
+};
+export function getDefaultTTL(provider) {
+    return DISCOVERY_TTLS[provider] ?? DISCOVERY_TTLS.default;
+}
+async function fetchWithTimeout(url, options = {}, timeoutMs = 5000) {
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), timeoutMs);
+    try {
+        return await fetch(url, { ...options, signal: controller.signal });
+    }
+    finally {
+        clearTimeout(timeout);
+    }
+}
+// ─── OpenAI Adapter ──────────────────────────────────────────────────────────
+const OPENAI_EXCLUDED_PREFIXES = ["embedding", "tts", "dall-e", "whisper", "text-embedding", "davinci", "babbage"];
+class OpenAIDiscoveryAdapter {
+    constructor() {
+        this.provider = "openai";
+        this.supportsDiscovery = true;
+    }
+    async fetchModels(apiKey, baseUrl) {
+        const url = `${baseUrl ?? "https://api.openai.com"}/v1/models`;
+        const response = await fetchWithTimeout(url, {
+            headers: { Authorization: `Bearer ${apiKey}` },
+        });
+        if (!response.ok) {
+            throw new Error(`OpenAI models API returned ${response.status}: ${response.statusText}`);
+        }
+        const data = (await response.json());
+        return data.data
+            .filter((m) => !OPENAI_EXCLUDED_PREFIXES.some((prefix) => m.id.startsWith(prefix)))
+            .map((m) => ({
+            id: m.id,
+            name: m.id,
+            input: ["text", "image"],
+        }));
+    }
+}
+// ─── Ollama Adapter ──────────────────────────────────────────────────────────
+class OllamaDiscoveryAdapter {
+    constructor() {
+        this.provider = "ollama";
+        this.supportsDiscovery = true;
+    }
+    async fetchModels(_apiKey, baseUrl) {
+        const url = `${baseUrl ?? "http://localhost:11434"}/api/tags`;
+        const response = await fetchWithTimeout(url);
+        if (!response.ok) {
+            throw new Error(`Ollama tags API returned ${response.status}: ${response.statusText}`);
+        }
+        const data = (await response.json());
+        return (data.models ?? []).map((m) => ({
+            id: m.name,
+            name: m.name,
+            input: ["text"],
+        }));
+    }
+}
+// ─── OpenRouter Adapter ──────────────────────────────────────────────────────
+class OpenRouterDiscoveryAdapter {
+    constructor() {
+        this.provider = "openrouter";
+        this.supportsDiscovery = true;
+    }
+    async fetchModels(apiKey, baseUrl) {
+        const url = `${baseUrl ?? "https://openrouter.ai"}/api/v1/models`;
+        const response = await fetchWithTimeout(url, {
+            headers: { Authorization: `Bearer ${apiKey}` },
+        });
+        if (!response.ok) {
+            throw new Error(`OpenRouter models API returned ${response.status}: ${response.statusText}`);
+        }
+        const data = (await response.json());
+        return (data.data ?? []).map((m) => {
+            const cost = m.pricing?.prompt !== undefined && m.pricing?.completion !== undefined
+                ? {
+                    input: parseFloat(m.pricing.prompt) * 1_000_000,
+                    output: parseFloat(m.pricing.completion) * 1_000_000,
+                    cacheRead: 0,
+                    cacheWrite: 0,
+                }
+                : undefined;
+            return {
+                id: m.id,
+                name: m.name,
+                contextWindow: m.context_length,
+                maxTokens: m.top_provider?.max_completion_tokens,
+                cost,
+                input: ["text", "image"],
+            };
+        });
+    }
+}
+// ─── Google/Gemini Adapter ───────────────────────────────────────────────────
+class GoogleDiscoveryAdapter {
+    constructor() {
+        this.provider = "google";
+        this.supportsDiscovery = true;
+    }
+    async fetchModels(apiKey, baseUrl) {
+        const url = `${baseUrl ?? "https://generativelanguage.googleapis.com"}/v1beta/models?key=${apiKey}`;
+        const response = await fetchWithTimeout(url);
+        if (!response.ok) {
+            throw new Error(`Google models API returned ${response.status}: ${response.statusText}`);
+        }
+        const data = (await response.json());
+        return (data.models ?? [])
+            .filter((m) => m.supportedGenerationMethods?.includes("generateContent"))
+            .map((m) => ({
+            id: m.name.replace("models/", ""),
+            name: m.displayName,
+            contextWindow: m.inputTokenLimit,
+            maxTokens: m.outputTokenLimit,
+            input: ["text", "image"],
+        }));
+    }
+}
+// ─── Static Adapter (no discovery) ───────────────────────────────────────────
+class StaticDiscoveryAdapter {
+    constructor(provider) {
+        this.supportsDiscovery = false;
+        this.provider = provider;
+    }
+    async fetchModels() {
+        return [];
+    }
+}
+// ─── Registry ────────────────────────────────────────────────────────────────
+const adapters = {
+    openai: new OpenAIDiscoveryAdapter(),
+    ollama: new OllamaDiscoveryAdapter(),
+    openrouter: new OpenRouterDiscoveryAdapter(),
+    google: new GoogleDiscoveryAdapter(),
+    anthropic: new StaticDiscoveryAdapter("anthropic"),
+    bedrock: new StaticDiscoveryAdapter("bedrock"),
+    "azure-openai": new StaticDiscoveryAdapter("azure-openai"),
+    groq: new StaticDiscoveryAdapter("groq"),
+    cerebras: new StaticDiscoveryAdapter("cerebras"),
+    xai: new StaticDiscoveryAdapter("xai"),
+    mistral: new StaticDiscoveryAdapter("mistral"),
+};
+export function getDiscoveryAdapter(provider) {
+    return adapters[provider] ?? new StaticDiscoveryAdapter(provider);
+}
+export function getDiscoverableProviders() {
+    return Object.entries(adapters)
+        .filter(([, adapter]) => adapter.supportsDiscovery)
+        .map(([name]) => name);
+}
+//# sourceMappingURL=model-discovery.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"model-discovery.js","sourceRoot":"","sources":["../../src/core/model-discovery.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAyBH,wCAAwC;AACxC,MAAM,CAAC,MAAM,cAAc,GAA2B;IACrD,MAAM,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,yCAAyC;IAChE,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,SAAS;IACjC,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,SAAS;IACjC,UAAU,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,SAAS;IACrC,OAAO,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,EAAE,WAAW;CACzC,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,QAAgB;IAC7C,OAAO,cAAc,CAAC,QAAQ,CAAC,IAAI,cAAc,CAAC,OAAO,CAAC;AAC3D,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,GAAW,EAAE,UAAuB,EAAE,EAAE,SAAS,GAAG,IAAI;IACvF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,SAAS,CAAC,CAAC;IAChE,IAAI,CAAC;QACJ,OAAO,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;IACpE,CAAC;YAAS,CAAC;QACV,YAAY,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;AACF,CAAC;AAED,gFAAgF;AAEhF,MAAM,wBAAwB,GAAG,CAAC,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,gBAAgB,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;AAEnH,MAAM,sBAAsB;IAA5B;QACC,aAAQ,GAAG,QAAQ,CAAC;QACpB,sBAAiB,GAAG,IAAI,CAAC;IAqB1B,CAAC;IAnBA,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,OAAgB;QACjD,MAAM,GAAG,GAAG,GAAG,OAAO,IAAI,wBAAwB,YAAY,CAAC;QAC/D,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE;YAC5C,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;SAC9C,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1F,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAuD,CAAC;QAC3F,OAAO,IAAI,CAAC,IAAI;aACd,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;aAClF,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACZ,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,EAAE;YACV,KAAK,EAAE,CAAC,MAAe,EAAE,OAAgB,CAAC;SAC1C,CAAC,CAAC,CAAC;IACN,CAAC;CACD;AAED,gFAAgF;AAEhF,MAAM,sBAAsB;IAA5B;QACC,aAAQ,GAAG,QAAQ,CAAC;QACpB,sBAAiB,GAAG,IAAI,CAAC;IAoB1B,CAAC;IAlBA,KAAK,CAAC,WAAW,CAAC,OAAe,EAAE,OAAgB;QAClD,MAAM,GAAG,GAAG,GAAG,OAAO,IAAI,wBAAwB,WAAW,CAAC;QAC9D,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAE7C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,4BAA4B,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAElC,CAAC;QAEF,OAAO,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtC,EAAE,EAAE,CAAC,CAAC,IAAI;YACV,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,KAAK,EAAE,CAAC,MAAe,CAAC;SACxB,CAAC,CAAC,CAAC;IACL,CAAC;CACD;AAED,gFAAgF;AAEhF,MAAM,0BAA0B;IAAhC;QACC,aAAQ,GAAG,YAAY,CAAC;QACxB,sBAAiB,GAAG,IAAI,CAAC;IA2C1B,CAAC;IAzCA,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,OAAgB;QACjD,MAAM,GAAG,GAAG,GAAG,OAAO,IAAI,uBAAuB,gBAAgB,CAAC;QAClE,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,GAAG,EAAE;YAC5C,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;SAC9C,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,kCAAkC,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAC9F,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAQlC,CAAC;QAEF,OAAO,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAClC,MAAM,IAAI,GACT,CAAC,CAAC,OAAO,EAAE,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,OAAO,EAAE,UAAU,KAAK,SAAS;gBACrE,CAAC,CAAC;oBACA,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,SAAS;oBAC/C,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,SAAS;oBACpD,SAAS,EAAE,CAAC;oBACZ,UAAU,EAAE,CAAC;iBACb;gBACF,CAAC,CAAC,SAAS,CAAC;YAEd,OAAO;gBACN,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,aAAa,EAAE,CAAC,CAAC,cAAc;gBAC/B,SAAS,EAAE,CAAC,CAAC,YAAY,EAAE,qBAAqB;gBAChD,IAAI;gBACJ,KAAK,EAAE,CAAC,MAAe,EAAE,OAAgB,CAAC;aAC1C,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC;CACD;AAED,gFAAgF;AAEhF,MAAM,sBAAsB;IAA5B;QACC,aAAQ,GAAG,QAAQ,CAAC;QACpB,sBAAiB,GAAG,IAAI,CAAC;IA8B1B,CAAC;IA5BA,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,OAAgB;QACjD,MAAM,GAAG,GAAG,GAAG,OAAO,IAAI,2CAA2C,sBAAsB,MAAM,EAAE,CAAC;QACpG,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAE7C,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,8BAA8B,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QAC1F,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAQlC,CAAC;QAEF,OAAO,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;aACxB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,0BAA0B,EAAE,QAAQ,CAAC,iBAAiB,CAAC,CAAC;aACxE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACZ,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;YACjC,IAAI,EAAE,CAAC,CAAC,WAAW;YACnB,aAAa,EAAE,CAAC,CAAC,eAAe;YAChC,SAAS,EAAE,CAAC,CAAC,gBAAgB;YAC7B,KAAK,EAAE,CAAC,MAAe,EAAE,OAAgB,CAAC;SAC1C,CAAC,CAAC,CAAC;IACN,CAAC;CACD;AAED,gFAAgF;AAEhF,MAAM,sBAAsB;IAI3B,YAAY,QAAgB;QAF5B,sBAAiB,GAAG,KAAK,CAAC;QAGzB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,WAAW;QAChB,OAAO,EAAE,CAAC;IACX,CAAC;CACD;AAED,gFAAgF;AAEhF,MAAM,QAAQ,GAA6C;IAC1D,MAAM,EAAE,IAAI,sBAAsB,EAAE;IACpC,MAAM,EAAE,IAAI,sBAAsB,EAAE;IACpC,UAAU,EAAE,IAAI,0BAA0B,EAAE;IAC5C,MAAM,EAAE,IAAI,sBAAsB,EAAE;IACpC,SAAS,EAAE,IAAI,sBAAsB,CAAC,WAAW,CAAC;IAClD,OAAO,EAAE,IAAI,sBAAsB,CAAC,SAAS,CAAC;IAC9C,cAAc,EAAE,IAAI,sBAAsB,CAAC,cAAc,CAAC;IAC1D,IAAI,EAAE,IAAI,sBAAsB,CAAC,MAAM,CAAC;IACxC,QAAQ,EAAE,IAAI,sBAAsB,CAAC,UAAU,CAAC;IAChD,GAAG,EAAE,IAAI,sBAAsB,CAAC,KAAK,CAAC;IACtC,OAAO,EAAE,IAAI,sBAAsB,CAAC,SAAS,CAAC;CAC9C,CAAC;AAEF,MAAM,UAAU,mBAAmB,CAAC,QAAgB;IACnD,OAAO,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,sBAAsB,CAAC,QAAQ,CAAC,CAAC;AACnE,CAAC;AAED,MAAM,UAAU,wBAAwB;IACvC,OAAO,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC;SAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC;SAClD,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;AACzB,CAAC","sourcesContent":["/**\n * Provider discovery adapters for runtime model enumeration.\n * Each adapter implements ProviderDiscoveryAdapter to fetch models from provider APIs.\n */\n\nexport interface DiscoveredModel {\n\tid: string;\n\tname?: string;\n\tcontextWindow?: number;\n\tmaxTokens?: number;\n\treasoning?: boolean;\n\tinput?: (\"text\" | \"image\")[];\n\tcost?: { input: number; output: number; cacheRead: number; cacheWrite: number };\n}\n\nexport interface DiscoveryResult {\n\tprovider: string;\n\tmodels: DiscoveredModel[];\n\tfetchedAt: number;\n\terror?: string;\n}\n\nexport interface ProviderDiscoveryAdapter {\n\tprovider: string;\n\tsupportsDiscovery: boolean;\n\tfetchModels(apiKey: string, baseUrl?: string): Promise<DiscoveredModel[]>;\n}\n\n/** Per-provider TTLs in milliseconds */\nexport const DISCOVERY_TTLS: Record<string, number> = {\n\tollama: 5 * 60 * 1000, // 5 minutes (local, models change often)\n\topenai: 60 * 60 * 1000, // 1 hour\n\tgoogle: 60 * 60 * 1000, // 1 hour\n\topenrouter: 60 * 60 * 1000, // 1 hour\n\tdefault: 24 * 60 * 60 * 1000, // 24 hours\n};\n\nexport function getDefaultTTL(provider: string): number {\n\treturn DISCOVERY_TTLS[provider] ?? DISCOVERY_TTLS.default;\n}\n\nasync function fetchWithTimeout(url: string, options: RequestInit = {}, timeoutMs = 5000): Promise<Response> {\n\tconst controller = new AbortController();\n\tconst timeout = setTimeout(() => controller.abort(), timeoutMs);\n\ttry {\n\t\treturn await fetch(url, { ...options, signal: controller.signal });\n\t} finally {\n\t\tclearTimeout(timeout);\n\t}\n}\n\n// ─── OpenAI Adapter ──────────────────────────────────────────────────────────\n\nconst OPENAI_EXCLUDED_PREFIXES = [\"embedding\", \"tts\", \"dall-e\", \"whisper\", \"text-embedding\", \"davinci\", \"babbage\"];\n\nclass OpenAIDiscoveryAdapter implements ProviderDiscoveryAdapter {\n\tprovider = \"openai\";\n\tsupportsDiscovery = true;\n\n\tasync fetchModels(apiKey: string, baseUrl?: string): Promise<DiscoveredModel[]> {\n\t\tconst url = `${baseUrl ?? \"https://api.openai.com\"}/v1/models`;\n\t\tconst response = await fetchWithTimeout(url, {\n\t\t\theaders: { Authorization: `Bearer ${apiKey}` },\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tthrow new Error(`OpenAI models API returned ${response.status}: ${response.statusText}`);\n\t\t}\n\n\t\tconst data = (await response.json()) as { data: Array<{ id: string; owned_by?: string }> };\n\t\treturn data.data\n\t\t\t.filter((m) => !OPENAI_EXCLUDED_PREFIXES.some((prefix) => m.id.startsWith(prefix)))\n\t\t\t.map((m) => ({\n\t\t\t\tid: m.id,\n\t\t\t\tname: m.id,\n\t\t\t\tinput: [\"text\" as const, \"image\" as const],\n\t\t\t}));\n\t}\n}\n\n// ─── Ollama Adapter ──────────────────────────────────────────────────────────\n\nclass OllamaDiscoveryAdapter implements ProviderDiscoveryAdapter {\n\tprovider = \"ollama\";\n\tsupportsDiscovery = true;\n\n\tasync fetchModels(_apiKey: string, baseUrl?: string): Promise<DiscoveredModel[]> {\n\t\tconst url = `${baseUrl ?? \"http://localhost:11434\"}/api/tags`;\n\t\tconst response = await fetchWithTimeout(url);\n\n\t\tif (!response.ok) {\n\t\t\tthrow new Error(`Ollama tags API returned ${response.status}: ${response.statusText}`);\n\t\t}\n\n\t\tconst data = (await response.json()) as {\n\t\t\tmodels: Array<{ name: string; size: number; details?: { parameter_size?: string } }>;\n\t\t};\n\n\t\treturn (data.models ?? []).map((m) => ({\n\t\t\tid: m.name,\n\t\t\tname: m.name,\n\t\t\tinput: [\"text\" as const],\n\t\t}));\n\t}\n}\n\n// ─── OpenRouter Adapter ──────────────────────────────────────────────────────\n\nclass OpenRouterDiscoveryAdapter implements ProviderDiscoveryAdapter {\n\tprovider = \"openrouter\";\n\tsupportsDiscovery = true;\n\n\tasync fetchModels(apiKey: string, baseUrl?: string): Promise<DiscoveredModel[]> {\n\t\tconst url = `${baseUrl ?? \"https://openrouter.ai\"}/api/v1/models`;\n\t\tconst response = await fetchWithTimeout(url, {\n\t\t\theaders: { Authorization: `Bearer ${apiKey}` },\n\t\t});\n\n\t\tif (!response.ok) {\n\t\t\tthrow new Error(`OpenRouter models API returned ${response.status}: ${response.statusText}`);\n\t\t}\n\n\t\tconst data = (await response.json()) as {\n\t\t\tdata: Array<{\n\t\t\t\tid: string;\n\t\t\t\tname: string;\n\t\t\t\tcontext_length?: number;\n\t\t\t\ttop_provider?: { max_completion_tokens?: number };\n\t\t\t\tpricing?: { prompt: string; completion: string };\n\t\t\t}>;\n\t\t};\n\n\t\treturn (data.data ?? []).map((m) => {\n\t\t\tconst cost =\n\t\t\t\tm.pricing?.prompt !== undefined && m.pricing?.completion !== undefined\n\t\t\t\t\t? {\n\t\t\t\t\t\t\tinput: parseFloat(m.pricing.prompt) * 1_000_000,\n\t\t\t\t\t\t\toutput: parseFloat(m.pricing.completion) * 1_000_000,\n\t\t\t\t\t\t\tcacheRead: 0,\n\t\t\t\t\t\t\tcacheWrite: 0,\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined;\n\n\t\t\treturn {\n\t\t\t\tid: m.id,\n\t\t\t\tname: m.name,\n\t\t\t\tcontextWindow: m.context_length,\n\t\t\t\tmaxTokens: m.top_provider?.max_completion_tokens,\n\t\t\t\tcost,\n\t\t\t\tinput: [\"text\" as const, \"image\" as const],\n\t\t\t};\n\t\t});\n\t}\n}\n\n// ─── Google/Gemini Adapter ───────────────────────────────────────────────────\n\nclass GoogleDiscoveryAdapter implements ProviderDiscoveryAdapter {\n\tprovider = \"google\";\n\tsupportsDiscovery = true;\n\n\tasync fetchModels(apiKey: string, baseUrl?: string): Promise<DiscoveredModel[]> {\n\t\tconst url = `${baseUrl ?? \"https://generativelanguage.googleapis.com\"}/v1beta/models?key=${apiKey}`;\n\t\tconst response = await fetchWithTimeout(url);\n\n\t\tif (!response.ok) {\n\t\t\tthrow new Error(`Google models API returned ${response.status}: ${response.statusText}`);\n\t\t}\n\n\t\tconst data = (await response.json()) as {\n\t\t\tmodels: Array<{\n\t\t\t\tname: string;\n\t\t\t\tdisplayName: string;\n\t\t\t\tsupportedGenerationMethods?: string[];\n\t\t\t\tinputTokenLimit?: number;\n\t\t\t\toutputTokenLimit?: number;\n\t\t\t}>;\n\t\t};\n\n\t\treturn (data.models ?? [])\n\t\t\t.filter((m) => m.supportedGenerationMethods?.includes(\"generateContent\"))\n\t\t\t.map((m) => ({\n\t\t\t\tid: m.name.replace(\"models/\", \"\"),\n\t\t\t\tname: m.displayName,\n\t\t\t\tcontextWindow: m.inputTokenLimit,\n\t\t\t\tmaxTokens: m.outputTokenLimit,\n\t\t\t\tinput: [\"text\" as const, \"image\" as const],\n\t\t\t}));\n\t}\n}\n\n// ─── Static Adapter (no discovery) ───────────────────────────────────────────\n\nclass StaticDiscoveryAdapter implements ProviderDiscoveryAdapter {\n\tprovider: string;\n\tsupportsDiscovery = false;\n\n\tconstructor(provider: string) {\n\t\tthis.provider = provider;\n\t}\n\n\tasync fetchModels(): Promise<DiscoveredModel[]> {\n\t\treturn [];\n\t}\n}\n\n// ─── Registry ────────────────────────────────────────────────────────────────\n\nconst adapters: Record<string, ProviderDiscoveryAdapter> = {\n\topenai: new OpenAIDiscoveryAdapter(),\n\tollama: new OllamaDiscoveryAdapter(),\n\topenrouter: new OpenRouterDiscoveryAdapter(),\n\tgoogle: new GoogleDiscoveryAdapter(),\n\tanthropic: new StaticDiscoveryAdapter(\"anthropic\"),\n\tbedrock: new StaticDiscoveryAdapter(\"bedrock\"),\n\t\"azure-openai\": new StaticDiscoveryAdapter(\"azure-openai\"),\n\tgroq: new StaticDiscoveryAdapter(\"groq\"),\n\tcerebras: new StaticDiscoveryAdapter(\"cerebras\"),\n\txai: new StaticDiscoveryAdapter(\"xai\"),\n\tmistral: new StaticDiscoveryAdapter(\"mistral\"),\n};\n\nexport function getDiscoveryAdapter(provider: string): ProviderDiscoveryAdapter {\n\treturn adapters[provider] ?? new StaticDiscoveryAdapter(provider);\n}\n\nexport function getDiscoverableProviders(): string[] {\n\treturn Object.entries(adapters)\n\t\t.filter(([, adapter]) => adapter.supportsDiscovery)\n\t\t.map(([name]) => name);\n}\n"]}

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

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

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

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

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

@@ -0,0 +1,100 @@
+import assert from "node:assert/strict";
+import { describe, it } from "node:test";
+import { DISCOVERY_TTLS, getDefaultTTL, getDiscoverableProviders, getDiscoveryAdapter, } from "./model-discovery.js";
+// ─── getDiscoveryAdapter ─────────────────────────────────────────────────────
+describe("getDiscoveryAdapter", () => {
+    it("returns an adapter for openai", () => {
+        const adapter = getDiscoveryAdapter("openai");
+        assert.equal(adapter.provider, "openai");
+        assert.equal(adapter.supportsDiscovery, true);
+    });
+    it("returns an adapter for ollama", () => {
+        const adapter = getDiscoveryAdapter("ollama");
+        assert.equal(adapter.provider, "ollama");
+        assert.equal(adapter.supportsDiscovery, true);
+    });
+    it("returns an adapter for openrouter", () => {
+        const adapter = getDiscoveryAdapter("openrouter");
+        assert.equal(adapter.provider, "openrouter");
+        assert.equal(adapter.supportsDiscovery, true);
+    });
+    it("returns an adapter for google", () => {
+        const adapter = getDiscoveryAdapter("google");
+        assert.equal(adapter.provider, "google");
+        assert.equal(adapter.supportsDiscovery, true);
+    });
+    it("returns a static adapter for anthropic", () => {
+        const adapter = getDiscoveryAdapter("anthropic");
+        assert.equal(adapter.provider, "anthropic");
+        assert.equal(adapter.supportsDiscovery, false);
+    });
+    it("returns a static adapter for bedrock", () => {
+        const adapter = getDiscoveryAdapter("bedrock");
+        assert.equal(adapter.provider, "bedrock");
+        assert.equal(adapter.supportsDiscovery, false);
+    });
+    it("returns a static adapter for unknown providers", () => {
+        const adapter = getDiscoveryAdapter("unknown-provider");
+        assert.equal(adapter.provider, "unknown-provider");
+        assert.equal(adapter.supportsDiscovery, false);
+    });
+    it("static adapter fetchModels returns empty array", async () => {
+        const adapter = getDiscoveryAdapter("anthropic");
+        const models = await adapter.fetchModels("key");
+        assert.deepEqual(models, []);
+    });
+});
+// ─── getDiscoverableProviders ────────────────────────────────────────────────
+describe("getDiscoverableProviders", () => {
+    it("returns only providers that support discovery", () => {
+        const providers = getDiscoverableProviders();
+        assert.ok(providers.includes("openai"));
+        assert.ok(providers.includes("ollama"));
+        assert.ok(providers.includes("openrouter"));
+        assert.ok(providers.includes("google"));
+        assert.ok(!providers.includes("anthropic"));
+        assert.ok(!providers.includes("bedrock"));
+    });
+    it("returns an array of strings", () => {
+        const providers = getDiscoverableProviders();
+        assert.ok(Array.isArray(providers));
+        for (const p of providers) {
+            assert.equal(typeof p, "string");
+        }
+    });
+});
+// ─── getDefaultTTL ───────────────────────────────────────────────────────────
+describe("getDefaultTTL", () => {
+    it("returns 5 minutes for ollama", () => {
+        assert.equal(getDefaultTTL("ollama"), 5 * 60 * 1000);
+    });
+    it("returns 1 hour for openai", () => {
+        assert.equal(getDefaultTTL("openai"), 60 * 60 * 1000);
+    });
+    it("returns 1 hour for google", () => {
+        assert.equal(getDefaultTTL("google"), 60 * 60 * 1000);
+    });
+    it("returns 1 hour for openrouter", () => {
+        assert.equal(getDefaultTTL("openrouter"), 60 * 60 * 1000);
+    });
+    it("returns 24 hours for unknown providers", () => {
+        assert.equal(getDefaultTTL("some-custom"), 24 * 60 * 60 * 1000);
+    });
+});
+// ─── DISCOVERY_TTLS ──────────────────────────────────────────────────────────
+describe("DISCOVERY_TTLS", () => {
+    it("has expected keys", () => {
+        assert.ok("ollama" in DISCOVERY_TTLS);
+        assert.ok("openai" in DISCOVERY_TTLS);
+        assert.ok("google" in DISCOVERY_TTLS);
+        assert.ok("openrouter" in DISCOVERY_TTLS);
+        assert.ok("default" in DISCOVERY_TTLS);
+    });
+    it("all values are positive numbers", () => {
+        for (const [, value] of Object.entries(DISCOVERY_TTLS)) {
+            assert.equal(typeof value, "number");
+            assert.ok(value > 0);
+        }
+    });
+});
+//# sourceMappingURL=model-discovery.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"model-discovery.test.js","sourceRoot":"","sources":["../../src/core/model-discovery.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EACN,cAAc,EACd,aAAa,EACb,wBAAwB,EACxB,mBAAmB,GACnB,MAAM,sBAAsB,CAAC;AAE9B,gFAAgF;AAEhF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC5C,MAAM,OAAO,GAAG,mBAAmB,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC7C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACjD,MAAM,OAAO,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC/C,MAAM,OAAO,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACzD,MAAM,OAAO,GAAG,mBAAmB,CAAC,kBAAkB,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,kBAAkB,CAAC,CAAC;QACnD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,OAAO,GAAG,mBAAmB,CAAC,WAAW,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACzC,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACxD,MAAM,SAAS,GAAG,wBAAwB,EAAE,CAAC;QAC7C,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACtC,MAAM,SAAS,GAAG,wBAAwB,EAAE,CAAC;QAC7C,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;QACpC,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YAC3B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,QAAQ,CAAC,CAAC;QAClC,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACvC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACpC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACpC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,YAAY,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,gFAAgF;AAEhF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC5B,MAAM,CAAC,EAAE,CAAC,QAAQ,IAAI,cAAc,CAAC,CAAC;QACtC,MAAM,CAAC,EAAE,CAAC,QAAQ,IAAI,cAAc,CAAC,CAAC;QACtC,MAAM,CAAC,EAAE,CAAC,QAAQ,IAAI,cAAc,CAAC,CAAC;QACtC,MAAM,CAAC,EAAE,CAAC,YAAY,IAAI,cAAc,CAAC,CAAC;QAC1C,MAAM,CAAC,EAAE,CAAC,SAAS,IAAI,cAAc,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC1C,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;YACxD,MAAM,CAAC,KAAK,CAAC,OAAO,KAAK,EAAE,QAAQ,CAAC,CAAC;YACrC,MAAM,CAAC,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QACtB,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { describe, it } from \"node:test\";\nimport {\n\tDISCOVERY_TTLS,\n\tgetDefaultTTL,\n\tgetDiscoverableProviders,\n\tgetDiscoveryAdapter,\n} from \"./model-discovery.js\";\n\n// ─── getDiscoveryAdapter ─────────────────────────────────────────────────────\n\ndescribe(\"getDiscoveryAdapter\", () => {\n\tit(\"returns an adapter for openai\", () => {\n\t\tconst adapter = getDiscoveryAdapter(\"openai\");\n\t\tassert.equal(adapter.provider, \"openai\");\n\t\tassert.equal(adapter.supportsDiscovery, true);\n\t});\n\n\tit(\"returns an adapter for ollama\", () => {\n\t\tconst adapter = getDiscoveryAdapter(\"ollama\");\n\t\tassert.equal(adapter.provider, \"ollama\");\n\t\tassert.equal(adapter.supportsDiscovery, true);\n\t});\n\n\tit(\"returns an adapter for openrouter\", () => {\n\t\tconst adapter = getDiscoveryAdapter(\"openrouter\");\n\t\tassert.equal(adapter.provider, \"openrouter\");\n\t\tassert.equal(adapter.supportsDiscovery, true);\n\t});\n\n\tit(\"returns an adapter for google\", () => {\n\t\tconst adapter = getDiscoveryAdapter(\"google\");\n\t\tassert.equal(adapter.provider, \"google\");\n\t\tassert.equal(adapter.supportsDiscovery, true);\n\t});\n\n\tit(\"returns a static adapter for anthropic\", () => {\n\t\tconst adapter = getDiscoveryAdapter(\"anthropic\");\n\t\tassert.equal(adapter.provider, \"anthropic\");\n\t\tassert.equal(adapter.supportsDiscovery, false);\n\t});\n\n\tit(\"returns a static adapter for bedrock\", () => {\n\t\tconst adapter = getDiscoveryAdapter(\"bedrock\");\n\t\tassert.equal(adapter.provider, \"bedrock\");\n\t\tassert.equal(adapter.supportsDiscovery, false);\n\t});\n\n\tit(\"returns a static adapter for unknown providers\", () => {\n\t\tconst adapter = getDiscoveryAdapter(\"unknown-provider\");\n\t\tassert.equal(adapter.provider, \"unknown-provider\");\n\t\tassert.equal(adapter.supportsDiscovery, false);\n\t});\n\n\tit(\"static adapter fetchModels returns empty array\", async () => {\n\t\tconst adapter = getDiscoveryAdapter(\"anthropic\");\n\t\tconst models = await adapter.fetchModels(\"key\");\n\t\tassert.deepEqual(models, []);\n\t});\n});\n\n// ─── getDiscoverableProviders ────────────────────────────────────────────────\n\ndescribe(\"getDiscoverableProviders\", () => {\n\tit(\"returns only providers that support discovery\", () => {\n\t\tconst providers = getDiscoverableProviders();\n\t\tassert.ok(providers.includes(\"openai\"));\n\t\tassert.ok(providers.includes(\"ollama\"));\n\t\tassert.ok(providers.includes(\"openrouter\"));\n\t\tassert.ok(providers.includes(\"google\"));\n\t\tassert.ok(!providers.includes(\"anthropic\"));\n\t\tassert.ok(!providers.includes(\"bedrock\"));\n\t});\n\n\tit(\"returns an array of strings\", () => {\n\t\tconst providers = getDiscoverableProviders();\n\t\tassert.ok(Array.isArray(providers));\n\t\tfor (const p of providers) {\n\t\t\tassert.equal(typeof p, \"string\");\n\t\t}\n\t});\n});\n\n// ─── getDefaultTTL ───────────────────────────────────────────────────────────\n\ndescribe(\"getDefaultTTL\", () => {\n\tit(\"returns 5 minutes for ollama\", () => {\n\t\tassert.equal(getDefaultTTL(\"ollama\"), 5 * 60 * 1000);\n\t});\n\n\tit(\"returns 1 hour for openai\", () => {\n\t\tassert.equal(getDefaultTTL(\"openai\"), 60 * 60 * 1000);\n\t});\n\n\tit(\"returns 1 hour for google\", () => {\n\t\tassert.equal(getDefaultTTL(\"google\"), 60 * 60 * 1000);\n\t});\n\n\tit(\"returns 1 hour for openrouter\", () => {\n\t\tassert.equal(getDefaultTTL(\"openrouter\"), 60 * 60 * 1000);\n\t});\n\n\tit(\"returns 24 hours for unknown providers\", () => {\n\t\tassert.equal(getDefaultTTL(\"some-custom\"), 24 * 60 * 60 * 1000);\n\t});\n});\n\n// ─── DISCOVERY_TTLS ──────────────────────────────────────────────────────────\n\ndescribe(\"DISCOVERY_TTLS\", () => {\n\tit(\"has expected keys\", () => {\n\t\tassert.ok(\"ollama\" in DISCOVERY_TTLS);\n\t\tassert.ok(\"openai\" in DISCOVERY_TTLS);\n\t\tassert.ok(\"google\" in DISCOVERY_TTLS);\n\t\tassert.ok(\"openrouter\" in DISCOVERY_TTLS);\n\t\tassert.ok(\"default\" in DISCOVERY_TTLS);\n\t});\n\n\tit(\"all values are positive numbers\", () => {\n\t\tfor (const [, value] of Object.entries(DISCOVERY_TTLS)) {\n\t\t\tassert.equal(typeof value, \"number\");\n\t\t\tassert.ok(value > 0);\n\t\t}\n\t});\n});\n"]}

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

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

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

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

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

@@ -0,0 +1,113 @@
+import assert from "node:assert/strict";
+import { mkdirSync, rmSync } 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";
+let testDir;
+beforeEach(() => {
+    testDir = join(tmpdir(), `model-registry-discovery-test-${Date.now()}-${Math.random().toString(36).slice(2)}`);
+    mkdirSync(testDir, { recursive: true });
+});
+afterEach(() => {
+    try {
+        rmSync(testDir, { recursive: true, force: true });
+    }
+    catch {
+        // Cleanup best-effort
+    }
+});
+// ─── discovery cache integration ─────────────────────────────────────────────
+describe("ModelDiscoveryCache — integration with discovery", () => {
+    it("cache respects provider-specific TTLs", () => {
+        const cachePath = join(testDir, "cache.json");
+        const cache = new ModelDiscoveryCache(cachePath);
+        cache.set("ollama", [{ id: "llama2" }]);
+        const entry = cache.get("ollama");
+        assert.ok(entry);
+        assert.equal(entry.ttlMs, getDefaultTTL("ollama"));
+    });
+    it("cache uses custom TTL when provided", () => {
+        const cachePath = join(testDir, "cache.json");
+        const cache = new ModelDiscoveryCache(cachePath);
+        cache.set("openai", [{ id: "gpt-4o" }], 999);
+        const entry = cache.get("openai");
+        assert.ok(entry);
+        assert.equal(entry.ttlMs, 999);
+    });
+});
+// ─── adapter resolution ─────────────────────────────────────────────────────
+describe("Discovery adapter resolution", () => {
+    it("all discoverable providers have adapters", () => {
+        const providers = getDiscoverableProviders();
+        for (const provider of providers) {
+            const adapter = getDiscoveryAdapter(provider);
+            assert.equal(adapter.supportsDiscovery, true, `${provider} should support discovery`);
+        }
+    });
+    it("static adapters return empty model lists", async () => {
+        const staticProviders = ["anthropic", "bedrock", "azure-openai", "groq", "cerebras"];
+        for (const provider of staticProviders) {
+            const adapter = getDiscoveryAdapter(provider);
+            assert.equal(adapter.supportsDiscovery, false, `${provider} should not support discovery`);
+            const models = await adapter.fetchModels("dummy-key");
+            assert.deepEqual(models, [], `${provider} should return empty models`);
+        }
+    });
+});
+// ─── AuthStorage hasAuth for discovery ───────────────────────────────────────
+describe("AuthStorage — hasAuth for discovery providers", () => {
+    it("returns false for providers without auth", () => {
+        const storage = AuthStorage.inMemory({});
+        assert.equal(storage.hasAuth("openai"), false);
+        assert.equal(storage.hasAuth("ollama"), false);
+    });
+    it("returns true for providers with stored keys", () => {
+        const storage = AuthStorage.inMemory({
+            openai: { type: "api_key", key: "sk-test" },
+        });
+        assert.equal(storage.hasAuth("openai"), true);
+        assert.equal(storage.hasAuth("ollama"), false);
+    });
+});
+// ─── cache persistence across instances ──────────────────────────────────────
+describe("ModelDiscoveryCache — persistence", () => {
+    it("data survives across cache instances", () => {
+        const cachePath = join(testDir, "persist.json");
+        const cache1 = new ModelDiscoveryCache(cachePath);
+        cache1.set("openai", [
+            { id: "gpt-4o", name: "GPT-4o", contextWindow: 128000 },
+            { id: "gpt-4o-mini", name: "GPT-4o Mini" },
+        ]);
+        const cache2 = new ModelDiscoveryCache(cachePath);
+        const entry = cache2.get("openai");
+        assert.ok(entry);
+        assert.equal(entry.models.length, 2);
+        assert.equal(entry.models[0].contextWindow, 128000);
+    });
+    it("clear persists across instances", () => {
+        const cachePath = join(testDir, "clear.json");
+        const cache1 = new ModelDiscoveryCache(cachePath);
+        cache1.set("openai", [{ id: "gpt-4o" }]);
+        cache1.clear("openai");
+        const cache2 = new ModelDiscoveryCache(cachePath);
+        assert.equal(cache2.get("openai"), undefined);
+    });
+});
+// ─── discovery TTL values ────────────────────────────────────────────────────
+describe("Discovery TTL configuration", () => {
+    it("ollama has shortest TTL (local models change often)", () => {
+        const ollamaTTL = getDefaultTTL("ollama");
+        const openaiTTL = getDefaultTTL("openai");
+        assert.ok(ollamaTTL < openaiTTL, "ollama TTL should be shorter than openai");
+    });
+    it("unknown providers get default TTL", () => {
+        const customTTL = getDefaultTTL("my-custom-provider");
+        const defaultTTL = getDefaultTTL("default");
+        // Unknown providers should get the same TTL as the explicit "default" key
+        assert.equal(customTTL, defaultTTL);
+    });
+});
+//# sourceMappingURL=model-registry-discovery.test.js.map

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

@@ -0,0 +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"]}

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

@@ -3,6 +3,8 @@
  */
 import { type Api, type AssistantMessageEventStream, type Context, type Model, type OAuthProviderInterface, type SimpleStreamOptions } from "@gsd/pi-ai";
 import type { AuthStorage } from "./auth-storage.js";
+import { ModelDiscoveryCache } from "./discovery-cache.js";
+import type { DiscoveryResult } from "./model-discovery.js";
 import { clearConfigValueCache } from "./resolve-config-value.js";
 /** Clear the config value command cache. Exported for testing. */
 export declare const clearApiKeyCache: typeof clearConfigValueCache;
@@ -13,6 +15,8 @@ export declare class ModelRegistry {
     readonly authStorage: AuthStorage;
     private modelsJsonPath;
     private models;
+    private discoveredModels;
+    private discoveryCache;
     private customProviderApiKeys;
     private registeredProviders;
     private loadError;
@@ -80,6 +84,28 @@ export declare class ModelRegistry {
      */
     unregisterProvider(providerName: string): void;
     private applyProviderConfig;
+    /**
+     * Discover models from all providers that support discovery.
+     * Results are cached and merged into the registry (never overrides existing models).
+     */
+    discoverModels(providers?: string[]): Promise<DiscoveryResult[]>;
+    /**
+     * Get all models including discovered ones.
+     * Discovered models are appended but never override existing models.
+     */
+    getAllWithDiscovered(): Model<Api>[];
+    /**
+     * Check if a model was added via discovery (not built-in or custom).
+     */
+    isDiscovered(model: Model<Api>): boolean;
+    /**
+     * Get the discovery cache instance.
+     */
+    getDiscoveryCache(): ModelDiscoveryCache;
+    /**
+     * Convert DiscoveryResult[] into Model<Api>[] with default values.
+     */
+    private convertDiscoveredModels;
 }
 /**
  * Input type for registerProvider API.

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,EACR,KAAK,2BAA2B,EAChC,KAAK,OAAO,EAIZ,KAAK,KAAK,EACV,KAAK,sBAAsB,EAK3B,KAAK,mBAAmB,EACxB,MAAM,YAAY,CAAC;AAOpB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,qBAAqB,EAAsC,MAAM,2BAA2B,CAAC;AA6LtG,kEAAkE;AAClE,eAAO,MAAM,gBAAgB,8BAAwB,CAAC;AAEtD;;GAEG;AACH,qBAAa,aAAa;IAOxB,QAAQ,CAAC,WAAW,EAAE,WAAW;IACjC,OAAO,CAAC,cAAc;IAPvB,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,qBAAqB,CAAkC;IAC/D,OAAO,CAAC,mBAAmB,CAA+C;IAC1E,OAAO,CAAC,SAAS,CAAiC;gBAGxC,WAAW,EAAE,WAAW,EACzB,cAAc,GAAE,MAAM,GAAG,SAA8C;IAehF;;OAEG;IACH,OAAO,IAAI,IAAI;IAef;;OAEG;IACH,QAAQ,IAAI,MAAM,GAAG,SAAS;IAI9B,OAAO,CAAC,UAAU;IA4BlB,8DAA8D;IAC9D,OAAO,CAAC,iBAAiB;IAiCzB,wFAAwF;IACxF,OAAO,CAAC,iBAAiB;IAazB,OAAO,CAAC,gBAAgB;IAuDxB,OAAO,CAAC,cAAc;IAyCtB,OAAO,CAAC,WAAW;IAqDnB;;;OAGG;IACH,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAItB;;;OAGG;IACH,YAAY,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAI5B;;OAEG;IACH,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS;IAI/D;;;OAGG;IACG,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAInF;;;OAGG;IACG,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAI7F;;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;CAmG3B;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IACnC,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;KAC9B,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,EACR,KAAK,2BAA2B,EAChC,KAAK,OAAO,EAIZ,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;AAE7E,OAAO,EAAE,qBAAqB,EAAsC,MAAM,2BAA2B,CAAC;AA6LtG,kEAAkE;AAClE,eAAO,MAAM,gBAAgB,8BAAwB,CAAC;AAEtD;;GAEG;AACH,qBAAa,aAAa;IASxB,QAAQ,CAAC,WAAW,EAAE,WAAW;IACjC,OAAO,CAAC,cAAc;IATvB,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,EACzB,cAAc,GAAE,MAAM,GAAG,SAA8C;IAiBhF;;OAEG;IACH,OAAO,IAAI,IAAI;IAef;;OAEG;IACH,QAAQ,IAAI,MAAM,GAAG,SAAS;IAI9B,OAAO,CAAC,UAAU;IA4BlB,8DAA8D;IAC9D,OAAO,CAAC,iBAAiB;IAiCzB,wFAAwF;IACxF,OAAO,CAAC,iBAAiB;IAazB,OAAO,CAAC,gBAAgB;IAuDxB,OAAO,CAAC,cAAc;IAyCtB,OAAO,CAAC,WAAW;IAqDnB;;;OAGG;IACH,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAItB;;;OAGG;IACH,YAAY,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE;IAI5B;;OAEG;IACH,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS;IAI/D;;;OAGG;IACG,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAInF;;;OAGG;IACG,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAI7F;;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;IAoG3B;;;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;CAqB/B;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IACnC,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;KAC9B,CAAC,CAAC;CACH"}