feat(core): Add support for custom aliases for model configs. (#13546)

This commit is contained in:
joshualitt
2025-11-21 16:13:10 -08:00
committed by GitHub
parent 1e715d1e5c
commit 42c2e1b217
5 changed files with 151 additions and 2 deletions

View File

@@ -43,6 +43,7 @@ export interface ModelConfigAlias {
export interface ModelConfigServiceConfig {
aliases?: Record<string, ModelConfigAlias>;
customAliases?: Record<string, ModelConfigAlias>;
overrides?: ModelConfigOverride[];
}
@@ -104,8 +105,12 @@ export class ModelConfigService {
generateContentConfig: GenerateContentConfig;
} {
const config = this.config || {};
const { aliases = {}, overrides = [] } = config;
const allAliases = { ...aliases, ...this.runtimeAliases };
const { aliases = {}, customAliases = {}, overrides = [] } = config;
const allAliases = {
...aliases,
...customAliases,
...this.runtimeAliases,
};
let baseModel: string | undefined = context.model;
let resolvedConfig: GenerateContentConfig = {};