diff --git a/packages/core/src/availability/policyCatalog.test.ts b/packages/core/src/availability/policyCatalog.test.ts index 04e35018d5..a926a6ea2f 100644 --- a/packages/core/src/availability/policyCatalog.test.ts +++ b/packages/core/src/availability/policyCatalog.test.ts @@ -11,6 +11,8 @@ import { validateModelPolicyChain, } from './policyCatalog.js'; import { + DEFAULT_GEMINI_FLASH_LITE_MODEL, + DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_MODEL, PREVIEW_GEMINI_3_1_CUSTOM_TOOLS_MODEL, PREVIEW_GEMINI_3_1_MODEL, @@ -50,7 +52,9 @@ describe('policyCatalog', () => { it('returns default chain when preview disabled', () => { const chain = getModelPolicyChain({ previewEnabled: false }); expect(chain[0]?.model).toBe(DEFAULT_GEMINI_MODEL); - expect(chain).toHaveLength(2); + expect(chain[1]?.model).toBe(DEFAULT_GEMINI_FLASH_MODEL); + expect(chain[2]?.model).toBe(DEFAULT_GEMINI_FLASH_LITE_MODEL); + expect(chain).toHaveLength(3); }); it('marks preview transients as sticky retries when auto-selected', () => { diff --git a/packages/core/src/availability/policyCatalog.ts b/packages/core/src/availability/policyCatalog.ts index a5694e94b8..811c81fa87 100644 --- a/packages/core/src/availability/policyCatalog.ts +++ b/packages/core/src/availability/policyCatalog.ts @@ -122,6 +122,10 @@ export function getModelPolicyChain( }), definePolicy({ model: DEFAULT_GEMINI_FLASH_MODEL, + maxAttempts: 10, + }), + definePolicy({ + model: DEFAULT_GEMINI_FLASH_LITE_MODEL, isLastResort: true, maxAttempts: 10, }),