@bastani/atomic 0.8.31-alpha.4 → 0.9.0-alpha.1

package/dist/core/model-registry-custom-loader.js

@@ -0,0 +1,213 @@
+import { getModels, getProviders } from "@earendil-works/pi-ai";
+import { existsSync, readFileSync } from "fs";
+import { normalizeContextWindowOptions, validateContextWindowValue } from "./context-window.js";
+import { mergeCompat, mergeCustomModels } from "./model-registry-builtins.js";
+import { formatValidationPath, stripJsonComments, validateModelsConfig, } from "./model-registry-schemas.js";
+function modelRequestKey(provider, modelId) {
+    return `${provider}:${modelId}`;
+}
+function emptyCustomModelsResult(error) {
+    return {
+        models: [],
+        overrides: new Map(),
+        modelOverrides: new Map(),
+        providerRequestConfigs: new Map(),
+        modelRequestHeaders: new Map(),
+        error,
+    };
+}
+function mergeCustomModelResults(base, incoming) {
+    return {
+        models: mergeCustomModels(base.models, incoming.models),
+        overrides: new Map([...base.overrides, ...incoming.overrides]),
+        modelOverrides: new Map([...base.modelOverrides, ...incoming.modelOverrides]),
+        providerRequestConfigs: new Map([...base.providerRequestConfigs, ...incoming.providerRequestConfigs]),
+        modelRequestHeaders: new Map([...base.modelRequestHeaders, ...incoming.modelRequestHeaders]),
+        error: undefined,
+    };
+}
+function collectProviderRequestConfig(providerName, config, requestConfigs) {
+    if (!config.apiKey && !config.headers && !config.authHeader)
+        return;
+    requestConfigs.set(providerName, {
+        apiKey: config.apiKey,
+        headers: config.headers,
+        authHeader: config.authHeader,
+    });
+}
+function collectModelHeaders(providerName, modelId, headers, modelHeaders) {
+    if (!headers || Object.keys(headers).length === 0)
+        return;
+    modelHeaders.set(modelRequestKey(providerName, modelId), headers);
+}
+function validateContextWindowOptions(providerName, modelId, options) {
+    for (const option of options ?? []) {
+        if (validateContextWindowValue(option)) {
+            throw new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindowOptions value`);
+        }
+    }
+}
+function validateConfig(config) {
+    const builtInProviders = new Set(getProviders());
+    for (const [providerName, providerConfig] of Object.entries(config.providers)) {
+        const isBuiltIn = builtInProviders.has(providerName);
+        const hasProviderApi = !!providerConfig.api;
+        const models = providerConfig.models ?? [];
+        const hasModelOverrides = providerConfig.modelOverrides && Object.keys(providerConfig.modelOverrides).length > 0;
+        if (models.length === 0) {
+            if (!providerConfig.baseUrl && !providerConfig.headers && !providerConfig.compat && !hasModelOverrides) {
+                throw new Error(`Provider ${providerName}: must specify "baseUrl", "headers", "compat", "modelOverrides", or "models".`);
+            }
+        }
+        else if (!isBuiltIn) {
+            if (!providerConfig.baseUrl) {
+                throw new Error(`Provider ${providerName}: "baseUrl" is required when defining custom models.`);
+            }
+            if (!providerConfig.apiKey) {
+                throw new Error(`Provider ${providerName}: "apiKey" is required when defining custom models.`);
+            }
+        }
+        for (const modelDef of models) {
+            const hasModelApi = !!modelDef.api;
+            if (!hasProviderApi && !hasModelApi && !isBuiltIn) {
+                throw new Error(`Provider ${providerName}, model ${modelDef.id}: no "api" specified. Set at provider or model level.`);
+            }
+            if (!modelDef.id)
+                throw new Error(`Provider ${providerName}: model missing "id"`);
+            if (modelDef.contextWindow !== undefined && validateContextWindowValue(modelDef.contextWindow)) {
+                throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);
+            }
+            validateContextWindowOptions(providerName, modelDef.id, modelDef.contextWindowOptions);
+            if (modelDef.maxTokens !== undefined && modelDef.maxTokens <= 0) {
+                throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);
+            }
+        }
+        for (const [modelId, modelOverride] of Object.entries(providerConfig.modelOverrides ?? {})) {
+            if (modelOverride.contextWindow !== undefined && validateContextWindowValue(modelOverride.contextWindow)) {
+                throw new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindow`);
+            }
+            validateContextWindowOptions(providerName, modelId, modelOverride.contextWindowOptions);
+            if (modelOverride.maxTokens !== undefined && modelOverride.maxTokens <= 0) {
+                throw new Error(`Provider ${providerName}, model ${modelId}: invalid maxTokens`);
+            }
+        }
+    }
+}
+function parseModels(config, modelHeaders) {
+    const models = [];
+    const builtInProviders = new Set(getProviders());
+    const builtInDefaultsCache = new Map();
+    const getBuiltInDefaults = (providerName) => {
+        if (!builtInProviders.has(providerName))
+            return undefined;
+        if (builtInDefaultsCache.has(providerName))
+            return builtInDefaultsCache.get(providerName);
+        const builtIn = getModels(providerName);
+        if (builtIn.length === 0)
+            return undefined;
+        const defaults = { api: builtIn[0].api, baseUrl: builtIn[0].baseUrl };
+        builtInDefaultsCache.set(providerName, defaults);
+        return defaults;
+    };
+    for (const [providerName, providerConfig] of Object.entries(config.providers)) {
+        const modelDefs = providerConfig.models ?? [];
+        if (modelDefs.length === 0)
+            continue;
+        const builtInDefaults = getBuiltInDefaults(providerName);
+        for (const modelDef of modelDefs) {
+            const api = modelDef.api ?? providerConfig.api ?? builtInDefaults?.api;
+            if (!api)
+                continue;
+            const baseUrl = modelDef.baseUrl ?? providerConfig.baseUrl ?? builtInDefaults?.baseUrl;
+            if (!baseUrl)
+                continue;
+            const compat = mergeCompat(providerConfig.compat, modelDef.compat);
+            collectModelHeaders(providerName, modelDef.id, modelDef.headers, modelHeaders);
+            const defaultCost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
+            const contextWindow = modelDef.contextWindow ?? 128000;
+            models.push({
+                id: modelDef.id,
+                name: modelDef.name ?? modelDef.id,
+                api: api,
+                provider: providerName,
+                baseUrl,
+                reasoning: modelDef.reasoning ?? false,
+                thinkingLevelMap: modelDef.thinkingLevelMap,
+                input: (modelDef.input ?? ["text"]),
+                cost: modelDef.cost ?? defaultCost,
+                contextWindow,
+                defaultContextWindow: contextWindow,
+                contextWindowOptions: normalizeContextWindowOptions([contextWindow, ...(modelDef.contextWindowOptions ?? [])]),
+                maxTokens: modelDef.maxTokens ?? 16384,
+                headers: undefined,
+                compat,
+            });
+        }
+    }
+    return models;
+}
+function loadCustomModels(modelsJsonPath) {
+    if (!existsSync(modelsJsonPath)) {
+        return emptyCustomModelsResult();
+    }
+    try {
+        const content = readFileSync(modelsJsonPath, "utf-8");
+        const parsed = JSON.parse(stripJsonComments(content));
+        if (!validateModelsConfig.Check(parsed)) {
+            const errors = validateModelsConfig
+                .Errors(parsed)
+                .map((error) => `  - ${formatValidationPath(error)}: ${error.message}`)
+                .join("\n") || "Unknown schema error";
+            return emptyCustomModelsResult(`Invalid models.json schema:\n${errors}\n\nFile: ${modelsJsonPath}`);
+        }
+        const config = parsed;
+        validateConfig(config);
+        const overrides = new Map();
+        const modelOverrides = new Map();
+        const providerRequestConfigs = new Map();
+        const modelRequestHeaders = new Map();
+        for (const [providerName, providerConfig] of Object.entries(config.providers)) {
+            if (providerConfig.baseUrl || providerConfig.compat) {
+                overrides.set(providerName, {
+                    baseUrl: providerConfig.baseUrl,
+                    compat: providerConfig.compat,
+                });
+            }
+            collectProviderRequestConfig(providerName, providerConfig, providerRequestConfigs);
+            if (providerConfig.modelOverrides) {
+                modelOverrides.set(providerName, new Map(Object.entries(providerConfig.modelOverrides)));
+                for (const [modelId, modelOverride] of Object.entries(providerConfig.modelOverrides)) {
+                    collectModelHeaders(providerName, modelId, modelOverride.headers, modelRequestHeaders);
+                }
+            }
+        }
+        return {
+            models: parseModels(config, modelRequestHeaders),
+            overrides,
+            modelOverrides,
+            providerRequestConfigs,
+            modelRequestHeaders,
+            error: undefined,
+        };
+    }
+    catch (error) {
+        if (error instanceof SyntaxError) {
+            return emptyCustomModelsResult(`Failed to parse models.json: ${error.message}\n\nFile: ${modelsJsonPath}`);
+        }
+        return emptyCustomModelsResult(`Failed to load models.json: ${error instanceof Error ? error.message : error}\n\nFile: ${modelsJsonPath}`);
+    }
+}
+export function loadCustomModelsFromPaths(modelsJsonPaths) {
+    let combined = emptyCustomModelsResult();
+    const errors = [];
+    for (let i = modelsJsonPaths.length - 1; i >= 0; i--) {
+        const result = loadCustomModels(modelsJsonPaths[i]);
+        if (result.error) {
+            errors.push(result.error);
+            continue;
+        }
+        combined = mergeCustomModelResults(combined, result);
+    }
+    return { ...combined, error: errors.length > 0 ? errors.join("\n\n") : undefined };
+}
+//# sourceMappingURL=model-registry-custom-loader.js.map

package/dist/core/model-registry-custom-loader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-custom-loader.js","sourceRoot":"","sources":["../../src/core/model-registry-custom-loader.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,SAAS,EAAE,YAAY,EAAkC,MAAM,uBAAuB,CAAC;AAC1G,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,6BAA6B,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAChG,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAC9E,OAAO,EACN,oBAAoB,EAGpB,iBAAiB,EACjB,oBAAoB,GACpB,MAAM,6BAA6B,CAAC;AAGrC,SAAS,eAAe,CAAC,QAAgB,EAAE,OAAe;IACzD,OAAO,GAAG,QAAQ,IAAI,OAAO,EAAE,CAAC;AACjC,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAc;IAC9C,OAAO;QACN,MAAM,EAAE,EAAE;QACV,SAAS,EAAE,IAAI,GAAG,EAAE;QACpB,cAAc,EAAE,IAAI,GAAG,EAAE;QACzB,sBAAsB,EAAE,IAAI,GAAG,EAAE;QACjC,mBAAmB,EAAE,IAAI,GAAG,EAAE;QAC9B,KAAK;KACL,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAAC,IAAwB,EAAE,QAA4B;IACtF,OAAO;QACN,MAAM,EAAE,iBAAiB,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC;QACvD,SAAS,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,EAAE,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC9D,cAAc,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,cAAc,EAAE,GAAG,QAAQ,CAAC,cAAc,CAAC,CAAC;QAC7E,sBAAsB,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,sBAAsB,EAAE,GAAG,QAAQ,CAAC,sBAAsB,CAAC,CAAC;QACrG,mBAAmB,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,mBAAmB,EAAE,GAAG,QAAQ,CAAC,mBAAmB,CAAC,CAAC;QAC5F,KAAK,EAAE,SAAS;KAChB,CAAC;AACH,CAAC;AAED,SAAS,4BAA4B,CACpC,YAAoB,EACpB,MAA6B,EAC7B,cAAkD;IAElD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU;QAAE,OAAO;IACpE,cAAc,CAAC,GAAG,CAAC,YAAY,EAAE;QAChC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,UAAU,EAAE,MAAM,CAAC,UAAU;KAC7B,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAC3B,YAAoB,EACpB,OAAe,EACf,OAA2C,EAC3C,YAAiD;IAEjD,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAC1D,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,YAAY,EAAE,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,4BAA4B,CAAC,YAAoB,EAAE,OAAe,EAAE,OAAsC;IAClH,KAAK,MAAM,MAAM,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;QACpC,IAAI,0BAA0B,CAAC,MAAM,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,OAAO,sCAAsC,CAAC,CAAC;QACnG,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,cAAc,CAAC,MAAoB;IAC3C,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAS,YAAY,EAAE,CAAC,CAAC;IAEzD,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/E,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACrD,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC;QAC5C,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,EAAE,CAAC;QAC3C,MAAM,iBAAiB,GACtB,cAAc,CAAC,cAAc,IAAI,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;QAExF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,cAAc,CAAC,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBACxG,MAAM,IAAI,KAAK,CACd,YAAY,YAAY,+EAA+E,CACvG,CAAC;YACH,CAAC;QACF,CAAC;aAAM,IAAI,CAAC,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;gBAC7B,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,sDAAsD,CAAC,CAAC;YACjG,CAAC;YACD,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,qDAAqD,CAAC,CAAC;YAChG,CAAC;QACF,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;YAC/B,MAAM,WAAW,GAAG,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC;YAEnC,IAAI,CAAC,cAAc,IAAI,CAAC,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnD,MAAM,IAAI,KAAK,CACd,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,uDAAuD,CACrG,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,sBAAsB,CAAC,CAAC;YAClF,IAAI,QAAQ,CAAC,aAAa,KAAK,SAAS,IAAI,0BAA0B,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBAChG,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,yBAAyB,CAAC,CAAC;YAC1F,CAAC;YACD,4BAA4B,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,oBAAoB,CAAC,CAAC;YACvF,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,IAAI,QAAQ,CAAC,SAAS,IAAI,CAAC,EAAE,CAAC;gBACjE,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACtF,CAAC;QACF,CAAC;QAED,KAAK,MAAM,CAAC,OAAO,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE,CAAC;YAC5F,IAAI,aAAa,CAAC,aAAa,KAAK,SAAS,IAAI,0BAA0B,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;gBAC1G,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,OAAO,yBAAyB,CAAC,CAAC;YACtF,CAAC;YACD,4BAA4B,CAAC,YAAY,EAAE,OAAO,EAAE,aAAa,CAAC,oBAAoB,CAAC,CAAC;YACxF,IAAI,aAAa,CAAC,SAAS,KAAK,SAAS,IAAI,aAAa,CAAC,SAAS,IAAI,CAAC,EAAE,CAAC;gBAC3E,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,OAAO,qBAAqB,CAAC,CAAC;YAClF,CAAC;QACF,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,WAAW,CACnB,MAAoB,EACpB,YAAiD;IAEjD,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAS,YAAY,EAAE,CAAC,CAAC;IACzD,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAA4C,CAAC;IACjF,MAAM,kBAAkB,GAAG,CAAC,YAAoB,EAAgD,EAAE;QACjG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC;YAAE,OAAO,SAAS,CAAC;QAC1D,IAAI,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC;YAAE,OAAO,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QAC1F,MAAM,OAAO,GAAG,SAAS,CAAC,YAA6B,CAAiB,CAAC;QACzE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAC3C,MAAM,QAAQ,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QACtE,oBAAoB,CAAC,GAAG,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;QACjD,OAAO,QAAQ,CAAC;IACjB,CAAC,CAAC;IAEF,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/E,MAAM,SAAS,GAAG,cAAc,CAAC,MAAM,IAAI,EAAE,CAAC;QAC9C,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAErC,MAAM,eAAe,GAAG,kBAAkB,CAAC,YAAY,CAAC,CAAC;QAEzD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAClC,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,cAAc,CAAC,GAAG,IAAI,eAAe,EAAE,GAAG,CAAC;YACvE,IAAI,CAAC,GAAG;gBAAE,SAAS;YAEnB,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,IAAI,cAAc,CAAC,OAAO,IAAI,eAAe,EAAE,OAAO,CAAC;YACvF,IAAI,CAAC,OAAO;gBAAE,SAAS;YAEvB,MAAM,MAAM,GAAG,WAAW,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YACnE,mBAAmB,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC;YAE/E,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;YACzE,MAAM,aAAa,GAAG,QAAQ,CAAC,aAAa,IAAI,MAAM,CAAC;YACvD,MAAM,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,EAAE;gBAClC,GAAG,EAAE,GAAU;gBACf,QAAQ,EAAE,YAAY;gBACtB,OAAO;gBACP,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,KAAK;gBACtC,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;gBAC3C,KAAK,EAAE,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,CAAyB;gBAC3D,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,WAAW;gBAClC,aAAa;gBACb,oBAAoB,EAAE,aAAa;gBACnC,oBAAoB,EAAE,6BAA6B,CAAC,CAAC,aAAa,EAAE,GAAG,CAAC,QAAQ,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC9G,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,KAAK;gBACtC,OAAO,EAAE,SAAS;gBAClB,MAAM;aACQ,CAAC,CAAC;QAClB,CAAC;IACF,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAED,SAAS,gBAAgB,CAAC,cAAsB;IAC/C,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACjC,OAAO,uBAAuB,EAAE,CAAC;IAClC,CAAC;IAED,IAAI,CAAC;QACJ,MAAM,OAAO,GAAG,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAY,CAAC;QAEjE,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;YACzC,MAAM,MAAM,GACX,oBAAoB;iBAClB,MAAM,CAAC,MAAM,CAAC;iBACd,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,oBAAoB,CAAC,KAAK,CAAC,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;iBACtE,IAAI,CAAC,IAAI,CAAC,IAAI,sBAAsB,CAAC;YACxC,OAAO,uBAAuB,CAAC,gCAAgC,MAAM,aAAa,cAAc,EAAE,CAAC,CAAC;QACrG,CAAC;QAED,MAAM,MAAM,GAAG,MAAsB,CAAC;QACtC,cAAc,CAAC,MAAM,CAAC,CAAC;QAEvB,MAAM,SAAS,GAAG,IAAI,GAAG,EAA4B,CAAC;QACtD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAsC,CAAC;QACrE,MAAM,sBAAsB,GAAG,IAAI,GAAG,EAAiC,CAAC;QACxE,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAkC,CAAC;QAEtE,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/E,IAAI,cAAc,CAAC,OAAO,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;gBACrD,SAAS,CAAC,GAAG,CAAC,YAAY,EAAE;oBAC3B,OAAO,EAAE,cAAc,CAAC,OAAO;oBAC/B,MAAM,EAAE,cAAc,CAAC,MAA8B;iBACrD,CAAC,CAAC;YACJ,CAAC;YAED,4BAA4B,CAAC,YAAY,EAAE,cAAc,EAAE,sBAAsB,CAAC,CAAC;YAEnF,IAAI,cAAc,CAAC,cAAc,EAAE,CAAC;gBACnC,cAAc,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;gBACzF,KAAK,MAAM,CAAC,OAAO,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,cAAc,CAAC,EAAE,CAAC;oBACtF,mBAAmB,CAAC,YAAY,EAAE,OAAO,EAAE,aAAa,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;gBACxF,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO;YACN,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,mBAAmB,CAAC;YAChD,SAAS;YACT,cAAc;YACd,sBAAsB;YACtB,mBAAmB;YACnB,KAAK,EAAE,SAAS;SAChB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;YAClC,OAAO,uBAAuB,CAAC,gCAAgC,KAAK,CAAC,OAAO,aAAa,cAAc,EAAE,CAAC,CAAC;QAC5G,CAAC;QACD,OAAO,uBAAuB,CAC7B,+BAA+B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,aAAa,cAAc,EAAE,CAC1G,CAAC;IACH,CAAC;AACF,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,eAAyB;IAClE,IAAI,QAAQ,GAAG,uBAAuB,EAAE,CAAC;IACzC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACtD,MAAM,MAAM,GAAG,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAE,CAAC,CAAC;QACrD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC1B,SAAS;QACV,CAAC;QACD,QAAQ,GAAG,uBAAuB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,EAAE,GAAG,QAAQ,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;AACpF,CAAC","sourcesContent":["import { type Api, getModels, getProviders, type KnownProvider, type Model } from \"@earendil-works/pi-ai\";\nimport { existsSync, readFileSync } from \"fs\";\nimport { normalizeContextWindowOptions, validateContextWindowValue } from \"./context-window.ts\";\nimport { mergeCompat, mergeCustomModels } from \"./model-registry-builtins.ts\";\nimport {\n\tformatValidationPath,\n\ttype ModelsConfig,\n\ttype ModelOverride,\n\tstripJsonComments,\n\tvalidateModelsConfig,\n} from \"./model-registry-schemas.ts\";\nimport type { CustomModelsResult, ProviderOverride, ProviderRequestConfig } from \"./model-registry-types.ts\";\n\nfunction modelRequestKey(provider: string, modelId: string): string {\n\treturn `${provider}:${modelId}`;\n}\n\nfunction emptyCustomModelsResult(error?: string): CustomModelsResult {\n\treturn {\n\t\tmodels: [],\n\t\toverrides: new Map(),\n\t\tmodelOverrides: new Map(),\n\t\tproviderRequestConfigs: new Map(),\n\t\tmodelRequestHeaders: new Map(),\n\t\terror,\n\t};\n}\n\nfunction mergeCustomModelResults(base: CustomModelsResult, incoming: CustomModelsResult): CustomModelsResult {\n\treturn {\n\t\tmodels: mergeCustomModels(base.models, incoming.models),\n\t\toverrides: new Map([...base.overrides, ...incoming.overrides]),\n\t\tmodelOverrides: new Map([...base.modelOverrides, ...incoming.modelOverrides]),\n\t\tproviderRequestConfigs: new Map([...base.providerRequestConfigs, ...incoming.providerRequestConfigs]),\n\t\tmodelRequestHeaders: new Map([...base.modelRequestHeaders, ...incoming.modelRequestHeaders]),\n\t\terror: undefined,\n\t};\n}\n\nfunction collectProviderRequestConfig(\n\tproviderName: string,\n\tconfig: ProviderRequestConfig,\n\trequestConfigs: Map<string, ProviderRequestConfig>,\n): void {\n\tif (!config.apiKey && !config.headers && !config.authHeader) return;\n\trequestConfigs.set(providerName, {\n\t\tapiKey: config.apiKey,\n\t\theaders: config.headers,\n\t\tauthHeader: config.authHeader,\n\t});\n}\n\nfunction collectModelHeaders(\n\tproviderName: string,\n\tmodelId: string,\n\theaders: Record<string, string> | undefined,\n\tmodelHeaders: Map<string, Record<string, string>>,\n): void {\n\tif (!headers || Object.keys(headers).length === 0) return;\n\tmodelHeaders.set(modelRequestKey(providerName, modelId), headers);\n}\n\nfunction validateContextWindowOptions(providerName: string, modelId: string, options: readonly number[] | undefined): void {\n\tfor (const option of options ?? []) {\n\t\tif (validateContextWindowValue(option)) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindowOptions value`);\n\t\t}\n\t}\n}\n\nfunction validateConfig(config: ModelsConfig): void {\n\tconst builtInProviders = new Set<string>(getProviders());\n\n\tfor (const [providerName, providerConfig] of Object.entries(config.providers)) {\n\t\tconst isBuiltIn = builtInProviders.has(providerName);\n\t\tconst hasProviderApi = !!providerConfig.api;\n\t\tconst models = providerConfig.models ?? [];\n\t\tconst hasModelOverrides =\n\t\t\tproviderConfig.modelOverrides && Object.keys(providerConfig.modelOverrides).length > 0;\n\n\t\tif (models.length === 0) {\n\t\t\tif (!providerConfig.baseUrl && !providerConfig.headers && !providerConfig.compat && !hasModelOverrides) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t`Provider ${providerName}: must specify \"baseUrl\", \"headers\", \"compat\", \"modelOverrides\", or \"models\".`,\n\t\t\t\t);\n\t\t\t}\n\t\t} else if (!isBuiltIn) {\n\t\t\tif (!providerConfig.baseUrl) {\n\t\t\t\tthrow new Error(`Provider ${providerName}: \"baseUrl\" is required when defining custom models.`);\n\t\t\t}\n\t\t\tif (!providerConfig.apiKey) {\n\t\t\t\tthrow new Error(`Provider ${providerName}: \"apiKey\" is required when defining custom models.`);\n\t\t\t}\n\t\t}\n\n\t\tfor (const modelDef of models) {\n\t\t\tconst hasModelApi = !!modelDef.api;\n\n\t\t\tif (!hasProviderApi && !hasModelApi && !isBuiltIn) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t`Provider ${providerName}, model ${modelDef.id}: no \"api\" specified. Set at provider or model level.`,\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tif (!modelDef.id) throw new Error(`Provider ${providerName}: model missing \"id\"`);\n\t\t\tif (modelDef.contextWindow !== undefined && validateContextWindowValue(modelDef.contextWindow)) {\n\t\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);\n\t\t\t}\n\t\t\tvalidateContextWindowOptions(providerName, modelDef.id, modelDef.contextWindowOptions);\n\t\t\tif (modelDef.maxTokens !== undefined && modelDef.maxTokens <= 0) {\n\t\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);\n\t\t\t}\n\t\t}\n\n\t\tfor (const [modelId, modelOverride] of Object.entries(providerConfig.modelOverrides ?? {})) {\n\t\t\tif (modelOverride.contextWindow !== undefined && validateContextWindowValue(modelOverride.contextWindow)) {\n\t\t\t\tthrow new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindow`);\n\t\t\t}\n\t\t\tvalidateContextWindowOptions(providerName, modelId, modelOverride.contextWindowOptions);\n\t\t\tif (modelOverride.maxTokens !== undefined && modelOverride.maxTokens <= 0) {\n\t\t\t\tthrow new Error(`Provider ${providerName}, model ${modelId}: invalid maxTokens`);\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunction parseModels(\n\tconfig: ModelsConfig,\n\tmodelHeaders: Map<string, Record<string, string>>,\n): Model<Api>[] {\n\tconst models: Model<Api>[] = [];\n\tconst builtInProviders = new Set<string>(getProviders());\n\tconst builtInDefaultsCache = new Map<string, { api: string; baseUrl: string }>();\n\tconst getBuiltInDefaults = (providerName: string): { api: string; baseUrl: string } | undefined => {\n\t\tif (!builtInProviders.has(providerName)) return undefined;\n\t\tif (builtInDefaultsCache.has(providerName)) return builtInDefaultsCache.get(providerName);\n\t\tconst builtIn = getModels(providerName as KnownProvider) as Model<Api>[];\n\t\tif (builtIn.length === 0) return undefined;\n\t\tconst defaults = { api: builtIn[0].api, baseUrl: builtIn[0].baseUrl };\n\t\tbuiltInDefaultsCache.set(providerName, defaults);\n\t\treturn defaults;\n\t};\n\n\tfor (const [providerName, providerConfig] of Object.entries(config.providers)) {\n\t\tconst modelDefs = providerConfig.models ?? [];\n\t\tif (modelDefs.length === 0) continue;\n\n\t\tconst builtInDefaults = getBuiltInDefaults(providerName);\n\n\t\tfor (const modelDef of modelDefs) {\n\t\t\tconst api = modelDef.api ?? providerConfig.api ?? builtInDefaults?.api;\n\t\t\tif (!api) continue;\n\n\t\t\tconst baseUrl = modelDef.baseUrl ?? providerConfig.baseUrl ?? builtInDefaults?.baseUrl;\n\t\t\tif (!baseUrl) continue;\n\n\t\t\tconst compat = mergeCompat(providerConfig.compat, modelDef.compat);\n\t\t\tcollectModelHeaders(providerName, modelDef.id, modelDef.headers, modelHeaders);\n\n\t\t\tconst defaultCost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };\n\t\t\tconst contextWindow = modelDef.contextWindow ?? 128000;\n\t\t\tmodels.push({\n\t\t\t\tid: modelDef.id,\n\t\t\t\tname: modelDef.name ?? modelDef.id,\n\t\t\t\tapi: api as Api,\n\t\t\t\tprovider: providerName,\n\t\t\t\tbaseUrl,\n\t\t\t\treasoning: modelDef.reasoning ?? false,\n\t\t\t\tthinkingLevelMap: modelDef.thinkingLevelMap,\n\t\t\t\tinput: (modelDef.input ?? [\"text\"]) as (\"text\" | \"image\")[],\n\t\t\t\tcost: modelDef.cost ?? defaultCost,\n\t\t\t\tcontextWindow,\n\t\t\t\tdefaultContextWindow: contextWindow,\n\t\t\t\tcontextWindowOptions: normalizeContextWindowOptions([contextWindow, ...(modelDef.contextWindowOptions ?? [])]),\n\t\t\t\tmaxTokens: modelDef.maxTokens ?? 16384,\n\t\t\t\theaders: undefined,\n\t\t\t\tcompat,\n\t\t\t} as Model<Api>);\n\t\t}\n\t}\n\n\treturn models;\n}\n\nfunction loadCustomModels(modelsJsonPath: string): CustomModelsResult {\n\tif (!existsSync(modelsJsonPath)) {\n\t\treturn emptyCustomModelsResult();\n\t}\n\n\ttry {\n\t\tconst content = readFileSync(modelsJsonPath, \"utf-8\");\n\t\tconst parsed = JSON.parse(stripJsonComments(content)) as unknown;\n\n\t\tif (!validateModelsConfig.Check(parsed)) {\n\t\t\tconst errors =\n\t\t\t\tvalidateModelsConfig\n\t\t\t\t\t.Errors(parsed)\n\t\t\t\t\t.map((error) => `  - ${formatValidationPath(error)}: ${error.message}`)\n\t\t\t\t\t.join(\"\\n\") || \"Unknown schema error\";\n\t\t\treturn emptyCustomModelsResult(`Invalid models.json schema:\\n${errors}\\n\\nFile: ${modelsJsonPath}`);\n\t\t}\n\n\t\tconst config = parsed as ModelsConfig;\n\t\tvalidateConfig(config);\n\n\t\tconst overrides = new Map<string, ProviderOverride>();\n\t\tconst modelOverrides = new Map<string, Map<string, ModelOverride>>();\n\t\tconst providerRequestConfigs = new Map<string, ProviderRequestConfig>();\n\t\tconst modelRequestHeaders = new Map<string, Record<string, string>>();\n\n\t\tfor (const [providerName, providerConfig] of Object.entries(config.providers)) {\n\t\t\tif (providerConfig.baseUrl || providerConfig.compat) {\n\t\t\t\toverrides.set(providerName, {\n\t\t\t\t\tbaseUrl: providerConfig.baseUrl,\n\t\t\t\t\tcompat: providerConfig.compat as Model<Api>[\"compat\"],\n\t\t\t\t});\n\t\t\t}\n\n\t\t\tcollectProviderRequestConfig(providerName, providerConfig, providerRequestConfigs);\n\n\t\t\tif (providerConfig.modelOverrides) {\n\t\t\t\tmodelOverrides.set(providerName, new Map(Object.entries(providerConfig.modelOverrides)));\n\t\t\t\tfor (const [modelId, modelOverride] of Object.entries(providerConfig.modelOverrides)) {\n\t\t\t\t\tcollectModelHeaders(providerName, modelId, modelOverride.headers, modelRequestHeaders);\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn {\n\t\t\tmodels: parseModels(config, modelRequestHeaders),\n\t\t\toverrides,\n\t\t\tmodelOverrides,\n\t\t\tproviderRequestConfigs,\n\t\t\tmodelRequestHeaders,\n\t\t\terror: undefined,\n\t\t};\n\t} catch (error) {\n\t\tif (error instanceof SyntaxError) {\n\t\t\treturn emptyCustomModelsResult(`Failed to parse models.json: ${error.message}\\n\\nFile: ${modelsJsonPath}`);\n\t\t}\n\t\treturn emptyCustomModelsResult(\n\t\t\t`Failed to load models.json: ${error instanceof Error ? error.message : error}\\n\\nFile: ${modelsJsonPath}`,\n\t\t);\n\t}\n}\n\nexport function loadCustomModelsFromPaths(modelsJsonPaths: string[]): CustomModelsResult {\n\tlet combined = emptyCustomModelsResult();\n\tconst errors: string[] = [];\n\tfor (let i = modelsJsonPaths.length - 1; i >= 0; i--) {\n\t\tconst result = loadCustomModels(modelsJsonPaths[i]!);\n\t\tif (result.error) {\n\t\t\terrors.push(result.error);\n\t\t\tcontinue;\n\t\t}\n\t\tcombined = mergeCustomModelResults(combined, result);\n\t}\n\treturn { ...combined, error: errors.length > 0 ? errors.join(\"\\n\\n\") : undefined };\n}\n"]}

package/dist/core/model-registry-dynamic.d.ts

@@ -0,0 +1,6 @@
+import { type Api, type Model } from "@earendil-works/pi-ai";
+import type { DynamicProviderApplyInput, ProviderConfigInput } from "./model-registry-types.ts";
+export declare function migrateLegacyRegisterProviderConfigValues(providerName: string, config: ProviderConfigInput): ProviderConfigInput;
+export declare function validateProviderConfig(providerName: string, config: ProviderConfigInput): void;
+export declare function applyProviderConfigToModels(input: DynamicProviderApplyInput): Model<Api>[];
+//# sourceMappingURL=model-registry-dynamic.d.ts.map

package/dist/core/model-registry-dynamic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-dynamic.d.ts","sourceRoot":"","sources":["../../src/core/model-registry-dynamic.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,KAAK,GAAG,EACR,KAAK,KAAK,EAIV,MAAM,uBAAuB,CAAC;AAG/B,OAAO,KAAK,EAAE,yBAAyB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAoChG,wBAAgB,yCAAyC,CACxD,YAAY,EAAE,MAAM,EACpB,MAAM,EAAE,mBAAmB,GACzB,mBAAmB,CA0CrB;AAED,wBAAgB,sBAAsB,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,GAAG,IAAI,CA6B9F;AAED,wBAAgB,2BAA2B,CAAC,KAAK,EAAE,yBAAyB,GAAG,KAAK,CAAC,GAAG,CAAC,EAAE,CAwE1F","sourcesContent":["import {\n\ttype Api,\n\ttype Model,\n\ttype OAuthProviderInterface,\n\tregisterApiProvider,\n\ttype SimpleStreamOptions,\n} from \"@earendil-works/pi-ai\";\nimport { registerOAuthProvider } from \"@earendil-works/pi-ai/oauth\";\nimport { normalizeContextWindowOptions, validateContextWindowValue } from \"./context-window.ts\";\nimport type { DynamicProviderApplyInput, ProviderConfigInput } from \"./model-registry-types.ts\";\nimport { warnDeprecation } from \"../utils/deprecation.ts\";\nimport { isLegacyEnvVarNameConfigValue } from \"./resolve-config-value.ts\";\n\nfunction validateContextWindowOptions(providerName: string, modelId: string, options: readonly number[] | undefined): void {\n\tfor (const option of options ?? []) {\n\t\tif (validateContextWindowValue(option)) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindowOptions value`);\n\t\t}\n\t}\n}\n\nfunction migrateLegacyRegisterProviderConfigValue(providerName: string, field: string, value: string): string {\n\tif (!isLegacyEnvVarNameConfigValue(value) || process.env[value] === undefined) return value;\n\twarnDeprecation(\n\t\t`registerProvider(\"${providerName}\") ${field} value \"${value}\" is treated as a legacy environment variable reference. This will no longer be detected as an environment variable reference in a future release. Pass \"$${value}\" instead.`,\n\t);\n\treturn `$${value}`;\n}\n\nfunction migrateLegacyRegisterProviderHeaders(\n\tproviderName: string,\n\tfield: string,\n\theaders: Record<string, string> | undefined,\n): Record<string, string> | undefined {\n\tif (!headers) return undefined;\n\tlet migratedHeaders: Record<string, string> | undefined;\n\tfor (const [key, value] of Object.entries(headers)) {\n\t\tconst migratedValue = migrateLegacyRegisterProviderConfigValue(providerName, `${field} header \"${key}\"`, value);\n\t\tif (migratedValue === value) continue;\n\t\tmigratedHeaders ??= { ...headers };\n\t\tmigratedHeaders[key] = migratedValue;\n\t}\n\treturn migratedHeaders ?? headers;\n}\n\nexport function migrateLegacyRegisterProviderConfigValues(\n\tproviderName: string,\n\tconfig: ProviderConfigInput,\n): ProviderConfigInput {\n\tlet migratedConfig: ProviderConfigInput | undefined;\n\n\tconst setMigratedConfigValue = <TKey extends keyof ProviderConfigInput>(\n\t\tkey: TKey,\n\t\tvalue: ProviderConfigInput[TKey],\n\t) => {\n\t\tmigratedConfig ??= { ...config };\n\t\tmigratedConfig[key] = value;\n\t};\n\n\tif (config.apiKey) {\n\t\tconst apiKey = migrateLegacyRegisterProviderConfigValue(providerName, \"apiKey\", config.apiKey);\n\t\tif (apiKey !== config.apiKey) {\n\t\t\tsetMigratedConfigValue(\"apiKey\", apiKey);\n\t\t}\n\t}\n\n\tconst headers = migrateLegacyRegisterProviderHeaders(providerName, \"headers\", config.headers);\n\tif (headers !== config.headers) {\n\t\tsetMigratedConfigValue(\"headers\", headers);\n\t}\n\n\tif (config.models) {\n\t\tlet models: ProviderConfigInput[\"models\"] | undefined;\n\t\tfor (let index = 0; index < config.models.length; index++) {\n\t\t\tconst model = config.models[index];\n\t\t\tconst modelHeaders = migrateLegacyRegisterProviderHeaders(\n\t\t\t\tproviderName,\n\t\t\t\t`model \"${model.id}\" headers`,\n\t\t\t\tmodel.headers,\n\t\t\t);\n\t\t\tif (modelHeaders === model.headers) continue;\n\t\t\tmodels ??= [...config.models];\n\t\t\tmodels[index] = { ...model, headers: modelHeaders };\n\t\t}\n\t\tif (models) {\n\t\t\tsetMigratedConfigValue(\"models\", models);\n\t\t}\n\t}\n\n\treturn migratedConfig ?? config;\n}\n\nexport function validateProviderConfig(providerName: string, config: ProviderConfigInput): void {\n\tif (config.streamSimple && !config.api) {\n\t\tthrow new Error(`Provider ${providerName}: \"api\" is required when registering streamSimple.`);\n\t}\n\n\tif (!config.models || config.models.length === 0) {\n\t\treturn;\n\t}\n\n\tif (!config.baseUrl) {\n\t\tthrow new Error(`Provider ${providerName}: \"baseUrl\" is required when defining models.`);\n\t}\n\tif (!config.apiKey && !config.oauth) {\n\t\tthrow new Error(`Provider ${providerName}: \"apiKey\" or \"oauth\" is required when defining models.`);\n\t}\n\n\tfor (const modelDef of config.models) {\n\t\tconst api = modelDef.api || config.api;\n\t\tif (!api) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: no \"api\" specified.`);\n\t\t}\n\t\tif (validateContextWindowValue(modelDef.contextWindow)) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);\n\t\t}\n\t\tvalidateContextWindowOptions(providerName, modelDef.id, modelDef.contextWindowOptions);\n\t\tif (modelDef.maxTokens <= 0) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);\n\t\t}\n\t}\n}\n\nexport function applyProviderConfigToModels(input: DynamicProviderApplyInput): Model<Api>[] {\n\tconst { providerName, config, authStorage, storeProviderRequestConfig, storeModelHeaders } = input;\n\tlet models = input.models;\n\n\tif (config.oauth) {\n\t\tconst oauthProvider: OAuthProviderInterface = {\n\t\t\t...config.oauth,\n\t\t\tid: providerName,\n\t\t};\n\t\tregisterOAuthProvider(oauthProvider);\n\t}\n\n\tif (config.streamSimple) {\n\t\tconst streamSimple = config.streamSimple;\n\t\tregisterApiProvider(\n\t\t\t{\n\t\t\t\tapi: config.api!,\n\t\t\t\tstream: (model, context, options) => streamSimple(model, context, options as SimpleStreamOptions),\n\t\t\t\tstreamSimple,\n\t\t\t},\n\t\t\t`provider:${providerName}`,\n\t\t);\n\t}\n\n\tstoreProviderRequestConfig(providerName, config);\n\n\tif (config.models && config.models.length > 0) {\n\t\tmodels = models.filter((m) => m.provider !== providerName);\n\n\t\tfor (const modelDef of config.models) {\n\t\t\tconst api = modelDef.api || config.api;\n\t\t\tstoreModelHeaders(providerName, modelDef.id, modelDef.headers);\n\n\t\t\tmodels.push({\n\t\t\t\tid: modelDef.id,\n\t\t\t\tname: modelDef.name,\n\t\t\t\tapi: api as Api,\n\t\t\t\tprovider: providerName,\n\t\t\t\tbaseUrl: modelDef.baseUrl ?? config.baseUrl!,\n\t\t\t\treasoning: modelDef.reasoning,\n\t\t\t\tthinkingLevelMap: modelDef.thinkingLevelMap,\n\t\t\t\tinput: modelDef.input as (\"text\" | \"image\")[],\n\t\t\t\tcost: modelDef.cost,\n\t\t\t\tcontextWindow: modelDef.contextWindow,\n\t\t\t\tdefaultContextWindow: modelDef.contextWindow,\n\t\t\t\tcontextWindowOptions: normalizeContextWindowOptions([\n\t\t\t\t\tmodelDef.contextWindow,\n\t\t\t\t\t...(modelDef.contextWindowOptions ?? []),\n\t\t\t\t]),\n\t\t\t\tmaxTokens: modelDef.maxTokens,\n\t\t\t\theaders: undefined,\n\t\t\t\tcompat: modelDef.compat,\n\t\t\t} as Model<Api>);\n\t\t}\n\n\t\tif (config.oauth?.modifyModels) {\n\t\t\tconst cred = authStorage.get(providerName);\n\t\t\tif (cred?.type === \"oauth\") {\n\t\t\t\tmodels = config.oauth.modifyModels(models, cred);\n\t\t\t}\n\t\t}\n\t} else if (config.baseUrl || config.headers) {\n\t\tmodels = models.map((m) => {\n\t\t\tif (m.provider !== providerName) return m;\n\t\t\treturn {\n\t\t\t\t...m,\n\t\t\t\tbaseUrl: config.baseUrl ?? m.baseUrl,\n\t\t\t};\n\t\t});\n\t}\n\n\treturn models;\n}\n"]}

package/dist/core/model-registry-dynamic.js

@@ -0,0 +1,155 @@
+import { registerApiProvider, } from "@earendil-works/pi-ai";
+import { registerOAuthProvider } from "@earendil-works/pi-ai/oauth";
+import { normalizeContextWindowOptions, validateContextWindowValue } from "./context-window.js";
+import { warnDeprecation } from "../utils/deprecation.js";
+import { isLegacyEnvVarNameConfigValue } from "./resolve-config-value.js";
+function validateContextWindowOptions(providerName, modelId, options) {
+    for (const option of options ?? []) {
+        if (validateContextWindowValue(option)) {
+            throw new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindowOptions value`);
+        }
+    }
+}
+function migrateLegacyRegisterProviderConfigValue(providerName, field, value) {
+    if (!isLegacyEnvVarNameConfigValue(value) || process.env[value] === undefined)
+        return value;
+    warnDeprecation(`registerProvider("${providerName}") ${field} value "${value}" is treated as a legacy environment variable reference. This will no longer be detected as an environment variable reference in a future release. Pass "$${value}" instead.`);
+    return `$${value}`;
+}
+function migrateLegacyRegisterProviderHeaders(providerName, field, headers) {
+    if (!headers)
+        return undefined;
+    let migratedHeaders;
+    for (const [key, value] of Object.entries(headers)) {
+        const migratedValue = migrateLegacyRegisterProviderConfigValue(providerName, `${field} header "${key}"`, value);
+        if (migratedValue === value)
+            continue;
+        migratedHeaders ??= { ...headers };
+        migratedHeaders[key] = migratedValue;
+    }
+    return migratedHeaders ?? headers;
+}
+export function migrateLegacyRegisterProviderConfigValues(providerName, config) {
+    let migratedConfig;
+    const setMigratedConfigValue = (key, value) => {
+        migratedConfig ??= { ...config };
+        migratedConfig[key] = value;
+    };
+    if (config.apiKey) {
+        const apiKey = migrateLegacyRegisterProviderConfigValue(providerName, "apiKey", config.apiKey);
+        if (apiKey !== config.apiKey) {
+            setMigratedConfigValue("apiKey", apiKey);
+        }
+    }
+    const headers = migrateLegacyRegisterProviderHeaders(providerName, "headers", config.headers);
+    if (headers !== config.headers) {
+        setMigratedConfigValue("headers", headers);
+    }
+    if (config.models) {
+        let models;
+        for (let index = 0; index < config.models.length; index++) {
+            const model = config.models[index];
+            const modelHeaders = migrateLegacyRegisterProviderHeaders(providerName, `model "${model.id}" headers`, model.headers);
+            if (modelHeaders === model.headers)
+                continue;
+            models ??= [...config.models];
+            models[index] = { ...model, headers: modelHeaders };
+        }
+        if (models) {
+            setMigratedConfigValue("models", models);
+        }
+    }
+    return migratedConfig ?? config;
+}
+export function validateProviderConfig(providerName, config) {
+    if (config.streamSimple && !config.api) {
+        throw new Error(`Provider ${providerName}: "api" is required when registering streamSimple.`);
+    }
+    if (!config.models || config.models.length === 0) {
+        return;
+    }
+    if (!config.baseUrl) {
+        throw new Error(`Provider ${providerName}: "baseUrl" is required when defining models.`);
+    }
+    if (!config.apiKey && !config.oauth) {
+        throw new Error(`Provider ${providerName}: "apiKey" or "oauth" is required when defining models.`);
+    }
+    for (const modelDef of config.models) {
+        const api = modelDef.api || config.api;
+        if (!api) {
+            throw new Error(`Provider ${providerName}, model ${modelDef.id}: no "api" specified.`);
+        }
+        if (validateContextWindowValue(modelDef.contextWindow)) {
+            throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);
+        }
+        validateContextWindowOptions(providerName, modelDef.id, modelDef.contextWindowOptions);
+        if (modelDef.maxTokens <= 0) {
+            throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);
+        }
+    }
+}
+export function applyProviderConfigToModels(input) {
+    const { providerName, config, authStorage, storeProviderRequestConfig, storeModelHeaders } = input;
+    let models = input.models;
+    if (config.oauth) {
+        const oauthProvider = {
+            ...config.oauth,
+            id: providerName,
+        };
+        registerOAuthProvider(oauthProvider);
+    }
+    if (config.streamSimple) {
+        const streamSimple = config.streamSimple;
+        registerApiProvider({
+            api: config.api,
+            stream: (model, context, options) => streamSimple(model, context, options),
+            streamSimple,
+        }, `provider:${providerName}`);
+    }
+    storeProviderRequestConfig(providerName, config);
+    if (config.models && config.models.length > 0) {
+        models = models.filter((m) => m.provider !== providerName);
+        for (const modelDef of config.models) {
+            const api = modelDef.api || config.api;
+            storeModelHeaders(providerName, modelDef.id, modelDef.headers);
+            models.push({
+                id: modelDef.id,
+                name: modelDef.name,
+                api: api,
+                provider: providerName,
+                baseUrl: modelDef.baseUrl ?? config.baseUrl,
+                reasoning: modelDef.reasoning,
+                thinkingLevelMap: modelDef.thinkingLevelMap,
+                input: modelDef.input,
+                cost: modelDef.cost,
+                contextWindow: modelDef.contextWindow,
+                defaultContextWindow: modelDef.contextWindow,
+                contextWindowOptions: normalizeContextWindowOptions([
+                    modelDef.contextWindow,
+                    ...(modelDef.contextWindowOptions ?? []),
+                ]),
+                maxTokens: modelDef.maxTokens,
+                headers: undefined,
+                compat: modelDef.compat,
+            });
+        }
+        if (config.oauth?.modifyModels) {
+            const cred = authStorage.get(providerName);
+            if (cred?.type === "oauth") {
+                models = config.oauth.modifyModels(models, cred);
+            }
+        }
+    }
+    else if (config.baseUrl || config.headers) {
+        models = models.map((m) => {
+            if (m.provider !== providerName)
+                return m;
+            return {
+                ...m,
+                baseUrl: config.baseUrl ?? m.baseUrl,
+            };
+        });
+    }
+    return models;
+}
+//# sourceMappingURL=model-registry-dynamic.js.map

package/dist/core/model-registry-dynamic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-dynamic.js","sourceRoot":"","sources":["../../src/core/model-registry-dynamic.ts"],"names":[],"mappings":"AAAA,OAAO,EAIN,mBAAmB,GAEnB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,6BAA6B,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAEhG,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAE1E,SAAS,4BAA4B,CAAC,YAAoB,EAAE,OAAe,EAAE,OAAsC;IAClH,KAAK,MAAM,MAAM,IAAI,OAAO,IAAI,EAAE,EAAE,CAAC;QACpC,IAAI,0BAA0B,CAAC,MAAM,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,OAAO,sCAAsC,CAAC,CAAC;QACnG,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,wCAAwC,CAAC,YAAoB,EAAE,KAAa,EAAE,KAAa;IACnG,IAAI,CAAC,6BAA6B,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IAC5F,eAAe,CACd,qBAAqB,YAAY,MAAM,KAAK,WAAW,KAAK,6JAA6J,KAAK,YAAY,CAC1O,CAAC;IACF,OAAO,IAAI,KAAK,EAAE,CAAC;AACpB,CAAC;AAED,SAAS,oCAAoC,CAC5C,YAAoB,EACpB,KAAa,EACb,OAA2C;IAE3C,IAAI,CAAC,OAAO;QAAE,OAAO,SAAS,CAAC;IAC/B,IAAI,eAAmD,CAAC;IACxD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACpD,MAAM,aAAa,GAAG,wCAAwC,CAAC,YAAY,EAAE,GAAG,KAAK,YAAY,GAAG,GAAG,EAAE,KAAK,CAAC,CAAC;QAChH,IAAI,aAAa,KAAK,KAAK;YAAE,SAAS;QACtC,eAAe,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC;QACnC,eAAe,CAAC,GAAG,CAAC,GAAG,aAAa,CAAC;IACtC,CAAC;IACD,OAAO,eAAe,IAAI,OAAO,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,yCAAyC,CACxD,YAAoB,EACpB,MAA2B;IAE3B,IAAI,cAA+C,CAAC;IAEpD,MAAM,sBAAsB,GAAG,CAC9B,GAAS,EACT,KAAgC,EAC/B,EAAE;QACH,cAAc,KAAK,EAAE,GAAG,MAAM,EAAE,CAAC;QACjC,cAAc,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IAC7B,CAAC,CAAC;IAEF,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,wCAAwC,CAAC,YAAY,EAAE,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/F,IAAI,MAAM,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;YAC9B,sBAAsB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;IACF,CAAC;IAED,MAAM,OAAO,GAAG,oCAAoC,CAAC,YAAY,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IAC9F,IAAI,OAAO,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC;QAChC,sBAAsB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,IAAI,MAAiD,CAAC;QACtD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;YAC3D,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACnC,MAAM,YAAY,GAAG,oCAAoC,CACxD,YAAY,EACZ,UAAU,KAAK,CAAC,EAAE,WAAW,EAC7B,KAAK,CAAC,OAAO,CACb,CAAC;YACF,IAAI,YAAY,KAAK,KAAK,CAAC,OAAO;gBAAE,SAAS;YAC7C,MAAM,KAAK,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;QACrD,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACZ,sBAAsB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC1C,CAAC;IACF,CAAC;IAED,OAAO,cAAc,IAAI,MAAM,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,YAAoB,EAAE,MAA2B;IACvF,IAAI,MAAM,CAAC,YAAY,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,oDAAoD,CAAC,CAAC;IAC/F,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAClD,OAAO;IACR,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,+CAA+C,CAAC,CAAC;IAC1F,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,yDAAyD,CAAC,CAAC;IACpG,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,MAAM,CAAC,GAAG,CAAC;QACvC,IAAI,CAAC,GAAG,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,uBAAuB,CAAC,CAAC;QACxF,CAAC;QACD,IAAI,0BAA0B,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,yBAAyB,CAAC,CAAC;QAC1F,CAAC;QACD,4BAA4B,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,oBAAoB,CAAC,CAAC;QACvF,IAAI,QAAQ,CAAC,SAAS,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,qBAAqB,CAAC,CAAC;QACtF,CAAC;IACF,CAAC;AACF,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,KAAgC;IAC3E,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,WAAW,EAAE,0BAA0B,EAAE,iBAAiB,EAAE,GAAG,KAAK,CAAC;IACnG,IAAI,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;IAE1B,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;QAClB,MAAM,aAAa,GAA2B;YAC7C,GAAG,MAAM,CAAC,KAAK;YACf,EAAE,EAAE,YAAY;SAChB,CAAC;QACF,qBAAqB,CAAC,aAAa,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACzB,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACzC,mBAAmB,CAClB;YACC,GAAG,EAAE,MAAM,CAAC,GAAI;YAChB,MAAM,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,OAA8B,CAAC;YACjG,YAAY;SACZ,EACD,YAAY,YAAY,EAAE,CAC1B,CAAC;IACH,CAAC;IAED,0BAA0B,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAEjD,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/C,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC;QAE3D,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACtC,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,MAAM,CAAC,GAAG,CAAC;YACvC,iBAAiB,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YAE/D,MAAM,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,GAAG,EAAE,GAAU;gBACf,QAAQ,EAAE,YAAY;gBACtB,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,MAAM,CAAC,OAAQ;gBAC5C,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,gBAAgB,EAAE,QAAQ,CAAC,gBAAgB;gBAC3C,KAAK,EAAE,QAAQ,CAAC,KAA6B;gBAC7C,IAAI,EAAE,QAAQ,CAAC,IAAI;gBACnB,aAAa,EAAE,QAAQ,CAAC,aAAa;gBACrC,oBAAoB,EAAE,QAAQ,CAAC,aAAa;gBAC5C,oBAAoB,EAAE,6BAA6B,CAAC;oBACnD,QAAQ,CAAC,aAAa;oBACtB,GAAG,CAAC,QAAQ,CAAC,oBAAoB,IAAI,EAAE,CAAC;iBACxC,CAAC;gBACF,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,OAAO,EAAE,SAAS;gBAClB,MAAM,EAAE,QAAQ,CAAC,MAAM;aACT,CAAC,CAAC;QAClB,CAAC;QAED,IAAI,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;YAC3C,IAAI,IAAI,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;gBAC5B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAClD,CAAC;QACF,CAAC;IACF,CAAC;SAAM,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QAC7C,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACzB,IAAI,CAAC,CAAC,QAAQ,KAAK,YAAY;gBAAE,OAAO,CAAC,CAAC;YAC1C,OAAO;gBACN,GAAG,CAAC;gBACJ,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO;aACpC,CAAC;QACH,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC","sourcesContent":["import {\n\ttype Api,\n\ttype Model,\n\ttype OAuthProviderInterface,\n\tregisterApiProvider,\n\ttype SimpleStreamOptions,\n} from \"@earendil-works/pi-ai\";\nimport { registerOAuthProvider } from \"@earendil-works/pi-ai/oauth\";\nimport { normalizeContextWindowOptions, validateContextWindowValue } from \"./context-window.ts\";\nimport type { DynamicProviderApplyInput, ProviderConfigInput } from \"./model-registry-types.ts\";\nimport { warnDeprecation } from \"../utils/deprecation.ts\";\nimport { isLegacyEnvVarNameConfigValue } from \"./resolve-config-value.ts\";\n\nfunction validateContextWindowOptions(providerName: string, modelId: string, options: readonly number[] | undefined): void {\n\tfor (const option of options ?? []) {\n\t\tif (validateContextWindowValue(option)) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindowOptions value`);\n\t\t}\n\t}\n}\n\nfunction migrateLegacyRegisterProviderConfigValue(providerName: string, field: string, value: string): string {\n\tif (!isLegacyEnvVarNameConfigValue(value) || process.env[value] === undefined) return value;\n\twarnDeprecation(\n\t\t`registerProvider(\"${providerName}\") ${field} value \"${value}\" is treated as a legacy environment variable reference. This will no longer be detected as an environment variable reference in a future release. Pass \"$${value}\" instead.`,\n\t);\n\treturn `$${value}`;\n}\n\nfunction migrateLegacyRegisterProviderHeaders(\n\tproviderName: string,\n\tfield: string,\n\theaders: Record<string, string> | undefined,\n): Record<string, string> | undefined {\n\tif (!headers) return undefined;\n\tlet migratedHeaders: Record<string, string> | undefined;\n\tfor (const [key, value] of Object.entries(headers)) {\n\t\tconst migratedValue = migrateLegacyRegisterProviderConfigValue(providerName, `${field} header \"${key}\"`, value);\n\t\tif (migratedValue === value) continue;\n\t\tmigratedHeaders ??= { ...headers };\n\t\tmigratedHeaders[key] = migratedValue;\n\t}\n\treturn migratedHeaders ?? headers;\n}\n\nexport function migrateLegacyRegisterProviderConfigValues(\n\tproviderName: string,\n\tconfig: ProviderConfigInput,\n): ProviderConfigInput {\n\tlet migratedConfig: ProviderConfigInput | undefined;\n\n\tconst setMigratedConfigValue = <TKey extends keyof ProviderConfigInput>(\n\t\tkey: TKey,\n\t\tvalue: ProviderConfigInput[TKey],\n\t) => {\n\t\tmigratedConfig ??= { ...config };\n\t\tmigratedConfig[key] = value;\n\t};\n\n\tif (config.apiKey) {\n\t\tconst apiKey = migrateLegacyRegisterProviderConfigValue(providerName, \"apiKey\", config.apiKey);\n\t\tif (apiKey !== config.apiKey) {\n\t\t\tsetMigratedConfigValue(\"apiKey\", apiKey);\n\t\t}\n\t}\n\n\tconst headers = migrateLegacyRegisterProviderHeaders(providerName, \"headers\", config.headers);\n\tif (headers !== config.headers) {\n\t\tsetMigratedConfigValue(\"headers\", headers);\n\t}\n\n\tif (config.models) {\n\t\tlet models: ProviderConfigInput[\"models\"] | undefined;\n\t\tfor (let index = 0; index < config.models.length; index++) {\n\t\t\tconst model = config.models[index];\n\t\t\tconst modelHeaders = migrateLegacyRegisterProviderHeaders(\n\t\t\t\tproviderName,\n\t\t\t\t`model \"${model.id}\" headers`,\n\t\t\t\tmodel.headers,\n\t\t\t);\n\t\t\tif (modelHeaders === model.headers) continue;\n\t\t\tmodels ??= [...config.models];\n\t\t\tmodels[index] = { ...model, headers: modelHeaders };\n\t\t}\n\t\tif (models) {\n\t\t\tsetMigratedConfigValue(\"models\", models);\n\t\t}\n\t}\n\n\treturn migratedConfig ?? config;\n}\n\nexport function validateProviderConfig(providerName: string, config: ProviderConfigInput): void {\n\tif (config.streamSimple && !config.api) {\n\t\tthrow new Error(`Provider ${providerName}: \"api\" is required when registering streamSimple.`);\n\t}\n\n\tif (!config.models || config.models.length === 0) {\n\t\treturn;\n\t}\n\n\tif (!config.baseUrl) {\n\t\tthrow new Error(`Provider ${providerName}: \"baseUrl\" is required when defining models.`);\n\t}\n\tif (!config.apiKey && !config.oauth) {\n\t\tthrow new Error(`Provider ${providerName}: \"apiKey\" or \"oauth\" is required when defining models.`);\n\t}\n\n\tfor (const modelDef of config.models) {\n\t\tconst api = modelDef.api || config.api;\n\t\tif (!api) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: no \"api\" specified.`);\n\t\t}\n\t\tif (validateContextWindowValue(modelDef.contextWindow)) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);\n\t\t}\n\t\tvalidateContextWindowOptions(providerName, modelDef.id, modelDef.contextWindowOptions);\n\t\tif (modelDef.maxTokens <= 0) {\n\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);\n\t\t}\n\t}\n}\n\nexport function applyProviderConfigToModels(input: DynamicProviderApplyInput): Model<Api>[] {\n\tconst { providerName, config, authStorage, storeProviderRequestConfig, storeModelHeaders } = input;\n\tlet models = input.models;\n\n\tif (config.oauth) {\n\t\tconst oauthProvider: OAuthProviderInterface = {\n\t\t\t...config.oauth,\n\t\t\tid: providerName,\n\t\t};\n\t\tregisterOAuthProvider(oauthProvider);\n\t}\n\n\tif (config.streamSimple) {\n\t\tconst streamSimple = config.streamSimple;\n\t\tregisterApiProvider(\n\t\t\t{\n\t\t\t\tapi: config.api!,\n\t\t\t\tstream: (model, context, options) => streamSimple(model, context, options as SimpleStreamOptions),\n\t\t\t\tstreamSimple,\n\t\t\t},\n\t\t\t`provider:${providerName}`,\n\t\t);\n\t}\n\n\tstoreProviderRequestConfig(providerName, config);\n\n\tif (config.models && config.models.length > 0) {\n\t\tmodels = models.filter((m) => m.provider !== providerName);\n\n\t\tfor (const modelDef of config.models) {\n\t\t\tconst api = modelDef.api || config.api;\n\t\t\tstoreModelHeaders(providerName, modelDef.id, modelDef.headers);\n\n\t\t\tmodels.push({\n\t\t\t\tid: modelDef.id,\n\t\t\t\tname: modelDef.name,\n\t\t\t\tapi: api as Api,\n\t\t\t\tprovider: providerName,\n\t\t\t\tbaseUrl: modelDef.baseUrl ?? config.baseUrl!,\n\t\t\t\treasoning: modelDef.reasoning,\n\t\t\t\tthinkingLevelMap: modelDef.thinkingLevelMap,\n\t\t\t\tinput: modelDef.input as (\"text\" | \"image\")[],\n\t\t\t\tcost: modelDef.cost,\n\t\t\t\tcontextWindow: modelDef.contextWindow,\n\t\t\t\tdefaultContextWindow: modelDef.contextWindow,\n\t\t\t\tcontextWindowOptions: normalizeContextWindowOptions([\n\t\t\t\t\tmodelDef.contextWindow,\n\t\t\t\t\t...(modelDef.contextWindowOptions ?? []),\n\t\t\t\t]),\n\t\t\t\tmaxTokens: modelDef.maxTokens,\n\t\t\t\theaders: undefined,\n\t\t\t\tcompat: modelDef.compat,\n\t\t\t} as Model<Api>);\n\t\t}\n\n\t\tif (config.oauth?.modifyModels) {\n\t\t\tconst cred = authStorage.get(providerName);\n\t\t\tif (cred?.type === \"oauth\") {\n\t\t\t\tmodels = config.oauth.modifyModels(models, cred);\n\t\t\t}\n\t\t}\n\t} else if (config.baseUrl || config.headers) {\n\t\tmodels = models.map((m) => {\n\t\t\tif (m.provider !== providerName) return m;\n\t\t\treturn {\n\t\t\t\t...m,\n\t\t\t\tbaseUrl: config.baseUrl ?? m.baseUrl,\n\t\t\t};\n\t\t});\n\t}\n\n\treturn models;\n}\n"]}

package/dist/core/model-registry-loader.d.ts

@@ -0,0 +1,4 @@
+import type { AuthStorage } from "./auth-storage.ts";
+import type { ModelRegistryLoadResult } from "./model-registry-types.ts";
+export declare function loadModelRegistryModels(authStorage: AuthStorage, modelsJsonPaths: string[]): ModelRegistryLoadResult;
+//# sourceMappingURL=model-registry-loader.d.ts.map

package/dist/core/model-registry-loader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-loader.d.ts","sourceRoot":"","sources":["../../src/core/model-registry-loader.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGrD,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAEzE,wBAAgB,uBAAuB,CACtC,WAAW,EAAE,WAAW,EACxB,eAAe,EAAE,MAAM,EAAE,GACvB,uBAAuB,CA0BzB","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\nimport type { AuthStorage } from \"./auth-storage.ts\";\nimport { loadBuiltInModels, mergeCustomModels } from \"./model-registry-builtins.ts\";\nimport { loadCustomModelsFromPaths } from \"./model-registry-custom-loader.ts\";\nimport type { ModelRegistryLoadResult } from \"./model-registry-types.ts\";\n\nexport function loadModelRegistryModels(\n\tauthStorage: AuthStorage,\n\tmodelsJsonPaths: string[],\n): ModelRegistryLoadResult {\n\tconst {\n\t\tmodels: customModels,\n\t\toverrides,\n\t\tmodelOverrides,\n\t\tproviderRequestConfigs,\n\t\tmodelRequestHeaders,\n\t\terror,\n\t} = loadCustomModelsFromPaths(modelsJsonPaths);\n\n\tconst builtInModels = loadBuiltInModels(overrides, modelOverrides);\n\tlet combined: Model<Api>[] = mergeCustomModels(builtInModels, customModels);\n\n\tfor (const oauthProvider of authStorage.getOAuthProviders()) {\n\t\tconst cred = authStorage.get(oauthProvider.id);\n\t\tif (cred?.type === \"oauth\" && oauthProvider.modifyModels) {\n\t\t\tcombined = oauthProvider.modifyModels(combined, cred);\n\t\t}\n\t}\n\n\treturn {\n\t\tmodels: combined,\n\t\tproviderRequestConfigs,\n\t\tmodelRequestHeaders,\n\t\tloadError: error,\n\t};\n}\n"]}

package/dist/core/model-registry-loader.js

@@ -0,0 +1,20 @@
+import { loadBuiltInModels, mergeCustomModels } from "./model-registry-builtins.js";
+import { loadCustomModelsFromPaths } from "./model-registry-custom-loader.js";
+export function loadModelRegistryModels(authStorage, modelsJsonPaths) {
+    const { models: customModels, overrides, modelOverrides, providerRequestConfigs, modelRequestHeaders, error, } = loadCustomModelsFromPaths(modelsJsonPaths);
+    const builtInModels = loadBuiltInModels(overrides, modelOverrides);
+    let combined = mergeCustomModels(builtInModels, customModels);
+    for (const oauthProvider of authStorage.getOAuthProviders()) {
+        const cred = authStorage.get(oauthProvider.id);
+        if (cred?.type === "oauth" && oauthProvider.modifyModels) {
+            combined = oauthProvider.modifyModels(combined, cred);
+        }
+    }
+    return {
+        models: combined,
+        providerRequestConfigs,
+        modelRequestHeaders,
+        loadError: error,
+    };
+}
+//# sourceMappingURL=model-registry-loader.js.map

package/dist/core/model-registry-loader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry-loader.js","sourceRoot":"","sources":["../../src/core/model-registry-loader.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACpF,OAAO,EAAE,yBAAyB,EAAE,MAAM,mCAAmC,CAAC;AAG9E,MAAM,UAAU,uBAAuB,CACtC,WAAwB,EACxB,eAAyB;IAEzB,MAAM,EACL,MAAM,EAAE,YAAY,EACpB,SAAS,EACT,cAAc,EACd,sBAAsB,EACtB,mBAAmB,EACnB,KAAK,GACL,GAAG,yBAAyB,CAAC,eAAe,CAAC,CAAC;IAE/C,MAAM,aAAa,GAAG,iBAAiB,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;IACnE,IAAI,QAAQ,GAAiB,iBAAiB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAE5E,KAAK,MAAM,aAAa,IAAI,WAAW,CAAC,iBAAiB,EAAE,EAAE,CAAC;QAC7D,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAC/C,IAAI,IAAI,EAAE,IAAI,KAAK,OAAO,IAAI,aAAa,CAAC,YAAY,EAAE,CAAC;YAC1D,QAAQ,GAAG,aAAa,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACvD,CAAC;IACF,CAAC;IAED,OAAO;QACN,MAAM,EAAE,QAAQ;QAChB,sBAAsB;QACtB,mBAAmB;QACnB,SAAS,EAAE,KAAK;KAChB,CAAC;AACH,CAAC","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai\";\nimport type { AuthStorage } from \"./auth-storage.ts\";\nimport { loadBuiltInModels, mergeCustomModels } from \"./model-registry-builtins.ts\";\nimport { loadCustomModelsFromPaths } from \"./model-registry-custom-loader.ts\";\nimport type { ModelRegistryLoadResult } from \"./model-registry-types.ts\";\n\nexport function loadModelRegistryModels(\n\tauthStorage: AuthStorage,\n\tmodelsJsonPaths: string[],\n): ModelRegistryLoadResult {\n\tconst {\n\t\tmodels: customModels,\n\t\toverrides,\n\t\tmodelOverrides,\n\t\tproviderRequestConfigs,\n\t\tmodelRequestHeaders,\n\t\terror,\n\t} = loadCustomModelsFromPaths(modelsJsonPaths);\n\n\tconst builtInModels = loadBuiltInModels(overrides, modelOverrides);\n\tlet combined: Model<Api>[] = mergeCustomModels(builtInModels, customModels);\n\n\tfor (const oauthProvider of authStorage.getOAuthProviders()) {\n\t\tconst cred = authStorage.get(oauthProvider.id);\n\t\tif (cred?.type === \"oauth\" && oauthProvider.modifyModels) {\n\t\t\tcombined = oauthProvider.modifyModels(combined, cred);\n\t\t}\n\t}\n\n\treturn {\n\t\tmodels: combined,\n\t\tproviderRequestConfigs,\n\t\tmodelRequestHeaders,\n\t\tloadError: error,\n\t};\n}\n"]}