@bastani/atomic 0.9.11-alpha.7 → 0.9.11-alpha.8
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +24 -0
- package/README.md +3 -5
- package/dist/builtin/intercom/package.json +1 -1
- package/dist/builtin/mcp/package.json +1 -1
- package/dist/builtin/mcp/sampling-handler.ts +1 -5
- package/dist/builtin/subagents/package.json +1 -1
- package/dist/builtin/web-access/package.json +1 -1
- package/dist/builtin/workflows/CHANGELOG.md +1 -0
- package/dist/builtin/workflows/README.md +1 -1
- package/dist/builtin/workflows/package.json +1 -1
- package/dist/builtin/workflows/src/extension/extension-runtime-state.ts +1 -24
- package/dist/builtin/workflows/src/extension/index.bundle.mjs +31 -28
- package/dist/builtin/workflows/src/extension/workflow-model-catalog.ts +26 -0
- package/dist/builtin/workflows/src/extension/workflow-prompts.ts +3 -2
- package/dist/builtin/workflows/src/runs/foreground/stage-runner-controller.ts +5 -5
- package/dist/builtin/workflows/src/runs/foreground/stage-runner-session-options.ts +3 -3
- package/dist/builtin/workflows/src/runs/foreground/stage-runner-session.ts +2 -2
- package/dist/cli/list-models.d.ts +2 -2
- package/dist/cli/list-models.d.ts.map +1 -1
- package/dist/cli/list-models.js +5 -5
- package/dist/cli/list-models.js.map +1 -1
- package/dist/core/agent-session-accessors.js +1 -1
- package/dist/core/agent-session-accessors.js.map +1 -1
- package/dist/core/agent-session-auto-compaction.d.ts.map +1 -1
- package/dist/core/agent-session-auto-compaction.js +2 -5
- package/dist/core/agent-session-auto-compaction.js.map +1 -1
- package/dist/core/agent-session-compaction.js +1 -1
- package/dist/core/agent-session-compaction.js.map +1 -1
- package/dist/core/agent-session-extension-bindings.js +5 -5
- package/dist/core/agent-session-extension-bindings.js.map +1 -1
- package/dist/core/agent-session-methods.d.ts +4 -4
- package/dist/core/agent-session-methods.d.ts.map +1 -1
- package/dist/core/agent-session-methods.js.map +1 -1
- package/dist/core/agent-session-models.d.ts +1 -1
- package/dist/core/agent-session-models.d.ts.map +1 -1
- package/dist/core/agent-session-models.js +18 -14
- package/dist/core/agent-session-models.js.map +1 -1
- package/dist/core/agent-session-post-tool-compaction.d.ts.map +1 -1
- package/dist/core/agent-session-post-tool-compaction.js +2 -4
- package/dist/core/agent-session-post-tool-compaction.js.map +1 -1
- package/dist/core/agent-session-prompt.d.ts.map +1 -1
- package/dist/core/agent-session-prompt.js +3 -7
- package/dist/core/agent-session-prompt.js.map +1 -1
- package/dist/core/agent-session-retry.js +1 -1
- package/dist/core/agent-session-retry.js.map +1 -1
- package/dist/core/agent-session-runtime-auth.d.ts +3 -3
- package/dist/core/agent-session-runtime-auth.d.ts.map +1 -1
- package/dist/core/agent-session-runtime-auth.js +4 -14
- package/dist/core/agent-session-runtime-auth.js.map +1 -1
- package/dist/core/agent-session-runtime.d.ts +1 -1
- package/dist/core/agent-session-runtime.d.ts.map +1 -1
- package/dist/core/agent-session-runtime.js +3 -3
- package/dist/core/agent-session-runtime.js.map +1 -1
- package/dist/core/agent-session-services.d.ts +3 -8
- package/dist/core/agent-session-services.d.ts.map +1 -1
- package/dist/core/agent-session-services.js +16 -23
- package/dist/core/agent-session-services.js.map +1 -1
- package/dist/core/agent-session-tool-registry.d.ts.map +1 -1
- package/dist/core/agent-session-tool-registry.js +2 -1
- package/dist/core/agent-session-tool-registry.js.map +1 -1
- package/dist/core/agent-session-types.d.ts +2 -2
- package/dist/core/agent-session-types.d.ts.map +1 -1
- package/dist/core/agent-session-types.js.map +1 -1
- package/dist/core/agent-session.d.ts +2 -2
- package/dist/core/agent-session.d.ts.map +1 -1
- package/dist/core/agent-session.js +1 -1
- package/dist/core/agent-session.js.map +1 -1
- package/dist/core/auth-storage-backends.d.ts +1 -0
- package/dist/core/auth-storage-backends.d.ts.map +1 -1
- package/dist/core/auth-storage-backends.js +18 -4
- package/dist/core/auth-storage-backends.js.map +1 -1
- package/dist/core/auth-storage.d.ts +12 -139
- package/dist/core/auth-storage.d.ts.map +1 -1
- package/dist/core/auth-storage.js +68 -374
- package/dist/core/auth-storage.js.map +1 -1
- package/dist/core/copilot-env-routing.d.ts +36 -0
- package/dist/core/copilot-env-routing.d.ts.map +1 -0
- package/dist/core/copilot-env-routing.js +94 -0
- package/dist/core/copilot-env-routing.js.map +1 -0
- package/dist/core/extensions/loader-virtual-modules.d.ts.map +1 -1
- package/dist/core/extensions/loader-virtual-modules.js +10 -2
- package/dist/core/extensions/loader-virtual-modules.js.map +1 -1
- package/dist/core/extensions/provider-types.d.ts +0 -40
- package/dist/core/extensions/provider-types.d.ts.map +1 -1
- package/dist/core/extensions/provider-types.js.map +1 -1
- package/dist/core/fallback-models.d.ts +4 -4
- package/dist/core/fallback-models.d.ts.map +1 -1
- package/dist/core/fallback-models.js +3 -3
- package/dist/core/fallback-models.js.map +1 -1
- package/dist/core/model-config.d.ts +542 -0
- package/dist/core/model-config.d.ts.map +1 -0
- package/dist/core/{model-registry-schemas.js → model-config.js} +85 -34
- package/dist/core/model-config.js.map +1 -0
- package/dist/core/model-registry.d.ts +25 -85
- package/dist/core/model-registry.d.ts.map +1 -1
- package/dist/core/model-registry.js +65 -388
- package/dist/core/model-registry.js.map +1 -1
- package/dist/core/model-resolver-cli.d.ts +2 -2
- package/dist/core/model-resolver-cli.d.ts.map +1 -1
- package/dist/core/model-resolver-cli.js +2 -2
- package/dist/core/model-resolver-cli.js.map +1 -1
- package/dist/core/model-resolver-initial.d.ts +4 -4
- package/dist/core/model-resolver-initial.d.ts.map +1 -1
- package/dist/core/model-resolver-initial.js +17 -17
- package/dist/core/model-resolver-initial.js.map +1 -1
- package/dist/core/model-resolver-scope.d.ts +3 -3
- package/dist/core/model-resolver-scope.d.ts.map +1 -1
- package/dist/core/model-resolver-scope.js +4 -4
- package/dist/core/model-resolver-scope.js.map +1 -1
- package/dist/core/model-runtime-auth.d.ts +7 -0
- package/dist/core/model-runtime-auth.d.ts.map +1 -0
- package/dist/core/model-runtime-auth.js +14 -0
- package/dist/core/model-runtime-auth.js.map +1 -0
- package/dist/core/model-runtime-providers.d.ts +6 -0
- package/dist/core/model-runtime-providers.d.ts.map +1 -0
- package/dist/core/model-runtime-providers.js +22 -0
- package/dist/core/model-runtime-providers.js.map +1 -0
- package/dist/core/model-runtime-restoration.d.ts +6 -0
- package/dist/core/model-runtime-restoration.d.ts.map +1 -0
- package/dist/core/model-runtime-restoration.js +21 -0
- package/dist/core/model-runtime-restoration.js.map +1 -0
- package/dist/core/model-runtime-snapshot.d.ts +16 -0
- package/dist/core/model-runtime-snapshot.d.ts.map +1 -0
- package/dist/core/model-runtime-snapshot.js +52 -0
- package/dist/core/model-runtime-snapshot.js.map +1 -0
- package/dist/core/model-runtime-streaming.d.ts +17 -0
- package/dist/core/model-runtime-streaming.d.ts.map +1 -0
- package/dist/core/model-runtime-streaming.js +66 -0
- package/dist/core/model-runtime-streaming.js.map +1 -0
- package/dist/core/model-runtime-types.d.ts +21 -0
- package/dist/core/model-runtime-types.d.ts.map +1 -0
- package/dist/core/model-runtime-types.js +2 -0
- package/dist/core/model-runtime-types.js.map +1 -0
- package/dist/core/model-runtime.d.ts +50 -29
- package/dist/core/model-runtime.d.ts.map +1 -1
- package/dist/core/model-runtime.js +330 -90
- package/dist/core/model-runtime.js.map +1 -1
- package/dist/core/models-store.d.ts +2 -10
- package/dist/core/models-store.d.ts.map +1 -1
- package/dist/core/models-store.js +2 -29
- package/dist/core/models-store.js.map +1 -1
- package/dist/core/oauth-login.d.ts +24 -0
- package/dist/core/oauth-login.d.ts.map +1 -0
- package/dist/core/oauth-login.js +87 -0
- package/dist/core/oauth-login.js.map +1 -0
- package/dist/core/oauth-provider-metadata.d.ts +5 -0
- package/dist/core/oauth-provider-metadata.d.ts.map +1 -0
- package/dist/core/oauth-provider-metadata.js +19 -0
- package/dist/core/oauth-provider-metadata.js.map +1 -0
- package/dist/core/provider-composer-internal.d.ts +53 -0
- package/dist/core/provider-composer-internal.d.ts.map +1 -0
- package/dist/core/provider-composer-internal.js +273 -0
- package/dist/core/provider-composer-internal.js.map +1 -0
- package/dist/core/provider-composer.d.ts +16 -0
- package/dist/core/provider-composer.d.ts.map +1 -0
- package/dist/core/provider-composer.js +108 -0
- package/dist/core/provider-composer.js.map +1 -0
- package/dist/core/remote-catalog-provider.d.ts.map +1 -1
- package/dist/core/remote-catalog-provider.js +74 -122
- package/dist/core/remote-catalog-provider.js.map +1 -1
- package/dist/core/resolve-config-value.d.ts +8 -7
- package/dist/core/resolve-config-value.d.ts.map +1 -1
- package/dist/core/resolve-config-value.js +26 -23
- package/dist/core/resolve-config-value.js.map +1 -1
- package/dist/core/runtime-credentials.d.ts +16 -0
- package/dist/core/runtime-credentials.d.ts.map +1 -0
- package/dist/core/runtime-credentials.js +42 -0
- package/dist/core/runtime-credentials.js.map +1 -0
- package/dist/core/sdk-types.d.ts +1 -7
- package/dist/core/sdk-types.d.ts.map +1 -1
- package/dist/core/sdk-types.js.map +1 -1
- package/dist/core/sdk.d.ts.map +1 -1
- package/dist/core/sdk.js +27 -28
- package/dist/core/sdk.js.map +1 -1
- package/dist/extensions/llama/index.js +1 -1
- package/dist/extensions/llama/index.js.map +1 -1
- package/dist/extensions/llama/provider.d.ts +2 -4
- package/dist/extensions/llama/provider.d.ts.map +1 -1
- package/dist/extensions/llama/provider.js +25 -17
- package/dist/extensions/llama/provider.js.map +1 -1
- package/dist/index.d.ts +2 -3
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +3 -4
- package/dist/index.js.map +1 -1
- package/dist/main-runtime-api-key.d.ts +6 -0
- package/dist/main-runtime-api-key.d.ts.map +1 -0
- package/dist/main-runtime-api-key.js +6 -0
- package/dist/main-runtime-api-key.js.map +1 -0
- package/dist/main-session-options.d.ts +2 -2
- package/dist/main-session-options.d.ts.map +1 -1
- package/dist/main-session-options.js +3 -3
- package/dist/main-session-options.js.map +1 -1
- package/dist/main.d.ts.map +1 -1
- package/dist/main.js +9 -11
- package/dist/main.js.map +1 -1
- package/dist/modes/interactive/components/footer.js +1 -1
- package/dist/modes/interactive/components/footer.js.map +1 -1
- package/dist/modes/interactive/components/login-dialog.d.ts.map +1 -1
- package/dist/modes/interactive/components/login-dialog.js +1 -3
- package/dist/modes/interactive/components/login-dialog.js.map +1 -1
- package/dist/modes/interactive/components/model-selector.d.ts +4 -3
- package/dist/modes/interactive/components/model-selector.d.ts.map +1 -1
- package/dist/modes/interactive/components/model-selector.js +56 -73
- package/dist/modes/interactive/components/model-selector.js.map +1 -1
- package/dist/modes/interactive/components/oauth-selector.d.ts +4 -3
- package/dist/modes/interactive/components/oauth-selector.d.ts.map +1 -1
- package/dist/modes/interactive/components/oauth-selector.js +15 -24
- package/dist/modes/interactive/components/oauth-selector.js.map +1 -1
- package/dist/modes/interactive/interactive-agent-events.js +1 -1
- package/dist/modes/interactive/interactive-agent-events.js.map +1 -1
- package/dist/modes/interactive/interactive-auth-login.js +20 -31
- package/dist/modes/interactive/interactive-auth-login.js.map +1 -1
- package/dist/modes/interactive/interactive-auth-routing.d.ts +1 -1
- package/dist/modes/interactive/interactive-auth-routing.d.ts.map +1 -1
- package/dist/modes/interactive/interactive-auth-routing.js +21 -47
- package/dist/modes/interactive/interactive-auth-routing.js.map +1 -1
- package/dist/modes/interactive/interactive-autocomplete.js +3 -3
- package/dist/modes/interactive/interactive-autocomplete.js.map +1 -1
- package/dist/modes/interactive/interactive-deferred-startup.js +8 -8
- package/dist/modes/interactive/interactive-deferred-startup.js.map +1 -1
- package/dist/modes/interactive/interactive-extension-runtime.js +2 -1
- package/dist/modes/interactive/interactive-extension-runtime.js.map +1 -1
- package/dist/modes/interactive/interactive-model-catalog-startup.js +2 -2
- package/dist/modes/interactive/interactive-model-catalog-startup.js.map +1 -1
- package/dist/modes/interactive/interactive-model-routing.js +11 -11
- package/dist/modes/interactive/interactive-model-routing.js.map +1 -1
- package/dist/modes/interactive/interactive-render-chat.js +1 -1
- package/dist/modes/interactive/interactive-render-chat.js.map +1 -1
- package/dist/modes/interactive/interactive-slash-commands.js +2 -2
- package/dist/modes/interactive/interactive-slash-commands.js.map +1 -1
- package/dist/modes/interactive/interactive-startup.js +1 -1
- package/dist/modes/interactive/interactive-startup.js.map +1 -1
- package/dist/modes/interactive-engine/isolated-auth.d.ts +2 -2
- package/dist/modes/interactive-engine/isolated-auth.d.ts.map +1 -1
- package/dist/modes/interactive-engine/isolated-auth.js +4 -3
- package/dist/modes/interactive-engine/isolated-auth.js.map +1 -1
- package/dist/modes/interactive-engine/isolated-runtime.d.ts.map +1 -1
- package/dist/modes/interactive-engine/isolated-runtime.js +4 -3
- package/dist/modes/interactive-engine/isolated-runtime.js.map +1 -1
- package/dist/modes/interactive-engine/remote-model-catalog.d.ts +0 -1
- package/dist/modes/interactive-engine/remote-model-catalog.d.ts.map +1 -1
- package/dist/modes/interactive-engine/remote-model-catalog.js +11 -50
- package/dist/modes/interactive-engine/remote-model-catalog.js.map +1 -1
- package/dist/modes/rpc/rpc-client-api.d.ts +2 -2
- package/dist/modes/rpc/rpc-client-api.d.ts.map +1 -1
- package/dist/modes/rpc/rpc-client-api.js.map +1 -1
- package/dist/modes/rpc/rpc-command-handler.d.ts.map +1 -1
- package/dist/modes/rpc/rpc-command-handler.js +26 -19
- package/dist/modes/rpc/rpc-command-handler.js.map +1 -1
- package/dist/modes/rpc/rpc-oauth-client.d.ts +1 -1
- package/dist/modes/rpc/rpc-oauth-client.d.ts.map +1 -1
- package/dist/modes/rpc/rpc-oauth-client.js +1 -1
- package/dist/modes/rpc/rpc-oauth-client.js.map +1 -1
- package/dist/modes/rpc/rpc-oauth-interaction.d.ts +1 -1
- package/dist/modes/rpc/rpc-oauth-interaction.d.ts.map +1 -1
- package/dist/modes/rpc/rpc-oauth-interaction.js +1 -1
- package/dist/modes/rpc/rpc-oauth-interaction.js.map +1 -1
- package/dist/modes/rpc/rpc-provider-auth.d.ts +4 -5
- package/dist/modes/rpc/rpc-provider-auth.d.ts.map +1 -1
- package/dist/modes/rpc/rpc-provider-auth.js +19 -66
- package/dist/modes/rpc/rpc-provider-auth.js.map +1 -1
- package/dist/modes/rpc/rpc-session-binding.js +1 -1
- package/dist/modes/rpc/rpc-session-binding.js.map +1 -1
- package/dist/modes/rpc/rpc-types.d.ts +7 -4
- package/dist/modes/rpc/rpc-types.d.ts.map +1 -1
- package/dist/modes/rpc/rpc-types.js.map +1 -1
- package/dist/package-manager-cli.d.ts.map +1 -1
- package/dist/package-manager-cli.js +5 -7
- package/dist/package-manager-cli.js.map +1 -1
- package/docs/changelog.mdx +1 -1
- package/docs/custom-provider.md +1 -1
- package/docs/models.md +4 -6
- package/docs/providers.md +13 -0
- package/docs/quickstart.md +2 -1
- package/docs/sdk.md +32 -49
- package/docs/workflows.md +3 -3
- package/examples/sdk/02-custom-model.ts +8 -10
- package/examples/sdk/09-api-keys-and-oauth.ts +17 -23
- package/examples/sdk/12-full-control.ts +9 -11
- package/examples/sdk/README.md +17 -21
- package/npm-shrinkwrap.json +23 -23
- package/package.json +2 -2
- package/dist/core/model-registry-auth.d.ts +0 -10
- package/dist/core/model-registry-auth.d.ts.map +0 -1
- package/dist/core/model-registry-auth.js +0 -98
- package/dist/core/model-registry-auth.js.map +0 -1
- package/dist/core/model-registry-builtins.d.ts +0 -8
- package/dist/core/model-registry-builtins.d.ts.map +0 -1
- package/dist/core/model-registry-builtins.js +0 -104
- package/dist/core/model-registry-builtins.js.map +0 -1
- package/dist/core/model-registry-custom-loader.d.ts +0 -3
- package/dist/core/model-registry-custom-loader.d.ts.map +0 -1
- package/dist/core/model-registry-custom-loader.js +0 -255
- package/dist/core/model-registry-custom-loader.js.map +0 -1
- package/dist/core/model-registry-dynamic.d.ts +0 -7
- package/dist/core/model-registry-dynamic.d.ts.map +0 -1
- package/dist/core/model-registry-dynamic.js +0 -189
- package/dist/core/model-registry-dynamic.js.map +0 -1
- package/dist/core/model-registry-extension-refresh.d.ts +0 -23
- package/dist/core/model-registry-extension-refresh.d.ts.map +0 -1
- package/dist/core/model-registry-extension-refresh.js +0 -32
- package/dist/core/model-registry-extension-refresh.js.map +0 -1
- package/dist/core/model-registry-loader.d.ts +0 -5
- package/dist/core/model-registry-loader.d.ts.map +0 -1
- package/dist/core/model-registry-loader.js +0 -29
- package/dist/core/model-registry-loader.js.map +0 -1
- package/dist/core/model-registry-schemas.d.ts +0 -1399
- package/dist/core/model-registry-schemas.d.ts.map +0 -1
- package/dist/core/model-registry-schemas.js.map +0 -1
- package/dist/core/model-registry-types.d.ts +0 -91
- package/dist/core/model-registry-types.d.ts.map +0 -1
- package/dist/core/model-registry-types.js +0 -2
- package/dist/core/model-registry-types.js.map +0 -1
- package/dist/core/model-registry-validation.d.ts +0 -7
- package/dist/core/model-registry-validation.d.ts.map +0 -1
- package/dist/core/model-registry-validation.js +0 -12
- package/dist/core/model-registry-validation.js.map +0 -1
- package/dist/core/oauth-compat.d.ts +0 -17
- package/dist/core/oauth-compat.d.ts.map +0 -1
- package/dist/core/oauth-compat.js +0 -2
- package/dist/core/oauth-compat.js.map +0 -1
- package/dist/core/oauth-provider-bridge.d.ts +0 -63
- package/dist/core/oauth-provider-bridge.d.ts.map +0 -1
- package/dist/core/oauth-provider-bridge.js +0 -232
- package/dist/core/oauth-provider-bridge.js.map +0 -1
|
@@ -1,255 +0,0 @@
|
|
|
1
|
-
import { getModels, getProviders } from "@earendil-works/pi-ai/compat";
|
|
2
|
-
import { radiusProvider } from "@earendil-works/pi-ai/providers/all";
|
|
3
|
-
import { existsSync, readFileSync } from "fs";
|
|
4
|
-
import { validateContextWindowValue } from "./model-registry-validation.js";
|
|
5
|
-
import { mergeCompat, mergeCustomModels } from "./model-registry-builtins.js";
|
|
6
|
-
import { formatValidationPath, stripJsonComments, validateModelsConfig, } from "./model-registry-schemas.js";
|
|
7
|
-
function modelRequestKey(provider, modelId) {
|
|
8
|
-
return `${provider}:${modelId}`;
|
|
9
|
-
}
|
|
10
|
-
function emptyCustomModelsResult(error) {
|
|
11
|
-
return {
|
|
12
|
-
models: [],
|
|
13
|
-
overrides: new Map(),
|
|
14
|
-
modelOverrides: new Map(),
|
|
15
|
-
providerRequestConfigs: new Map(),
|
|
16
|
-
modelRequestHeaders: new Map(),
|
|
17
|
-
modelRequestHeaderSources: new Map(),
|
|
18
|
-
configuredProviders: new Map(),
|
|
19
|
-
error,
|
|
20
|
-
};
|
|
21
|
-
}
|
|
22
|
-
function mergeModelOverrides(base, incoming) {
|
|
23
|
-
const merged = new Map([...base].map(([providerName, overrides]) => [providerName, new Map(overrides)]));
|
|
24
|
-
for (const [providerName, incomingOverrides] of incoming) {
|
|
25
|
-
const providerOverrides = new Map(merged.get(providerName) ?? []);
|
|
26
|
-
for (const [modelId, modelOverride] of incomingOverrides) {
|
|
27
|
-
providerOverrides.set(modelId, modelOverride);
|
|
28
|
-
}
|
|
29
|
-
merged.set(providerName, providerOverrides);
|
|
30
|
-
}
|
|
31
|
-
return merged;
|
|
32
|
-
}
|
|
33
|
-
function mergeModelRequestHeaders(base, incoming) {
|
|
34
|
-
const headers = new Map(base.modelRequestHeaders);
|
|
35
|
-
const sources = new Map(base.modelRequestHeaderSources);
|
|
36
|
-
const invalidate = (key, source) => {
|
|
37
|
-
if (sources.get(key) !== source)
|
|
38
|
-
return;
|
|
39
|
-
headers.delete(key);
|
|
40
|
-
sources.delete(key);
|
|
41
|
-
};
|
|
42
|
-
for (const model of incoming.models) {
|
|
43
|
-
invalidate(modelRequestKey(model.provider, model.id), "model");
|
|
44
|
-
}
|
|
45
|
-
for (const [providerName, overrides] of incoming.modelOverrides) {
|
|
46
|
-
for (const modelId of overrides.keys()) {
|
|
47
|
-
invalidate(modelRequestKey(providerName, modelId), "modelOverride");
|
|
48
|
-
}
|
|
49
|
-
}
|
|
50
|
-
for (const [key, value] of incoming.modelRequestHeaders)
|
|
51
|
-
headers.set(key, value);
|
|
52
|
-
for (const [key, source] of incoming.modelRequestHeaderSources)
|
|
53
|
-
sources.set(key, source);
|
|
54
|
-
return { headers, sources };
|
|
55
|
-
}
|
|
56
|
-
function mergeCustomModelResults(base, incoming) {
|
|
57
|
-
const mergedHeaders = mergeModelRequestHeaders(base, incoming);
|
|
58
|
-
return {
|
|
59
|
-
models: mergeCustomModels(base.models, incoming.models),
|
|
60
|
-
overrides: new Map([...base.overrides, ...incoming.overrides]),
|
|
61
|
-
modelOverrides: mergeModelOverrides(base.modelOverrides, incoming.modelOverrides),
|
|
62
|
-
providerRequestConfigs: new Map([...base.providerRequestConfigs, ...incoming.providerRequestConfigs]),
|
|
63
|
-
modelRequestHeaders: mergedHeaders.headers,
|
|
64
|
-
modelRequestHeaderSources: mergedHeaders.sources,
|
|
65
|
-
configuredProviders: new Map([...base.configuredProviders, ...incoming.configuredProviders]),
|
|
66
|
-
error: undefined,
|
|
67
|
-
};
|
|
68
|
-
}
|
|
69
|
-
function collectProviderRequestConfig(providerName, config, requestConfigs) {
|
|
70
|
-
if (!config.apiKey && !config.headers && !config.authHeader)
|
|
71
|
-
return;
|
|
72
|
-
requestConfigs.set(providerName, {
|
|
73
|
-
apiKey: config.apiKey,
|
|
74
|
-
headers: config.headers,
|
|
75
|
-
authHeader: config.authHeader,
|
|
76
|
-
});
|
|
77
|
-
}
|
|
78
|
-
function collectModelHeaders(providerName, modelId, headers, modelHeaders, modelHeaderSources, source) {
|
|
79
|
-
if (!headers || Object.keys(headers).length === 0)
|
|
80
|
-
return;
|
|
81
|
-
const key = modelRequestKey(providerName, modelId);
|
|
82
|
-
modelHeaders.set(key, headers);
|
|
83
|
-
modelHeaderSources.set(key, source);
|
|
84
|
-
}
|
|
85
|
-
function validateConfig(config) {
|
|
86
|
-
const builtInProviders = new Set(getProviders());
|
|
87
|
-
for (const [providerName, providerConfig] of Object.entries(config.providers)) {
|
|
88
|
-
const isBuiltIn = builtInProviders.has(providerName);
|
|
89
|
-
const hasProviderApi = !!providerConfig.api;
|
|
90
|
-
const models = providerConfig.models ?? [];
|
|
91
|
-
const hasModelOverrides = providerConfig.modelOverrides && Object.keys(providerConfig.modelOverrides).length > 0;
|
|
92
|
-
if (models.length === 0) {
|
|
93
|
-
if (!providerConfig.baseUrl && !providerConfig.headers && !providerConfig.compat && !hasModelOverrides) {
|
|
94
|
-
throw new Error(`Provider ${providerName}: must specify "baseUrl", "headers", "compat", "modelOverrides", or "models".`);
|
|
95
|
-
}
|
|
96
|
-
}
|
|
97
|
-
else if (!isBuiltIn) {
|
|
98
|
-
if (!providerConfig.baseUrl) {
|
|
99
|
-
throw new Error(`Provider ${providerName}: "baseUrl" is required when defining custom models.`);
|
|
100
|
-
}
|
|
101
|
-
if (!providerConfig.apiKey) {
|
|
102
|
-
throw new Error(`Provider ${providerName}: "apiKey" is required when defining custom models.`);
|
|
103
|
-
}
|
|
104
|
-
}
|
|
105
|
-
for (const modelDef of models) {
|
|
106
|
-
const hasModelApi = !!modelDef.api;
|
|
107
|
-
if (!hasProviderApi && !hasModelApi && !isBuiltIn) {
|
|
108
|
-
throw new Error(`Provider ${providerName}, model ${modelDef.id}: no "api" specified. Set at provider or model level.`);
|
|
109
|
-
}
|
|
110
|
-
if (!modelDef.id)
|
|
111
|
-
throw new Error(`Provider ${providerName}: model missing "id"`);
|
|
112
|
-
if (modelDef.contextWindow !== undefined && validateContextWindowValue(modelDef.contextWindow)) {
|
|
113
|
-
throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);
|
|
114
|
-
}
|
|
115
|
-
if (modelDef.maxTokens !== undefined && modelDef.maxTokens <= 0) {
|
|
116
|
-
throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);
|
|
117
|
-
}
|
|
118
|
-
}
|
|
119
|
-
for (const [modelId, modelOverride] of Object.entries(providerConfig.modelOverrides ?? {})) {
|
|
120
|
-
if (modelOverride.contextWindow !== undefined && validateContextWindowValue(modelOverride.contextWindow)) {
|
|
121
|
-
throw new Error(`Provider ${providerName}, model ${modelId}: invalid contextWindow`);
|
|
122
|
-
}
|
|
123
|
-
if (modelOverride.maxTokens !== undefined && modelOverride.maxTokens <= 0) {
|
|
124
|
-
throw new Error(`Provider ${providerName}, model ${modelId}: invalid maxTokens`);
|
|
125
|
-
}
|
|
126
|
-
}
|
|
127
|
-
}
|
|
128
|
-
}
|
|
129
|
-
function parseModels(config, modelHeaders, modelHeaderSources) {
|
|
130
|
-
const models = [];
|
|
131
|
-
const builtInProviders = new Set(getProviders());
|
|
132
|
-
const builtInDefaultsCache = new Map();
|
|
133
|
-
const getBuiltInDefaults = (providerName) => {
|
|
134
|
-
if (!builtInProviders.has(providerName))
|
|
135
|
-
return undefined;
|
|
136
|
-
if (builtInDefaultsCache.has(providerName))
|
|
137
|
-
return builtInDefaultsCache.get(providerName);
|
|
138
|
-
const builtIn = getModels(providerName);
|
|
139
|
-
if (builtIn.length === 0)
|
|
140
|
-
return undefined;
|
|
141
|
-
const defaults = { api: builtIn[0].api, baseUrl: builtIn[0].baseUrl };
|
|
142
|
-
builtInDefaultsCache.set(providerName, defaults);
|
|
143
|
-
return defaults;
|
|
144
|
-
};
|
|
145
|
-
for (const [providerName, providerConfig] of Object.entries(config.providers)) {
|
|
146
|
-
const modelDefs = providerConfig.models ?? [];
|
|
147
|
-
if (modelDefs.length === 0)
|
|
148
|
-
continue;
|
|
149
|
-
const builtInDefaults = getBuiltInDefaults(providerName);
|
|
150
|
-
for (const modelDef of modelDefs) {
|
|
151
|
-
const api = modelDef.api ?? providerConfig.api ?? builtInDefaults?.api;
|
|
152
|
-
if (!api)
|
|
153
|
-
continue;
|
|
154
|
-
const baseUrl = modelDef.baseUrl ?? providerConfig.baseUrl ?? builtInDefaults?.baseUrl;
|
|
155
|
-
if (!baseUrl)
|
|
156
|
-
continue;
|
|
157
|
-
const compat = mergeCompat(providerConfig.compat, modelDef.compat);
|
|
158
|
-
collectModelHeaders(providerName, modelDef.id, modelDef.headers, modelHeaders, modelHeaderSources, "model");
|
|
159
|
-
const defaultCost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
|
|
160
|
-
const contextWindow = modelDef.contextWindow ?? 128000;
|
|
161
|
-
models.push({
|
|
162
|
-
id: modelDef.id,
|
|
163
|
-
name: modelDef.name ?? modelDef.id,
|
|
164
|
-
api: api,
|
|
165
|
-
provider: providerName,
|
|
166
|
-
baseUrl,
|
|
167
|
-
reasoning: modelDef.reasoning ?? false,
|
|
168
|
-
thinkingLevelMap: modelDef.thinkingLevelMap,
|
|
169
|
-
input: (modelDef.input ?? ["text"]),
|
|
170
|
-
cost: modelDef.cost ?? defaultCost,
|
|
171
|
-
contextWindow,
|
|
172
|
-
maxTokens: modelDef.maxTokens ?? 16384,
|
|
173
|
-
headers: undefined,
|
|
174
|
-
compat,
|
|
175
|
-
});
|
|
176
|
-
}
|
|
177
|
-
}
|
|
178
|
-
return models;
|
|
179
|
-
}
|
|
180
|
-
function loadCustomModels(modelsJsonPath) {
|
|
181
|
-
if (!existsSync(modelsJsonPath)) {
|
|
182
|
-
return emptyCustomModelsResult();
|
|
183
|
-
}
|
|
184
|
-
try {
|
|
185
|
-
const content = readFileSync(modelsJsonPath, "utf-8");
|
|
186
|
-
const parsed = JSON.parse(stripJsonComments(content));
|
|
187
|
-
if (!validateModelsConfig.Check(parsed)) {
|
|
188
|
-
const errors = validateModelsConfig
|
|
189
|
-
.Errors(parsed)
|
|
190
|
-
.map((error) => ` - ${formatValidationPath(error)}: ${error.message}`)
|
|
191
|
-
.join("\n") || "Unknown schema error";
|
|
192
|
-
return emptyCustomModelsResult(`Invalid models.json schema:\n${errors}\n\nFile: ${modelsJsonPath}`);
|
|
193
|
-
}
|
|
194
|
-
const config = parsed;
|
|
195
|
-
validateConfig(config);
|
|
196
|
-
const overrides = new Map();
|
|
197
|
-
const modelOverrides = new Map();
|
|
198
|
-
const providerRequestConfigs = new Map();
|
|
199
|
-
const modelRequestHeaders = new Map();
|
|
200
|
-
const modelRequestHeaderSources = new Map();
|
|
201
|
-
const configuredProviders = new Map();
|
|
202
|
-
for (const [providerName, providerConfig] of Object.entries(config.providers)) {
|
|
203
|
-
if (providerConfig.oauth === "radius" && providerConfig.baseUrl) {
|
|
204
|
-
configuredProviders.set(providerName, radiusProvider({
|
|
205
|
-
id: providerName,
|
|
206
|
-
name: providerConfig.name ?? providerName,
|
|
207
|
-
gateway: providerConfig.baseUrl.replace(/\/v1\/?$/u, ""),
|
|
208
|
-
}));
|
|
209
|
-
}
|
|
210
|
-
if ((providerConfig.baseUrl && providerConfig.oauth !== "radius") || providerConfig.compat) {
|
|
211
|
-
overrides.set(providerName, {
|
|
212
|
-
baseUrl: providerConfig.oauth === "radius" ? undefined : providerConfig.baseUrl,
|
|
213
|
-
compat: providerConfig.compat,
|
|
214
|
-
});
|
|
215
|
-
}
|
|
216
|
-
collectProviderRequestConfig(providerName, providerConfig, providerRequestConfigs);
|
|
217
|
-
if (providerConfig.modelOverrides) {
|
|
218
|
-
modelOverrides.set(providerName, new Map(Object.entries(providerConfig.modelOverrides)));
|
|
219
|
-
for (const [modelId, modelOverride] of Object.entries(providerConfig.modelOverrides)) {
|
|
220
|
-
collectModelHeaders(providerName, modelId, modelOverride.headers, modelRequestHeaders, modelRequestHeaderSources, "modelOverride");
|
|
221
|
-
}
|
|
222
|
-
}
|
|
223
|
-
}
|
|
224
|
-
return {
|
|
225
|
-
models: parseModels(config, modelRequestHeaders, modelRequestHeaderSources),
|
|
226
|
-
overrides,
|
|
227
|
-
modelOverrides,
|
|
228
|
-
providerRequestConfigs,
|
|
229
|
-
modelRequestHeaders,
|
|
230
|
-
modelRequestHeaderSources,
|
|
231
|
-
configuredProviders,
|
|
232
|
-
error: undefined,
|
|
233
|
-
};
|
|
234
|
-
}
|
|
235
|
-
catch (error) {
|
|
236
|
-
if (error instanceof SyntaxError) {
|
|
237
|
-
return emptyCustomModelsResult(`Failed to parse models.json: ${error.message}\n\nFile: ${modelsJsonPath}`);
|
|
238
|
-
}
|
|
239
|
-
return emptyCustomModelsResult(`Failed to load models.json: ${error instanceof Error ? error.message : error}\n\nFile: ${modelsJsonPath}`);
|
|
240
|
-
}
|
|
241
|
-
}
|
|
242
|
-
export function loadCustomModelsFromPaths(modelsJsonPaths) {
|
|
243
|
-
let combined = emptyCustomModelsResult();
|
|
244
|
-
const errors = [];
|
|
245
|
-
for (let i = modelsJsonPaths.length - 1; i >= 0; i--) {
|
|
246
|
-
const result = loadCustomModels(modelsJsonPaths[i]);
|
|
247
|
-
if (result.error) {
|
|
248
|
-
errors.push(result.error);
|
|
249
|
-
continue;
|
|
250
|
-
}
|
|
251
|
-
combined = mergeCustomModelResults(combined, result);
|
|
252
|
-
}
|
|
253
|
-
return { ...combined, error: errors.length > 0 ? errors.join("\n\n") : undefined };
|
|
254
|
-
}
|
|
255
|
-
//# sourceMappingURL=model-registry-custom-loader.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"model-registry-custom-loader.js","sourceRoot":"","sources":["../../src/core/model-registry-custom-loader.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkC,SAAS,EAAE,YAAY,EAAc,MAAM,8BAA8B,CAAC;AACnH,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,0BAA0B,EAAE,MAAM,gCAAgC,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAC9E,OAAO,EACN,oBAAoB,EAGpB,iBAAiB,EACjB,oBAAoB,GACpB,MAAM,6BAA6B,CAAC;AAQrC,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,yBAAyB,EAAE,IAAI,GAAG,EAAE;QACpC,mBAAmB,EAAE,IAAI,GAAG,EAAE;QAC9B,KAAK;KACL,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAC3B,IAA6C,EAC7C,QAAiD;IAEjD,MAAM,MAAM,GAAG,IAAI,GAAG,CACrB,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,EAAE,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,CAChF,CAAC;IACF,KAAK,MAAM,CAAC,YAAY,EAAE,iBAAiB,CAAC,IAAI,QAAQ,EAAE,CAAC;QAC1D,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;QAClE,KAAK,MAAM,CAAC,OAAO,EAAE,aAAa,CAAC,IAAI,iBAAiB,EAAE,CAAC;YAC1D,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAC/C,CAAC;QACD,MAAM,CAAC,GAAG,CAAC,YAAY,EAAE,iBAAiB,CAAC,CAAC;IAC7C,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC;AAOD,SAAS,wBAAwB,CAChC,IAAwB,EACxB,QAA4B;IAE5B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACxD,MAAM,UAAU,GAAG,CAAC,GAAW,EAAE,MAAgC,EAAQ,EAAE;QAC1E,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,MAAM;YAAE,OAAO;QACxC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACpB,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACrB,CAAC,CAAC;IAEF,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QACrC,UAAU,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,CAAC,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IACD,KAAK,MAAM,CAAC,YAAY,EAAE,SAAS,CAAC,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;QACjE,KAAK,MAAM,OAAO,IAAI,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC;YACxC,UAAU,CAAC,eAAe,CAAC,YAAY,EAAE,OAAO,CAAC,EAAE,eAAe,CAAC,CAAC;QACrE,CAAC;IACF,CAAC;IACD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,QAAQ,CAAC,mBAAmB;QAAE,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACjF,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,QAAQ,CAAC,yBAAyB;QAAE,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IACzF,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AAC7B,CAAC;AAED,SAAS,uBAAuB,CAAC,IAAwB,EAAE,QAA4B;IACtF,MAAM,aAAa,GAAG,wBAAwB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC/D,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,mBAAmB,CAAC,IAAI,CAAC,cAAc,EAAE,QAAQ,CAAC,cAAc,CAAC;QACjF,sBAAsB,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,sBAAsB,EAAE,GAAG,QAAQ,CAAC,sBAAsB,CAAC,CAAC;QACrG,mBAAmB,EAAE,aAAa,CAAC,OAAO;QAC1C,yBAAyB,EAAE,aAAa,CAAC,OAAO;QAChD,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,EACjD,kBAAyD,EACzD,MAAgC;IAEhC,IAAI,CAAC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAC1D,MAAM,GAAG,GAAG,eAAe,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACnD,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC/B,kBAAkB,CAAC,GAAG,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;AACrC,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,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,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,EACjD,kBAAyD;IAEzD,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,YAA+B,CAAiB,CAAC;QAC3E,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,EAAE,kBAAkB,EAAE,OAAO,CAAC,CAAC;YAE5G,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,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;QACtE,MAAM,yBAAyB,GAAG,IAAI,GAAG,EAAoC,CAAC;QAC9E,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA6C,CAAC;QAEjF,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/E,IAAI,cAAc,CAAC,KAAK,KAAK,QAAQ,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC;gBACjE,mBAAmB,CAAC,GAAG,CAAC,YAAY,EAAE,cAAc,CAAC;oBACpD,EAAE,EAAE,YAAY;oBAChB,IAAI,EAAE,cAAc,CAAC,IAAI,IAAI,YAAY;oBACzC,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;iBACxD,CAAC,CAAC,CAAC;YACL,CAAC;YACD,IAAI,CAAC,cAAc,CAAC,OAAO,IAAI,cAAc,CAAC,KAAK,KAAK,QAAQ,CAAC,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;gBAC5F,SAAS,CAAC,GAAG,CAAC,YAAY,EAAE;oBAC3B,OAAO,EAAE,cAAc,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc,CAAC,OAAO;oBAC/E,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,CAClB,YAAY,EACZ,OAAO,EACP,aAAa,CAAC,OAAO,EACrB,mBAAmB,EACnB,yBAAyB,EACzB,eAAe,CACf,CAAC;gBACH,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO;YACN,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,mBAAmB,EAAE,yBAAyB,CAAC;YAC3E,SAAS;YACT,cAAc;YACd,sBAAsB;YACtB,mBAAmB;YACnB,yBAAyB;YACzB,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, type BuiltinProvider, getModels, getProviders, type Model } from \"@earendil-works/pi-ai/compat\";\nimport { radiusProvider } from \"@earendil-works/pi-ai/providers/all\";\nimport { existsSync, readFileSync } from \"fs\";\nimport { validateContextWindowValue } from \"./model-registry-validation.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 {\n\tCustomModelsResult,\n\tModelRequestHeaderSource,\n\tProviderOverride,\n\tProviderRequestConfig,\n} 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\tmodelRequestHeaderSources: new Map(),\n\t\tconfiguredProviders: new Map(),\n\t\terror,\n\t};\n}\n\nfunction mergeModelOverrides(\n\tbase: Map<string, Map<string, ModelOverride>>,\n\tincoming: Map<string, Map<string, ModelOverride>>,\n): Map<string, Map<string, ModelOverride>> {\n\tconst merged = new Map<string, Map<string, ModelOverride>>(\n\t\t[...base].map(([providerName, overrides]) => [providerName, new Map(overrides)]),\n\t);\n\tfor (const [providerName, incomingOverrides] of incoming) {\n\t\tconst providerOverrides = new Map(merged.get(providerName) ?? []);\n\t\tfor (const [modelId, modelOverride] of incomingOverrides) {\n\t\t\tproviderOverrides.set(modelId, modelOverride);\n\t\t}\n\t\tmerged.set(providerName, providerOverrides);\n\t}\n\treturn merged;\n}\n\ninterface MergedModelRequestHeaders {\n\theaders: Map<string, Record<string, string>>;\n\tsources: Map<string, ModelRequestHeaderSource>;\n}\n\nfunction mergeModelRequestHeaders(\n\tbase: CustomModelsResult,\n\tincoming: CustomModelsResult,\n): MergedModelRequestHeaders {\n\tconst headers = new Map(base.modelRequestHeaders);\n\tconst sources = new Map(base.modelRequestHeaderSources);\n\tconst invalidate = (key: string, source: ModelRequestHeaderSource): void => {\n\t\tif (sources.get(key) !== source) return;\n\t\theaders.delete(key);\n\t\tsources.delete(key);\n\t};\n\n\tfor (const model of incoming.models) {\n\t\tinvalidate(modelRequestKey(model.provider, model.id), \"model\");\n\t}\n\tfor (const [providerName, overrides] of incoming.modelOverrides) {\n\t\tfor (const modelId of overrides.keys()) {\n\t\t\tinvalidate(modelRequestKey(providerName, modelId), \"modelOverride\");\n\t\t}\n\t}\n\tfor (const [key, value] of incoming.modelRequestHeaders) headers.set(key, value);\n\tfor (const [key, source] of incoming.modelRequestHeaderSources) sources.set(key, source);\n\treturn { headers, sources };\n}\n\nfunction mergeCustomModelResults(base: CustomModelsResult, incoming: CustomModelsResult): CustomModelsResult {\n\tconst mergedHeaders = mergeModelRequestHeaders(base, incoming);\n\treturn {\n\t\tmodels: mergeCustomModels(base.models, incoming.models),\n\t\toverrides: new Map([...base.overrides, ...incoming.overrides]),\n\t\tmodelOverrides: mergeModelOverrides(base.modelOverrides, incoming.modelOverrides),\n\t\tproviderRequestConfigs: new Map([...base.providerRequestConfigs, ...incoming.providerRequestConfigs]),\n\t\tmodelRequestHeaders: mergedHeaders.headers,\n\t\tmodelRequestHeaderSources: mergedHeaders.sources,\n\t\tconfiguredProviders: new Map([...base.configuredProviders, ...incoming.configuredProviders]),\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\tmodelHeaderSources: Map<string, ModelRequestHeaderSource>,\n\tsource: ModelRequestHeaderSource,\n): void {\n\tif (!headers || Object.keys(headers).length === 0) return;\n\tconst key = modelRequestKey(providerName, modelId);\n\tmodelHeaders.set(key, headers);\n\tmodelHeaderSources.set(key, source);\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\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\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\tmodelHeaderSources: Map<string, ModelRequestHeaderSource>,\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 BuiltinProvider) 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, modelHeaderSources, \"model\");\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\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\t\tconst modelRequestHeaderSources = new Map<string, ModelRequestHeaderSource>();\n\t\tconst configuredProviders = new Map<string, ReturnType<typeof radiusProvider>>();\n\n\t\tfor (const [providerName, providerConfig] of Object.entries(config.providers)) {\n\t\t\tif (providerConfig.oauth === \"radius\" && providerConfig.baseUrl) {\n\t\t\t\tconfiguredProviders.set(providerName, radiusProvider({\n\t\t\t\t\tid: providerName,\n\t\t\t\t\tname: providerConfig.name ?? providerName,\n\t\t\t\t\tgateway: providerConfig.baseUrl.replace(/\\/v1\\/?$/u, \"\"),\n\t\t\t\t}));\n\t\t\t}\n\t\t\tif ((providerConfig.baseUrl && providerConfig.oauth !== \"radius\") || providerConfig.compat) {\n\t\t\t\toverrides.set(providerName, {\n\t\t\t\t\tbaseUrl: providerConfig.oauth === \"radius\" ? undefined : 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(\n\t\t\t\t\t\tproviderName,\n\t\t\t\t\t\tmodelId,\n\t\t\t\t\t\tmodelOverride.headers,\n\t\t\t\t\t\tmodelRequestHeaders,\n\t\t\t\t\t\tmodelRequestHeaderSources,\n\t\t\t\t\t\t\"modelOverride\",\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn {\n\t\t\tmodels: parseModels(config, modelRequestHeaders, modelRequestHeaderSources),\n\t\t\toverrides,\n\t\t\tmodelOverrides,\n\t\t\tproviderRequestConfigs,\n\t\t\tmodelRequestHeaders,\n\t\t\tmodelRequestHeaderSources,\n\t\t\tconfiguredProviders,\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"]}
|
|
@@ -1,7 +0,0 @@
|
|
|
1
|
-
import { type Api, type Model } from "@earendil-works/pi-ai/compat";
|
|
2
|
-
import type { DynamicProviderApplyInput, ProviderConfigInput } from "./model-registry-types.ts";
|
|
3
|
-
export declare function unregisterProviderRuntime(sourceId: string): void;
|
|
4
|
-
export declare function migrateLegacyRegisterProviderConfigValues(providerName: string, config: ProviderConfigInput): ProviderConfigInput;
|
|
5
|
-
export declare function validateProviderConfig(providerName: string, config: ProviderConfigInput): void;
|
|
6
|
-
export declare function applyProviderConfigToModels(input: DynamicProviderApplyInput): Model<Api>[];
|
|
7
|
-
//# sourceMappingURL=model-registry-dynamic.d.ts.map
|
|
@@ -1 +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,EAER,KAAK,KAAK,EAIV,MAAM,8BAA8B,CAAC;AAKtC,OAAO,KAAK,EAAE,yBAAyB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAsDhG,wBAAgB,yBAAyB,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAGhE;AAyBD,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,CA4B9F;AAED,wBAAgB,2BAA2B,CAAC,KAAK,EAAE,yBAAyB,GAAG,KAAK,CAAC,GAAG,CAAC,EAAE,CA+D1F"}
|
|
@@ -1,189 +0,0 @@
|
|
|
1
|
-
import { getApiProvider, registerApiProvider, unregisterApiProviders, } from "@earendil-works/pi-ai/compat";
|
|
2
|
-
import { warnDeprecation } from "../utils/deprecation.js";
|
|
3
|
-
import { validateContextWindowValue } from "./model-registry-validation.js";
|
|
4
|
-
import { normalizeGrammarToolCapability } from "./model-capabilities.js";
|
|
5
|
-
import { applyModelOverride } from "./model-registry-builtins.js";
|
|
6
|
-
import { registerLegacyOAuthProvider, unregisterLegacyOAuthProviders } from "./oauth-provider-bridge.js";
|
|
7
|
-
import { isLegacyEnvVarNameConfigValue } from "./resolve-config-value.js";
|
|
8
|
-
const runtimeApiRegistrations = new Map();
|
|
9
|
-
const runtimeApiFallbacks = new Map();
|
|
10
|
-
function unregisterRuntimeApiProvider(sourceId) {
|
|
11
|
-
for (const [api, registrations] of runtimeApiRegistrations) {
|
|
12
|
-
const index = registrations.findIndex((entry) => entry.sourceId === sourceId);
|
|
13
|
-
if (index < 0)
|
|
14
|
-
continue;
|
|
15
|
-
const [removed] = registrations.splice(index, 1);
|
|
16
|
-
if (removed && getApiProvider(api) === removed.activeProvider) {
|
|
17
|
-
unregisterApiProviders(sourceId);
|
|
18
|
-
const previous = registrations.at(-1);
|
|
19
|
-
if (previous) {
|
|
20
|
-
registerApiProvider(previous.provider, previous.sourceId);
|
|
21
|
-
previous.activeProvider = getApiProvider(api);
|
|
22
|
-
}
|
|
23
|
-
else {
|
|
24
|
-
const fallback = runtimeApiFallbacks.get(api);
|
|
25
|
-
if (fallback)
|
|
26
|
-
registerApiProvider(fallback, `atomic:restored-api:${api}`);
|
|
27
|
-
}
|
|
28
|
-
}
|
|
29
|
-
if (registrations.length === 0) {
|
|
30
|
-
runtimeApiRegistrations.delete(api);
|
|
31
|
-
runtimeApiFallbacks.delete(api);
|
|
32
|
-
}
|
|
33
|
-
}
|
|
34
|
-
}
|
|
35
|
-
function registerRuntimeApiProvider(provider, sourceId) {
|
|
36
|
-
unregisterRuntimeApiProvider(sourceId);
|
|
37
|
-
const registrations = runtimeApiRegistrations.get(provider.api) ?? [];
|
|
38
|
-
if (registrations.length === 0) {
|
|
39
|
-
const fallback = getApiProvider(provider.api);
|
|
40
|
-
if (fallback)
|
|
41
|
-
runtimeApiFallbacks.set(provider.api, fallback);
|
|
42
|
-
}
|
|
43
|
-
const registration = { sourceId, provider };
|
|
44
|
-
registrations.push(registration);
|
|
45
|
-
runtimeApiRegistrations.set(provider.api, registrations);
|
|
46
|
-
registerApiProvider(provider, sourceId);
|
|
47
|
-
registration.activeProvider = getApiProvider(provider.api);
|
|
48
|
-
}
|
|
49
|
-
export function unregisterProviderRuntime(sourceId) {
|
|
50
|
-
unregisterRuntimeApiProvider(sourceId);
|
|
51
|
-
unregisterLegacyOAuthProviders(sourceId);
|
|
52
|
-
}
|
|
53
|
-
function migrateLegacyRegisterProviderConfigValue(providerName, field, value) {
|
|
54
|
-
if (!isLegacyEnvVarNameConfigValue(value) || process.env[value] === undefined)
|
|
55
|
-
return value;
|
|
56
|
-
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.`);
|
|
57
|
-
return `$${value}`;
|
|
58
|
-
}
|
|
59
|
-
function migrateLegacyRegisterProviderHeaders(providerName, field, headers) {
|
|
60
|
-
if (!headers)
|
|
61
|
-
return undefined;
|
|
62
|
-
let migratedHeaders;
|
|
63
|
-
for (const [key, value] of Object.entries(headers)) {
|
|
64
|
-
const migratedValue = migrateLegacyRegisterProviderConfigValue(providerName, `${field} header "${key}"`, value);
|
|
65
|
-
if (migratedValue === value)
|
|
66
|
-
continue;
|
|
67
|
-
migratedHeaders ??= { ...headers };
|
|
68
|
-
migratedHeaders[key] = migratedValue;
|
|
69
|
-
}
|
|
70
|
-
return migratedHeaders ?? headers;
|
|
71
|
-
}
|
|
72
|
-
export function migrateLegacyRegisterProviderConfigValues(providerName, config) {
|
|
73
|
-
let migratedConfig;
|
|
74
|
-
const setMigratedConfigValue = (key, value) => {
|
|
75
|
-
migratedConfig ??= { ...config };
|
|
76
|
-
migratedConfig[key] = value;
|
|
77
|
-
};
|
|
78
|
-
if (config.apiKey) {
|
|
79
|
-
const apiKey = migrateLegacyRegisterProviderConfigValue(providerName, "apiKey", config.apiKey);
|
|
80
|
-
if (apiKey !== config.apiKey) {
|
|
81
|
-
setMigratedConfigValue("apiKey", apiKey);
|
|
82
|
-
}
|
|
83
|
-
}
|
|
84
|
-
const headers = migrateLegacyRegisterProviderHeaders(providerName, "headers", config.headers);
|
|
85
|
-
if (headers !== config.headers) {
|
|
86
|
-
setMigratedConfigValue("headers", headers);
|
|
87
|
-
}
|
|
88
|
-
if (config.models) {
|
|
89
|
-
let models;
|
|
90
|
-
for (let index = 0; index < config.models.length; index++) {
|
|
91
|
-
const model = config.models[index];
|
|
92
|
-
const modelHeaders = migrateLegacyRegisterProviderHeaders(providerName, `model "${model.id}" headers`, model.headers);
|
|
93
|
-
if (modelHeaders === model.headers)
|
|
94
|
-
continue;
|
|
95
|
-
models ??= [...config.models];
|
|
96
|
-
models[index] = { ...model, headers: modelHeaders };
|
|
97
|
-
}
|
|
98
|
-
if (models) {
|
|
99
|
-
setMigratedConfigValue("models", models);
|
|
100
|
-
}
|
|
101
|
-
}
|
|
102
|
-
return migratedConfig ?? config;
|
|
103
|
-
}
|
|
104
|
-
export function validateProviderConfig(providerName, config) {
|
|
105
|
-
if (config.streamSimple && !config.api) {
|
|
106
|
-
throw new Error(`Provider ${providerName}: "api" is required when registering streamSimple.`);
|
|
107
|
-
}
|
|
108
|
-
if (!config.models || config.models.length === 0) {
|
|
109
|
-
return;
|
|
110
|
-
}
|
|
111
|
-
if (!config.baseUrl) {
|
|
112
|
-
throw new Error(`Provider ${providerName}: "baseUrl" is required when defining models.`);
|
|
113
|
-
}
|
|
114
|
-
if (!config.apiKey && !config.oauth && !config.auth?.apiKey) {
|
|
115
|
-
throw new Error(`Provider ${providerName}: "apiKey", "oauth", or "auth.apiKey" is required when defining models.`);
|
|
116
|
-
}
|
|
117
|
-
for (const modelDef of config.models) {
|
|
118
|
-
const api = modelDef.api || config.api;
|
|
119
|
-
if (!api) {
|
|
120
|
-
throw new Error(`Provider ${providerName}, model ${modelDef.id}: no "api" specified.`);
|
|
121
|
-
}
|
|
122
|
-
if (validateContextWindowValue(modelDef.contextWindow)) {
|
|
123
|
-
throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);
|
|
124
|
-
}
|
|
125
|
-
if (modelDef.maxTokens <= 0) {
|
|
126
|
-
throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);
|
|
127
|
-
}
|
|
128
|
-
}
|
|
129
|
-
}
|
|
130
|
-
export function applyProviderConfigToModels(input) {
|
|
131
|
-
const { providerName, registrationSource, config, authStorage, modelOverrides, storeProviderRequestConfig, storeModelHeaders } = input;
|
|
132
|
-
let models = input.models;
|
|
133
|
-
if (input.registerRuntime && config.oauth)
|
|
134
|
-
registerLegacyOAuthProvider(providerName, config.oauth, registrationSource);
|
|
135
|
-
if (input.registerRuntime && config.streamSimple) {
|
|
136
|
-
const streamSimple = config.streamSimple;
|
|
137
|
-
registerRuntimeApiProvider({
|
|
138
|
-
api: config.api,
|
|
139
|
-
stream: (model, context, options) => streamSimple(model, context, options),
|
|
140
|
-
streamSimple,
|
|
141
|
-
}, registrationSource);
|
|
142
|
-
}
|
|
143
|
-
storeProviderRequestConfig(providerName, config);
|
|
144
|
-
if (config.models && config.models.length > 0) {
|
|
145
|
-
models = models.filter((m) => m.provider !== providerName);
|
|
146
|
-
for (const modelDef of config.models) {
|
|
147
|
-
const api = modelDef.api || config.api;
|
|
148
|
-
const modelOverride = modelOverrides.get(providerName)?.get(modelDef.id);
|
|
149
|
-
storeModelHeaders(providerName, modelDef.id, {
|
|
150
|
-
...modelDef.headers,
|
|
151
|
-
...modelOverride?.headers,
|
|
152
|
-
});
|
|
153
|
-
const model = {
|
|
154
|
-
id: modelDef.id,
|
|
155
|
-
name: modelDef.name,
|
|
156
|
-
api: api,
|
|
157
|
-
provider: providerName,
|
|
158
|
-
baseUrl: modelDef.baseUrl ?? config.baseUrl,
|
|
159
|
-
reasoning: modelDef.reasoning,
|
|
160
|
-
thinkingLevelMap: modelDef.thinkingLevelMap,
|
|
161
|
-
input: modelDef.input,
|
|
162
|
-
cost: modelDef.cost,
|
|
163
|
-
contextWindow: modelDef.contextWindow,
|
|
164
|
-
maxTokens: modelDef.maxTokens,
|
|
165
|
-
headers: undefined,
|
|
166
|
-
compat: normalizeGrammarToolCapability(modelDef.compat),
|
|
167
|
-
};
|
|
168
|
-
models.push(modelOverride ? applyModelOverride(model, modelOverride) : model);
|
|
169
|
-
}
|
|
170
|
-
if (config.oauth?.modifyModels) {
|
|
171
|
-
const cred = authStorage.get(providerName);
|
|
172
|
-
if (cred?.type === "oauth") {
|
|
173
|
-
models = config.oauth.modifyModels(models, cred);
|
|
174
|
-
}
|
|
175
|
-
}
|
|
176
|
-
}
|
|
177
|
-
else if (config.baseUrl || config.headers) {
|
|
178
|
-
models = models.map((m) => {
|
|
179
|
-
if (m.provider !== providerName)
|
|
180
|
-
return m;
|
|
181
|
-
return {
|
|
182
|
-
...m,
|
|
183
|
-
baseUrl: config.baseUrl ?? m.baseUrl,
|
|
184
|
-
};
|
|
185
|
-
});
|
|
186
|
-
}
|
|
187
|
-
return models;
|
|
188
|
-
}
|
|
189
|
-
//# sourceMappingURL=model-registry-dynamic.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"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,0BAA0B,EAAE,MAAM,gCAAgC,CAAC;AAC5E,OAAO,EAAE,8BAA8B,EAAE,MAAM,yBAAyB,CAAC;AACzE,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,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,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;QAC7D,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,yEAAyE,CAAC,CAAC;IACpH,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,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,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,OAAO,EAAE,SAAS;gBAClB,MAAM,EAAE,8BAA8B,CAAC,QAAQ,CAAC,MAAM,CAAC;aACzC,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 { validateContextWindowValue } from \"./model-registry-validation.ts\";\nimport { normalizeGrammarToolCapability } from \"./model-capabilities.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 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 && !config.auth?.apiKey) {\n\t\tthrow new Error(`Provider ${providerName}: \"apiKey\", \"oauth\", or \"auth.apiKey\" 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\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\tmaxTokens: modelDef.maxTokens,\n\t\t\t\theaders: undefined,\n\t\t\t\tcompat: normalizeGrammarToolCapability(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"]}
|
|
@@ -1,23 +0,0 @@
|
|
|
1
|
-
import type { Credential, RefreshModelsContext } from "@earendil-works/pi-ai";
|
|
2
|
-
import type { ProviderConfigInput } from "./model-registry-types.ts";
|
|
3
|
-
type ProviderModels = NonNullable<ProviderConfigInput["models"]>;
|
|
4
|
-
export interface ExtensionRefreshInput {
|
|
5
|
-
config: ProviderConfigInput;
|
|
6
|
-
credential: Credential | undefined;
|
|
7
|
-
store: RefreshModelsContext["store"];
|
|
8
|
-
allowNetwork: boolean;
|
|
9
|
-
force?: boolean;
|
|
10
|
-
signal: AbortSignal;
|
|
11
|
-
isCurrent(): boolean;
|
|
12
|
-
}
|
|
13
|
-
export interface ExtensionRefreshResult {
|
|
14
|
-
models?: ProviderModels;
|
|
15
|
-
error?: Error;
|
|
16
|
-
}
|
|
17
|
-
/**
|
|
18
|
-
* Refresh an extension provider, falling back to its validated persisted catalog
|
|
19
|
-
* when an online refresh fails. The online error remains authoritative.
|
|
20
|
-
*/
|
|
21
|
-
export declare function refreshExtensionProvider(input: ExtensionRefreshInput): Promise<ExtensionRefreshResult>;
|
|
22
|
-
export {};
|
|
23
|
-
//# sourceMappingURL=model-registry-extension-refresh.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"model-registry-extension-refresh.d.ts","sourceRoot":"","sources":["../../src/core/model-registry-extension-refresh.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC9E,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAErE,KAAK,cAAc,GAAG,WAAW,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC;AAEjE,MAAM,WAAW,qBAAqB;IACrC,MAAM,EAAE,mBAAmB,CAAC;IAC5B,UAAU,EAAE,UAAU,GAAG,SAAS,CAAC;IACnC,KAAK,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACrC,YAAY,EAAE,OAAO,CAAC;IACtB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,EAAE,WAAW,CAAC;IACpB,SAAS,IAAI,OAAO,CAAC;CACrB;AAED,MAAM,WAAW,sBAAsB;IACtC,MAAM,CAAC,EAAE,cAAc,CAAC;IACxB,KAAK,CAAC,EAAE,KAAK,CAAC;CACd;AAMD;;;GAGG;AACH,wBAAsB,wBAAwB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,sBAAsB,CAAC,CAoB5G"}
|
|
@@ -1,32 +0,0 @@
|
|
|
1
|
-
function toError(error) {
|
|
2
|
-
return error instanceof Error ? error : new Error(String(error));
|
|
3
|
-
}
|
|
4
|
-
/**
|
|
5
|
-
* Refresh an extension provider, falling back to its validated persisted catalog
|
|
6
|
-
* when an online refresh fails. The online error remains authoritative.
|
|
7
|
-
*/
|
|
8
|
-
export async function refreshExtensionProvider(input) {
|
|
9
|
-
const refresh = (allowNetwork, force) => input.config.refreshModels({
|
|
10
|
-
credential: input.credential,
|
|
11
|
-
store: input.store,
|
|
12
|
-
allowNetwork,
|
|
13
|
-
force,
|
|
14
|
-
signal: input.signal,
|
|
15
|
-
});
|
|
16
|
-
try {
|
|
17
|
-
return { models: await refresh(input.allowNetwork, input.force) };
|
|
18
|
-
}
|
|
19
|
-
catch (error) {
|
|
20
|
-
const result = { error: toError(error) };
|
|
21
|
-
if (!input.allowNetwork || !input.isCurrent())
|
|
22
|
-
return result;
|
|
23
|
-
try {
|
|
24
|
-
result.models = await refresh(false, false);
|
|
25
|
-
}
|
|
26
|
-
catch {
|
|
27
|
-
// Preserve the original online error; cache recovery is best-effort.
|
|
28
|
-
}
|
|
29
|
-
return result;
|
|
30
|
-
}
|
|
31
|
-
}
|
|
32
|
-
//# sourceMappingURL=model-registry-extension-refresh.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"model-registry-extension-refresh.js","sourceRoot":"","sources":["../../src/core/model-registry-extension-refresh.ts"],"names":[],"mappings":"AAoBA,SAAS,OAAO,CAAC,KAAc;IAC9B,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AAClE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,KAA4B;IAC1E,MAAM,OAAO,GAAG,CAAC,YAAqB,EAAE,KAA0B,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,aAAc,CAAC;QAClG,UAAU,EAAE,KAAK,CAAC,UAAU;QAC5B,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,YAAY;QACZ,KAAK;QACL,MAAM,EAAE,KAAK,CAAC,MAAM;KACpB,CAAC,CAAC;IACH,IAAI,CAAC;QACJ,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;IACnE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,MAAM,GAA2B,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACjE,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE;YAAE,OAAO,MAAM,CAAC;QAC7D,IAAI,CAAC;YACJ,MAAM,CAAC,MAAM,GAAG,MAAM,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;QAAC,MAAM,CAAC;YACR,qEAAqE;QACtE,CAAC;QACD,OAAO,MAAM,CAAC;IACf,CAAC;AACF,CAAC","sourcesContent":["import type { Credential, RefreshModelsContext } from \"@earendil-works/pi-ai\";\nimport type { ProviderConfigInput } from \"./model-registry-types.ts\";\n\ntype ProviderModels = NonNullable<ProviderConfigInput[\"models\"]>;\n\nexport interface ExtensionRefreshInput {\n\tconfig: ProviderConfigInput;\n\tcredential: Credential | undefined;\n\tstore: RefreshModelsContext[\"store\"];\n\tallowNetwork: boolean;\n\tforce?: boolean;\n\tsignal: AbortSignal;\n\tisCurrent(): boolean;\n}\n\nexport interface ExtensionRefreshResult {\n\tmodels?: ProviderModels;\n\terror?: Error;\n}\n\nfunction toError(error: unknown): Error {\n\treturn error instanceof Error ? error : new Error(String(error));\n}\n\n/**\n * Refresh an extension provider, falling back to its validated persisted catalog\n * when an online refresh fails. The online error remains authoritative.\n */\nexport async function refreshExtensionProvider(input: ExtensionRefreshInput): Promise<ExtensionRefreshResult> {\n\tconst refresh = (allowNetwork: boolean, force: boolean | undefined) => input.config.refreshModels!({\n\t\tcredential: input.credential,\n\t\tstore: input.store,\n\t\tallowNetwork,\n\t\tforce,\n\t\tsignal: input.signal,\n\t});\n\ttry {\n\t\treturn { models: await refresh(input.allowNetwork, input.force) };\n\t} catch (error) {\n\t\tconst result: ExtensionRefreshResult = { error: toError(error) };\n\t\tif (!input.allowNetwork || !input.isCurrent()) return result;\n\t\ttry {\n\t\t\tresult.models = await refresh(false, false);\n\t\t} catch {\n\t\t\t// Preserve the original online error; cache recovery is best-effort.\n\t\t}\n\t\treturn result;\n\t}\n}\n"]}
|
|
@@ -1,5 +0,0 @@
|
|
|
1
|
-
import type { Api, Model } from "@earendil-works/pi-ai/compat";
|
|
2
|
-
import type { AuthStorage } from "./auth-storage.ts";
|
|
3
|
-
import type { ModelRegistryLoadResult } from "./model-registry-types.ts";
|
|
4
|
-
export declare function loadModelRegistryModels(authStorage: AuthStorage, modelsJsonPaths: string[], baseModels?: readonly Model<Api>[]): ModelRegistryLoadResult;
|
|
5
|
-
//# sourceMappingURL=model-registry-loader.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"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,CAqCzB"}
|