diff --git a/packages/cli/src/ui/components/ProQuotaDialog.test.tsx b/packages/cli/src/ui/components/ProQuotaDialog.test.tsx index 42746678e2..2520036246 100644 --- a/packages/cli/src/ui/components/ProQuotaDialog.test.tsx +++ b/packages/cli/src/ui/components/ProQuotaDialog.test.tsx @@ -169,7 +169,7 @@ describe('ProQuotaDialog', () => { }); describe('when it is a capacity error', () => { - it('should render keep trying and stop options', () => { + it('should render keep trying, switch, and stop options', () => { const { unmount } = render( { value: 'retry_once', key: 'retry_once', }, + { + label: 'Switch to gemini-2.5-flash', + value: 'retry_always', + key: 'retry_always', + }, { label: 'Stop', value: 'retry_later', key: 'retry_later' }, ], }), diff --git a/packages/cli/src/ui/components/ProQuotaDialog.tsx b/packages/cli/src/ui/components/ProQuotaDialog.tsx index cda3937a7f..0dbf134c5b 100644 --- a/packages/cli/src/ui/components/ProQuotaDialog.tsx +++ b/packages/cli/src/ui/components/ProQuotaDialog.tsx @@ -91,6 +91,11 @@ export function ProQuotaDialog({ value: 'retry_once' as const, key: 'retry_once', }, + { + label: `Switch to ${fallbackModel}`, + value: 'retry_always' as const, + key: 'retry_always', + }, { label: 'Stop', value: 'retry_later' as const,