@bastani/atomic 0.9.9 → 0.9.10

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

@@ -1 +1 @@
-{"version":3,"file":"model-registry-dynamic.js","sourceRoot":"","sources":["../../src/core/model-registry-dynamic.ts"],"names":[],"mappings":"AAAA,OAAO,EAIN,mBAAmB,GAEnB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,6BAA6B,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAChG,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAElE,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,cAAc,EAAE,0BAA0B,EAAE,iBAAiB,EAAE,GAAG,KAAK,CAAC;IACnH,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,MAAM,aAAa,GAAG,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACzE,iBAAiB,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE;gBAC5C,GAAG,QAAQ,CAAC,OAAO;gBACnB,GAAG,aAAa,EAAE,OAAO;aACzB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG;gBACb,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;YAChB,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAC/E,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/compat\";\nimport { registerOAuthProvider } from \"@earendil-works/pi-ai/oauth\";\nimport { warnDeprecation } from \"../utils/deprecation.ts\";\nimport { normalizeContextWindowOptions, validateContextWindowValue } from \"./context-window.ts\";\nimport { applyModelOverride } from \"./model-registry-builtins.ts\";\nimport type { DynamicProviderApplyInput, ProviderConfigInput } from \"./model-registry-types.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, modelOverrides, 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\tconst modelOverride = modelOverrides.get(providerName)?.get(modelDef.id);\n\t\t\tstoreModelHeaders(providerName, modelDef.id, {\n\t\t\t\t...modelDef.headers,\n\t\t\t\t...modelOverride?.headers,\n\t\t\t});\n\n\t\t\tconst model = {\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\tmodels.push(modelOverride ? applyModelOverride(model, modelOverride) : model);\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"]}
+{"version":3,"file":"model-registry-dynamic.js","sourceRoot":"","sources":["../../src/core/model-registry-dynamic.ts"],"names":[],"mappings":"AAAA,OAAO,EAEN,cAAc,EAEd,mBAAmB,EAEnB,sBAAsB,GACtB,MAAM,8BAA8B,CAAC;AACtC,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,6BAA6B,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAChG,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAElE,OAAO,EAAE,2BAA2B,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AACzG,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAY1E,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAiC,CAAC;AACzE,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA0B,CAAC;AAE9D,SAAS,4BAA4B,CAAC,QAAgB;IACrD,KAAK,MAAM,CAAC,GAAG,EAAE,aAAa,CAAC,IAAI,uBAAuB,EAAE,CAAC;QAC5D,MAAM,KAAK,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC;QAC9E,IAAI,KAAK,GAAG,CAAC;YAAE,SAAS;QACxB,MAAM,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACjD,IAAI,OAAO,IAAI,cAAc,CAAC,GAAG,CAAC,KAAK,OAAO,CAAC,cAAc,EAAE,CAAC;YAC/D,sBAAsB,CAAC,QAAQ,CAAC,CAAC;YACjC,MAAM,QAAQ,GAAG,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACtC,IAAI,QAAQ,EAAE,CAAC;gBACd,mBAAmB,CAAC,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBAC1D,QAAQ,CAAC,cAAc,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;YAC/C,CAAC;iBAAM,CAAC;gBACP,MAAM,QAAQ,GAAG,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAC9C,IAAI,QAAQ;oBAAE,mBAAmB,CAAC,QAAQ,EAAE,uBAAuB,GAAG,EAAE,CAAC,CAAC;YAC3E,CAAC;QACF,CAAC;QACD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAChC,uBAAuB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACpC,mBAAmB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjC,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,0BAA0B,CAAC,QAA4B,EAAE,QAAgB;IACjF,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,aAAa,GAAG,uBAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;IACtE,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC9C,IAAI,QAAQ;YAAE,mBAAmB,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;IAC/D,CAAC;IACD,MAAM,YAAY,GAA2B,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC;IACpE,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACjC,uBAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;IACzD,mBAAmB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACxC,YAAY,CAAC,cAAc,GAAG,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;AAC5D,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,QAAgB;IACzD,4BAA4B,CAAC,QAAQ,CAAC,CAAC;IACvC,8BAA8B,CAAC,QAAQ,CAAC,CAAC;AAC1C,CAAC;AACD,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,kBAAkB,EAAE,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,0BAA0B,EAAE,iBAAiB,EAAE,GAAG,KAAK,CAAC;IACvI,IAAI,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;IAE1B,IAAI,KAAK,CAAC,eAAe,IAAI,MAAM,CAAC,KAAK;QAAE,2BAA2B,CAAC,YAAY,EAAE,MAAM,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAC;IAEvH,IAAI,KAAK,CAAC,eAAe,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QAClD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QACzC,0BAA0B,CAAC;YAC1B,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,EAAE,kBAAkB,CAAC,CAAC;IACxB,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,MAAM,aAAa,GAAG,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACzE,iBAAiB,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,EAAE;gBAC5C,GAAG,QAAQ,CAAC,OAAO;gBACnB,GAAG,aAAa,EAAE,OAAO;aACzB,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG;gBACb,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;YAChB,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QAC/E,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\tgetApiProvider,\n\ttype Model,\n\tregisterApiProvider,\n\ttype SimpleStreamOptions,\n\tunregisterApiProviders,\n} from \"@earendil-works/pi-ai/compat\";\nimport { warnDeprecation } from \"../utils/deprecation.ts\";\nimport { normalizeContextWindowOptions, validateContextWindowValue } from \"./context-window.ts\";\nimport { applyModelOverride } from \"./model-registry-builtins.ts\";\nimport type { DynamicProviderApplyInput, ProviderConfigInput } from \"./model-registry-types.ts\";\nimport { registerLegacyOAuthProvider, unregisterLegacyOAuthProviders } from \"./oauth-provider-bridge.ts\";\nimport { isLegacyEnvVarNameConfigValue } from \"./resolve-config-value.ts\";\n\n\ntype RuntimeApiProvider = Parameters<typeof registerApiProvider>[0];\ntype ActiveApiProvider = NonNullable<ReturnType<typeof getApiProvider>>;\n\ninterface RuntimeApiRegistration {\n\tsourceId: string;\n\tprovider: RuntimeApiProvider;\n\tactiveProvider?: ActiveApiProvider;\n}\n\nconst runtimeApiRegistrations = new Map<Api, RuntimeApiRegistration[]>();\nconst runtimeApiFallbacks = new Map<Api, ActiveApiProvider>();\n\nfunction unregisterRuntimeApiProvider(sourceId: string): void {\n\tfor (const [api, registrations] of runtimeApiRegistrations) {\n\t\tconst index = registrations.findIndex((entry) => entry.sourceId === sourceId);\n\t\tif (index < 0) continue;\n\t\tconst [removed] = registrations.splice(index, 1);\n\t\tif (removed && getApiProvider(api) === removed.activeProvider) {\n\t\t\tunregisterApiProviders(sourceId);\n\t\t\tconst previous = registrations.at(-1);\n\t\t\tif (previous) {\n\t\t\t\tregisterApiProvider(previous.provider, previous.sourceId);\n\t\t\t\tprevious.activeProvider = getApiProvider(api);\n\t\t\t} else {\n\t\t\t\tconst fallback = runtimeApiFallbacks.get(api);\n\t\t\t\tif (fallback) registerApiProvider(fallback, `atomic:restored-api:${api}`);\n\t\t\t}\n\t\t}\n\t\tif (registrations.length === 0) {\n\t\t\truntimeApiRegistrations.delete(api);\n\t\t\truntimeApiFallbacks.delete(api);\n\t\t}\n\t}\n}\n\nfunction registerRuntimeApiProvider(provider: RuntimeApiProvider, sourceId: string): void {\n\tunregisterRuntimeApiProvider(sourceId);\n\tconst registrations = runtimeApiRegistrations.get(provider.api) ?? [];\n\tif (registrations.length === 0) {\n\t\tconst fallback = getApiProvider(provider.api);\n\t\tif (fallback) runtimeApiFallbacks.set(provider.api, fallback);\n\t}\n\tconst registration: RuntimeApiRegistration = { sourceId, provider };\n\tregistrations.push(registration);\n\truntimeApiRegistrations.set(provider.api, registrations);\n\tregisterApiProvider(provider, sourceId);\n\tregistration.activeProvider = getApiProvider(provider.api);\n}\n\nexport function unregisterProviderRuntime(sourceId: string): void {\n\tunregisterRuntimeApiProvider(sourceId);\n\tunregisterLegacyOAuthProviders(sourceId);\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, registrationSource, config, authStorage, modelOverrides, storeProviderRequestConfig, storeModelHeaders } = input;\n\tlet models = input.models;\n\n\tif (input.registerRuntime && config.oauth) registerLegacyOAuthProvider(providerName, config.oauth, registrationSource);\n\n\tif (input.registerRuntime && config.streamSimple) {\n\t\tconst streamSimple = config.streamSimple;\n\t\tregisterRuntimeApiProvider({\n\t\t\tapi: config.api!,\n\t\t\tstream: (model, context, options) => streamSimple(model, context, options as SimpleStreamOptions),\n\t\t\tstreamSimple,\n\t\t}, registrationSource);\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\tconst modelOverride = modelOverrides.get(providerName)?.get(modelDef.id);\n\t\t\tstoreModelHeaders(providerName, modelDef.id, {\n\t\t\t\t...modelDef.headers,\n\t\t\t\t...modelOverride?.headers,\n\t\t\t});\n\n\t\t\tconst model = {\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\tmodels.push(modelOverride ? applyModelOverride(model, modelOverride) : model);\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

@@ -1,4 +1,5 @@
+import type { Api, Model } from "@earendil-works/pi-ai/compat";
 import type { AuthStorage } from "./auth-storage.ts";
 import type { ModelRegistryLoadResult } from "./model-registry-types.ts";
-export declare function loadModelRegistryModels(authStorage: AuthStorage, modelsJsonPaths: string[]): ModelRegistryLoadResult;
+export declare function loadModelRegistryModels(authStorage: AuthStorage, modelsJsonPaths: string[], baseModels?: readonly Model<Api>[]): ModelRegistryLoadResult;
 //# sourceMappingURL=model-registry-loader.d.ts.map

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

@@ -1 +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;AAIzE,wBAAgB,uBAAuB,CACtC,WAAW,EAAE,WAAW,EACxB,eAAe,EAAE,MAAM,EAAE,GACvB,uBAAuB,CAmCzB"}
+{"version":3,"file":"model-registry-loader.d.ts","sourceRoot":"","sources":["../../src/core/model-registry-loader.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGrD,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AAIzE,wBAAgB,uBAAuB,CACtC,WAAW,EAAE,WAAW,EACxB,eAAe,EAAE,MAAM,EAAE,EACzB,UAAU,CAAC,EAAE,SAAS,KAAK,CAAC,GAAG,CAAC,EAAE,GAChC,uBAAuB,CAmCzB"}

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

@@ -1,9 +1,9 @@
 import { loadBuiltInModels, mergeCustomModels } from "./model-registry-builtins.js";
 import { loadCustomModelsFromPaths } from "./model-registry-custom-loader.js";
 const OPENAI_COMPATIBLE_APIS = new Set(["openai-completions", "openai-responses"]);
-export function loadModelRegistryModels(authStorage, modelsJsonPaths) {
+export function loadModelRegistryModels(authStorage, modelsJsonPaths, baseModels) {
     const { models: customModels, overrides, modelOverrides, providerRequestConfigs, modelRequestHeaders, error, } = loadCustomModelsFromPaths(modelsJsonPaths);
-    const builtInModels = loadBuiltInModels(overrides, modelOverrides);
+    const builtInModels = loadBuiltInModels(overrides, modelOverrides, baseModels);
     const builtInProviders = new Set(builtInModels.map((model) => model.provider));
     const customOpenAICompatibleProviders = new Set(customModels
         .filter((model) => !builtInProviders.has(model.provider) && OPENAI_COMPATIBLE_APIS.has(model.api))

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

@@ -1 +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,sBAAsB,GAAG,IAAI,GAAG,CAAM,CAAC,oBAAoB,EAAE,kBAAkB,CAAC,CAAC,CAAC;AAExF,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,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/E,MAAM,+BAA+B,GAAG,IAAI,GAAG,CAC9C,YAAY;SACV,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,sBAAsB,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;SACjG,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAChC,CAAC;IACF,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,cAAc;QACd,MAAM,EAAE,QAAQ;QAChB,sBAAsB;QACtB,mBAAmB;QACnB,gBAAgB;QAChB,+BAA+B;QAC/B,SAAS,EAAE,KAAK;KAChB,CAAC;AACH,CAAC","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai/compat\";\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\nconst OPENAI_COMPATIBLE_APIS = new Set<Api>([\"openai-completions\", \"openai-responses\"]);\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\tconst builtInProviders = new Set(builtInModels.map((model) => model.provider));\n\tconst customOpenAICompatibleProviders = new Set(\n\t\tcustomModels\n\t\t\t.filter((model) => !builtInProviders.has(model.provider) && OPENAI_COMPATIBLE_APIS.has(model.api))\n\t\t\t.map((model) => model.provider),\n\t);\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\tmodelOverrides,\n\t\tmodels: combined,\n\t\tproviderRequestConfigs,\n\t\tmodelRequestHeaders,\n\t\tbuiltInProviders,\n\t\tcustomOpenAICompatibleProviders,\n\t\tloadError: error,\n\t};\n}\n"]}
+{"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,sBAAsB,GAAG,IAAI,GAAG,CAAM,CAAC,oBAAoB,EAAE,kBAAkB,CAAC,CAAC,CAAC;AAExF,MAAM,UAAU,uBAAuB,CACtC,WAAwB,EACxB,eAAyB,EACzB,UAAkC;IAElC,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,EAAE,UAAU,CAAC,CAAC;IAC/E,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/E,MAAM,+BAA+B,GAAG,IAAI,GAAG,CAC9C,YAAY;SACV,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,sBAAsB,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;SACjG,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAChC,CAAC;IACF,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,cAAc;QACd,MAAM,EAAE,QAAQ;QAChB,sBAAsB;QACtB,mBAAmB;QACnB,gBAAgB;QAChB,+BAA+B;QAC/B,SAAS,EAAE,KAAK;KAChB,CAAC;AACH,CAAC","sourcesContent":["import type { Api, Model } from \"@earendil-works/pi-ai/compat\";\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\nconst OPENAI_COMPATIBLE_APIS = new Set<Api>([\"openai-completions\", \"openai-responses\"]);\n\nexport function loadModelRegistryModels(\n\tauthStorage: AuthStorage,\n\tmodelsJsonPaths: string[],\n\tbaseModels?: readonly Model<Api>[],\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, baseModels);\n\tconst builtInProviders = new Set(builtInModels.map((model) => model.provider));\n\tconst customOpenAICompatibleProviders = new Set(\n\t\tcustomModels\n\t\t\t.filter((model) => !builtInProviders.has(model.provider) && OPENAI_COMPATIBLE_APIS.has(model.api))\n\t\t\t.map((model) => model.provider),\n\t);\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\tmodelOverrides,\n\t\tmodels: combined,\n\t\tproviderRequestConfigs,\n\t\tmodelRequestHeaders,\n\t\tbuiltInProviders,\n\t\tcustomOpenAICompatibleProviders,\n\t\tloadError: error,\n\t};\n}\n"]}

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

@@ -1,4 +1,7 @@
-import type { AnthropicMessagesCompat, Api, AssistantMessageEventStream, Context, Model, OAuthProviderInterface, OpenAICompletionsCompat, OpenAIResponsesCompat, SimpleStreamOptions } from "@earendil-works/pi-ai/compat";
+import type { AnthropicMessagesCompat, Api, AssistantMessageEventStream, Context, Model, OpenAICompletionsCompat, OpenAIResponsesCompat, SimpleStreamOptions } from "@earendil-works/pi-ai/compat";
+import type { ProviderHeaders } from "@earendil-works/pi-ai";
+import type { LegacyOAuthProvider } from "./oauth-provider-bridge.ts";
+import type { RefreshModelsContext } from "@earendil-works/pi-ai";
 import type { AuthStorage } from "./auth-storage.ts";
 import type { ModelOverride } from "./model-registry-schemas.ts";
 export interface ProviderOverride {
@@ -13,7 +16,8 @@ export interface ProviderRequestConfig {
 export type ResolvedRequestAuth = {
     ok: true;
     apiKey?: string;
-    headers?: Record<string, string>;
+    headers?: ProviderHeaders;
+    baseUrl?: string;
 } | {
     ok: false;
     error: string;
@@ -39,6 +43,8 @@ export interface ModelRegistryLoadResult {
 }
 export interface DynamicProviderApplyInput {
     providerName: string;
+    registrationSource: string;
+    registerRuntime?: boolean;
     config: ProviderConfigInput;
     models: Model<Api>[];
     modelOverrides: Map<string, Map<string, ModelOverride>>;
@@ -55,7 +61,8 @@ export interface ProviderConfigInput {
     streamSimple?: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;
     headers?: Record<string, string>;
     authHeader?: boolean;
-    oauth?: Omit<OAuthProviderInterface, "id">;
+    refreshModels?(context: RefreshModelsContext): Promise<NonNullable<ProviderConfigInput["models"]>>;
+    oauth?: LegacyOAuthProvider;
     models?: Array<{
         id: string;
         name: string;

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

@@ -1 +1 @@
-{"version":3,"file":"model-registry-types.d.ts","sourceRoot":"","sources":["../../src/core/model-registry-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,uBAAuB,EACvB,GAAG,EACH,2BAA2B,EAC3B,OAAO,EACP,KAAK,EACL,sBAAsB,EACtB,uBAAuB,EACvB,qBAAqB,EACrB,mBAAmB,EACnB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAEjE,MAAM,WAAW,gBAAgB;IAChC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;CAC9B;AAED,MAAM,WAAW,qBAAqB;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,UAAU,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,MAAM,mBAAmB,GAC5B;IACA,EAAE,EAAE,IAAI,CAAC;IACT,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAChC,GACD;IACA,EAAE,EAAE,KAAK,CAAC;IACV,KAAK,EAAE,MAAM,CAAC;CACb,CAAC;AAEL,MAAM,MAAM,wBAAwB,GAAG,OAAO,GAAG,eAAe,CAAC;AAEjE,MAAM,WAAW,kBAAkB;IAClC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IACrB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACzC,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IACxD,sBAAsB,EAAE,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC;IAC3D,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACzD,yBAAyB,EAAE,GAAG,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;IACjE,KAAK,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1B;AAED,MAAM,WAAW,uBAAuB;IACvC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IACrB,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IACxD,sBAAsB,EAAE,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC;IAC3D,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACzD,gBAAgB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,+BAA+B,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC7C,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B;AAED,MAAM,WAAW,yBAAyB;IACzC,YAAY,EAAE,MAAM,CAAC;IACrB,MAAM,EAAE,mBAAmB,CAAC;IAC5B,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IACrB,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IACxD,WAAW,EAAE,WAAW,CAAC;IACzB,0BAA0B,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,qBAAqB,KAAK,IAAI,CAAC;IAC1F,iBAAiB,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC;CACrG;AAED,MAAM,MAAM,cAAc,GACvB,uBAAuB,GACvB,qBAAqB,GACrB,uBAAuB,CAAC;AAE3B,MAAM,WAAW,mBAAmB;IACnC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,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,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,gBAAgB,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,kBAAkB,CAAC,CAAC;QAClD,KAAK,EAAE,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;QAC5B,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;QACzB,aAAa,EAAE,MAAM,CAAC;QACtB,oBAAoB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;QACzC,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-types.d.ts","sourceRoot":"","sources":["../../src/core/model-registry-types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACX,uBAAuB,EACvB,GAAG,EACH,2BAA2B,EAC3B,OAAO,EACP,KAAK,EACL,uBAAuB,EACvB,qBAAqB,EACrB,mBAAmB,EACnB,MAAM,8BAA8B,CAAC;AACtC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAEjE,MAAM,WAAW,gBAAgB;IAChC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;CAC9B;AAED,MAAM,WAAW,qBAAqB;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACjC,UAAU,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,MAAM,mBAAmB,GAC5B;IACA,EAAE,EAAE,IAAI,CAAC;IACT,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,eAAe,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;CAChB,GACD;IACA,EAAE,EAAE,KAAK,CAAC;IACV,KAAK,EAAE,MAAM,CAAC;CACb,CAAC;AAEL,MAAM,MAAM,wBAAwB,GAAG,OAAO,GAAG,eAAe,CAAC;AAEjE,MAAM,WAAW,kBAAkB;IAClC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IACrB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACzC,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IACxD,sBAAsB,EAAE,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC;IAC3D,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACzD,yBAAyB,EAAE,GAAG,CAAC,MAAM,EAAE,wBAAwB,CAAC,CAAC;IACjE,KAAK,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1B;AAED,MAAM,WAAW,uBAAuB;IACvC,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IACrB,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IACxD,sBAAsB,EAAE,GAAG,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC;IAC3D,mBAAmB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACzD,gBAAgB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,+BAA+B,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC7C,SAAS,EAAE,MAAM,GAAG,SAAS,CAAC;CAC9B;AAED,MAAM,WAAW,yBAAyB;IACzC,YAAY,EAAE,MAAM,CAAC;IACrB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,MAAM,EAAE,mBAAmB,CAAC;IAC5B,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;IACrB,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;IACxD,WAAW,EAAE,WAAW,CAAC;IACzB,0BAA0B,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,qBAAqB,KAAK,IAAI,CAAC;IAC1F,iBAAiB,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC;CACrG;AAED,MAAM,MAAM,cAAc,GACvB,uBAAuB,GACvB,qBAAqB,GACrB,uBAAuB,CAAC;AAE3B,MAAM,WAAW,mBAAmB;IACnC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,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,aAAa,CAAC,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,WAAW,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IACnG,KAAK,CAAC,EAAE,mBAAmB,CAAC;IAC5B,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,gBAAgB,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,kBAAkB,CAAC,CAAC;QAClD,KAAK,EAAE,CAAC,MAAM,GAAG,OAAO,CAAC,EAAE,CAAC;QAC5B,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;QACzB,aAAa,EAAE,MAAM,CAAC;QACtB,oBAAoB,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;QACzC,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"}

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

@@ -1 +1 @@
-{"version":3,"file":"model-registry-types.js","sourceRoot":"","sources":["../../src/core/model-registry-types.ts"],"names":[],"mappings":"","sourcesContent":["import type {\n\tAnthropicMessagesCompat,\n\tApi,\n\tAssistantMessageEventStream,\n\tContext,\n\tModel,\n\tOAuthProviderInterface,\n\tOpenAICompletionsCompat,\n\tOpenAIResponsesCompat,\n\tSimpleStreamOptions,\n} from \"@earendil-works/pi-ai/compat\";\nimport type { AuthStorage } from \"./auth-storage.ts\";\nimport type { ModelOverride } from \"./model-registry-schemas.ts\";\n\nexport interface ProviderOverride {\n\tbaseUrl?: string;\n\tcompat?: Model<Api>[\"compat\"];\n}\n\nexport interface ProviderRequestConfig {\n\tapiKey?: string;\n\theaders?: Record<string, string>;\n\tauthHeader?: boolean;\n}\n\nexport type ResolvedRequestAuth =\n\t| {\n\t\t\tok: true;\n\t\t\tapiKey?: string;\n\t\t\theaders?: Record<string, string>;\n\t  }\n\t| {\n\t\t\tok: false;\n\t\t\terror: string;\n\t  };\n\nexport type ModelRequestHeaderSource = \"model\" | \"modelOverride\";\n\nexport interface CustomModelsResult {\n\tmodels: Model<Api>[];\n\toverrides: Map<string, ProviderOverride>;\n\tmodelOverrides: Map<string, Map<string, ModelOverride>>;\n\tproviderRequestConfigs: Map<string, ProviderRequestConfig>;\n\tmodelRequestHeaders: Map<string, Record<string, string>>;\n\tmodelRequestHeaderSources: Map<string, ModelRequestHeaderSource>;\n\terror: string | undefined;\n}\n\nexport interface ModelRegistryLoadResult {\n\tmodels: Model<Api>[];\n\tmodelOverrides: Map<string, Map<string, ModelOverride>>;\n\tproviderRequestConfigs: Map<string, ProviderRequestConfig>;\n\tmodelRequestHeaders: Map<string, Record<string, string>>;\n\tbuiltInProviders: Set<string>;\n\tcustomOpenAICompatibleProviders: Set<string>;\n\tloadError: string | undefined;\n}\n\nexport interface DynamicProviderApplyInput {\n\tproviderName: string;\n\tconfig: ProviderConfigInput;\n\tmodels: Model<Api>[];\n\tmodelOverrides: Map<string, Map<string, ModelOverride>>;\n\tauthStorage: AuthStorage;\n\tstoreProviderRequestConfig: (providerName: string, config: ProviderRequestConfig) => void;\n\tstoreModelHeaders: (providerName: string, modelId: string, headers?: Record<string, string>) => void;\n}\n\nexport type ProviderCompat =\n\t| OpenAICompletionsCompat\n\t| OpenAIResponsesCompat\n\t| AnthropicMessagesCompat;\n\nexport interface ProviderConfigInput {\n\tname?: string;\n\tbaseUrl?: string;\n\tapiKey?: string;\n\tapi?: Api;\n\tstreamSimple?: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;\n\theaders?: Record<string, string>;\n\tauthHeader?: boolean;\n\toauth?: Omit<OAuthProviderInterface, \"id\">;\n\tmodels?: Array<{\n\t\tid: string;\n\t\tname: string;\n\t\tapi?: Api;\n\t\tbaseUrl?: string;\n\t\treasoning: boolean;\n\t\tthinkingLevelMap?: Model<Api>[\"thinkingLevelMap\"];\n\t\tinput: (\"text\" | \"image\")[];\n\t\tcost: Model<Api>[\"cost\"];\n\t\tcontextWindow: number;\n\t\tcontextWindowOptions?: readonly number[];\n\t\tmaxTokens: number;\n\t\theaders?: Record<string, string>;\n\t\tcompat?: Model<Api>[\"compat\"];\n\t}>;\n}\n"]}
+{"version":3,"file":"model-registry-types.js","sourceRoot":"","sources":["../../src/core/model-registry-types.ts"],"names":[],"mappings":"","sourcesContent":["import type {\n\tAnthropicMessagesCompat,\n\tApi,\n\tAssistantMessageEventStream,\n\tContext,\n\tModel,\n\tOpenAICompletionsCompat,\n\tOpenAIResponsesCompat,\n\tSimpleStreamOptions,\n} from \"@earendil-works/pi-ai/compat\";\nimport type { ProviderHeaders } from \"@earendil-works/pi-ai\";\nimport type { LegacyOAuthProvider } from \"./oauth-provider-bridge.ts\";\nimport type { RefreshModelsContext } from \"@earendil-works/pi-ai\";\nimport type { AuthStorage } from \"./auth-storage.ts\";\nimport type { ModelOverride } from \"./model-registry-schemas.ts\";\n\nexport interface ProviderOverride {\n\tbaseUrl?: string;\n\tcompat?: Model<Api>[\"compat\"];\n}\n\nexport interface ProviderRequestConfig {\n\tapiKey?: string;\n\theaders?: Record<string, string>;\n\tauthHeader?: boolean;\n}\n\nexport type ResolvedRequestAuth =\n\t| {\n\t\t\tok: true;\n\t\t\tapiKey?: string;\n\t\t\theaders?: ProviderHeaders;\n\t\t\tbaseUrl?: string;\n\t  }\n\t| {\n\t\t\tok: false;\n\t\t\terror: string;\n\t  };\n\nexport type ModelRequestHeaderSource = \"model\" | \"modelOverride\";\n\nexport interface CustomModelsResult {\n\tmodels: Model<Api>[];\n\toverrides: Map<string, ProviderOverride>;\n\tmodelOverrides: Map<string, Map<string, ModelOverride>>;\n\tproviderRequestConfigs: Map<string, ProviderRequestConfig>;\n\tmodelRequestHeaders: Map<string, Record<string, string>>;\n\tmodelRequestHeaderSources: Map<string, ModelRequestHeaderSource>;\n\terror: string | undefined;\n}\n\nexport interface ModelRegistryLoadResult {\n\tmodels: Model<Api>[];\n\tmodelOverrides: Map<string, Map<string, ModelOverride>>;\n\tproviderRequestConfigs: Map<string, ProviderRequestConfig>;\n\tmodelRequestHeaders: Map<string, Record<string, string>>;\n\tbuiltInProviders: Set<string>;\n\tcustomOpenAICompatibleProviders: Set<string>;\n\tloadError: string | undefined;\n}\n\nexport interface DynamicProviderApplyInput {\n\tproviderName: string;\n\tregistrationSource: string;\n\tregisterRuntime?: boolean;\n\tconfig: ProviderConfigInput;\n\tmodels: Model<Api>[];\n\tmodelOverrides: Map<string, Map<string, ModelOverride>>;\n\tauthStorage: AuthStorage;\n\tstoreProviderRequestConfig: (providerName: string, config: ProviderRequestConfig) => void;\n\tstoreModelHeaders: (providerName: string, modelId: string, headers?: Record<string, string>) => void;\n}\n\nexport type ProviderCompat =\n\t| OpenAICompletionsCompat\n\t| OpenAIResponsesCompat\n\t| AnthropicMessagesCompat;\n\nexport interface ProviderConfigInput {\n\tname?: string;\n\tbaseUrl?: string;\n\tapiKey?: string;\n\tapi?: Api;\n\tstreamSimple?: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;\n\theaders?: Record<string, string>;\n\tauthHeader?: boolean;\n\trefreshModels?(context: RefreshModelsContext): Promise<NonNullable<ProviderConfigInput[\"models\"]>>;\n\toauth?: LegacyOAuthProvider;\n\tmodels?: Array<{\n\t\tid: string;\n\t\tname: string;\n\t\tapi?: Api;\n\t\tbaseUrl?: string;\n\t\treasoning: boolean;\n\t\tthinkingLevelMap?: Model<Api>[\"thinkingLevelMap\"];\n\t\tinput: (\"text\" | \"image\")[];\n\t\tcost: Model<Api>[\"cost\"];\n\t\tcontextWindow: number;\n\t\tcontextWindowOptions?: readonly number[];\n\t\tmaxTokens: number;\n\t\theaders?: Record<string, string>;\n\t\tcompat?: Model<Api>[\"compat\"];\n\t}>;\n}\n"]}

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

@@ -1,6 +1,7 @@
 /**
  * Model registry - manages built-in and custom models, provides API key resolution.
  */
+import { type ModelsRefreshResult } from "@earendil-works/pi-ai";
 import { type Api, type Model } from "@earendil-works/pi-ai/compat";
 import type { AuthStatus, AuthStorage } from "./auth-storage.ts";
 import type { ProviderConfigInput, ResolvedRequestAuth } from "./model-registry-types.ts";
@@ -20,6 +21,11 @@ export declare class ModelRegistry {
     private builtInProviders;
     private customOpenAICompatibleProviders;
     private loadError;
+    private refreshGeneration;
+    private readonly registrationSource;
+    private readonly modelsStore;
+    private readonly credentialStore;
+    private readonly providerModels;
     readonly authStorage: AuthStorage;
     private modelsJsonPaths;
     private constructor();
@@ -29,7 +35,15 @@ export declare class ModelRegistry {
     /**
      * Reload models from disk (built-in + custom from models.json).
      */
-    refresh(): void;
+    refresh(options?: {
+        signal?: AbortSignal;
+        timeoutMs?: number;
+        force?: boolean;
+        allowNetwork?: boolean;
+    }): Promise<ModelsRefreshResult>;
+    private publishProviderModels;
+    private rebuildProviderModels;
+    private providerRegistrationSource;
     /**
      * Get any error from loading models.json (undefined if no error).
      */

package/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,EAAE,KAAK,GAAG,EAAE,KAAK,KAAK,EAAqB,MAAM,8BAA8B,CAAC;AAKvF,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAKjE,OAAO,KAAK,EAAE,mBAAmB,EAAyB,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAGjH,OAAO,EAAE,qBAAqB,EAA2B,MAAM,2BAA2B,CAAC;AAK3F,YAAY,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAE1F,kEAAkE;AAClE,eAAO,MAAM,gBAAgB,8BAAwB,CAAC;AAEtD;;GAEG;AACH,qBAAa,aAAa;IACzB,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,cAAc,CAAsD;IAC5E,OAAO,CAAC,sBAAsB,CAAiD;IAC/E,OAAO,CAAC,mBAAmB,CAAkD;IAC7E,OAAO,CAAC,mBAAmB,CAA+C;IAC1E,OAAO,CAAC,gBAAgB,CAA0B;IAClD,OAAO,CAAC,+BAA+B,CAA0B;IACjE,OAAO,CAAC,SAAS,CAAiC;IAElD,SAAiB,WAAW,EAAE,WAAW,CAAC;IAC1C,QAAgB,eAAe,CAAW;IAE1C,OAAO,eAQN;IAED,OAAO,CAAC,gCAAgC;IAQxC,MAAM,CAAC,MAAM,CACZ,WAAW,EAAE,WAAW,EACxB,cAAc,GAAE,MAAM,GAAG,MAAM,EAAuC,GACpE,aAAa,CAEf;IAED,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,GAAG,aAAa,CAEvD;IAED;;OAEG;IACH,OAAO,IAAI,IAAI,CAad;IAED;;OAEG;IACH,QAAQ,IAAI,MAAM,GAAG,SAAS,CAE7B;IAED,OAAO,CAAC,UAAU;IAWlB;;;OAGG;IACH,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAErB;IAED;;;OAGG;IACH,YAAY,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAE3B;IAED;;OAEG;IACH,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS,CAE9D;IAED,kFAAkF;IAClF,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAYhD;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAM5C;IAED,OAAO,CAAC,kBAAkB;IAI1B,OAAO,CAAC,0BAA0B;IAmBlC,OAAO,CAAC,iBAAiB;IASzB;;OAEG;IACG,mBAAmB,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAEzE;IAED;;;OAGG;IACH,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,UAAU,CAElD;IAED;;OAEG;IACH,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAW/C;IAED;;OAEG;IACG,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAExE;IAED;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAGvC;IAED;;OAEG;IACH,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,GAAG,IAAI,CAKxE;IAED;;OAEG;IACH,+BAA+B,CAAC,GAAG,EAAE,GAAG,GAAG,OAAO,CAOjD;IAED;;OAEG;IACH,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAI7C;IAED,OAAO,CAAC,wBAAwB;IAahC,OAAO,CAAC,mBAAmB;CAW3B"}
+{"version":3,"file":"model-registry.d.ts","sourceRoot":"","sources":["../../src/core/model-registry.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAKN,KAAK,mBAAmB,EAExB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EAAE,KAAK,GAAG,EAAE,KAAK,KAAK,EAAE,MAAM,8BAA8B,CAAC;AAIpE,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAKjE,OAAO,KAAK,EAAE,mBAAmB,EAAyB,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAMjH,OAAO,EAAE,qBAAqB,EAA2B,MAAM,2BAA2B,CAAC;AAwC3F,YAAY,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAE1F,kEAAkE;AAClE,eAAO,MAAM,gBAAgB,8BAAwB,CAAC;AAEtD;;GAEG;AACH,qBAAa,aAAa;IACzB,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,cAAc,CAAsD;IAC5E,OAAO,CAAC,sBAAsB,CAAiD;IAC/E,OAAO,CAAC,mBAAmB,CAAkD;IAC7E,OAAO,CAAC,mBAAmB,CAA+C;IAC1E,OAAO,CAAC,gBAAgB,CAA0B;IAClD,OAAO,CAAC,+BAA+B,CAA0B;IACjE,OAAO,CAAC,SAAS,CAAiC;IAClD,OAAO,CAAC,iBAAiB,CAAK;IAC9B,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAA2D;IAC9F,iBAAyB,WAAW,CAAyB;IAC7D,iBAAyB,eAAe,CAAkB;IAC1D,iBAAyB,cAAc,CAAgB;IAEvD,SAAiB,WAAW,EAAE,WAAW,CAAC;IAC1C,QAAgB,eAAe,CAAW;IAE1C,OAAO,eAmBN;IAED,OAAO,CAAC,gCAAgC;IAQxC,MAAM,CAAC,MAAM,CACZ,WAAW,EAAE,WAAW,EACxB,cAAc,GAAE,MAAM,GAAG,MAAM,EAAuC,GACpE,aAAa,CAEf;IAED,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE,WAAW,GAAG,aAAa,CAEvD;IAED;;OAEG;IACG,OAAO,CAAC,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAO,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAmG/I;IAED,OAAO,CAAC,qBAAqB;IAI7B,OAAO,CAAC,qBAAqB;IAO7B,OAAO,CAAC,0BAA0B;IAIlC;;OAEG;IACH,QAAQ,IAAI,MAAM,GAAG,SAAS,CAE7B;IAED,OAAO,CAAC,UAAU;IAWlB;;;OAGG;IACH,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAErB;IAED;;;OAGG;IACH,YAAY,IAAI,KAAK,CAAC,GAAG,CAAC,EAAE,CAiB3B;IAED;;OAEG;IACH,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,SAAS,CAE9D;IAED,kFAAkF;IAClF,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAYhD;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAM5C;IAED,OAAO,CAAC,kBAAkB;IAI1B,OAAO,CAAC,0BAA0B;IAmBlC,OAAO,CAAC,iBAAiB;IASzB;;OAEG;IACG,mBAAmB,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,mBAAmB,CAAC,CA+BzE;IAED;;;OAGG;IACH,qBAAqB,CAAC,QAAQ,EAAE,MAAM,GAAG,UAAU,CAElD;IAED;;OAEG;IACH,sBAAsB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,CAW/C;IAED;;OAEG;IACG,oBAAoB,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAExE;IAED;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAGvC;IAED;;OAEG;IACH,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,mBAAmB,GAAG,IAAI,CAOxE;IAED;;OAEG;IACH,+BAA+B,CAAC,GAAG,EAAE,GAAG,GAAG,OAAO,CAOjD;IAED;;OAEG;IACH,kBAAkB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAK7C;IAED,OAAO,CAAC,wBAAwB;IAShC,OAAO,CAAC,mBAAmB;CAa3B"}

package/dist/core/model-registry.js

@@ -1,19 +1,50 @@
 /**
  * Model registry - manages built-in and custom models, provides API key resolution.
  */
-import { resetApiProviders } from "@earendil-works/pi-ai/compat";
-import { resetOAuthProviders } from "@earendil-works/pi-ai/oauth";
-import { dirname } from "node:path";
+import { createModels, } from "@earendil-works/pi-ai";
+import { builtinProviders } from "@earendil-works/pi-ai/providers/all";
+import { dirname, join } from "node:path";
 import { getAgentConfigPaths } from "../config.js";
 import { normalizePath } from "../utils/paths.js";
-import { copilotCatalogCachePath, copilotTokenFromEnvironment, seedActiveCopilotModelCatalogFromCache } from "./copilot-model-catalog.js";
+import { copilotApiBaseUrlFromToken, copilotCatalogCachePath, copilotTokenFromEnvironment, seedActiveCopilotModelCatalogFromCache } from "./copilot-model-catalog.js";
 import { getModelRequestAuth, getApiKeyForProviderFromConfig, getProviderAuthStatusFromConfig } from "./model-registry-auth.js";
-import { applyProviderConfigToModels, migrateLegacyRegisterProviderConfigValues, validateProviderConfig } from "./model-registry-dynamic.js";
+import { applyProviderConfigToModels, migrateLegacyRegisterProviderConfigValues, unregisterProviderRuntime, validateProviderConfig } from "./model-registry-dynamic.js";
 import { loadModelRegistryModels } from "./model-registry-loader.js";
 import { BUILT_IN_PROVIDER_DISPLAY_NAMES } from "./provider-display-names.js";
+import { FileModelsStore, InMemoryCodingAgentModelsStore } from "./models-store.js";
+import { getLegacyOAuthProvider, oauthCredentialToAuth } from "./oauth-provider-bridge.js";
+import { withRemoteCatalog } from "./remote-catalog-provider.js";
 import { clearConfigValueCache, isConfigValueConfigured } from "./resolve-config-value.js";
 const REMOTE_CATALOG_PROVIDERS = new Set(["cursor", "github-copilot", "openrouter", "vercel-ai-gateway"]);
 const OPENAI_COMPATIBLE_APIS = new Set(["openai-completions", "openai-responses"]);
+let nextRegistryRegistrationId = 0;
+function copilotBaseUrlFromRuntimeToken(apiKey) {
+    return /(?:^|;)proxy-ep=[^;]+/.test(apiKey) ? copilotApiBaseUrlFromToken(apiKey) : undefined;
+}
+function overlayRuntimeApiKey(runtimeApiKey, resolvedAuth, storedOAuthAuth, runtimeBaseUrl) {
+    const headers = storedOAuthAuth?.headers || resolvedAuth?.headers
+        ? { ...storedOAuthAuth?.headers, ...resolvedAuth?.headers }
+        : undefined;
+    return {
+        ...storedOAuthAuth,
+        ...resolvedAuth,
+        apiKey: runtimeApiKey,
+        headers,
+        baseUrl: runtimeBaseUrl ?? storedOAuthAuth?.baseUrl ?? resolvedAuth?.baseUrl,
+    };
+}
+function createProviderCredentialStore(credentials) {
+    return {
+        ...credentials,
+        read: async (providerId) => {
+            const credential = await credentials.read(providerId);
+            if (credential?.type !== "oauth" || !getLegacyOAuthProvider(providerId))
+                return credential;
+            const auth = await oauthCredentialToAuth(providerId, credential);
+            return auth?.apiKey === undefined ? undefined : { type: "api_key", key: auth.apiKey };
+        },
+    };
+}
 /** Clear the config value command cache. Exported for testing. */
 export const clearApiKeyCache = clearConfigValueCache;
 /**
@@ -29,8 +60,21 @@ export class ModelRegistry {
         this.builtInProviders = new Set();
         this.customOpenAICompatibleProviders = new Set();
         this.loadError = undefined;
+        this.refreshGeneration = 0;
+        this.registrationSource = `atomic:model-registry:${++nextRegistryRegistrationId}`;
         this.authStorage = authStorage;
         this.modelsJsonPaths = modelsJsonPaths.map((path) => normalizePath(path));
+        this.modelsStore = this.modelsJsonPaths.length > 0
+            ? new FileModelsStore(join(dirname(this.modelsJsonPaths[0]), "models-store.json"))
+            : new InMemoryCodingAgentModelsStore();
+        this.credentialStore = authStorage.asCredentialStore();
+        this.providerModels = createModels({
+            credentials: createProviderCredentialStore(this.credentialStore),
+            modelsStore: this.modelsStore,
+        });
+        for (const provider of builtinProviders()) {
+            this.providerModels.setProvider(provider.id === "radius" ? provider : withRemoteCatalog(provider));
+        }
         this.seedCopilotModelCatalogFromCache();
         this.loadModels();
     }
@@ -52,25 +96,132 @@ export class ModelRegistry {
     /**
      * Reload models from disk (built-in + custom from models.json).
      */
-    refresh() {
-        this.providerRequestConfigs.clear();
-        this.modelRequestHeaders.clear();
+    async refresh(options = {}) {
+        const generation = ++this.refreshGeneration;
         this.loadError = undefined;
-        resetApiProviders();
-        resetOAuthProviders();
-        this.loadModels();
-        for (const [providerName, config] of this.registeredProviders.entries()) {
+        this.rebuildProviderModels();
+        const controller = new AbortController();
+        const abort = () => controller.abort();
+        options.signal?.addEventListener("abort", abort, { once: true });
+        if (options.signal?.aborted)
+            controller.abort(options.signal.reason);
+        const timeout = setTimeout(abort, options.timeoutMs ?? 15_000);
+        const errors = new Map();
+        const aborted = new Promise((resolve) => {
+            if (controller.signal.aborted)
+                resolve();
+            else
+                controller.signal.addEventListener("abort", () => resolve(), { once: true });
+        });
+        try {
+            const restore = this.providerModels
+                .refresh({ allowNetwork: false, signal: controller.signal })
+                .then((result) => {
+                if (controller.signal.aborted || generation !== this.refreshGeneration)
+                    return;
+                for (const [provider, error] of result.errors)
+                    errors.set(provider, error);
+                this.publishProviderModels();
+            });
+            await Promise.race([restore, aborted]);
+            if (options.allowNetwork !== false && !controller.signal.aborted) {
+                const legacyOAuthProviders = new Set([
+                    ...this.models.map((model) => model.provider),
+                    ...[...this.registeredProviders]
+                        .filter(([, config]) => config.oauth !== undefined)
+                        .map(([providerId]) => providerId),
+                ].filter((providerId) => getLegacyOAuthProvider(providerId) !== undefined));
+                const refreshLegacyOAuth = Promise.all([...legacyOAuthProviders].map(async (providerId) => {
+                    const credential = this.authStorage.get(providerId);
+                    if (credential?.type !== "oauth" || Date.now() < credential.expires)
+                        return;
+                    await this.authStorage.getModelAuth(providerId, { includeFallback: false });
+                }));
+                await Promise.race([refreshLegacyOAuth, aborted]);
+            }
+            const extensionRefreshes = controller.signal.aborted ? [] : [...this.registeredProviders].map(async ([providerName, config]) => {
+                if (!config.refreshModels)
+                    return;
+                const isCurrentExtensionRefresh = () => !controller.signal.aborted
+                    && generation === this.refreshGeneration
+                    && this.registeredProviders.get(providerName) === config;
+                const store = {
+                    read: () => this.modelsStore.read(providerName),
+                    write: (entry) => this.modelsStore.writeIf(providerName, entry, isCurrentExtensionRefresh),
+                    delete: () => this.modelsStore.deleteIf(providerName, isCurrentExtensionRefresh),
+                };
+                try {
+                    let credential = await this.credentialStore.read(providerName);
+                    if (credential?.type === "api_key") {
+                        const effectiveApiKey = await this.authStorage.getApiKey(providerName, { includeFallback: false });
+                        credential = effectiveApiKey === undefined ? undefined : { type: "api_key", key: effectiveApiKey };
+                    }
+                    if (credential === undefined) {
+                        const configuredApiKey = await getApiKeyForProviderFromConfig(providerName, this.authStorage, this.providerRequestConfigs);
+                        if (configuredApiKey !== undefined)
+                            credential = { type: "api_key", key: configuredApiKey };
+                    }
+                    if (!isCurrentExtensionRefresh())
+                        return;
+                    const models = await config.refreshModels({
+                        credential,
+                        store,
+                        allowNetwork: options.allowNetwork ?? true,
+                        force: options.force,
+                        signal: controller.signal,
+                    });
+                    if (!isCurrentExtensionRefresh())
+                        return;
+                    const refreshed = { ...config, models };
+                    this.registeredProviders.set(providerName, refreshed);
+                    this.rebuildProviderModels();
+                }
+                catch (error) {
+                    if (generation === this.refreshGeneration && !controller.signal.aborted) {
+                        errors.set(providerName, error instanceof Error ? error : new Error(String(error)));
+                    }
+                }
+            });
+            const builtinRefresh = controller.signal.aborted
+                ? Promise.resolve()
+                : this.providerModels
+                    .refresh({ allowNetwork: options.allowNetwork ?? true, force: options.force, signal: controller.signal })
+                    .then((result) => {
+                    if (controller.signal.aborted || generation !== this.refreshGeneration)
+                        return;
+                    for (const [provider, error] of result.errors)
+                        errors.set(provider, error);
+                    this.publishProviderModels();
+                });
+            await Promise.race([Promise.all(extensionRefreshes), aborted]);
+            await Promise.race([builtinRefresh, aborted]);
+        }
+        finally {
+            clearTimeout(timeout);
+            options.signal?.removeEventListener("abort", abort);
+        }
+        return { aborted: controller.signal.aborted, errors };
+    }
+    publishProviderModels() {
+        this.rebuildProviderModels();
+    }
+    rebuildProviderModels() {
+        this.loadModels(this.providerModels.getModels());
+        for (const [providerName, config] of this.registeredProviders) {
             this.applyProviderConfig(providerName, config);
         }
     }
+    providerRegistrationSource(providerName) {
+        return `${this.registrationSource}:${providerName}`;
+    }
     /**
      * Get any error from loading models.json (undefined if no error).
      */
     getError() {
         return this.loadError;
     }
-    loadModels() {
-        const loaded = loadModelRegistryModels(this.authStorage, this.modelsJsonPaths);
+    loadModels(baseModels) {
+        const loaded = loadModelRegistryModels(this.authStorage, this.modelsJsonPaths, baseModels);
         this.modelOverrides = loaded.modelOverrides;
         this.models = loaded.models;
         this.providerRequestConfigs = loaded.providerRequestConfigs;
@@ -91,7 +242,20 @@ export class ModelRegistry {
      * This is a fast check that doesn't refresh OAuth tokens.
      */
     getAvailable() {
-        return this.models.filter((m) => this.hasConfiguredAuth(m));
+        const configured = this.models.filter((model) => this.hasConfiguredAuth(model));
+        const allowedByProvider = new Map();
+        for (const provider of this.providerModels.getProviders()) {
+            const extension = this.registeredProviders.get(provider.id);
+            if (!provider.filterModels || extension?.models || extension?.oauth)
+                continue;
+            const providerModels = configured.filter((model) => model.provider === provider.id);
+            const runtimeApiKey = this.authStorage.getRuntimeApiKey(provider.id);
+            const credential = runtimeApiKey === undefined
+                ? this.authStorage.get(provider.id)
+                : { type: "api_key", key: runtimeApiKey };
+            allowedByProvider.set(provider.id, new Set(provider.filterModels(providerModels, credential).map((model) => model.id)));
+        }
+        return configured.filter((model) => allowedByProvider.get(model.provider)?.has(model.id) ?? true);
     }
     /**
      * Find a model by provider and ID.
@@ -146,7 +310,28 @@ export class ModelRegistry {
      * Get API key and request headers for a model.
      */
     async getApiKeyAndHeaders(model) {
-        return getModelRequestAuth(model, this.authStorage, this.providerRequestConfigs, this.modelRequestHeaders);
+        try {
+            const runtimeApiKey = this.authStorage.getRuntimeApiKey(model.provider);
+            const extensionReplacesOAuth = this.registeredProviders.get(model.provider)?.oauth !== undefined
+                || getLegacyOAuthProvider(model.provider) !== undefined;
+            const resolvedProviderAuth = this.providerModels.getProvider(model.provider) && !extensionReplacesOAuth
+                ? (await this.providerModels.getAuth(model, runtimeApiKey === undefined ? undefined : { apiKey: runtimeApiKey }))?.auth
+                : undefined;
+            const storedCredential = this.authStorage.get(model.provider);
+            const storedOAuthAuth = runtimeApiKey !== undefined && !extensionReplacesOAuth && storedCredential?.type === "oauth"
+                ? await oauthCredentialToAuth(model.provider, storedCredential)
+                : undefined;
+            const runtimeBaseUrl = runtimeApiKey !== undefined && model.provider === "github-copilot"
+                ? copilotBaseUrlFromRuntimeToken(runtimeApiKey)
+                : undefined;
+            const providerAuth = runtimeApiKey === undefined
+                ? resolvedProviderAuth
+                : overlayRuntimeApiKey(runtimeApiKey, resolvedProviderAuth, storedOAuthAuth, runtimeBaseUrl);
+            return getModelRequestAuth(model, this.authStorage, this.providerRequestConfigs, this.modelRequestHeaders, providerAuth);
+        }
+        catch (error) {
+            return { ok: false, error: error instanceof Error ? error.message : String(error) };
+        }
     }
     /**
      * Return auth status for a provider, including request auth configured in models.json.
@@ -186,8 +371,10 @@ export class ModelRegistry {
     registerProvider(providerName, config) {
         const migratedConfig = migrateLegacyRegisterProviderConfigValues(providerName, config);
         validateProviderConfig(providerName, migratedConfig);
-        this.applyProviderConfig(providerName, migratedConfig);
-        this.upsertRegisteredProvider(providerName, migratedConfig);
+        const mergedConfig = this.upsertRegisteredProvider(providerName, migratedConfig);
+        unregisterProviderRuntime(this.providerRegistrationSource(providerName));
+        this.rebuildProviderModels();
+        this.applyProviderConfig(providerName, mergedConfig, true);
     }
     /**
      * Check whether extensions have registered custom streamSimple dispatch for an API.
@@ -207,22 +394,19 @@ export class ModelRegistry {
         if (!this.registeredProviders.has(providerName))
             return;
         this.registeredProviders.delete(providerName);
-        this.refresh();
+        unregisterProviderRuntime(this.providerRegistrationSource(providerName));
+        this.rebuildProviderModels();
     }
     upsertRegisteredProvider(providerName, config) {
-        const existing = this.registeredProviders.get(providerName);
-        if (!existing) {
-            this.registeredProviders.set(providerName, config);
-            return;
-        }
-        for (const k of Object.keys(config)) {
-            if (config[k] !== undefined) {
-                existing[k] = config[k];
-            }
-        }
+        const definedConfig = Object.fromEntries(Object.entries(config).filter(([, value]) => value !== undefined));
+        const merged = { ...this.registeredProviders.get(providerName), ...definedConfig };
+        this.registeredProviders.set(providerName, merged);
+        return merged;
     }
-    applyProviderConfig(providerName, config) {
+    applyProviderConfig(providerName, config, registerRuntime = false) {
         this.models = applyProviderConfigToModels({
+            registrationSource: this.providerRegistrationSource(providerName),
+            registerRuntime,
             providerName,
             config,
             models: this.models,