@google/gemini-cli-core 0.1.0-development.1

package/dist/src/utils/retry.test.js

@@ -0,0 +1,322 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { retryWithBackoff } from './retry.js';
+import { setSimulate429 } from './testUtils.js';
+// Helper to create a mock function that fails a certain number of times
+const createFailingFunction = (failures, successValue = 'success') => {
+    let attempts = 0;
+    return vi.fn(async () => {
+        attempts++;
+        if (attempts <= failures) {
+            // Simulate a retryable error
+            const error = new Error(`Simulated error attempt ${attempts}`);
+            error.status = 500; // Simulate a server error
+            throw error;
+        }
+        return successValue;
+    });
+};
+// Custom error for testing non-retryable conditions
+class NonRetryableError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = 'NonRetryableError';
+    }
+}
+describe('retryWithBackoff', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+        // Disable 429 simulation for tests
+        setSimulate429(false);
+        // Suppress unhandled promise rejection warnings for tests that expect errors
+        console.warn = vi.fn();
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+        vi.useRealTimers();
+    });
+    it('should return the result on the first attempt if successful', async () => {
+        const mockFn = createFailingFunction(0);
+        const result = await retryWithBackoff(mockFn);
+        expect(result).toBe('success');
+        expect(mockFn).toHaveBeenCalledTimes(1);
+    });
+    it('should retry and succeed if failures are within maxAttempts', async () => {
+        const mockFn = createFailingFunction(2);
+        const promise = retryWithBackoff(mockFn, {
+            maxAttempts: 3,
+            initialDelayMs: 10,
+        });
+        await vi.runAllTimersAsync(); // Ensure all delays and retries complete
+        const result = await promise;
+        expect(result).toBe('success');
+        expect(mockFn).toHaveBeenCalledTimes(3);
+    });
+    it('should throw an error if all attempts fail', async () => {
+        const mockFn = createFailingFunction(3);
+        // 1. Start the retryable operation, which returns a promise.
+        const promise = retryWithBackoff(mockFn, {
+            maxAttempts: 3,
+            initialDelayMs: 10,
+        });
+        // 2. IMPORTANT: Attach the rejection expectation to the promise *immediately*.
+        //    This ensures a 'catch' handler is present before the promise can reject.
+        //    The result is a new promise that resolves when the assertion is met.
+        const assertionPromise = expect(promise).rejects.toThrow('Simulated error attempt 3');
+        // 3. Now, advance the timers. This will trigger the retries and the
+        //    eventual rejection. The handler attached in step 2 will catch it.
+        await vi.runAllTimersAsync();
+        // 4. Await the assertion promise itself to ensure the test was successful.
+        await assertionPromise;
+        // 5. Finally, assert the number of calls.
+        expect(mockFn).toHaveBeenCalledTimes(3);
+    });
+    it('should not retry if shouldRetry returns false', async () => {
+        const mockFn = vi.fn(async () => {
+            throw new NonRetryableError('Non-retryable error');
+        });
+        const shouldRetry = (error) => !(error instanceof NonRetryableError);
+        const promise = retryWithBackoff(mockFn, {
+            shouldRetry,
+            initialDelayMs: 10,
+        });
+        await expect(promise).rejects.toThrow('Non-retryable error');
+        expect(mockFn).toHaveBeenCalledTimes(1);
+    });
+    it('should use default shouldRetry if not provided, retrying on 429', async () => {
+        const mockFn = vi.fn(async () => {
+            const error = new Error('Too Many Requests');
+            error.status = 429;
+            throw error;
+        });
+        const promise = retryWithBackoff(mockFn, {
+            maxAttempts: 2,
+            initialDelayMs: 10,
+        });
+        // Attach the rejection expectation *before* running timers
+        const assertionPromise = expect(promise).rejects.toThrow('Too Many Requests');
+        // Run timers to trigger retries and eventual rejection
+        await vi.runAllTimersAsync();
+        // Await the assertion
+        await assertionPromise;
+        expect(mockFn).toHaveBeenCalledTimes(2);
+    });
+    it('should use default shouldRetry if not provided, not retrying on 400', async () => {
+        const mockFn = vi.fn(async () => {
+            const error = new Error('Bad Request');
+            error.status = 400;
+            throw error;
+        });
+        const promise = retryWithBackoff(mockFn, {
+            maxAttempts: 2,
+            initialDelayMs: 10,
+        });
+        await expect(promise).rejects.toThrow('Bad Request');
+        expect(mockFn).toHaveBeenCalledTimes(1);
+    });
+    it('should respect maxDelayMs', async () => {
+        const mockFn = createFailingFunction(3);
+        const setTimeoutSpy = vi.spyOn(global, 'setTimeout');
+        const promise = retryWithBackoff(mockFn, {
+            maxAttempts: 4,
+            initialDelayMs: 100,
+            maxDelayMs: 250, // Max delay is less than 100 * 2 * 2 = 400
+        });
+        await vi.advanceTimersByTimeAsync(1000); // Advance well past all delays
+        await promise;
+        const delays = setTimeoutSpy.mock.calls.map((call) => call[1]);
+        // Delays should be around initial, initial*2, maxDelay (due to cap)
+        // Jitter makes exact assertion hard, so we check ranges / caps
+        expect(delays.length).toBe(3);
+        expect(delays[0]).toBeGreaterThanOrEqual(100 * 0.7);
+        expect(delays[0]).toBeLessThanOrEqual(100 * 1.3);
+        expect(delays[1]).toBeGreaterThanOrEqual(200 * 0.7);
+        expect(delays[1]).toBeLessThanOrEqual(200 * 1.3);
+        // The third delay should be capped by maxDelayMs (250ms), accounting for jitter
+        expect(delays[2]).toBeGreaterThanOrEqual(250 * 0.7);
+        expect(delays[2]).toBeLessThanOrEqual(250 * 1.3);
+    });
+    it('should handle jitter correctly, ensuring varied delays', async () => {
+        let mockFn = createFailingFunction(5);
+        const setTimeoutSpy = vi.spyOn(global, 'setTimeout');
+        // Run retryWithBackoff multiple times to observe jitter
+        const runRetry = () => retryWithBackoff(mockFn, {
+            maxAttempts: 2, // Only one retry, so one delay
+            initialDelayMs: 100,
+            maxDelayMs: 1000,
+        });
+        // We expect rejections as mockFn fails 5 times
+        const promise1 = runRetry();
+        // Attach the rejection expectation *before* running timers
+        const assertionPromise1 = expect(promise1).rejects.toThrow();
+        await vi.runAllTimersAsync(); // Advance for the delay in the first runRetry
+        await assertionPromise1;
+        const firstDelaySet = setTimeoutSpy.mock.calls.map((call) => call[1]);
+        setTimeoutSpy.mockClear(); // Clear calls for the next run
+        // Reset mockFn to reset its internal attempt counter for the next run
+        mockFn = createFailingFunction(5); // Re-initialize with 5 failures
+        const promise2 = runRetry();
+        // Attach the rejection expectation *before* running timers
+        const assertionPromise2 = expect(promise2).rejects.toThrow();
+        await vi.runAllTimersAsync(); // Advance for the delay in the second runRetry
+        await assertionPromise2;
+        const secondDelaySet = setTimeoutSpy.mock.calls.map((call) => call[1]);
+        // Check that the delays are not exactly the same due to jitter
+        // This is a probabilistic test, but with +/-30% jitter, it's highly likely they differ.
+        if (firstDelaySet.length > 0 && secondDelaySet.length > 0) {
+            // Check the first delay of each set
+            expect(firstDelaySet[0]).not.toBe(secondDelaySet[0]);
+        }
+        else {
+            // If somehow no delays were captured (e.g. test setup issue), fail explicitly
+            throw new Error('Delays were not captured for jitter test');
+        }
+        // Ensure delays are within the expected jitter range [70, 130] for initialDelayMs = 100
+        [...firstDelaySet, ...secondDelaySet].forEach((d) => {
+            expect(d).toBeGreaterThanOrEqual(100 * 0.7);
+            expect(d).toBeLessThanOrEqual(100 * 1.3);
+        });
+    });
+    describe('Flash model fallback for OAuth users', () => {
+        it('should trigger fallback for OAuth personal users after persistent 429 errors', async () => {
+            const fallbackCallback = vi.fn().mockResolvedValue('gemini-2.5-flash');
+            let fallbackOccurred = false;
+            const mockFn = vi.fn().mockImplementation(async () => {
+                if (!fallbackOccurred) {
+                    const error = new Error('Rate limit exceeded');
+                    error.status = 429;
+                    throw error;
+                }
+                return 'success';
+            });
+            const promise = retryWithBackoff(mockFn, {
+                maxAttempts: 3,
+                initialDelayMs: 100,
+                onPersistent429: async (authType) => {
+                    fallbackOccurred = true;
+                    return await fallbackCallback(authType);
+                },
+                authType: 'oauth-personal',
+            });
+            // Advance all timers to complete retries
+            await vi.runAllTimersAsync();
+            // Should succeed after fallback
+            await expect(promise).resolves.toBe('success');
+            // Verify callback was called with correct auth type
+            expect(fallbackCallback).toHaveBeenCalledWith('oauth-personal');
+            // Should retry again after fallback
+            expect(mockFn).toHaveBeenCalledTimes(3); // 2 initial attempts + 1 after fallback
+        });
+        it('should NOT trigger fallback for API key users', async () => {
+            const fallbackCallback = vi.fn();
+            const mockFn = vi.fn(async () => {
+                const error = new Error('Rate limit exceeded');
+                error.status = 429;
+                throw error;
+            });
+            const promise = retryWithBackoff(mockFn, {
+                maxAttempts: 3,
+                initialDelayMs: 100,
+                onPersistent429: fallbackCallback,
+                authType: 'gemini-api-key',
+            });
+            // Handle the promise properly to avoid unhandled rejections
+            const resultPromise = promise.catch((error) => error);
+            await vi.runAllTimersAsync();
+            const result = await resultPromise;
+            // Should fail after all retries without fallback
+            expect(result).toBeInstanceOf(Error);
+            expect(result.message).toBe('Rate limit exceeded');
+            // Callback should not be called for API key users
+            expect(fallbackCallback).not.toHaveBeenCalled();
+        });
+        it('should reset attempt counter and continue after successful fallback', async () => {
+            let fallbackCalled = false;
+            const fallbackCallback = vi.fn().mockImplementation(async () => {
+                fallbackCalled = true;
+                return 'gemini-2.5-flash';
+            });
+            const mockFn = vi.fn().mockImplementation(async () => {
+                if (!fallbackCalled) {
+                    const error = new Error('Rate limit exceeded');
+                    error.status = 429;
+                    throw error;
+                }
+                return 'success';
+            });
+            const promise = retryWithBackoff(mockFn, {
+                maxAttempts: 3,
+                initialDelayMs: 100,
+                onPersistent429: fallbackCallback,
+                authType: 'oauth-personal',
+            });
+            await vi.runAllTimersAsync();
+            await expect(promise).resolves.toBe('success');
+            expect(fallbackCallback).toHaveBeenCalledOnce();
+        });
+        it('should continue with original error if fallback is rejected', async () => {
+            const fallbackCallback = vi.fn().mockResolvedValue(null); // User rejected fallback
+            const mockFn = vi.fn(async () => {
+                const error = new Error('Rate limit exceeded');
+                error.status = 429;
+                throw error;
+            });
+            const promise = retryWithBackoff(mockFn, {
+                maxAttempts: 3,
+                initialDelayMs: 100,
+                onPersistent429: fallbackCallback,
+                authType: 'oauth-personal',
+            });
+            // Handle the promise properly to avoid unhandled rejections
+            const resultPromise = promise.catch((error) => error);
+            await vi.runAllTimersAsync();
+            const result = await resultPromise;
+            // Should fail with original error when fallback is rejected
+            expect(result).toBeInstanceOf(Error);
+            expect(result.message).toBe('Rate limit exceeded');
+            expect(fallbackCallback).toHaveBeenCalledWith('oauth-personal');
+        });
+        it('should handle mixed error types (only count consecutive 429s)', async () => {
+            const fallbackCallback = vi.fn().mockResolvedValue('gemini-2.5-flash');
+            let attempts = 0;
+            let fallbackOccurred = false;
+            const mockFn = vi.fn().mockImplementation(async () => {
+                attempts++;
+                if (fallbackOccurred) {
+                    return 'success';
+                }
+                if (attempts === 1) {
+                    // First attempt: 500 error (resets consecutive count)
+                    const error = new Error('Server error');
+                    error.status = 500;
+                    throw error;
+                }
+                else {
+                    // Remaining attempts: 429 errors
+                    const error = new Error('Rate limit exceeded');
+                    error.status = 429;
+                    throw error;
+                }
+            });
+            const promise = retryWithBackoff(mockFn, {
+                maxAttempts: 5,
+                initialDelayMs: 100,
+                onPersistent429: async (authType) => {
+                    fallbackOccurred = true;
+                    return await fallbackCallback(authType);
+                },
+                authType: 'oauth-personal',
+            });
+            await vi.runAllTimersAsync();
+            await expect(promise).resolves.toBe('success');
+            // Should trigger fallback after 2 consecutive 429s (attempts 2-3)
+            expect(fallbackCallback).toHaveBeenCalledWith('oauth-personal');
+        });
+    });
+});
+//# sourceMappingURL=retry.test.js.map

package/dist/src/utils/retry.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry.test.js","sourceRoot":"","sources":["../../../src/utils/retry.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,uDAAuD;AACvD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAC9C,OAAO,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAOhD,wEAAwE;AACxE,MAAM,qBAAqB,GAAG,CAC5B,QAAgB,EAChB,eAAuB,SAAS,EAChC,EAAE;IACF,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,OAAO,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;QACtB,QAAQ,EAAE,CAAC;QACX,IAAI,QAAQ,IAAI,QAAQ,EAAE,CAAC;YACzB,6BAA6B;YAC7B,MAAM,KAAK,GAAc,IAAI,KAAK,CAAC,2BAA2B,QAAQ,EAAE,CAAC,CAAC;YAC1E,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,0BAA0B;YAC9C,MAAM,KAAK,CAAC;QACd,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,oDAAoD;AACpD,MAAM,iBAAkB,SAAQ,KAAK;IACnC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,mBAAmB,CAAC;IAClC,CAAC;CACF;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,mCAAmC;QACnC,cAAc,CAAC,KAAK,CAAC,CAAC;QACtB,6EAA6E;QAC7E,OAAO,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;YACvC,WAAW,EAAE,CAAC;YACd,cAAc,EAAE,EAAE;SACnB,CAAC,CAAC;QAEH,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,yCAAyC;QAEvE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAExC,6DAA6D;QAC7D,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;YACvC,WAAW,EAAE,CAAC;YACd,cAAc,EAAE,EAAE;SACnB,CAAC,CAAC;QAEH,+EAA+E;QAC/E,8EAA8E;QAC9E,0EAA0E;QAC1E,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CACtD,2BAA2B,CAC5B,CAAC;QAEF,oEAAoE;QACpE,uEAAuE;QACvE,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAE7B,2EAA2E;QAC3E,MAAM,gBAAgB,CAAC;QAEvB,0CAA0C;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,IAAI,iBAAiB,CAAC,qBAAqB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,CAAC,KAAY,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,YAAY,iBAAiB,CAAC,CAAC;QAE5E,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;YACvC,WAAW;YACX,cAAc,EAAE,EAAE;SACnB,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,mBAAmB,CAAQ,CAAC;YACpD,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;YACnB,MAAM,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;YACvC,WAAW,EAAE,CAAC;YACd,cAAc,EAAE,EAAE;SACnB,CAAC,CAAC;QAEH,2DAA2D;QAC3D,MAAM,gBAAgB,GACpB,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAEvD,uDAAuD;QACvD,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAE7B,sBAAsB;QACtB,MAAM,gBAAgB,CAAC;QAEvB,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,aAAa,CAAQ,CAAC;YAC9C,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;YACnB,MAAM,KAAK,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;YACvC,WAAW,EAAE,CAAC;YACd,cAAc,EAAE,EAAE;SACnB,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,aAAa,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAErD,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;YACvC,WAAW,EAAE,CAAC;YACd,cAAc,EAAE,GAAG;YACnB,UAAU,EAAE,GAAG,EAAE,2CAA2C;SAC7D,CAAC,CAAC;QAEH,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,+BAA+B;QACxE,MAAM,OAAO,CAAC;QAEd,MAAM,MAAM,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAW,CAAC,CAAC;QAEzE,oEAAoE;QACpE,+DAA+D;QAC/D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QACjD,gFAAgF;QAChF,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,IAAI,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACtC,MAAM,aAAa,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAErD,wDAAwD;QACxD,MAAM,QAAQ,GAAG,GAAG,EAAE,CACpB,gBAAgB,CAAC,MAAM,EAAE;YACvB,WAAW,EAAE,CAAC,EAAE,+BAA+B;YAC/C,cAAc,EAAE,GAAG;YACnB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QAEL,+CAA+C;QAC/C,MAAM,QAAQ,GAAG,QAAQ,EAAE,CAAC;QAC5B,2DAA2D;QAC3D,MAAM,iBAAiB,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC7D,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,8CAA8C;QAC5E,MAAM,iBAAiB,CAAC;QAExB,MAAM,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAChD,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAW,CAC5B,CAAC;QACF,aAAa,CAAC,SAAS,EAAE,CAAC,CAAC,+BAA+B;QAE1D,sEAAsE;QACtE,MAAM,GAAG,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,gCAAgC;QAEnE,MAAM,QAAQ,GAAG,QAAQ,EAAE,CAAC;QAC5B,2DAA2D;QAC3D,MAAM,iBAAiB,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC7D,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC,+CAA+C;QAC7E,MAAM,iBAAiB,CAAC;QAExB,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CACjD,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAW,CAC5B,CAAC;QAEF,+DAA+D;QAC/D,wFAAwF;QACxF,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1D,oCAAoC;YACpC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,8EAA8E;YAC9E,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC9D,CAAC;QAED,wFAAwF;QACxF,CAAC,GAAG,aAAa,EAAE,GAAG,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YAClD,MAAM,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;YAC5C,MAAM,CAAC,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;QACpD,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAEvE,IAAI,gBAAgB,GAAG,KAAK,CAAC;YAC7B,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACnD,IAAI,CAAC,gBAAgB,EAAE,CAAC;oBACtB,MAAM,KAAK,GAAc,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;oBAC1D,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;oBACnB,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;gBACvC,WAAW,EAAE,CAAC;gBACd,cAAc,EAAE,GAAG;gBACnB,eAAe,EAAE,KAAK,EAAE,QAAiB,EAAE,EAAE;oBAC3C,gBAAgB,GAAG,IAAI,CAAC;oBACxB,OAAO,MAAM,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAC1C,CAAC;gBACD,QAAQ,EAAE,gBAAgB;aAC3B,CAAC,CAAC;YAEH,yCAAyC;YACzC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAE7B,gCAAgC;YAChC,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAE/C,oDAAoD;YACpD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;YAEhE,oCAAoC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,wCAAwC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YAEjC,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;gBAC9B,MAAM,KAAK,GAAc,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;gBAC1D,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;gBACnB,MAAM,KAAK,CAAC;YACd,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;gBACvC,WAAW,EAAE,CAAC;gBACd,cAAc,EAAE,GAAG;gBACnB,eAAe,EAAE,gBAAgB;gBACjC,QAAQ,EAAE,gBAAgB;aAC3B,CAAC,CAAC;YAEH,4DAA4D;YAC5D,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;YACtD,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;YAEnC,iDAAiD;YACjD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YAEnD,kDAAkD;YAClD,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,IAAI,cAAc,GAAG,KAAK,CAAC;YAC3B,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBAC7D,cAAc,GAAG,IAAI,CAAC;gBACtB,OAAO,kBAAkB,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACnD,IAAI,CAAC,cAAc,EAAE,CAAC;oBACpB,MAAM,KAAK,GAAc,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;oBAC1D,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;oBACnB,MAAM,KAAK,CAAC;gBACd,CAAC;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;gBACvC,WAAW,EAAE,CAAC;gBACd,cAAc,EAAE,GAAG;gBACnB,eAAe,EAAE,gBAAgB;gBACjC,QAAQ,EAAE,gBAAgB;aAC3B,CAAC,CAAC;YAEH,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAE7B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/C,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,yBAAyB;YAEnF,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;gBAC9B,MAAM,KAAK,GAAc,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;gBAC1D,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;gBACnB,MAAM,KAAK,CAAC;YACd,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;gBACvC,WAAW,EAAE,CAAC;gBACd,cAAc,EAAE,GAAG;gBACnB,eAAe,EAAE,gBAAgB;gBACjC,QAAQ,EAAE,gBAAgB;aAC3B,CAAC,CAAC;YAEH,4DAA4D;YAC5D,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;YACtD,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;YAEnC,4DAA4D;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACnD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YACvE,IAAI,QAAQ,GAAG,CAAC,CAAC;YACjB,IAAI,gBAAgB,GAAG,KAAK,CAAC;YAE7B,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;gBACnD,QAAQ,EAAE,CAAC;gBACX,IAAI,gBAAgB,EAAE,CAAC;oBACrB,OAAO,SAAS,CAAC;gBACnB,CAAC;gBACD,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;oBACnB,sDAAsD;oBACtD,MAAM,KAAK,GAAc,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC;oBACnD,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;oBACnB,MAAM,KAAK,CAAC;gBACd,CAAC;qBAAM,CAAC;oBACN,iCAAiC;oBACjC,MAAM,KAAK,GAAc,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;oBAC1D,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;oBACnB,MAAM,KAAK,CAAC;gBACd,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,gBAAgB,CAAC,MAAM,EAAE;gBACvC,WAAW,EAAE,CAAC;gBACd,cAAc,EAAE,GAAG;gBACnB,eAAe,EAAE,KAAK,EAAE,QAAiB,EAAE,EAAE;oBAC3C,gBAAgB,GAAG,IAAI,CAAC;oBACxB,OAAO,MAAM,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAC1C,CAAC;gBACD,QAAQ,EAAE,gBAAgB;aAC3B,CAAC,CAAC;YAEH,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAC;YAE7B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAE/C,kEAAkE;YAClE,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/utils/schemaValidator.d.ts

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Simple utility to validate objects against JSON Schemas
+ */
+export declare class SchemaValidator {
+    /**
+     * Validates data against a JSON schema
+     * @param schema JSON Schema to validate against
+     * @param data Data to validate
+     * @returns True if valid, false otherwise
+     */
+    static validate(schema: Record<string, unknown>, data: unknown): boolean;
+}

package/dist/src/utils/schemaValidator.js

@@ -0,0 +1,50 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Simple utility to validate objects against JSON Schemas
+ */
+export class SchemaValidator {
+    /**
+     * Validates data against a JSON schema
+     * @param schema JSON Schema to validate against
+     * @param data Data to validate
+     * @returns True if valid, false otherwise
+     */
+    static validate(schema, data) {
+        // This is a simplified implementation
+        // In a real application, you would use a library like Ajv for proper validation
+        // Check for required fields
+        if (schema.required && Array.isArray(schema.required)) {
+            const required = schema.required;
+            const dataObj = data;
+            for (const field of required) {
+                if (dataObj[field] === undefined) {
+                    console.error(`Missing required field: ${field}`);
+                    return false;
+                }
+            }
+        }
+        // Check property types if properties are defined
+        if (schema.properties && typeof schema.properties === 'object') {
+            const properties = schema.properties;
+            const dataObj = data;
+            for (const [key, prop] of Object.entries(properties)) {
+                if (dataObj[key] !== undefined && prop.type) {
+                    const expectedType = prop.type;
+                    const actualType = Array.isArray(dataObj[key])
+                        ? 'array'
+                        : typeof dataObj[key];
+                    if (expectedType !== actualType) {
+                        console.error(`Type mismatch for property "${key}": expected ${expectedType}, got ${actualType}`);
+                        return false;
+                    }
+                }
+            }
+        }
+        return true;
+    }
+}
+//# sourceMappingURL=schemaValidator.js.map

package/dist/src/utils/schemaValidator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemaValidator.js","sourceRoot":"","sources":["../../../src/utils/schemaValidator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AACH,MAAM,OAAO,eAAe;IAC1B;;;;;OAKG;IACH,MAAM,CAAC,QAAQ,CAAC,MAA+B,EAAE,IAAa;QAC5D,sCAAsC;QACtC,gFAAgF;QAEhF,4BAA4B;QAC5B,IAAI,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YACtD,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAoB,CAAC;YAC7C,MAAM,OAAO,GAAG,IAA+B,CAAC;YAEhD,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;gBAC7B,IAAI,OAAO,CAAC,KAAK,CAAC,KAAK,SAAS,EAAE,CAAC;oBACjC,OAAO,CAAC,KAAK,CAAC,2BAA2B,KAAK,EAAE,CAAC,CAAC;oBAClD,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;QACH,CAAC;QAED,iDAAiD;QACjD,IAAI,MAAM,CAAC,UAAU,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;YAC/D,MAAM,UAAU,GAAG,MAAM,CAAC,UAA+C,CAAC;YAC1E,MAAM,OAAO,GAAG,IAA+B,CAAC;YAEhD,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;gBACrD,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC;oBAC/B,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;wBAC5C,CAAC,CAAC,OAAO;wBACT,CAAC,CAAC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;oBAExB,IAAI,YAAY,KAAK,UAAU,EAAE,CAAC;wBAChC,OAAO,CAAC,KAAK,CACX,+BAA+B,GAAG,eAAe,YAAY,SAAS,UAAU,EAAE,CACnF,CAAC;wBACF,OAAO,KAAK,CAAC;oBACf,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}

package/dist/src/utils/session.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const sessionId: `${string}-${string}-${string}-${string}-${string}`;

package/dist/src/utils/session.js

@@ -0,0 +1,8 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { randomUUID } from 'crypto';
+export const sessionId = randomUUID();
+//# sourceMappingURL=session.js.map

package/dist/src/utils/session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../src/utils/session.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,MAAM,CAAC,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC"}

package/dist/src/utils/testUtils.d.ts

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Check if we should simulate a 429 error for the current request
+ */
+export declare function shouldSimulate429(authType?: string): boolean;
+/**
+ * Reset the request counter (useful for tests)
+ */
+export declare function resetRequestCounter(): void;
+/**
+ * Disable 429 simulation after successful fallback
+ */
+export declare function disableSimulationAfterFallback(): void;
+/**
+ * Create a simulated 429 error response
+ */
+export declare function createSimulated429Error(): Error;
+/**
+ * Reset simulation state when switching auth methods
+ */
+export declare function resetSimulationState(): void;
+/**
+ * Enable/disable 429 simulation programmatically (for tests)
+ */
+export declare function setSimulate429(enabled: boolean, afterRequests?: number, forAuthType?: string): void;

package/dist/src/utils/testUtils.js

@@ -0,0 +1,70 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Testing utilities for simulating 429 errors in unit tests
+ */
+let requestCounter = 0;
+let simulate429Enabled = false;
+let simulate429AfterRequests = 0;
+let simulate429ForAuthType;
+let fallbackOccurred = false;
+/**
+ * Check if we should simulate a 429 error for the current request
+ */
+export function shouldSimulate429(authType) {
+    if (!simulate429Enabled || fallbackOccurred) {
+        return false;
+    }
+    // If auth type filter is set, only simulate for that auth type
+    if (simulate429ForAuthType && authType !== simulate429ForAuthType) {
+        return false;
+    }
+    requestCounter++;
+    // If afterRequests is set, only simulate after that many requests
+    if (simulate429AfterRequests > 0) {
+        return requestCounter > simulate429AfterRequests;
+    }
+    // Otherwise, simulate for every request
+    return true;
+}
+/**
+ * Reset the request counter (useful for tests)
+ */
+export function resetRequestCounter() {
+    requestCounter = 0;
+}
+/**
+ * Disable 429 simulation after successful fallback
+ */
+export function disableSimulationAfterFallback() {
+    fallbackOccurred = true;
+}
+/**
+ * Create a simulated 429 error response
+ */
+export function createSimulated429Error() {
+    const error = new Error('Rate limit exceeded (simulated)');
+    error.status = 429;
+    return error;
+}
+/**
+ * Reset simulation state when switching auth methods
+ */
+export function resetSimulationState() {
+    fallbackOccurred = false;
+    resetRequestCounter();
+}
+/**
+ * Enable/disable 429 simulation programmatically (for tests)
+ */
+export function setSimulate429(enabled, afterRequests = 0, forAuthType) {
+    simulate429Enabled = enabled;
+    simulate429AfterRequests = afterRequests;
+    simulate429ForAuthType = forAuthType;
+    fallbackOccurred = false; // Reset fallback state when simulation is re-enabled
+    resetRequestCounter();
+}
+//# sourceMappingURL=testUtils.js.map

package/dist/src/utils/testUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testUtils.js","sourceRoot":"","sources":["../../../src/utils/testUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;GAEG;AAEH,IAAI,cAAc,GAAG,CAAC,CAAC;AACvB,IAAI,kBAAkB,GAAG,KAAK,CAAC;AAC/B,IAAI,wBAAwB,GAAG,CAAC,CAAC;AACjC,IAAI,sBAA0C,CAAC;AAC/C,IAAI,gBAAgB,GAAG,KAAK,CAAC;AAE7B;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,QAAiB;IACjD,IAAI,CAAC,kBAAkB,IAAI,gBAAgB,EAAE,CAAC;QAC5C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,+DAA+D;IAC/D,IAAI,sBAAsB,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;QAClE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,cAAc,EAAE,CAAC;IAEjB,kEAAkE;IAClE,IAAI,wBAAwB,GAAG,CAAC,EAAE,CAAC;QACjC,OAAO,cAAc,GAAG,wBAAwB,CAAC;IACnD,CAAC;IAED,wCAAwC;IACxC,OAAO,IAAI,CAAC;AACd,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB;IACjC,cAAc,GAAG,CAAC,CAAC;AACrB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,8BAA8B;IAC5C,gBAAgB,GAAG,IAAI,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB;IACrC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,iCAAiC,CAExD,CAAC;IACF,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC;IACnB,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,gBAAgB,GAAG,KAAK,CAAC;IACzB,mBAAmB,EAAE,CAAC;AACxB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAC5B,OAAgB,EAChB,aAAa,GAAG,CAAC,EACjB,WAAoB;IAEpB,kBAAkB,GAAG,OAAO,CAAC;IAC7B,wBAAwB,GAAG,aAAa,CAAC;IACzC,sBAAsB,GAAG,WAAW,CAAC;IACrC,gBAAgB,GAAG,KAAK,CAAC,CAAC,qDAAqD;IAC/E,mBAAmB,EAAE,CAAC;AACxB,CAAC"}

package/dist/src/utils/user_id.d.ts

@@ -0,0 +1,17 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Retrieves the installation ID from a file, creating it if it doesn't exist.
+ * This ID is used for unique user installation tracking.
+ * @returns A UUID string for the user.
+ */
+export declare function getInstallationId(): string;
+/**
+ * Retrieves the obfuscated Google Account ID for the currently authenticated user.
+ * When OAuth is available, returns the user's cached Google Account ID. Otherwise, returns the installation ID.
+ * @returns A string ID for the user (Google Account ID if available, otherwise installation ID).
+ */
+export declare function getGoogleAccountId(): Promise<string>;

package/dist/src/utils/user_id.js

@@ -0,0 +1,70 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import * as os from 'os';
+import * as fs from 'fs';
+import * as path from 'path';
+import { randomUUID } from 'crypto';
+import { GEMINI_DIR } from './paths.js';
+const homeDir = os.homedir() ?? '';
+const geminiDir = path.join(homeDir, GEMINI_DIR);
+const installationIdFile = path.join(geminiDir, 'installation_id');
+function ensureGeminiDirExists() {
+    if (!fs.existsSync(geminiDir)) {
+        fs.mkdirSync(geminiDir, { recursive: true });
+    }
+}
+function readInstallationIdFromFile() {
+    if (fs.existsSync(installationIdFile)) {
+        const installationid = fs.readFileSync(installationIdFile, 'utf-8').trim();
+        return installationid || null;
+    }
+    return null;
+}
+function writeInstallationIdToFile(installationId) {
+    fs.writeFileSync(installationIdFile, installationId, 'utf-8');
+}
+/**
+ * Retrieves the installation ID from a file, creating it if it doesn't exist.
+ * This ID is used for unique user installation tracking.
+ * @returns A UUID string for the user.
+ */
+export function getInstallationId() {
+    try {
+        ensureGeminiDirExists();
+        let installationId = readInstallationIdFromFile();
+        if (!installationId) {
+            installationId = randomUUID();
+            writeInstallationIdToFile(installationId);
+        }
+        return installationId;
+    }
+    catch (error) {
+        console.error('Error accessing installation ID file, generating ephemeral ID:', error);
+        return '123456789';
+    }
+}
+/**
+ * Retrieves the obfuscated Google Account ID for the currently authenticated user.
+ * When OAuth is available, returns the user's cached Google Account ID. Otherwise, returns the installation ID.
+ * @returns A string ID for the user (Google Account ID if available, otherwise installation ID).
+ */
+export async function getGoogleAccountId() {
+    // Try to get cached Google Account ID first
+    try {
+        // Dynamic import to avoid circular dependencies
+        const { getCachedGoogleAccountId } = await import('../code_assist/oauth2.js');
+        const googleAccountId = getCachedGoogleAccountId();
+        if (googleAccountId) {
+            return googleAccountId;
+        }
+    }
+    catch (error) {
+        // If there's any error accessing Google Account ID, just return empty string
+        console.debug('Could not get cached Google Account ID:', error);
+    }
+    return '';
+}
+//# sourceMappingURL=user_id.js.map

package/dist/src/utils/user_id.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"user_id.js","sourceRoot":"","sources":["../../../src/utils/user_id.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpC,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAExC,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;AACnC,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;AACjD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC;AAEnE,SAAS,qBAAqB;IAC5B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B;IACjC,IAAI,EAAE,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;QACtC,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;QAC3E,OAAO,cAAc,IAAI,IAAI,CAAC;IAChC,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,yBAAyB,CAAC,cAAsB;IACvD,EAAE,CAAC,aAAa,CAAC,kBAAkB,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;AAChE,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,iBAAiB;IAC/B,IAAI,CAAC;QACH,qBAAqB,EAAE,CAAC;QACxB,IAAI,cAAc,GAAG,0BAA0B,EAAE,CAAC;QAElD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,cAAc,GAAG,UAAU,EAAE,CAAC;YAC9B,yBAAyB,CAAC,cAAc,CAAC,CAAC;QAC5C,CAAC;QAED,OAAO,cAAc,CAAC;IACxB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CACX,gEAAgE,EAChE,KAAK,CACN,CAAC;QACF,OAAO,WAAW,CAAC;IACrB,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB;IACtC,4CAA4C;IAC5C,IAAI,CAAC;QACH,gDAAgD;QAChD,MAAM,EAAE,wBAAwB,EAAE,GAAG,MAAM,MAAM,CAC/C,0BAA0B,CAC3B,CAAC;QACF,MAAM,eAAe,GAAG,wBAAwB,EAAE,CAAC;QACnD,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,eAAe,CAAC;QACzB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,6EAA6E;QAC7E,OAAO,CAAC,KAAK,CAAC,yCAAyC,EAAE,KAAK,CAAC,CAAC;IAClE,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC"}

package/dist/src/utils/user_id.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};