refactor(commands): dedupe provider config + default model helpers

This commit is contained in:
Peter Steinberger
2026-02-15 06:33:37 +00:00
parent 2bd672f3ab
commit 6217561931
3 changed files with 107 additions and 146 deletions
+36 -121
View File
@@ -48,7 +48,11 @@ export {
LITELLM_BASE_URL, LITELLM_BASE_URL,
LITELLM_DEFAULT_MODEL_ID, LITELLM_DEFAULT_MODEL_ID,
} from "./onboard-auth.config-litellm.js"; } from "./onboard-auth.config-litellm.js";
import { applyProviderConfigWithDefaultModel } from "./onboard-auth.config-shared.js"; import {
applyAgentDefaultModelPrimary,
applyProviderConfigWithDefaultModel,
applyProviderConfigWithDefaultModels,
} from "./onboard-auth.config-shared.js";
import { import {
buildZaiModelDefinition, buildZaiModelDefinition,
buildMoonshotModelDefinition, buildMoonshotModelDefinition,
@@ -236,46 +240,12 @@ function applyMoonshotProviderConfigWithBaseUrl(
export function applyMoonshotConfig(cfg: OpenClawConfig): OpenClawConfig { export function applyMoonshotConfig(cfg: OpenClawConfig): OpenClawConfig {
const next = applyMoonshotProviderConfig(cfg); const next = applyMoonshotProviderConfig(cfg);
const existingModel = next.agents?.defaults?.model; return applyAgentDefaultModelPrimary(next, MOONSHOT_DEFAULT_MODEL_REF);
return {
...next,
agents: {
...next.agents,
defaults: {
...next.agents?.defaults,
model: {
...(existingModel && "fallbacks" in (existingModel as Record<string, unknown>)
? {
fallbacks: (existingModel as { fallbacks?: string[] }).fallbacks,
}
: undefined),
primary: MOONSHOT_DEFAULT_MODEL_REF,
},
},
},
};
} }
export function applyMoonshotConfigCn(cfg: OpenClawConfig): OpenClawConfig { export function applyMoonshotConfigCn(cfg: OpenClawConfig): OpenClawConfig {
const next = applyMoonshotProviderConfigCn(cfg); const next = applyMoonshotProviderConfigCn(cfg);
const existingModel = next.agents?.defaults?.model; return applyAgentDefaultModelPrimary(next, MOONSHOT_DEFAULT_MODEL_REF);
return {
...next,
agents: {
...next.agents,
defaults: {
...next.agents?.defaults,
model: {
...(existingModel && "fallbacks" in (existingModel as Record<string, unknown>)
? {
fallbacks: (existingModel as { fallbacks?: string[] }).fallbacks,
}
: undefined),
primary: MOONSHOT_DEFAULT_MODEL_REF,
},
},
},
};
} }
export function applyKimiCodeProviderConfig(cfg: OpenClawConfig): OpenClawConfig { export function applyKimiCodeProviderConfig(cfg: OpenClawConfig): OpenClawConfig {
@@ -394,47 +364,16 @@ export function applyXiaomiProviderConfig(cfg: OpenClawConfig): OpenClawConfig {
...models[XIAOMI_DEFAULT_MODEL_REF], ...models[XIAOMI_DEFAULT_MODEL_REF],
alias: models[XIAOMI_DEFAULT_MODEL_REF]?.alias ?? "Xiaomi", alias: models[XIAOMI_DEFAULT_MODEL_REF]?.alias ?? "Xiaomi",
}; };
const providers = { ...cfg.models?.providers };
const existingProvider = providers.xiaomi;
const defaultProvider = buildXiaomiProvider(); const defaultProvider = buildXiaomiProvider();
const existingModels = Array.isArray(existingProvider?.models) ? existingProvider.models : []; const resolvedApi = defaultProvider.api ?? "openai-completions";
const defaultModels = defaultProvider.models ?? []; return applyProviderConfigWithDefaultModels(cfg, {
const hasDefaultModel = existingModels.some((model) => model.id === XIAOMI_DEFAULT_MODEL_ID); agentModels: models,
const mergedModels = providerId: "xiaomi",
existingModels.length > 0 api: resolvedApi,
? hasDefaultModel
? existingModels
: [...existingModels, ...defaultModels]
: defaultModels;
const { apiKey: existingApiKey, ...existingProviderRest } = (existingProvider ?? {}) as Record<
string,
unknown
> as { apiKey?: string };
const resolvedApiKey = typeof existingApiKey === "string" ? existingApiKey : undefined;
const normalizedApiKey = resolvedApiKey?.trim();
providers.xiaomi = {
...existingProviderRest,
baseUrl: defaultProvider.baseUrl, baseUrl: defaultProvider.baseUrl,
api: defaultProvider.api, defaultModels: defaultProvider.models ?? [],
...(normalizedApiKey ? { apiKey: normalizedApiKey } : {}), defaultModelId: XIAOMI_DEFAULT_MODEL_ID,
models: mergedModels.length > 0 ? mergedModels : defaultProvider.models, });
};
return {
...cfg,
agents: {
...cfg.agents,
defaults: {
...cfg.agents?.defaults,
models,
},
},
models: {
mode: cfg.models?.mode ?? "merge",
providers,
},
};
} }
export function applyXiaomiConfig(cfg: OpenClawConfig): OpenClawConfig { export function applyXiaomiConfig(cfg: OpenClawConfig): OpenClawConfig {
@@ -780,53 +719,29 @@ export function applyQianfanProviderConfig(cfg: OpenClawConfig): OpenClawConfig
...models[QIANFAN_DEFAULT_MODEL_REF], ...models[QIANFAN_DEFAULT_MODEL_REF],
alias: models[QIANFAN_DEFAULT_MODEL_REF]?.alias ?? "QIANFAN", alias: models[QIANFAN_DEFAULT_MODEL_REF]?.alias ?? "QIANFAN",
}; };
const providers = { ...cfg.models?.providers };
const existingProvider = providers.qianfan;
const defaultProvider = buildQianfanProvider(); const defaultProvider = buildQianfanProvider();
const existingModels = Array.isArray(existingProvider?.models) ? existingProvider.models : []; const existingProvider = cfg.models?.providers?.qianfan as
const defaultModels = defaultProvider.models ?? []; | {
const hasDefaultModel = existingModels.some((model) => model.id === QIANFAN_DEFAULT_MODEL_ID); baseUrl?: unknown;
const mergedModels = api?: unknown;
existingModels.length > 0 }
? hasDefaultModel | undefined;
? existingModels const existingBaseUrl =
: [...existingModels, ...defaultModels] typeof existingProvider?.baseUrl === "string" ? existingProvider.baseUrl.trim() : "";
: defaultModels; const resolvedBaseUrl = existingBaseUrl || QIANFAN_BASE_URL;
const { const resolvedApi =
apiKey: existingApiKey, typeof existingProvider?.api === "string"
baseUrl: existingBaseUrl, ? (existingProvider.api as ModelApi)
api: existingApi, : "openai-completions";
...existingProviderRest
} = (existingProvider ?? {}) as Record<string, unknown> as {
apiKey?: string;
baseUrl?: string;
api?: ModelApi;
};
const resolvedApiKey = typeof existingApiKey === "string" ? existingApiKey : undefined;
const normalizedApiKey = resolvedApiKey?.trim();
providers.qianfan = {
...existingProviderRest,
baseUrl: existingBaseUrl ?? QIANFAN_BASE_URL,
api: existingApi ?? "openai-completions",
...(normalizedApiKey ? { apiKey: normalizedApiKey } : {}),
models: mergedModels.length > 0 ? mergedModels : defaultProvider.models,
};
return { return applyProviderConfigWithDefaultModels(cfg, {
...cfg, agentModels: models,
agents: { providerId: "qianfan",
...cfg.agents, api: resolvedApi,
defaults: { baseUrl: resolvedBaseUrl,
...cfg.agents?.defaults, defaultModels: defaultProvider.models ?? [],
models, defaultModelId: QIANFAN_DEFAULT_MODEL_ID,
}, });
},
models: {
mode: cfg.models?.mode ?? "merge",
providers,
},
};
} }
export function applyQianfanConfig(cfg: OpenClawConfig): OpenClawConfig { export function applyQianfanConfig(cfg: OpenClawConfig): OpenClawConfig {
+5 -19
View File
@@ -1,5 +1,8 @@
import type { OpenClawConfig } from "../config/config.js"; import type { OpenClawConfig } from "../config/config.js";
import { applyProviderConfigWithDefaultModel } from "./onboard-auth.config-shared.js"; import {
applyAgentDefaultModelPrimary,
applyProviderConfigWithDefaultModel,
} from "./onboard-auth.config-shared.js";
import { LITELLM_DEFAULT_MODEL_REF } from "./onboard-auth.credentials.js"; import { LITELLM_DEFAULT_MODEL_REF } from "./onboard-auth.credentials.js";
export const LITELLM_BASE_URL = "http://localhost:4000"; export const LITELLM_BASE_URL = "http://localhost:4000";
@@ -58,22 +61,5 @@ export function applyLitellmProviderConfig(cfg: OpenClawConfig): OpenClawConfig
export function applyLitellmConfig(cfg: OpenClawConfig): OpenClawConfig { export function applyLitellmConfig(cfg: OpenClawConfig): OpenClawConfig {
const next = applyLitellmProviderConfig(cfg); const next = applyLitellmProviderConfig(cfg);
const existingModel = next.agents?.defaults?.model; return applyAgentDefaultModelPrimary(next, LITELLM_DEFAULT_MODEL_REF);
return {
...next,
agents: {
...next.agents,
defaults: {
...next.agents?.defaults,
model: {
...(existingModel && "fallbacks" in (existingModel as Record<string, unknown>)
? {
fallbacks: (existingModel as { fallbacks?: string[] }).fallbacks,
}
: undefined),
primary: LITELLM_DEFAULT_MODEL_REF,
},
},
},
};
} }
+66 -6
View File
@@ -6,14 +6,45 @@ import type {
ModelProviderConfig, ModelProviderConfig,
} from "../config/types.models.js"; } from "../config/types.models.js";
export function applyProviderConfigWithDefaultModel( function extractAgentDefaultModelFallbacks(model: unknown): string[] | undefined {
if (!model || typeof model !== "object") {
return undefined;
}
if (!("fallbacks" in model)) {
return undefined;
}
const fallbacks = (model as { fallbacks?: unknown }).fallbacks;
return Array.isArray(fallbacks) ? fallbacks.map((v) => String(v)) : undefined;
}
export function applyAgentDefaultModelPrimary(
cfg: OpenClawConfig,
primary: string,
): OpenClawConfig {
const existingFallbacks = extractAgentDefaultModelFallbacks(cfg.agents?.defaults?.model);
return {
...cfg,
agents: {
...cfg.agents,
defaults: {
...cfg.agents?.defaults,
model: {
...(existingFallbacks ? { fallbacks: existingFallbacks } : undefined),
primary,
},
},
},
};
}
export function applyProviderConfigWithDefaultModels(
cfg: OpenClawConfig, cfg: OpenClawConfig,
params: { params: {
agentModels: Record<string, AgentModelEntryConfig>; agentModels: Record<string, AgentModelEntryConfig>;
providerId: string; providerId: string;
api: ModelApi; api: ModelApi;
baseUrl: string; baseUrl: string;
defaultModel: ModelDefinitionConfig; defaultModels: ModelDefinitionConfig[];
defaultModelId?: string; defaultModelId?: string;
}, },
): OpenClawConfig { ): OpenClawConfig {
@@ -24,9 +55,17 @@ export function applyProviderConfigWithDefaultModel(
? existingProvider.models ? existingProvider.models
: []; : [];
const defaultModelId = params.defaultModelId ?? params.defaultModel.id; const defaultModels = params.defaultModels;
const hasDefaultModel = existingModels.some((model) => model.id === defaultModelId); const defaultModelId = params.defaultModelId ?? defaultModels[0]?.id;
const mergedModels = hasDefaultModel ? existingModels : [...existingModels, params.defaultModel]; const hasDefaultModel = defaultModelId
? existingModels.some((model) => model.id === defaultModelId)
: true;
const mergedModels =
existingModels.length > 0
? hasDefaultModel || defaultModels.length === 0
? existingModels
: [...existingModels, ...defaultModels]
: defaultModels;
const { apiKey: existingApiKey, ...existingProviderRest } = (existingProvider ?? {}) as { const { apiKey: existingApiKey, ...existingProviderRest } = (existingProvider ?? {}) as {
apiKey?: string; apiKey?: string;
@@ -39,7 +78,7 @@ export function applyProviderConfigWithDefaultModel(
baseUrl: params.baseUrl, baseUrl: params.baseUrl,
api: params.api, api: params.api,
...(normalizedApiKey ? { apiKey: normalizedApiKey } : {}), ...(normalizedApiKey ? { apiKey: normalizedApiKey } : {}),
models: mergedModels.length > 0 ? mergedModels : [params.defaultModel], models: mergedModels.length > 0 ? mergedModels : defaultModels,
}; };
return { return {
@@ -57,3 +96,24 @@ export function applyProviderConfigWithDefaultModel(
}, },
}; };
} }
export function applyProviderConfigWithDefaultModel(
cfg: OpenClawConfig,
params: {
agentModels: Record<string, AgentModelEntryConfig>;
providerId: string;
api: ModelApi;
baseUrl: string;
defaultModel: ModelDefinitionConfig;
defaultModelId?: string;
},
): OpenClawConfig {
return applyProviderConfigWithDefaultModels(cfg, {
agentModels: params.agentModels,
providerId: params.providerId,
api: params.api,
baseUrl: params.baseUrl,
defaultModels: [params.defaultModel],
defaultModelId: params.defaultModelId ?? params.defaultModel.id,
});
}