diff --git a/packages/core/src/availability/policyCatalog.test.ts b/packages/core/src/availability/policyCatalog.test.ts index 04e35018d5..d57d9c139f 100644 --- a/packages/core/src/availability/policyCatalog.test.ts +++ b/packages/core/src/availability/policyCatalog.test.ts @@ -50,7 +50,7 @@ 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).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, }),