feat(core): Preliminary changes for subagent model routing. (#16035)

This commit is contained in:
joshualitt
2026-01-07 13:21:10 -08:00
committed by GitHub
parent 17b3eb730a
commit a1dd19738e
12 changed files with 200 additions and 15 deletions

View File

@@ -65,6 +65,7 @@ export interface _ResolvedModelConfig {
export class ModelConfigService {
private readonly runtimeAliases: Record<string, ModelConfigAlias> = {};
private readonly runtimeOverrides: ModelConfigOverride[] = [];
// TODO(12597): Process config to build a typed alias hierarchy.
constructor(private readonly config: ModelConfigServiceConfig) {}
@@ -73,6 +74,10 @@ export class ModelConfigService {
this.runtimeAliases[aliasName] = alias;
}
registerRuntimeModelOverride(override: ModelConfigOverride): void {
this.runtimeOverrides.push(override);
}
private resolveAlias(
aliasName: string,
aliases: Record<string, ModelConfigAlias>,
@@ -123,7 +128,11 @@ export class ModelConfigService {
...customAliases,
...this.runtimeAliases,
};
const allOverrides = [...overrides, ...customOverrides];
const allOverrides = [
...overrides,
...customOverrides,
...this.runtimeOverrides,
];
let baseModel: string | undefined = context.model;
let resolvedConfig: GenerateContentConfig = {};