From 246410947d4ab2d2bcd39fc3b13f12fbebaa4adf Mon Sep 17 00:00:00 2001 From: Kaiyi Date: Thu, 5 Mar 2026 00:21:02 +0800 Subject: [PATCH] Model: honor configured provider API over stale registry model --- src/agents/pi-embedded-runner/model.test.ts | 52 +++++++++++++++ src/agents/pi-embedded-runner/model.ts | 72 ++++++++++++++++----- 2 files changed, 107 insertions(+), 17 deletions(-) diff --git a/src/agents/pi-embedded-runner/model.test.ts b/src/agents/pi-embedded-runner/model.test.ts index 54fa48cf17a..a04fdbbe412 100644 --- a/src/agents/pi-embedded-runner/model.test.ts +++ b/src/agents/pi-embedded-runner/model.test.ts @@ -278,6 +278,58 @@ describe("resolveModel", () => { expect(result.model?.reasoning).toBe(true); }); + it("prefers configured provider api metadata over discovered registry model", () => { + mockDiscoveredModel({ + provider: "onehub", + modelId: "glm-5", + templateModel: { + id: "glm-5", + name: "GLM-5 (cached)", + provider: "onehub", + api: "anthropic-messages", + baseUrl: "https://old-provider.example.com/v1", + reasoning: false, + input: ["text"], + cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 }, + contextWindow: 8192, + maxTokens: 2048, + }, + }); + + const cfg = { + models: { + providers: { + onehub: { + baseUrl: "http://new-provider.example.com/v1", + api: "openai-completions", + models: [ + { + ...makeModel("glm-5"), + api: "openai-completions", + reasoning: true, + contextWindow: 198000, + maxTokens: 16000, + }, + ], + }, + }, + }, + } as OpenClawConfig; + + const result = resolveModel("onehub", "glm-5", "/tmp/agent", cfg); + + expect(result.error).toBeUndefined(); + expect(result.model).toMatchObject({ + provider: "onehub", + id: "glm-5", + api: "openai-completions", + baseUrl: "http://new-provider.example.com/v1", + reasoning: true, + contextWindow: 198000, + maxTokens: 16000, + }); + }); + it("builds an openai-codex fallback for gpt-5.3-codex", () => { mockOpenAICodexTemplateModel(); diff --git a/src/agents/pi-embedded-runner/model.ts b/src/agents/pi-embedded-runner/model.ts index 0b7fc61ed01..8769102762a 100644 --- a/src/agents/pi-embedded-runner/model.ts +++ b/src/agents/pi-embedded-runner/model.ts @@ -7,7 +7,7 @@ import { DEFAULT_CONTEXT_TOKENS } from "../defaults.js"; import { buildModelAliasLines } from "../model-alias-lines.js"; import { normalizeModelCompat } from "../model-compat.js"; import { resolveForwardCompatModel } from "../model-forward-compat.js"; -import { normalizeProviderId } from "../model-selection.js"; +import { findNormalizedProviderValue, normalizeProviderId } from "../model-selection.js"; import { discoverAuthStorage, discoverModels } from "../pi-model-discovery.js"; type InlineModelEntry = ModelDefinitionConfig & { @@ -24,6 +24,47 @@ type InlineProviderConfig = { export { buildModelAliasLines }; +function resolveConfiguredProviderConfig( + cfg: OpenClawConfig | undefined, + provider: string, +): InlineProviderConfig | undefined { + return findNormalizedProviderValue(cfg?.models?.providers, provider); +} + +function applyConfiguredProviderOverrides(params: { + discoveredModel: Model; + providerConfig?: InlineProviderConfig; + modelId: string; +}): Model { + const { discoveredModel, providerConfig, modelId } = params; + if (!providerConfig) { + return discoveredModel; + } + const configuredModel = providerConfig.models?.find((candidate) => candidate.id === modelId); + if (!configuredModel && !providerConfig.baseUrl && !providerConfig.api && !providerConfig.headers) { + return discoveredModel; + } + return { + ...discoveredModel, + api: configuredModel?.api ?? providerConfig.api ?? discoveredModel.api, + baseUrl: providerConfig.baseUrl ?? discoveredModel.baseUrl, + reasoning: configuredModel?.reasoning ?? discoveredModel.reasoning, + input: configuredModel?.input ?? discoveredModel.input, + cost: configuredModel?.cost ?? discoveredModel.cost, + contextWindow: configuredModel?.contextWindow ?? discoveredModel.contextWindow, + maxTokens: configuredModel?.maxTokens ?? discoveredModel.maxTokens, + headers: + providerConfig.headers || configuredModel?.headers + ? { + ...(discoveredModel.headers ?? {}), + ...(providerConfig.headers ?? {}), + ...(configuredModel?.headers ?? {}), + } + : discoveredModel.headers, + compat: configuredModel?.compat ?? discoveredModel.compat, + }; +} + export function buildInlineProviderModels( providers: Record, ): InlineModelEntry[] { @@ -59,6 +100,7 @@ export function resolveModel( const resolvedAgentDir = agentDir ?? resolveOpenClawAgentDir(); const authStorage = discoverAuthStorage(resolvedAgentDir); const modelRegistry = discoverModels(authStorage, resolvedAgentDir); + const providerConfig = resolveConfiguredProviderConfig(cfg, provider); const model = modelRegistry.find(provider, modelId) as Model | null; if (!model) { @@ -100,7 +142,7 @@ export function resolveModel( } as Model); return { model: fallbackModel, authStorage, modelRegistry }; } - const providerCfg = providers[provider]; + const providerCfg = providerConfig; if (providerCfg || modelId.startsWith("mock-")) { const configuredModel = providerCfg?.models?.find((candidate) => candidate.id === modelId); const fallbackModel: Model = normalizeModelCompat({ @@ -133,21 +175,17 @@ export function resolveModel( modelRegistry, }; } - const providerOverride = cfg?.models?.providers?.[provider] as InlineProviderConfig | undefined; - if (providerOverride?.baseUrl || providerOverride?.headers) { - const overridden: Model & { headers?: Record } = { ...model }; - if (providerOverride.baseUrl) { - overridden.baseUrl = providerOverride.baseUrl; - } - if (providerOverride.headers) { - overridden.headers = { - ...(model as Model & { headers?: Record }).headers, - ...providerOverride.headers, - }; - } - return { model: normalizeModelCompat(overridden), authStorage, modelRegistry }; - } - return { model: normalizeModelCompat(model), authStorage, modelRegistry }; + return { + model: normalizeModelCompat( + applyConfiguredProviderOverrides({ + discoveredModel: model, + providerConfig, + modelId, + }), + ), + authStorage, + modelRegistry, + }; } /**