mirror of
https://github.com/farcasclaudiu/openclaw.git
synced 2026-06-22 07:01:44 +03:00
refactor(commands): dedupe provider config + default model helpers
This commit is contained in:
@@ -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 {
|
||||||
|
|||||||
@@ -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,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
},
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user