diff --git a/src/agents/live-model-filter.ts b/src/agents/live-model-filter.ts index 32d4b94a5aa..efce653db2b 100644 --- a/src/agents/live-model-filter.ts +++ b/src/agents/live-model-filter.ts @@ -35,7 +35,12 @@ const HIGH_SIGNAL_LIVE_MODEL_PRIORITY = [ export const DEFAULT_HIGH_SIGNAL_LIVE_MODEL_LIMIT = HIGH_SIGNAL_LIVE_MODEL_PRIORITY.length; const DEFAULT_HIGH_SIGNAL_LIVE_EXCLUDED_PROVIDERS = new Set(["codex", "codex-cli", "openai-codex"]); -const CURATED_ONLY_HIGH_SIGNAL_LIVE_PROVIDERS = new Set(["fireworks", "openrouter", "xai"]); +const CURATED_ONLY_HIGH_SIGNAL_LIVE_PROVIDERS = new Set([ + "fireworks", + "google", + "openrouter", + "xai", +]); const HIGH_SIGNAL_LIVE_MODEL_PRIORITY_INDEX = new Map( HIGH_SIGNAL_LIVE_MODEL_PRIORITY.map((key, index) => [key, index]), diff --git a/src/agents/model-compat.test.ts b/src/agents/model-compat.test.ts index 8e7a7eba8a1..a3222dc3f99 100644 --- a/src/agents/model-compat.test.ts +++ b/src/agents/model-compat.test.ts @@ -462,7 +462,7 @@ describe("isHighSignalLiveModelRef", () => { ).toBe(false); }); - it("drops Gemini families older than major version 3 from the default live matrix", () => { + it("keeps only curated Gemini routes in the default live matrix", () => { providerRuntimeMocks.resolveProviderModernModelRef.mockReturnValue(true); expect(isHighSignalLiveModelRef({ provider: "google", id: "gemini-2.5-flash-lite" })).toBe( @@ -474,6 +474,13 @@ describe("isHighSignalLiveModelRef", () => { expect(isHighSignalLiveModelRef({ provider: "google", id: "gemini-3-flash-preview" })).toBe( true, ); + expect(isHighSignalLiveModelRef({ provider: "google", id: "gemini-3-pro-preview" })).toBe( + false, + ); + expect( + isHighSignalLiveModelRef({ provider: "google", id: "gemini-3.1-pro-preview-customtools" }), + ).toBe(false); + expect(isHighSignalLiveModelRef({ provider: "google", id: "gemma-4-31b-it" })).toBe(false); expect(isHighSignalLiveModelRef({ provider: "google", id: "gemini-flash-latest" })).toBe(false); expect(isHighSignalLiveModelRef({ provider: "google", id: "gemini-flash-lite-latest" })).toBe( false,