diff options
Diffstat (limited to 'packages/core/src/utils')
| -rw-r--r-- | packages/core/src/utils/flashFallback.integration.test.ts | 4 | ||||
| -rw-r--r-- | packages/core/src/utils/retry.ts | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/packages/core/src/utils/flashFallback.integration.test.ts b/packages/core/src/utils/flashFallback.integration.test.ts index cffe9aa5..6554425f 100644 --- a/packages/core/src/utils/flashFallback.integration.test.ts +++ b/packages/core/src/utils/flashFallback.integration.test.ts @@ -78,14 +78,14 @@ describe('Flash Fallback Integration', () => { return status === 429; }, onPersistent429: mockFallbackHandler, - authType: AuthType.LOGIN_WITH_GOOGLE_PERSONAL, + authType: AuthType.LOGIN_WITH_GOOGLE, }); // Verify fallback was triggered expect(fallbackCalled).toBe(true); expect(fallbackModel).toBe(DEFAULT_GEMINI_FLASH_MODEL); expect(mockFallbackHandler).toHaveBeenCalledWith( - AuthType.LOGIN_WITH_GOOGLE_PERSONAL, + AuthType.LOGIN_WITH_GOOGLE, ); expect(result).toBe('success after fallback'); // Should have: 2 failures, then fallback triggered, then 1 success after retry reset diff --git a/packages/core/src/utils/retry.ts b/packages/core/src/utils/retry.ts index 372a7976..f3f5f2d2 100644 --- a/packages/core/src/utils/retry.ts +++ b/packages/core/src/utils/retry.ts @@ -97,7 +97,7 @@ export async function retryWithBackoff<T>( if ( consecutive429Count >= 2 && onPersistent429 && - authType === AuthType.LOGIN_WITH_GOOGLE_PERSONAL + authType === AuthType.LOGIN_WITH_GOOGLE ) { try { const fallbackModel = await onPersistent429(authType); |