@imdigitalashish/zpi 0.1.1

package/dist/core/model-registry.js

@@ -0,0 +1,515 @@
+/**
+ * Model registry - manages built-in and custom models, provides API key resolution.
+ */
+import { getModels, getProviders, registerApiProvider, registerOAuthProvider, } from "@mariozechner/pi-ai";
+import { Type } from "@sinclair/typebox";
+import AjvModule from "ajv";
+import { existsSync, readFileSync } from "fs";
+import { join } from "path";
+import { getAgentDir } from "../config.js";
+import { clearConfigValueCache, resolveConfigValue, resolveHeaders } from "./resolve-config-value.js";
+const Ajv = AjvModule.default || AjvModule;
+// Schema for OpenRouter routing preferences
+const OpenRouterRoutingSchema = Type.Object({
+    only: Type.Optional(Type.Array(Type.String())),
+    order: Type.Optional(Type.Array(Type.String())),
+});
+// Schema for Vercel AI Gateway routing preferences
+const VercelGatewayRoutingSchema = Type.Object({
+    only: Type.Optional(Type.Array(Type.String())),
+    order: Type.Optional(Type.Array(Type.String())),
+});
+// Schema for OpenAI compatibility settings
+const OpenAICompletionsCompatSchema = Type.Object({
+    supportsStore: Type.Optional(Type.Boolean()),
+    supportsDeveloperRole: Type.Optional(Type.Boolean()),
+    supportsReasoningEffort: Type.Optional(Type.Boolean()),
+    supportsUsageInStreaming: Type.Optional(Type.Boolean()),
+    maxTokensField: Type.Optional(Type.Union([Type.Literal("max_completion_tokens"), Type.Literal("max_tokens")])),
+    requiresToolResultName: Type.Optional(Type.Boolean()),
+    requiresAssistantAfterToolResult: Type.Optional(Type.Boolean()),
+    requiresThinkingAsText: Type.Optional(Type.Boolean()),
+    requiresMistralToolIds: Type.Optional(Type.Boolean()),
+    thinkingFormat: Type.Optional(Type.Union([Type.Literal("openai"), Type.Literal("zai"), Type.Literal("qwen")])),
+    openRouterRouting: Type.Optional(OpenRouterRoutingSchema),
+    vercelGatewayRouting: Type.Optional(VercelGatewayRoutingSchema),
+});
+const OpenAIResponsesCompatSchema = Type.Object({});
+const OpenAICompatSchema = Type.Union([OpenAICompletionsCompatSchema, OpenAIResponsesCompatSchema]);
+// Schema for custom model definition
+// Most fields are optional with sensible defaults for local models (Ollama, LM Studio, etc.)
+const ModelDefinitionSchema = Type.Object({
+    id: Type.String({ minLength: 1 }),
+    name: Type.Optional(Type.String({ minLength: 1 })),
+    api: Type.Optional(Type.String({ minLength: 1 })),
+    reasoning: Type.Optional(Type.Boolean()),
+    input: Type.Optional(Type.Array(Type.Union([Type.Literal("text"), Type.Literal("image")]))),
+    cost: Type.Optional(Type.Object({
+        input: Type.Number(),
+        output: Type.Number(),
+        cacheRead: Type.Number(),
+        cacheWrite: Type.Number(),
+    })),
+    contextWindow: Type.Optional(Type.Number()),
+    maxTokens: Type.Optional(Type.Number()),
+    headers: Type.Optional(Type.Record(Type.String(), Type.String())),
+    compat: Type.Optional(OpenAICompatSchema),
+});
+// Schema for per-model overrides (all fields optional, merged with built-in model)
+const ModelOverrideSchema = Type.Object({
+    name: Type.Optional(Type.String({ minLength: 1 })),
+    reasoning: Type.Optional(Type.Boolean()),
+    input: Type.Optional(Type.Array(Type.Union([Type.Literal("text"), Type.Literal("image")]))),
+    cost: Type.Optional(Type.Object({
+        input: Type.Optional(Type.Number()),
+        output: Type.Optional(Type.Number()),
+        cacheRead: Type.Optional(Type.Number()),
+        cacheWrite: Type.Optional(Type.Number()),
+    })),
+    contextWindow: Type.Optional(Type.Number()),
+    maxTokens: Type.Optional(Type.Number()),
+    headers: Type.Optional(Type.Record(Type.String(), Type.String())),
+    compat: Type.Optional(OpenAICompatSchema),
+});
+const ProviderConfigSchema = Type.Object({
+    baseUrl: Type.Optional(Type.String({ minLength: 1 })),
+    apiKey: Type.Optional(Type.String({ minLength: 1 })),
+    api: Type.Optional(Type.String({ minLength: 1 })),
+    headers: Type.Optional(Type.Record(Type.String(), Type.String())),
+    authHeader: Type.Optional(Type.Boolean()),
+    models: Type.Optional(Type.Array(ModelDefinitionSchema)),
+    modelOverrides: Type.Optional(Type.Record(Type.String(), ModelOverrideSchema)),
+});
+const ModelsConfigSchema = Type.Object({
+    providers: Type.Record(Type.String(), ProviderConfigSchema),
+});
+function emptyCustomModelsResult(error) {
+    return { models: [], overrides: new Map(), modelOverrides: new Map(), error };
+}
+function mergeCompat(baseCompat, overrideCompat) {
+    if (!overrideCompat)
+        return baseCompat;
+    const base = baseCompat;
+    const override = overrideCompat;
+    const merged = { ...base, ...override };
+    const baseCompletions = base;
+    const overrideCompletions = override;
+    const mergedCompletions = merged;
+    if (baseCompletions?.openRouterRouting || overrideCompletions.openRouterRouting) {
+        mergedCompletions.openRouterRouting = {
+            ...baseCompletions?.openRouterRouting,
+            ...overrideCompletions.openRouterRouting,
+        };
+    }
+    if (baseCompletions?.vercelGatewayRouting || overrideCompletions.vercelGatewayRouting) {
+        mergedCompletions.vercelGatewayRouting = {
+            ...baseCompletions?.vercelGatewayRouting,
+            ...overrideCompletions.vercelGatewayRouting,
+        };
+    }
+    return merged;
+}
+/**
+ * Deep merge a model override into a model.
+ * Handles nested objects (cost, compat) by merging rather than replacing.
+ */
+function applyModelOverride(model, override) {
+    const result = { ...model };
+    // Simple field overrides
+    if (override.name !== undefined)
+        result.name = override.name;
+    if (override.reasoning !== undefined)
+        result.reasoning = override.reasoning;
+    if (override.input !== undefined)
+        result.input = override.input;
+    if (override.contextWindow !== undefined)
+        result.contextWindow = override.contextWindow;
+    if (override.maxTokens !== undefined)
+        result.maxTokens = override.maxTokens;
+    // Merge cost (partial override)
+    if (override.cost) {
+        result.cost = {
+            input: override.cost.input ?? model.cost.input,
+            output: override.cost.output ?? model.cost.output,
+            cacheRead: override.cost.cacheRead ?? model.cost.cacheRead,
+            cacheWrite: override.cost.cacheWrite ?? model.cost.cacheWrite,
+        };
+    }
+    // Merge headers
+    if (override.headers) {
+        const resolvedHeaders = resolveHeaders(override.headers);
+        result.headers = resolvedHeaders ? { ...model.headers, ...resolvedHeaders } : model.headers;
+    }
+    // Deep merge compat
+    result.compat = mergeCompat(model.compat, override.compat);
+    return result;
+}
+/** Clear the config value command cache. Exported for testing. */
+export const clearApiKeyCache = clearConfigValueCache;
+/**
+ * Model registry - loads and manages models, resolves API keys via AuthStorage.
+ */
+export class ModelRegistry {
+    authStorage;
+    modelsJsonPath;
+    models = [];
+    customProviderApiKeys = new Map();
+    registeredProviders = new Map();
+    loadError = undefined;
+    constructor(authStorage, modelsJsonPath = join(getAgentDir(), "models.json")) {
+        this.authStorage = authStorage;
+        this.modelsJsonPath = modelsJsonPath;
+        // Set up fallback resolver for custom provider API keys
+        this.authStorage.setFallbackResolver((provider) => {
+            const keyConfig = this.customProviderApiKeys.get(provider);
+            if (keyConfig) {
+                return resolveConfigValue(keyConfig);
+            }
+            return undefined;
+        });
+        // Load models
+        this.loadModels();
+    }
+    /**
+     * Reload models from disk (built-in + custom from models.json).
+     */
+    refresh() {
+        this.customProviderApiKeys.clear();
+        this.loadError = undefined;
+        this.loadModels();
+        for (const [providerName, config] of this.registeredProviders.entries()) {
+            this.applyProviderConfig(providerName, config);
+        }
+    }
+    /**
+     * Get any error from loading models.json (undefined if no error).
+     */
+    getError() {
+        return this.loadError;
+    }
+    loadModels() {
+        // Load custom models and overrides from models.json
+        const { models: customModels, overrides, modelOverrides, error, } = this.modelsJsonPath ? this.loadCustomModels(this.modelsJsonPath) : emptyCustomModelsResult();
+        if (error) {
+            this.loadError = error;
+            // Keep built-in models even if custom models failed to load
+        }
+        const builtInModels = this.loadBuiltInModels(overrides, modelOverrides);
+        let combined = this.mergeCustomModels(builtInModels, customModels);
+        // Let OAuth providers modify their models (e.g., update baseUrl)
+        for (const oauthProvider of this.authStorage.getOAuthProviders()) {
+            const cred = this.authStorage.get(oauthProvider.id);
+            if (cred?.type === "oauth" && oauthProvider.modifyModels) {
+                combined = oauthProvider.modifyModels(combined, cred);
+            }
+        }
+        this.models = combined;
+    }
+    /** Load built-in models and apply provider/model overrides */
+    loadBuiltInModels(overrides, modelOverrides) {
+        return getProviders().flatMap((provider) => {
+            const models = getModels(provider);
+            const providerOverride = overrides.get(provider);
+            const perModelOverrides = modelOverrides.get(provider);
+            return models.map((m) => {
+                let model = m;
+                // Apply provider-level baseUrl/headers override
+                if (providerOverride) {
+                    const resolvedHeaders = resolveHeaders(providerOverride.headers);
+                    model = {
+                        ...model,
+                        baseUrl: providerOverride.baseUrl ?? model.baseUrl,
+                        headers: resolvedHeaders ? { ...model.headers, ...resolvedHeaders } : model.headers,
+                    };
+                }
+                // Apply per-model override
+                const modelOverride = perModelOverrides?.get(m.id);
+                if (modelOverride) {
+                    model = applyModelOverride(model, modelOverride);
+                }
+                return model;
+            });
+        });
+    }
+    /** Merge custom models into built-in list by provider+id (custom wins on conflicts). */
+    mergeCustomModels(builtInModels, customModels) {
+        const merged = [...builtInModels];
+        for (const customModel of customModels) {
+            const existingIndex = merged.findIndex((m) => m.provider === customModel.provider && m.id === customModel.id);
+            if (existingIndex >= 0) {
+                merged[existingIndex] = customModel;
+            }
+            else {
+                merged.push(customModel);
+            }
+        }
+        return merged;
+    }
+    loadCustomModels(modelsJsonPath) {
+        if (!existsSync(modelsJsonPath)) {
+            return emptyCustomModelsResult();
+        }
+        try {
+            const content = readFileSync(modelsJsonPath, "utf-8");
+            const config = JSON.parse(content);
+            // Validate schema
+            const ajv = new Ajv();
+            const validate = ajv.compile(ModelsConfigSchema);
+            if (!validate(config)) {
+                const errors = validate.errors?.map((e) => `  - ${e.instancePath || "root"}: ${e.message}`).join("\n") ||
+                    "Unknown schema error";
+                return emptyCustomModelsResult(`Invalid models.json schema:\n${errors}\n\nFile: ${modelsJsonPath}`);
+            }
+            // Additional validation
+            this.validateConfig(config);
+            const overrides = new Map();
+            const modelOverrides = new Map();
+            for (const [providerName, providerConfig] of Object.entries(config.providers)) {
+                // Apply provider-level baseUrl/headers/apiKey override to built-in models when configured.
+                if (providerConfig.baseUrl || providerConfig.headers || providerConfig.apiKey) {
+                    overrides.set(providerName, {
+                        baseUrl: providerConfig.baseUrl,
+                        headers: providerConfig.headers,
+                        apiKey: providerConfig.apiKey,
+                    });
+                }
+                // Store API key for fallback resolver.
+                if (providerConfig.apiKey) {
+                    this.customProviderApiKeys.set(providerName, providerConfig.apiKey);
+                }
+                if (providerConfig.modelOverrides) {
+                    modelOverrides.set(providerName, new Map(Object.entries(providerConfig.modelOverrides)));
+                }
+            }
+            return { models: this.parseModels(config), overrides, modelOverrides, error: undefined };
+        }
+        catch (error) {
+            if (error instanceof SyntaxError) {
+                return emptyCustomModelsResult(`Failed to parse models.json: ${error.message}\n\nFile: ${modelsJsonPath}`);
+            }
+            return emptyCustomModelsResult(`Failed to load models.json: ${error instanceof Error ? error.message : error}\n\nFile: ${modelsJsonPath}`);
+        }
+    }
+    validateConfig(config) {
+        for (const [providerName, providerConfig] of Object.entries(config.providers)) {
+            const hasProviderApi = !!providerConfig.api;
+            const models = providerConfig.models ?? [];
+            const hasModelOverrides = providerConfig.modelOverrides && Object.keys(providerConfig.modelOverrides).length > 0;
+            if (models.length === 0) {
+                // Override-only config: needs baseUrl OR modelOverrides (or both)
+                if (!providerConfig.baseUrl && !hasModelOverrides) {
+                    throw new Error(`Provider ${providerName}: must specify "baseUrl", "modelOverrides", or "models".`);
+                }
+            }
+            else {
+                // Custom models are merged into provider models and require endpoint + auth.
+                if (!providerConfig.baseUrl) {
+                    throw new Error(`Provider ${providerName}: "baseUrl" is required when defining custom models.`);
+                }
+                if (!providerConfig.apiKey) {
+                    throw new Error(`Provider ${providerName}: "apiKey" is required when defining custom models.`);
+                }
+            }
+            for (const modelDef of models) {
+                const hasModelApi = !!modelDef.api;
+                if (!hasProviderApi && !hasModelApi) {
+                    throw new Error(`Provider ${providerName}, model ${modelDef.id}: no "api" specified. Set at provider or model level.`);
+                }
+                if (!modelDef.id)
+                    throw new Error(`Provider ${providerName}: model missing "id"`);
+                // Validate contextWindow/maxTokens only if provided (they have defaults)
+                if (modelDef.contextWindow !== undefined && modelDef.contextWindow <= 0)
+                    throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);
+                if (modelDef.maxTokens !== undefined && modelDef.maxTokens <= 0)
+                    throw new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);
+            }
+        }
+    }
+    parseModels(config) {
+        const models = [];
+        for (const [providerName, providerConfig] of Object.entries(config.providers)) {
+            const modelDefs = providerConfig.models ?? [];
+            if (modelDefs.length === 0)
+                continue; // Override-only, no custom models
+            // Store API key config for fallback resolver
+            if (providerConfig.apiKey) {
+                this.customProviderApiKeys.set(providerName, providerConfig.apiKey);
+            }
+            for (const modelDef of modelDefs) {
+                const api = modelDef.api || providerConfig.api;
+                if (!api)
+                    continue;
+                // Merge headers: provider headers are base, model headers override
+                // Resolve env vars and shell commands in header values
+                const providerHeaders = resolveHeaders(providerConfig.headers);
+                const modelHeaders = resolveHeaders(modelDef.headers);
+                let headers = providerHeaders || modelHeaders ? { ...providerHeaders, ...modelHeaders } : undefined;
+                // If authHeader is true, add Authorization header with resolved API key
+                if (providerConfig.authHeader && providerConfig.apiKey) {
+                    const resolvedKey = resolveConfigValue(providerConfig.apiKey);
+                    if (resolvedKey) {
+                        headers = { ...headers, Authorization: `Bearer ${resolvedKey}` };
+                    }
+                }
+                // baseUrl is validated to exist for providers with models
+                // Apply defaults for optional fields
+                const defaultCost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
+                models.push({
+                    id: modelDef.id,
+                    name: modelDef.name ?? modelDef.id,
+                    api: api,
+                    provider: providerName,
+                    baseUrl: providerConfig.baseUrl,
+                    reasoning: modelDef.reasoning ?? false,
+                    input: (modelDef.input ?? ["text"]),
+                    cost: modelDef.cost ?? defaultCost,
+                    contextWindow: modelDef.contextWindow ?? 128000,
+                    maxTokens: modelDef.maxTokens ?? 16384,
+                    headers,
+                    compat: modelDef.compat,
+                });
+            }
+        }
+        return models;
+    }
+    /**
+     * Get all models (built-in + custom).
+     * If models.json had errors, returns only built-in models.
+     */
+    getAll() {
+        return this.models;
+    }
+    /**
+     * Get only models that have auth configured.
+     * This is a fast check that doesn't refresh OAuth tokens.
+     */
+    getAvailable() {
+        return this.models.filter((m) => this.authStorage.hasAuth(m.provider));
+    }
+    /**
+     * Find a model by provider and ID.
+     */
+    find(provider, modelId) {
+        return this.models.find((m) => m.provider === provider && m.id === modelId);
+    }
+    /**
+     * Get API key for a model.
+     */
+    async getApiKey(model) {
+        return this.authStorage.getApiKey(model.provider);
+    }
+    /**
+     * Get API key for a provider.
+     */
+    async getApiKeyForProvider(provider) {
+        return this.authStorage.getApiKey(provider);
+    }
+    /**
+     * Check if a model is using OAuth credentials (subscription).
+     */
+    isUsingOAuth(model) {
+        const cred = this.authStorage.get(model.provider);
+        return cred?.type === "oauth";
+    }
+    /**
+     * Register a provider dynamically (from extensions).
+     *
+     * If provider has models: replaces all existing models for this provider.
+     * If provider has only baseUrl/headers: overrides existing models' URLs.
+     * If provider has oauth: registers OAuth provider for /login support.
+     */
+    registerProvider(providerName, config) {
+        this.registeredProviders.set(providerName, config);
+        this.applyProviderConfig(providerName, config);
+    }
+    applyProviderConfig(providerName, config) {
+        // Register OAuth provider if provided
+        if (config.oauth) {
+            // Ensure the OAuth provider ID matches the provider name
+            const oauthProvider = {
+                ...config.oauth,
+                id: providerName,
+            };
+            registerOAuthProvider(oauthProvider);
+        }
+        if (config.streamSimple) {
+            if (!config.api) {
+                throw new Error(`Provider ${providerName}: "api" is required when registering streamSimple.`);
+            }
+            const streamSimple = config.streamSimple;
+            registerApiProvider({
+                api: config.api,
+                stream: (model, context, options) => streamSimple(model, context, options),
+                streamSimple,
+            });
+        }
+        // Store API key for auth resolution
+        if (config.apiKey) {
+            this.customProviderApiKeys.set(providerName, config.apiKey);
+        }
+        if (config.models && config.models.length > 0) {
+            // Full replacement: remove existing models for this provider
+            this.models = this.models.filter((m) => m.provider !== providerName);
+            // Validate required fields
+            if (!config.baseUrl) {
+                throw new Error(`Provider ${providerName}: "baseUrl" is required when defining models.`);
+            }
+            if (!config.apiKey && !config.oauth) {
+                throw new Error(`Provider ${providerName}: "apiKey" or "oauth" is required when defining models.`);
+            }
+            // Parse and add new models
+            for (const modelDef of config.models) {
+                const api = modelDef.api || config.api;
+                if (!api) {
+                    throw new Error(`Provider ${providerName}, model ${modelDef.id}: no "api" specified.`);
+                }
+                // Merge headers
+                const providerHeaders = resolveHeaders(config.headers);
+                const modelHeaders = resolveHeaders(modelDef.headers);
+                let headers = providerHeaders || modelHeaders ? { ...providerHeaders, ...modelHeaders } : undefined;
+                // If authHeader is true, add Authorization header
+                if (config.authHeader && config.apiKey) {
+                    const resolvedKey = resolveConfigValue(config.apiKey);
+                    if (resolvedKey) {
+                        headers = { ...headers, Authorization: `Bearer ${resolvedKey}` };
+                    }
+                }
+                this.models.push({
+                    id: modelDef.id,
+                    name: modelDef.name,
+                    api: api,
+                    provider: providerName,
+                    baseUrl: config.baseUrl,
+                    reasoning: modelDef.reasoning,
+                    input: modelDef.input,
+                    cost: modelDef.cost,
+                    contextWindow: modelDef.contextWindow,
+                    maxTokens: modelDef.maxTokens,
+                    headers,
+                    compat: modelDef.compat,
+                });
+            }
+            // Apply OAuth modifyModels if credentials exist (e.g., to update baseUrl)
+            if (config.oauth?.modifyModels) {
+                const cred = this.authStorage.get(providerName);
+                if (cred?.type === "oauth") {
+                    this.models = config.oauth.modifyModels(this.models, cred);
+                }
+            }
+        }
+        else if (config.baseUrl) {
+            // Override-only: update baseUrl/headers for existing models
+            const resolvedHeaders = resolveHeaders(config.headers);
+            this.models = this.models.map((m) => {
+                if (m.provider !== providerName)
+                    return m;
+                return {
+                    ...m,
+                    baseUrl: config.baseUrl ?? m.baseUrl,
+                    headers: resolvedHeaders ? { ...m.headers, ...resolvedHeaders } : m.headers,
+                };
+            });
+        }
+    }
+}
+//# sourceMappingURL=model-registry.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../../src/core/model-registry.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAIN,SAAS,EACT,YAAY,EAMZ,mBAAmB,EACnB,qBAAqB,GAErB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAe,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,SAAS,MAAM,KAAK,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAC5B,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAE3C,OAAO,EAAE,qBAAqB,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAEtG,MAAM,GAAG,GAAI,SAAiB,CAAC,OAAO,IAAI,SAAS,CAAC;AAEpD,4CAA4C;AAC5C,MAAM,uBAAuB,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3C,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9C,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;CAC/C,CAAC,CAAC;AAEH,mDAAmD;AACnD,MAAM,0BAA0B,GAAG,IAAI,CAAC,MAAM,CAAC;IAC9C,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9C,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;CAC/C,CAAC,CAAC;AAEH,2CAA2C;AAC3C,MAAM,6BAA6B,GAAG,IAAI,CAAC,MAAM,CAAC;IACjD,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IAC5C,qBAAqB,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACpD,uBAAuB,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACtD,wBAAwB,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACvD,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,uBAAuB,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC9G,sBAAsB,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACrD,gCAAgC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IAC/D,sBAAsB,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACrD,sBAAsB,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACrD,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAC9G,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IACzD,oBAAoB,EAAE,IAAI,CAAC,QAAQ,CAAC,0BAA0B,CAAC;CAC/D,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,IAAI,CAAC,MAAM,CAAC,EAE/C,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,6BAA6B,EAAE,2BAA2B,CAAC,CAAC,CAAC;AAEpG,qCAAqC;AACrC,6FAA6F;AAC7F,MAAM,qBAAqB,GAAG,IAAI,CAAC,MAAM,CAAC;IACzC,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IACjC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IAClD,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACjD,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACxC,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3F,IAAI,EAAE,IAAI,CAAC,QAAQ,CAClB,IAAI,CAAC,MAAM,CAAC;QACX,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE;QACpB,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE;QACrB,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE;QACxB,UAAU,EAAE,IAAI,CAAC,MAAM,EAAE;KACzB,CAAC,CACF;IACD,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;IAC3C,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;IACvC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC;CACzC,CAAC,CAAC;AAEH,mFAAmF;AACnF,MAAM,mBAAmB,GAAG,IAAI,CAAC,MAAM,CAAC;IACvC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IAClD,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACxC,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3F,IAAI,EAAE,IAAI,CAAC,QAAQ,CAClB,IAAI,CAAC,MAAM,CAAC;QACX,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACpC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;QACvC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;KACxC,CAAC,CACF;IACD,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;IAC3C,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;IACvC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjE,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC;CACzC,CAAC,CAAC;AAIH,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC;IACxC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACrD,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACpD,GAAG,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACjD,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IACjE,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;IACzC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC;CAC9E,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC;IACtC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC;CAC3D,CAAC,CAAC;AAqBH,SAAS,uBAAuB,CAAC,KAAc,EAAsB;IACpE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC;AAAA,CAC9E;AAED,SAAS,WAAW,CACnB,UAAgC,EAChC,cAAuC,EACJ;IACnC,IAAI,CAAC,cAAc;QAAE,OAAO,UAAU,CAAC;IAEvC,MAAM,IAAI,GAAG,UAAyE,CAAC;IACvF,MAAM,QAAQ,GAAG,cAAiE,CAAC;IACnF,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,EAAE,GAAG,QAAQ,EAAqD,CAAC;IAE3F,MAAM,eAAe,GAAG,IAA2C,CAAC;IACpE,MAAM,mBAAmB,GAAG,QAAmC,CAAC;IAChE,MAAM,iBAAiB,GAAG,MAAiC,CAAC;IAE5D,IAAI,eAAe,EAAE,iBAAiB,IAAI,mBAAmB,CAAC,iBAAiB,EAAE,CAAC;QACjF,iBAAiB,CAAC,iBAAiB,GAAG;YACrC,GAAG,eAAe,EAAE,iBAAiB;YACrC,GAAG,mBAAmB,CAAC,iBAAiB;SACxC,CAAC;IACH,CAAC;IAED,IAAI,eAAe,EAAE,oBAAoB,IAAI,mBAAmB,CAAC,oBAAoB,EAAE,CAAC;QACvF,iBAAiB,CAAC,oBAAoB,GAAG;YACxC,GAAG,eAAe,EAAE,oBAAoB;YACxC,GAAG,mBAAmB,CAAC,oBAAoB;SAC3C,CAAC;IACH,CAAC;IAED,OAAO,MAA8B,CAAC;AAAA,CACtC;AAED;;;GAGG;AACH,SAAS,kBAAkB,CAAC,KAAiB,EAAE,QAAuB,EAAc;IACnF,MAAM,MAAM,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC;IAE5B,yBAAyB;IACzB,IAAI,QAAQ,CAAC,IAAI,KAAK,SAAS;QAAE,MAAM,CAAC,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IAC7D,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS;QAAE,MAAM,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;IAC5E,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS;QAAE,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC,KAA6B,CAAC;IACxF,IAAI,QAAQ,CAAC,aAAa,KAAK,SAAS;QAAE,MAAM,CAAC,aAAa,GAAG,QAAQ,CAAC,aAAa,CAAC;IACxF,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS;QAAE,MAAM,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;IAE5E,gCAAgC;IAChC,IAAI,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnB,MAAM,CAAC,IAAI,GAAG;YACb,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK;YAC9C,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM;YACjD,SAAS,EAAE,QAAQ,CAAC,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC,IAAI,CAAC,SAAS;YAC1D,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU;SAC7D,CAAC;IACH,CAAC;IAED,gBAAgB;IAChB,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;QACtB,MAAM,eAAe,GAAG,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;IAC7F,CAAC;IAED,oBAAoB;IACpB,MAAM,CAAC,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IAE3D,OAAO,MAAM,CAAC;AAAA,CACd;AAED,kEAAkE;AAClE,MAAM,CAAC,MAAM,gBAAgB,GAAG,qBAAqB,CAAC;AAEtD;;GAEG;AACH,MAAM,OAAO,aAAa;IAOf,WAAW;IACZ,cAAc;IAPf,MAAM,GAAiB,EAAE,CAAC;IAC1B,qBAAqB,GAAwB,IAAI,GAAG,EAAE,CAAC;IACvD,mBAAmB,GAAqC,IAAI,GAAG,EAAE,CAAC;IAClE,SAAS,GAAuB,SAAS,CAAC;IAElD,YACU,WAAwB,EACzB,cAAc,GAAuB,IAAI,CAAC,WAAW,EAAE,EAAE,aAAa,CAAC,EAC9E;2BAFQ,WAAW;8BACZ,cAAc;QAEtB,wDAAwD;QACxD,IAAI,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC;YAClD,MAAM,SAAS,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3D,IAAI,SAAS,EAAE,CAAC;gBACf,OAAO,kBAAkB,CAAC,SAAS,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,SAAS,CAAC;QAAA,CACjB,CAAC,CAAC;QAEH,cAAc;QACd,IAAI,CAAC,UAAU,EAAE,CAAC;IAAA,CAClB;IAED;;OAEG;IACH,OAAO,GAAS;QACf,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC;QACnC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,KAAK,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAE,CAAC;YACzE,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAChD,CAAC;IAAA,CACD;IAED;;OAEG;IACH,QAAQ,GAAuB;QAC9B,OAAO,IAAI,CAAC,SAAS,CAAC;IAAA,CACtB;IAEO,UAAU,GAAS;QAC1B,oDAAoD;QACpD,MAAM,EACL,MAAM,EAAE,YAAY,EACpB,SAAS,EACT,cAAc,EACd,KAAK,GACL,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,uBAAuB,EAAE,CAAC;QAEjG,IAAI,KAAK,EAAE,CAAC;YACX,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,4DAA4D;QAC7D,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;QACxE,IAAI,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;QAEnE,iEAAiE;QACjE,KAAK,MAAM,aAAa,IAAI,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,EAAE,CAAC;YAClE,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YACpD,IAAI,IAAI,EAAE,IAAI,KAAK,OAAO,IAAI,aAAa,CAAC,YAAY,EAAE,CAAC;gBAC1D,QAAQ,GAAG,aAAa,CAAC,YAAY,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACvD,CAAC;QACF,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;IAAA,CACvB;IAED,8DAA8D;IACtD,iBAAiB,CACxB,SAAwC,EACxC,cAAuD,EACxC;QACf,OAAO,YAAY,EAAE,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC;YAC3C,MAAM,MAAM,GAAG,SAAS,CAAC,QAAyB,CAAiB,CAAC;YACpE,MAAM,gBAAgB,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,iBAAiB,GAAG,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAEvD,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;gBACxB,IAAI,KAAK,GAAG,CAAC,CAAC;gBAEd,gDAAgD;gBAChD,IAAI,gBAAgB,EAAE,CAAC;oBACtB,MAAM,eAAe,GAAG,cAAc,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBACjE,KAAK,GAAG;wBACP,GAAG,KAAK;wBACR,OAAO,EAAE,gBAAgB,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO;wBAClD,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO;qBACnF,CAAC;gBACH,CAAC;gBAED,2BAA2B;gBAC3B,MAAM,aAAa,GAAG,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACnD,IAAI,aAAa,EAAE,CAAC;oBACnB,KAAK,GAAG,kBAAkB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;gBAClD,CAAC;gBAED,OAAO,KAAK,CAAC;YAAA,CACb,CAAC,CAAC;QAAA,CACH,CAAC,CAAC;IAAA,CACH;IAED,wFAAwF;IAChF,iBAAiB,CAAC,aAA2B,EAAE,YAA0B,EAAgB;QAChG,MAAM,MAAM,GAAG,CAAC,GAAG,aAAa,CAAC,CAAC;QAClC,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;YACxC,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,WAAW,CAAC,QAAQ,IAAI,CAAC,CAAC,EAAE,KAAK,WAAW,CAAC,EAAE,CAAC,CAAC;YAC9G,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;gBACxB,MAAM,CAAC,aAAa,CAAC,GAAG,WAAW,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACP,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC;QACD,OAAO,MAAM,CAAC;IAAA,CACd;IAEO,gBAAgB,CAAC,cAAsB,EAAsB;QACpE,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YACjC,OAAO,uBAAuB,EAAE,CAAC;QAClC,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,MAAM,GAAiB,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAEjD,kBAAkB;YAClB,MAAM,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC;YACtB,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YACjD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBACvB,MAAM,MAAM,GACX,QAAQ,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,YAAY,IAAI,MAAM,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;oBAC5F,sBAAsB,CAAC;gBACxB,OAAO,uBAAuB,CAAC,gCAAgC,MAAM,aAAa,cAAc,EAAE,CAAC,CAAC;YACrG,CAAC;YAED,wBAAwB;YACxB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAE5B,MAAM,SAAS,GAAG,IAAI,GAAG,EAA4B,CAAC;YACtD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAsC,CAAC;YAErE,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC/E,2FAA2F;gBAC3F,IAAI,cAAc,CAAC,OAAO,IAAI,cAAc,CAAC,OAAO,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;oBAC/E,SAAS,CAAC,GAAG,CAAC,YAAY,EAAE;wBAC3B,OAAO,EAAE,cAAc,CAAC,OAAO;wBAC/B,OAAO,EAAE,cAAc,CAAC,OAAO;wBAC/B,MAAM,EAAE,cAAc,CAAC,MAAM;qBAC7B,CAAC,CAAC;gBACJ,CAAC;gBAED,uCAAuC;gBACvC,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;oBAC3B,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,YAAY,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC;gBACrE,CAAC;gBAED,IAAI,cAAc,CAAC,cAAc,EAAE,CAAC;oBACnC,cAAc,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;gBAC1F,CAAC;YACF,CAAC;YAED,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC1F,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;gBAClC,OAAO,uBAAuB,CAAC,gCAAgC,KAAK,CAAC,OAAO,aAAa,cAAc,EAAE,CAAC,CAAC;YAC5G,CAAC;YACD,OAAO,uBAAuB,CAC7B,+BAA+B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,aAAa,cAAc,EAAE,CAC1G,CAAC;QACH,CAAC;IAAA,CACD;IAEO,cAAc,CAAC,MAAoB,EAAQ;QAClD,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/E,MAAM,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC;YAC5C,MAAM,MAAM,GAAG,cAAc,CAAC,MAAM,IAAI,EAAE,CAAC;YAC3C,MAAM,iBAAiB,GACtB,cAAc,CAAC,cAAc,IAAI,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;YAExF,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,kEAAkE;gBAClE,IAAI,CAAC,cAAc,CAAC,OAAO,IAAI,CAAC,iBAAiB,EAAE,CAAC;oBACnD,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,0DAA0D,CAAC,CAAC;gBACrG,CAAC;YACF,CAAC;iBAAM,CAAC;gBACP,6EAA6E;gBAC7E,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;oBAC7B,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,sDAAsD,CAAC,CAAC;gBACjG,CAAC;gBACD,IAAI,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC;oBAC5B,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,qDAAqD,CAAC,CAAC;gBAChG,CAAC;YACF,CAAC;YAED,KAAK,MAAM,QAAQ,IAAI,MAAM,EAAE,CAAC;gBAC/B,MAAM,WAAW,GAAG,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC;gBAEnC,IAAI,CAAC,cAAc,IAAI,CAAC,WAAW,EAAE,CAAC;oBACrC,MAAM,IAAI,KAAK,CACd,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,uDAAuD,CACrG,CAAC;gBACH,CAAC;gBAED,IAAI,CAAC,QAAQ,CAAC,EAAE;oBAAE,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,sBAAsB,CAAC,CAAC;gBAClF,yEAAyE;gBACzE,IAAI,QAAQ,CAAC,aAAa,KAAK,SAAS,IAAI,QAAQ,CAAC,aAAa,IAAI,CAAC;oBACtE,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,yBAAyB,CAAC,CAAC;gBAC1F,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,IAAI,QAAQ,CAAC,SAAS,IAAI,CAAC;oBAC9D,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,qBAAqB,CAAC,CAAC;YACvF,CAAC;QACF,CAAC;IAAA,CACD;IAEO,WAAW,CAAC,MAAoB,EAAgB;QACvD,MAAM,MAAM,GAAiB,EAAE,CAAC;QAEhC,KAAK,MAAM,CAAC,YAAY,EAAE,cAAc,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/E,MAAM,SAAS,GAAG,cAAc,CAAC,MAAM,IAAI,EAAE,CAAC;YAC9C,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS,CAAC,kCAAkC;YAExE,6CAA6C;YAC7C,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;gBAC3B,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,YAAY,EAAE,cAAc,CAAC,MAAM,CAAC,CAAC;YACrE,CAAC;YAED,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBAClC,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,cAAc,CAAC,GAAG,CAAC;gBAC/C,IAAI,CAAC,GAAG;oBAAE,SAAS;gBAEnB,mEAAmE;gBACnE,uDAAuD;gBACvD,MAAM,eAAe,GAAG,cAAc,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;gBAC/D,MAAM,YAAY,GAAG,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBACtD,IAAI,OAAO,GAAG,eAAe,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,GAAG,eAAe,EAAE,GAAG,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBAEpG,wEAAwE;gBACxE,IAAI,cAAc,CAAC,UAAU,IAAI,cAAc,CAAC,MAAM,EAAE,CAAC;oBACxD,MAAM,WAAW,GAAG,kBAAkB,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;oBAC9D,IAAI,WAAW,EAAE,CAAC;wBACjB,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,aAAa,EAAE,UAAU,WAAW,EAAE,EAAE,CAAC;oBAClE,CAAC;gBACF,CAAC;gBAED,0DAA0D;gBAC1D,qCAAqC;gBACrC,MAAM,WAAW,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC;gBACzE,MAAM,CAAC,IAAI,CAAC;oBACX,EAAE,EAAE,QAAQ,CAAC,EAAE;oBACf,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,EAAE;oBAClC,GAAG,EAAE,GAAU;oBACf,QAAQ,EAAE,YAAY;oBACtB,OAAO,EAAE,cAAc,CAAC,OAAQ;oBAChC,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,KAAK;oBACtC,KAAK,EAAE,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,CAAyB;oBAC3D,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,WAAW;oBAClC,aAAa,EAAE,QAAQ,CAAC,aAAa,IAAI,MAAM;oBAC/C,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,KAAK;oBACtC,OAAO;oBACP,MAAM,EAAE,QAAQ,CAAC,MAAM;iBACT,CAAC,CAAC;YAClB,CAAC;QACF,CAAC;QAED,OAAO,MAAM,CAAC;IAAA,CACd;IAED;;;OAGG;IACH,MAAM,GAAiB;QACtB,OAAO,IAAI,CAAC,MAAM,CAAC;IAAA,CACnB;IAED;;;OAGG;IACH,YAAY,GAAiB;QAC5B,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;IAAA,CACvE;IAED;;OAEG;IACH,IAAI,CAAC,QAAgB,EAAE,OAAe,EAA0B;QAC/D,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,QAAQ,IAAI,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;IAAA,CAC5E;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,KAAiB,EAA+B;QAC/D,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAAA,CAClD;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB,CAAC,QAAgB,EAA+B;QACzE,OAAO,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAAA,CAC5C;IAED;;OAEG;IACH,YAAY,CAAC,KAAiB,EAAW;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAClD,OAAO,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC;IAAA,CAC9B;IAED;;;;;;OAMG;IACH,gBAAgB,CAAC,YAAoB,EAAE,MAA2B,EAAQ;QACzE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QACnD,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IAAA,CAC/C;IAEO,mBAAmB,CAAC,YAAoB,EAAE,MAA2B,EAAQ;QACpF,sCAAsC;QACtC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;YAClB,yDAAyD;YACzD,MAAM,aAAa,GAA2B;gBAC7C,GAAG,MAAM,CAAC,KAAK;gBACf,EAAE,EAAE,YAAY;aAChB,CAAC;YACF,qBAAqB,CAAC,aAAa,CAAC,CAAC;QACtC,CAAC;QAED,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,oDAAoD,CAAC,CAAC;YAC/F,CAAC;YACD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;YACzC,mBAAmB,CAAC;gBACnB,GAAG,EAAE,MAAM,CAAC,GAAG;gBACf,MAAM,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,OAA8B,CAAC;gBACjG,YAAY;aACZ,CAAC,CAAC;QACJ,CAAC;QAED,oCAAoC;QACpC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QAC7D,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/C,6DAA6D;YAC7D,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,YAAY,CAAC,CAAC;YAErE,2BAA2B;YAC3B,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,+CAA+C,CAAC,CAAC;YAC1F,CAAC;YACD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;gBACrC,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,yDAAyD,CAAC,CAAC;YACpG,CAAC;YAED,2BAA2B;YAC3B,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACtC,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,MAAM,CAAC,GAAG,CAAC;gBACvC,IAAI,CAAC,GAAG,EAAE,CAAC;oBACV,MAAM,IAAI,KAAK,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,EAAE,uBAAuB,CAAC,CAAC;gBACxF,CAAC;gBAED,gBAAgB;gBAChB,MAAM,eAAe,GAAG,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACvD,MAAM,YAAY,GAAG,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBACtD,IAAI,OAAO,GAAG,eAAe,IAAI,YAAY,CAAC,CAAC,CAAC,EAAE,GAAG,eAAe,EAAE,GAAG,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;gBAEpG,kDAAkD;gBAClD,IAAI,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBACxC,MAAM,WAAW,GAAG,kBAAkB,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;oBACtD,IAAI,WAAW,EAAE,CAAC;wBACjB,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,aAAa,EAAE,UAAU,WAAW,EAAE,EAAE,CAAC;oBAClE,CAAC;gBACF,CAAC;gBAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBAChB,EAAE,EAAE,QAAQ,CAAC,EAAE;oBACf,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,GAAG,EAAE,GAAU;oBACf,QAAQ,EAAE,YAAY;oBACtB,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,SAAS,EAAE,QAAQ,CAAC,SAAS;oBAC7B,KAAK,EAAE,QAAQ,CAAC,KAA6B;oBAC7C,IAAI,EAAE,QAAQ,CAAC,IAAI;oBACnB,aAAa,EAAE,QAAQ,CAAC,aAAa;oBACrC,SAAS,EAAE,QAAQ,CAAC,SAAS;oBAC7B,OAAO;oBACP,MAAM,EAAE,QAAQ,CAAC,MAAM;iBACT,CAAC,CAAC;YAClB,CAAC;YAED,0EAA0E;YAC1E,IAAI,MAAM,CAAC,KAAK,EAAE,YAAY,EAAE,CAAC;gBAChC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAChD,IAAI,IAAI,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;oBAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;gBAC5D,CAAC;YACF,CAAC;QACF,CAAC;aAAM,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YAC3B,4DAA4D;YAC5D,MAAM,eAAe,GAAG,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACvD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;gBACpC,IAAI,CAAC,CAAC,QAAQ,KAAK,YAAY;oBAAE,OAAO,CAAC,CAAC;gBAC1C,OAAO;oBACN,GAAG,CAAC;oBACJ,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO;oBACpC,OAAO,EAAE,eAAe,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,OAAO,EAAE,GAAG,eAAe,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO;iBAC3E,CAAC;YAAA,CACF,CAAC,CAAC;QACJ,CAAC;IAAA,CACD;CACD","sourcesContent":["/**\r\n * Model registry - manages built-in and custom models, provides API key resolution.\r\n */\r\n\r\nimport {\r\n\ttype Api,\r\n\ttype AssistantMessageEventStream,\r\n\ttype Context,\r\n\tgetModels,\r\n\tgetProviders,\r\n\ttype KnownProvider,\r\n\ttype Model,\r\n\ttype OAuthProviderInterface,\r\n\ttype OpenAICompletionsCompat,\r\n\ttype OpenAIResponsesCompat,\r\n\tregisterApiProvider,\r\n\tregisterOAuthProvider,\r\n\ttype SimpleStreamOptions,\r\n} from \"@mariozechner/pi-ai\";\r\nimport { type Static, Type } from \"@sinclair/typebox\";\r\nimport AjvModule from \"ajv\";\r\nimport { existsSync, readFileSync } from \"fs\";\r\nimport { join } from \"path\";\r\nimport { getAgentDir } from \"../config.js\";\r\nimport type { AuthStorage } from \"./auth-storage.js\";\r\nimport { clearConfigValueCache, resolveConfigValue, resolveHeaders } from \"./resolve-config-value.js\";\r\n\r\nconst Ajv = (AjvModule as any).default || AjvModule;\r\n\r\n// Schema for OpenRouter routing preferences\r\nconst OpenRouterRoutingSchema = Type.Object({\r\n\tonly: Type.Optional(Type.Array(Type.String())),\r\n\torder: Type.Optional(Type.Array(Type.String())),\r\n});\r\n\r\n// Schema for Vercel AI Gateway routing preferences\r\nconst VercelGatewayRoutingSchema = Type.Object({\r\n\tonly: Type.Optional(Type.Array(Type.String())),\r\n\torder: Type.Optional(Type.Array(Type.String())),\r\n});\r\n\r\n// Schema for OpenAI compatibility settings\r\nconst OpenAICompletionsCompatSchema = Type.Object({\r\n\tsupportsStore: Type.Optional(Type.Boolean()),\r\n\tsupportsDeveloperRole: Type.Optional(Type.Boolean()),\r\n\tsupportsReasoningEffort: Type.Optional(Type.Boolean()),\r\n\tsupportsUsageInStreaming: Type.Optional(Type.Boolean()),\r\n\tmaxTokensField: Type.Optional(Type.Union([Type.Literal(\"max_completion_tokens\"), Type.Literal(\"max_tokens\")])),\r\n\trequiresToolResultName: Type.Optional(Type.Boolean()),\r\n\trequiresAssistantAfterToolResult: Type.Optional(Type.Boolean()),\r\n\trequiresThinkingAsText: Type.Optional(Type.Boolean()),\r\n\trequiresMistralToolIds: Type.Optional(Type.Boolean()),\r\n\tthinkingFormat: Type.Optional(Type.Union([Type.Literal(\"openai\"), Type.Literal(\"zai\"), Type.Literal(\"qwen\")])),\r\n\topenRouterRouting: Type.Optional(OpenRouterRoutingSchema),\r\n\tvercelGatewayRouting: Type.Optional(VercelGatewayRoutingSchema),\r\n});\r\n\r\nconst OpenAIResponsesCompatSchema = Type.Object({\r\n\t// Reserved for future use\r\n});\r\n\r\nconst OpenAICompatSchema = Type.Union([OpenAICompletionsCompatSchema, OpenAIResponsesCompatSchema]);\r\n\r\n// Schema for custom model definition\r\n// Most fields are optional with sensible defaults for local models (Ollama, LM Studio, etc.)\r\nconst ModelDefinitionSchema = Type.Object({\r\n\tid: Type.String({ minLength: 1 }),\r\n\tname: Type.Optional(Type.String({ minLength: 1 })),\r\n\tapi: Type.Optional(Type.String({ minLength: 1 })),\r\n\treasoning: Type.Optional(Type.Boolean()),\r\n\tinput: Type.Optional(Type.Array(Type.Union([Type.Literal(\"text\"), Type.Literal(\"image\")]))),\r\n\tcost: Type.Optional(\r\n\t\tType.Object({\r\n\t\t\tinput: Type.Number(),\r\n\t\t\toutput: Type.Number(),\r\n\t\t\tcacheRead: Type.Number(),\r\n\t\t\tcacheWrite: Type.Number(),\r\n\t\t}),\r\n\t),\r\n\tcontextWindow: Type.Optional(Type.Number()),\r\n\tmaxTokens: Type.Optional(Type.Number()),\r\n\theaders: Type.Optional(Type.Record(Type.String(), Type.String())),\r\n\tcompat: Type.Optional(OpenAICompatSchema),\r\n});\r\n\r\n// Schema for per-model overrides (all fields optional, merged with built-in model)\r\nconst ModelOverrideSchema = Type.Object({\r\n\tname: Type.Optional(Type.String({ minLength: 1 })),\r\n\treasoning: Type.Optional(Type.Boolean()),\r\n\tinput: Type.Optional(Type.Array(Type.Union([Type.Literal(\"text\"), Type.Literal(\"image\")]))),\r\n\tcost: Type.Optional(\r\n\t\tType.Object({\r\n\t\t\tinput: Type.Optional(Type.Number()),\r\n\t\t\toutput: Type.Optional(Type.Number()),\r\n\t\t\tcacheRead: Type.Optional(Type.Number()),\r\n\t\t\tcacheWrite: Type.Optional(Type.Number()),\r\n\t\t}),\r\n\t),\r\n\tcontextWindow: Type.Optional(Type.Number()),\r\n\tmaxTokens: Type.Optional(Type.Number()),\r\n\theaders: Type.Optional(Type.Record(Type.String(), Type.String())),\r\n\tcompat: Type.Optional(OpenAICompatSchema),\r\n});\r\n\r\ntype ModelOverride = Static<typeof ModelOverrideSchema>;\r\n\r\nconst ProviderConfigSchema = Type.Object({\r\n\tbaseUrl: Type.Optional(Type.String({ minLength: 1 })),\r\n\tapiKey: Type.Optional(Type.String({ minLength: 1 })),\r\n\tapi: Type.Optional(Type.String({ minLength: 1 })),\r\n\theaders: Type.Optional(Type.Record(Type.String(), Type.String())),\r\n\tauthHeader: Type.Optional(Type.Boolean()),\r\n\tmodels: Type.Optional(Type.Array(ModelDefinitionSchema)),\r\n\tmodelOverrides: Type.Optional(Type.Record(Type.String(), ModelOverrideSchema)),\r\n});\r\n\r\nconst ModelsConfigSchema = Type.Object({\r\n\tproviders: Type.Record(Type.String(), ProviderConfigSchema),\r\n});\r\n\r\ntype ModelsConfig = Static<typeof ModelsConfigSchema>;\r\n\r\n/** Provider override config (baseUrl, headers, apiKey) without custom models */\r\ninterface ProviderOverride {\r\n\tbaseUrl?: string;\r\n\theaders?: Record<string, string>;\r\n\tapiKey?: string;\r\n}\r\n\r\n/** Result of loading custom models from models.json */\r\ninterface CustomModelsResult {\r\n\tmodels: Model<Api>[];\r\n\t/** Providers with baseUrl/headers/apiKey overrides for built-in models */\r\n\toverrides: Map<string, ProviderOverride>;\r\n\t/** Per-model overrides: provider -> modelId -> override */\r\n\tmodelOverrides: Map<string, Map<string, ModelOverride>>;\r\n\terror: string | undefined;\r\n}\r\n\r\nfunction emptyCustomModelsResult(error?: string): CustomModelsResult {\r\n\treturn { models: [], overrides: new Map(), modelOverrides: new Map(), error };\r\n}\r\n\r\nfunction mergeCompat(\r\n\tbaseCompat: Model<Api>[\"compat\"],\r\n\toverrideCompat: ModelOverride[\"compat\"],\r\n): Model<Api>[\"compat\"] | undefined {\r\n\tif (!overrideCompat) return baseCompat;\r\n\r\n\tconst base = baseCompat as OpenAICompletionsCompat | OpenAIResponsesCompat | undefined;\r\n\tconst override = overrideCompat as OpenAICompletionsCompat | OpenAIResponsesCompat;\r\n\tconst merged = { ...base, ...override } as OpenAICompletionsCompat | OpenAIResponsesCompat;\r\n\r\n\tconst baseCompletions = base as OpenAICompletionsCompat | undefined;\r\n\tconst overrideCompletions = override as OpenAICompletionsCompat;\r\n\tconst mergedCompletions = merged as OpenAICompletionsCompat;\r\n\r\n\tif (baseCompletions?.openRouterRouting || overrideCompletions.openRouterRouting) {\r\n\t\tmergedCompletions.openRouterRouting = {\r\n\t\t\t...baseCompletions?.openRouterRouting,\r\n\t\t\t...overrideCompletions.openRouterRouting,\r\n\t\t};\r\n\t}\r\n\r\n\tif (baseCompletions?.vercelGatewayRouting || overrideCompletions.vercelGatewayRouting) {\r\n\t\tmergedCompletions.vercelGatewayRouting = {\r\n\t\t\t...baseCompletions?.vercelGatewayRouting,\r\n\t\t\t...overrideCompletions.vercelGatewayRouting,\r\n\t\t};\r\n\t}\r\n\r\n\treturn merged as Model<Api>[\"compat\"];\r\n}\r\n\r\n/**\r\n * Deep merge a model override into a model.\r\n * Handles nested objects (cost, compat) by merging rather than replacing.\r\n */\r\nfunction applyModelOverride(model: Model<Api>, override: ModelOverride): Model<Api> {\r\n\tconst result = { ...model };\r\n\r\n\t// Simple field overrides\r\n\tif (override.name !== undefined) result.name = override.name;\r\n\tif (override.reasoning !== undefined) result.reasoning = override.reasoning;\r\n\tif (override.input !== undefined) result.input = override.input as (\"text\" | \"image\")[];\r\n\tif (override.contextWindow !== undefined) result.contextWindow = override.contextWindow;\r\n\tif (override.maxTokens !== undefined) result.maxTokens = override.maxTokens;\r\n\r\n\t// Merge cost (partial override)\r\n\tif (override.cost) {\r\n\t\tresult.cost = {\r\n\t\t\tinput: override.cost.input ?? model.cost.input,\r\n\t\t\toutput: override.cost.output ?? model.cost.output,\r\n\t\t\tcacheRead: override.cost.cacheRead ?? model.cost.cacheRead,\r\n\t\t\tcacheWrite: override.cost.cacheWrite ?? model.cost.cacheWrite,\r\n\t\t};\r\n\t}\r\n\r\n\t// Merge headers\r\n\tif (override.headers) {\r\n\t\tconst resolvedHeaders = resolveHeaders(override.headers);\r\n\t\tresult.headers = resolvedHeaders ? { ...model.headers, ...resolvedHeaders } : model.headers;\r\n\t}\r\n\r\n\t// Deep merge compat\r\n\tresult.compat = mergeCompat(model.compat, override.compat);\r\n\r\n\treturn result;\r\n}\r\n\r\n/** Clear the config value command cache. Exported for testing. */\r\nexport const clearApiKeyCache = clearConfigValueCache;\r\n\r\n/**\r\n * Model registry - loads and manages models, resolves API keys via AuthStorage.\r\n */\r\nexport class ModelRegistry {\r\n\tprivate models: Model<Api>[] = [];\r\n\tprivate customProviderApiKeys: Map<string, string> = new Map();\r\n\tprivate registeredProviders: Map<string, ProviderConfigInput> = new Map();\r\n\tprivate loadError: string | undefined = undefined;\r\n\r\n\tconstructor(\r\n\t\treadonly authStorage: AuthStorage,\r\n\t\tprivate modelsJsonPath: string | undefined = join(getAgentDir(), \"models.json\"),\r\n\t) {\r\n\t\t// Set up fallback resolver for custom provider API keys\r\n\t\tthis.authStorage.setFallbackResolver((provider) => {\r\n\t\t\tconst keyConfig = this.customProviderApiKeys.get(provider);\r\n\t\t\tif (keyConfig) {\r\n\t\t\t\treturn resolveConfigValue(keyConfig);\r\n\t\t\t}\r\n\t\t\treturn undefined;\r\n\t\t});\r\n\r\n\t\t// Load models\r\n\t\tthis.loadModels();\r\n\t}\r\n\r\n\t/**\r\n\t * Reload models from disk (built-in + custom from models.json).\r\n\t */\r\n\trefresh(): void {\r\n\t\tthis.customProviderApiKeys.clear();\r\n\t\tthis.loadError = undefined;\r\n\t\tthis.loadModels();\r\n\r\n\t\tfor (const [providerName, config] of this.registeredProviders.entries()) {\r\n\t\t\tthis.applyProviderConfig(providerName, config);\r\n\t\t}\r\n\t}\r\n\r\n\t/**\r\n\t * Get any error from loading models.json (undefined if no error).\r\n\t */\r\n\tgetError(): string | undefined {\r\n\t\treturn this.loadError;\r\n\t}\r\n\r\n\tprivate loadModels(): void {\r\n\t\t// Load custom models and overrides from models.json\r\n\t\tconst {\r\n\t\t\tmodels: customModels,\r\n\t\t\toverrides,\r\n\t\t\tmodelOverrides,\r\n\t\t\terror,\r\n\t\t} = this.modelsJsonPath ? this.loadCustomModels(this.modelsJsonPath) : emptyCustomModelsResult();\r\n\r\n\t\tif (error) {\r\n\t\t\tthis.loadError = error;\r\n\t\t\t// Keep built-in models even if custom models failed to load\r\n\t\t}\r\n\r\n\t\tconst builtInModels = this.loadBuiltInModels(overrides, modelOverrides);\r\n\t\tlet combined = this.mergeCustomModels(builtInModels, customModels);\r\n\r\n\t\t// Let OAuth providers modify their models (e.g., update baseUrl)\r\n\t\tfor (const oauthProvider of this.authStorage.getOAuthProviders()) {\r\n\t\t\tconst cred = this.authStorage.get(oauthProvider.id);\r\n\t\t\tif (cred?.type === \"oauth\" && oauthProvider.modifyModels) {\r\n\t\t\t\tcombined = oauthProvider.modifyModels(combined, cred);\r\n\t\t\t}\r\n\t\t}\r\n\r\n\t\tthis.models = combined;\r\n\t}\r\n\r\n\t/** Load built-in models and apply provider/model overrides */\r\n\tprivate loadBuiltInModels(\r\n\t\toverrides: Map<string, ProviderOverride>,\r\n\t\tmodelOverrides: Map<string, Map<string, ModelOverride>>,\r\n\t): Model<Api>[] {\r\n\t\treturn getProviders().flatMap((provider) => {\r\n\t\t\tconst models = getModels(provider as KnownProvider) as Model<Api>[];\r\n\t\t\tconst providerOverride = overrides.get(provider);\r\n\t\t\tconst perModelOverrides = modelOverrides.get(provider);\r\n\r\n\t\t\treturn models.map((m) => {\r\n\t\t\t\tlet model = m;\r\n\r\n\t\t\t\t// Apply provider-level baseUrl/headers override\r\n\t\t\t\tif (providerOverride) {\r\n\t\t\t\t\tconst resolvedHeaders = resolveHeaders(providerOverride.headers);\r\n\t\t\t\t\tmodel = {\r\n\t\t\t\t\t\t...model,\r\n\t\t\t\t\t\tbaseUrl: providerOverride.baseUrl ?? model.baseUrl,\r\n\t\t\t\t\t\theaders: resolvedHeaders ? { ...model.headers, ...resolvedHeaders } : model.headers,\r\n\t\t\t\t\t};\r\n\t\t\t\t}\r\n\r\n\t\t\t\t// Apply per-model override\r\n\t\t\t\tconst modelOverride = perModelOverrides?.get(m.id);\r\n\t\t\t\tif (modelOverride) {\r\n\t\t\t\t\tmodel = applyModelOverride(model, modelOverride);\r\n\t\t\t\t}\r\n\r\n\t\t\t\treturn model;\r\n\t\t\t});\r\n\t\t});\r\n\t}\r\n\r\n\t/** Merge custom models into built-in list by provider+id (custom wins on conflicts). */\r\n\tprivate mergeCustomModels(builtInModels: Model<Api>[], customModels: Model<Api>[]): Model<Api>[] {\r\n\t\tconst merged = [...builtInModels];\r\n\t\tfor (const customModel of customModels) {\r\n\t\t\tconst existingIndex = merged.findIndex((m) => m.provider === customModel.provider && m.id === customModel.id);\r\n\t\t\tif (existingIndex >= 0) {\r\n\t\t\t\tmerged[existingIndex] = customModel;\r\n\t\t\t} else {\r\n\t\t\t\tmerged.push(customModel);\r\n\t\t\t}\r\n\t\t}\r\n\t\treturn merged;\r\n\t}\r\n\r\n\tprivate loadCustomModels(modelsJsonPath: string): CustomModelsResult {\r\n\t\tif (!existsSync(modelsJsonPath)) {\r\n\t\t\treturn emptyCustomModelsResult();\r\n\t\t}\r\n\r\n\t\ttry {\r\n\t\t\tconst content = readFileSync(modelsJsonPath, \"utf-8\");\r\n\t\t\tconst config: ModelsConfig = JSON.parse(content);\r\n\r\n\t\t\t// Validate schema\r\n\t\t\tconst ajv = new Ajv();\r\n\t\t\tconst validate = ajv.compile(ModelsConfigSchema);\r\n\t\t\tif (!validate(config)) {\r\n\t\t\t\tconst errors =\r\n\t\t\t\t\tvalidate.errors?.map((e: any) => `  - ${e.instancePath || \"root\"}: ${e.message}`).join(\"\\n\") ||\r\n\t\t\t\t\t\"Unknown schema error\";\r\n\t\t\t\treturn emptyCustomModelsResult(`Invalid models.json schema:\\n${errors}\\n\\nFile: ${modelsJsonPath}`);\r\n\t\t\t}\r\n\r\n\t\t\t// Additional validation\r\n\t\t\tthis.validateConfig(config);\r\n\r\n\t\t\tconst overrides = new Map<string, ProviderOverride>();\r\n\t\t\tconst modelOverrides = new Map<string, Map<string, ModelOverride>>();\r\n\r\n\t\t\tfor (const [providerName, providerConfig] of Object.entries(config.providers)) {\r\n\t\t\t\t// Apply provider-level baseUrl/headers/apiKey override to built-in models when configured.\r\n\t\t\t\tif (providerConfig.baseUrl || providerConfig.headers || providerConfig.apiKey) {\r\n\t\t\t\t\toverrides.set(providerName, {\r\n\t\t\t\t\t\tbaseUrl: providerConfig.baseUrl,\r\n\t\t\t\t\t\theaders: providerConfig.headers,\r\n\t\t\t\t\t\tapiKey: providerConfig.apiKey,\r\n\t\t\t\t\t});\r\n\t\t\t\t}\r\n\r\n\t\t\t\t// Store API key for fallback resolver.\r\n\t\t\t\tif (providerConfig.apiKey) {\r\n\t\t\t\t\tthis.customProviderApiKeys.set(providerName, providerConfig.apiKey);\r\n\t\t\t\t}\r\n\r\n\t\t\t\tif (providerConfig.modelOverrides) {\r\n\t\t\t\t\tmodelOverrides.set(providerName, new Map(Object.entries(providerConfig.modelOverrides)));\r\n\t\t\t\t}\r\n\t\t\t}\r\n\r\n\t\t\treturn { models: this.parseModels(config), overrides, modelOverrides, error: undefined };\r\n\t\t} catch (error) {\r\n\t\t\tif (error instanceof SyntaxError) {\r\n\t\t\t\treturn emptyCustomModelsResult(`Failed to parse models.json: ${error.message}\\n\\nFile: ${modelsJsonPath}`);\r\n\t\t\t}\r\n\t\t\treturn emptyCustomModelsResult(\r\n\t\t\t\t`Failed to load models.json: ${error instanceof Error ? error.message : error}\\n\\nFile: ${modelsJsonPath}`,\r\n\t\t\t);\r\n\t\t}\r\n\t}\r\n\r\n\tprivate validateConfig(config: ModelsConfig): void {\r\n\t\tfor (const [providerName, providerConfig] of Object.entries(config.providers)) {\r\n\t\t\tconst hasProviderApi = !!providerConfig.api;\r\n\t\t\tconst models = providerConfig.models ?? [];\r\n\t\t\tconst hasModelOverrides =\r\n\t\t\t\tproviderConfig.modelOverrides && Object.keys(providerConfig.modelOverrides).length > 0;\r\n\r\n\t\t\tif (models.length === 0) {\r\n\t\t\t\t// Override-only config: needs baseUrl OR modelOverrides (or both)\r\n\t\t\t\tif (!providerConfig.baseUrl && !hasModelOverrides) {\r\n\t\t\t\t\tthrow new Error(`Provider ${providerName}: must specify \"baseUrl\", \"modelOverrides\", or \"models\".`);\r\n\t\t\t\t}\r\n\t\t\t} else {\r\n\t\t\t\t// Custom models are merged into provider models and require endpoint + auth.\r\n\t\t\t\tif (!providerConfig.baseUrl) {\r\n\t\t\t\t\tthrow new Error(`Provider ${providerName}: \"baseUrl\" is required when defining custom models.`);\r\n\t\t\t\t}\r\n\t\t\t\tif (!providerConfig.apiKey) {\r\n\t\t\t\t\tthrow new Error(`Provider ${providerName}: \"apiKey\" is required when defining custom models.`);\r\n\t\t\t\t}\r\n\t\t\t}\r\n\r\n\t\t\tfor (const modelDef of models) {\r\n\t\t\t\tconst hasModelApi = !!modelDef.api;\r\n\r\n\t\t\t\tif (!hasProviderApi && !hasModelApi) {\r\n\t\t\t\t\tthrow new Error(\r\n\t\t\t\t\t\t`Provider ${providerName}, model ${modelDef.id}: no \"api\" specified. Set at provider or model level.`,\r\n\t\t\t\t\t);\r\n\t\t\t\t}\r\n\r\n\t\t\t\tif (!modelDef.id) throw new Error(`Provider ${providerName}: model missing \"id\"`);\r\n\t\t\t\t// Validate contextWindow/maxTokens only if provided (they have defaults)\r\n\t\t\t\tif (modelDef.contextWindow !== undefined && modelDef.contextWindow <= 0)\r\n\t\t\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid contextWindow`);\r\n\t\t\t\tif (modelDef.maxTokens !== undefined && modelDef.maxTokens <= 0)\r\n\t\t\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: invalid maxTokens`);\r\n\t\t\t}\r\n\t\t}\r\n\t}\r\n\r\n\tprivate parseModels(config: ModelsConfig): Model<Api>[] {\r\n\t\tconst models: Model<Api>[] = [];\r\n\r\n\t\tfor (const [providerName, providerConfig] of Object.entries(config.providers)) {\r\n\t\t\tconst modelDefs = providerConfig.models ?? [];\r\n\t\t\tif (modelDefs.length === 0) continue; // Override-only, no custom models\r\n\r\n\t\t\t// Store API key config for fallback resolver\r\n\t\t\tif (providerConfig.apiKey) {\r\n\t\t\t\tthis.customProviderApiKeys.set(providerName, providerConfig.apiKey);\r\n\t\t\t}\r\n\r\n\t\t\tfor (const modelDef of modelDefs) {\r\n\t\t\t\tconst api = modelDef.api || providerConfig.api;\r\n\t\t\t\tif (!api) continue;\r\n\r\n\t\t\t\t// Merge headers: provider headers are base, model headers override\r\n\t\t\t\t// Resolve env vars and shell commands in header values\r\n\t\t\t\tconst providerHeaders = resolveHeaders(providerConfig.headers);\r\n\t\t\t\tconst modelHeaders = resolveHeaders(modelDef.headers);\r\n\t\t\t\tlet headers = providerHeaders || modelHeaders ? { ...providerHeaders, ...modelHeaders } : undefined;\r\n\r\n\t\t\t\t// If authHeader is true, add Authorization header with resolved API key\r\n\t\t\t\tif (providerConfig.authHeader && providerConfig.apiKey) {\r\n\t\t\t\t\tconst resolvedKey = resolveConfigValue(providerConfig.apiKey);\r\n\t\t\t\t\tif (resolvedKey) {\r\n\t\t\t\t\t\theaders = { ...headers, Authorization: `Bearer ${resolvedKey}` };\r\n\t\t\t\t\t}\r\n\t\t\t\t}\r\n\r\n\t\t\t\t// baseUrl is validated to exist for providers with models\r\n\t\t\t\t// Apply defaults for optional fields\r\n\t\t\t\tconst defaultCost = { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };\r\n\t\t\t\tmodels.push({\r\n\t\t\t\t\tid: modelDef.id,\r\n\t\t\t\t\tname: modelDef.name ?? modelDef.id,\r\n\t\t\t\t\tapi: api as Api,\r\n\t\t\t\t\tprovider: providerName,\r\n\t\t\t\t\tbaseUrl: providerConfig.baseUrl!,\r\n\t\t\t\t\treasoning: modelDef.reasoning ?? false,\r\n\t\t\t\t\tinput: (modelDef.input ?? [\"text\"]) as (\"text\" | \"image\")[],\r\n\t\t\t\t\tcost: modelDef.cost ?? defaultCost,\r\n\t\t\t\t\tcontextWindow: modelDef.contextWindow ?? 128000,\r\n\t\t\t\t\tmaxTokens: modelDef.maxTokens ?? 16384,\r\n\t\t\t\t\theaders,\r\n\t\t\t\t\tcompat: modelDef.compat,\r\n\t\t\t\t} as Model<Api>);\r\n\t\t\t}\r\n\t\t}\r\n\r\n\t\treturn models;\r\n\t}\r\n\r\n\t/**\r\n\t * Get all models (built-in + custom).\r\n\t * If models.json had errors, returns only built-in models.\r\n\t */\r\n\tgetAll(): Model<Api>[] {\r\n\t\treturn this.models;\r\n\t}\r\n\r\n\t/**\r\n\t * Get only models that have auth configured.\r\n\t * This is a fast check that doesn't refresh OAuth tokens.\r\n\t */\r\n\tgetAvailable(): Model<Api>[] {\r\n\t\treturn this.models.filter((m) => this.authStorage.hasAuth(m.provider));\r\n\t}\r\n\r\n\t/**\r\n\t * Find a model by provider and ID.\r\n\t */\r\n\tfind(provider: string, modelId: string): Model<Api> | undefined {\r\n\t\treturn this.models.find((m) => m.provider === provider && m.id === modelId);\r\n\t}\r\n\r\n\t/**\r\n\t * Get API key for a model.\r\n\t */\r\n\tasync getApiKey(model: Model<Api>): Promise<string | undefined> {\r\n\t\treturn this.authStorage.getApiKey(model.provider);\r\n\t}\r\n\r\n\t/**\r\n\t * Get API key for a provider.\r\n\t */\r\n\tasync getApiKeyForProvider(provider: string): Promise<string | undefined> {\r\n\t\treturn this.authStorage.getApiKey(provider);\r\n\t}\r\n\r\n\t/**\r\n\t * Check if a model is using OAuth credentials (subscription).\r\n\t */\r\n\tisUsingOAuth(model: Model<Api>): boolean {\r\n\t\tconst cred = this.authStorage.get(model.provider);\r\n\t\treturn cred?.type === \"oauth\";\r\n\t}\r\n\r\n\t/**\r\n\t * Register a provider dynamically (from extensions).\r\n\t *\r\n\t * If provider has models: replaces all existing models for this provider.\r\n\t * If provider has only baseUrl/headers: overrides existing models' URLs.\r\n\t * If provider has oauth: registers OAuth provider for /login support.\r\n\t */\r\n\tregisterProvider(providerName: string, config: ProviderConfigInput): void {\r\n\t\tthis.registeredProviders.set(providerName, config);\r\n\t\tthis.applyProviderConfig(providerName, config);\r\n\t}\r\n\r\n\tprivate applyProviderConfig(providerName: string, config: ProviderConfigInput): void {\r\n\t\t// Register OAuth provider if provided\r\n\t\tif (config.oauth) {\r\n\t\t\t// Ensure the OAuth provider ID matches the provider name\r\n\t\t\tconst oauthProvider: OAuthProviderInterface = {\r\n\t\t\t\t...config.oauth,\r\n\t\t\t\tid: providerName,\r\n\t\t\t};\r\n\t\t\tregisterOAuthProvider(oauthProvider);\r\n\t\t}\r\n\r\n\t\tif (config.streamSimple) {\r\n\t\t\tif (!config.api) {\r\n\t\t\t\tthrow new Error(`Provider ${providerName}: \"api\" is required when registering streamSimple.`);\r\n\t\t\t}\r\n\t\t\tconst streamSimple = config.streamSimple;\r\n\t\t\tregisterApiProvider({\r\n\t\t\t\tapi: config.api,\r\n\t\t\t\tstream: (model, context, options) => streamSimple(model, context, options as SimpleStreamOptions),\r\n\t\t\t\tstreamSimple,\r\n\t\t\t});\r\n\t\t}\r\n\r\n\t\t// Store API key for auth resolution\r\n\t\tif (config.apiKey) {\r\n\t\t\tthis.customProviderApiKeys.set(providerName, config.apiKey);\r\n\t\t}\r\n\r\n\t\tif (config.models && config.models.length > 0) {\r\n\t\t\t// Full replacement: remove existing models for this provider\r\n\t\t\tthis.models = this.models.filter((m) => m.provider !== providerName);\r\n\r\n\t\t\t// Validate required fields\r\n\t\t\tif (!config.baseUrl) {\r\n\t\t\t\tthrow new Error(`Provider ${providerName}: \"baseUrl\" is required when defining models.`);\r\n\t\t\t}\r\n\t\t\tif (!config.apiKey && !config.oauth) {\r\n\t\t\t\tthrow new Error(`Provider ${providerName}: \"apiKey\" or \"oauth\" is required when defining models.`);\r\n\t\t\t}\r\n\r\n\t\t\t// Parse and add new models\r\n\t\t\tfor (const modelDef of config.models) {\r\n\t\t\t\tconst api = modelDef.api || config.api;\r\n\t\t\t\tif (!api) {\r\n\t\t\t\t\tthrow new Error(`Provider ${providerName}, model ${modelDef.id}: no \"api\" specified.`);\r\n\t\t\t\t}\r\n\r\n\t\t\t\t// Merge headers\r\n\t\t\t\tconst providerHeaders = resolveHeaders(config.headers);\r\n\t\t\t\tconst modelHeaders = resolveHeaders(modelDef.headers);\r\n\t\t\t\tlet headers = providerHeaders || modelHeaders ? { ...providerHeaders, ...modelHeaders } : undefined;\r\n\r\n\t\t\t\t// If authHeader is true, add Authorization header\r\n\t\t\t\tif (config.authHeader && config.apiKey) {\r\n\t\t\t\t\tconst resolvedKey = resolveConfigValue(config.apiKey);\r\n\t\t\t\t\tif (resolvedKey) {\r\n\t\t\t\t\t\theaders = { ...headers, Authorization: `Bearer ${resolvedKey}` };\r\n\t\t\t\t\t}\r\n\t\t\t\t}\r\n\r\n\t\t\t\tthis.models.push({\r\n\t\t\t\t\tid: modelDef.id,\r\n\t\t\t\t\tname: modelDef.name,\r\n\t\t\t\t\tapi: api as Api,\r\n\t\t\t\t\tprovider: providerName,\r\n\t\t\t\t\tbaseUrl: config.baseUrl,\r\n\t\t\t\t\treasoning: modelDef.reasoning,\r\n\t\t\t\t\tinput: modelDef.input as (\"text\" | \"image\")[],\r\n\t\t\t\t\tcost: modelDef.cost,\r\n\t\t\t\t\tcontextWindow: modelDef.contextWindow,\r\n\t\t\t\t\tmaxTokens: modelDef.maxTokens,\r\n\t\t\t\t\theaders,\r\n\t\t\t\t\tcompat: modelDef.compat,\r\n\t\t\t\t} as Model<Api>);\r\n\t\t\t}\r\n\r\n\t\t\t// Apply OAuth modifyModels if credentials exist (e.g., to update baseUrl)\r\n\t\t\tif (config.oauth?.modifyModels) {\r\n\t\t\t\tconst cred = this.authStorage.get(providerName);\r\n\t\t\t\tif (cred?.type === \"oauth\") {\r\n\t\t\t\t\tthis.models = config.oauth.modifyModels(this.models, cred);\r\n\t\t\t\t}\r\n\t\t\t}\r\n\t\t} else if (config.baseUrl) {\r\n\t\t\t// Override-only: update baseUrl/headers for existing models\r\n\t\t\tconst resolvedHeaders = resolveHeaders(config.headers);\r\n\t\t\tthis.models = this.models.map((m) => {\r\n\t\t\t\tif (m.provider !== providerName) return m;\r\n\t\t\t\treturn {\r\n\t\t\t\t\t...m,\r\n\t\t\t\t\tbaseUrl: config.baseUrl ?? m.baseUrl,\r\n\t\t\t\t\theaders: resolvedHeaders ? { ...m.headers, ...resolvedHeaders } : m.headers,\r\n\t\t\t\t};\r\n\t\t\t});\r\n\t\t}\r\n\t}\r\n}\r\n\r\n/**\r\n * Input type for registerProvider API.\r\n */\r\nexport interface ProviderConfigInput {\r\n\tbaseUrl?: string;\r\n\tapiKey?: string;\r\n\tapi?: Api;\r\n\tstreamSimple?: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;\r\n\theaders?: Record<string, string>;\r\n\tauthHeader?: boolean;\r\n\t/** OAuth provider for /login support */\r\n\toauth?: Omit<OAuthProviderInterface, \"id\">;\r\n\tmodels?: Array<{\r\n\t\tid: string;\r\n\t\tname: string;\r\n\t\tapi?: Api;\r\n\t\treasoning: boolean;\r\n\t\tinput: (\"text\" | \"image\")[];\r\n\t\tcost: { input: number; output: number; cacheRead: number; cacheWrite: number };\r\n\t\tcontextWindow: number;\r\n\t\tmaxTokens: number;\r\n\t\theaders?: Record<string, string>;\r\n\t\tcompat?: Model<Api>[\"compat\"];\r\n\t}>;\r\n}\r\n"]}