@google/gemini-cli-core 0.0.1

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

@@ -0,0 +1,225 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeAll, afterAll } from 'vitest';
+import { escapePath, unescapePath, isSubpath } from './paths.js';
+describe('escapePath', () => {
+    it('should escape spaces', () => {
+        expect(escapePath('my file.txt')).toBe('my\\ file.txt');
+    });
+    it('should escape tabs', () => {
+        expect(escapePath('file\twith\ttabs.txt')).toBe('file\\\twith\\\ttabs.txt');
+    });
+    it('should escape parentheses', () => {
+        expect(escapePath('file(1).txt')).toBe('file\\(1\\).txt');
+    });
+    it('should escape square brackets', () => {
+        expect(escapePath('file[backup].txt')).toBe('file\\[backup\\].txt');
+    });
+    it('should escape curly braces', () => {
+        expect(escapePath('file{temp}.txt')).toBe('file\\{temp\\}.txt');
+    });
+    it('should escape semicolons', () => {
+        expect(escapePath('file;name.txt')).toBe('file\\;name.txt');
+    });
+    it('should escape ampersands', () => {
+        expect(escapePath('file&name.txt')).toBe('file\\&name.txt');
+    });
+    it('should escape pipes', () => {
+        expect(escapePath('file|name.txt')).toBe('file\\|name.txt');
+    });
+    it('should escape asterisks', () => {
+        expect(escapePath('file*.txt')).toBe('file\\*.txt');
+    });
+    it('should escape question marks', () => {
+        expect(escapePath('file?.txt')).toBe('file\\?.txt');
+    });
+    it('should escape dollar signs', () => {
+        expect(escapePath('file$name.txt')).toBe('file\\$name.txt');
+    });
+    it('should escape backticks', () => {
+        expect(escapePath('file`name.txt')).toBe('file\\`name.txt');
+    });
+    it('should escape single quotes', () => {
+        expect(escapePath("file'name.txt")).toBe("file\\'name.txt");
+    });
+    it('should escape double quotes', () => {
+        expect(escapePath('file"name.txt')).toBe('file\\"name.txt');
+    });
+    it('should escape hash symbols', () => {
+        expect(escapePath('file#name.txt')).toBe('file\\#name.txt');
+    });
+    it('should escape exclamation marks', () => {
+        expect(escapePath('file!name.txt')).toBe('file\\!name.txt');
+    });
+    it('should escape tildes', () => {
+        expect(escapePath('file~name.txt')).toBe('file\\~name.txt');
+    });
+    it('should escape less than and greater than signs', () => {
+        expect(escapePath('file<name>.txt')).toBe('file\\<name\\>.txt');
+    });
+    it('should handle multiple special characters', () => {
+        expect(escapePath('my file (backup) [v1.2].txt')).toBe('my\\ file\\ \\(backup\\)\\ \\[v1.2\\].txt');
+    });
+    it('should not double-escape already escaped characters', () => {
+        expect(escapePath('my\\ file.txt')).toBe('my\\ file.txt');
+        expect(escapePath('file\\(name\\).txt')).toBe('file\\(name\\).txt');
+    });
+    it('should handle escaped backslashes correctly', () => {
+        // Double backslash (escaped backslash) followed by space should escape the space
+        expect(escapePath('path\\\\ file.txt')).toBe('path\\\\\\ file.txt');
+        // Triple backslash (escaped backslash + escaping backslash) followed by space should not double-escape
+        expect(escapePath('path\\\\\\ file.txt')).toBe('path\\\\\\ file.txt');
+        // Quadruple backslash (two escaped backslashes) followed by space should escape the space
+        expect(escapePath('path\\\\\\\\ file.txt')).toBe('path\\\\\\\\\\ file.txt');
+    });
+    it('should handle complex escaped backslash scenarios', () => {
+        // Escaped backslash before special character that needs escaping
+        expect(escapePath('file\\\\(test).txt')).toBe('file\\\\\\(test\\).txt');
+        // Multiple escaped backslashes
+        expect(escapePath('path\\\\\\\\with space.txt')).toBe('path\\\\\\\\with\\ space.txt');
+    });
+    it('should handle paths without special characters', () => {
+        expect(escapePath('normalfile.txt')).toBe('normalfile.txt');
+        expect(escapePath('path/to/normalfile.txt')).toBe('path/to/normalfile.txt');
+    });
+    it('should handle complex real-world examples', () => {
+        expect(escapePath('My Documents/Project (2024)/file [backup].txt')).toBe('My\\ Documents/Project\\ \\(2024\\)/file\\ \\[backup\\].txt');
+        expect(escapePath('file with $special &chars!.txt')).toBe('file\\ with\\ \\$special\\ \\&chars\\!.txt');
+    });
+    it('should handle empty strings', () => {
+        expect(escapePath('')).toBe('');
+    });
+    it('should handle paths with only special characters', () => {
+        expect(escapePath(' ()[]{};&|*?$`\'"#!~<>')).toBe('\\ \\(\\)\\[\\]\\{\\}\\;\\&\\|\\*\\?\\$\\`\\\'\\"\\#\\!\\~\\<\\>');
+    });
+});
+describe('unescapePath', () => {
+    it('should unescape spaces', () => {
+        expect(unescapePath('my\\ file.txt')).toBe('my file.txt');
+    });
+    it('should unescape tabs', () => {
+        expect(unescapePath('file\\\twith\\\ttabs.txt')).toBe('file\twith\ttabs.txt');
+    });
+    it('should unescape parentheses', () => {
+        expect(unescapePath('file\\(1\\).txt')).toBe('file(1).txt');
+    });
+    it('should unescape square brackets', () => {
+        expect(unescapePath('file\\[backup\\].txt')).toBe('file[backup].txt');
+    });
+    it('should unescape curly braces', () => {
+        expect(unescapePath('file\\{temp\\}.txt')).toBe('file{temp}.txt');
+    });
+    it('should unescape multiple special characters', () => {
+        expect(unescapePath('my\\ file\\ \\(backup\\)\\ \\[v1.2\\].txt')).toBe('my file (backup) [v1.2].txt');
+    });
+    it('should handle paths without escaped characters', () => {
+        expect(unescapePath('normalfile.txt')).toBe('normalfile.txt');
+        expect(unescapePath('path/to/normalfile.txt')).toBe('path/to/normalfile.txt');
+    });
+    it('should handle all special characters', () => {
+        expect(unescapePath('\\ \\(\\)\\[\\]\\{\\}\\;\\&\\|\\*\\?\\$\\`\\\'\\"\\#\\!\\~\\<\\>')).toBe(' ()[]{};&|*?$`\'"#!~<>');
+    });
+    it('should be the inverse of escapePath', () => {
+        const testCases = [
+            'my file.txt',
+            'file(1).txt',
+            'file[backup].txt',
+            'My Documents/Project (2024)/file [backup].txt',
+            'file with $special &chars!.txt',
+            ' ()[]{};&|*?$`\'"#!~<>',
+            'file\twith\ttabs.txt',
+        ];
+        testCases.forEach((testCase) => {
+            expect(unescapePath(escapePath(testCase))).toBe(testCase);
+        });
+    });
+    it('should handle empty strings', () => {
+        expect(unescapePath('')).toBe('');
+    });
+    it('should not affect backslashes not followed by special characters', () => {
+        expect(unescapePath('file\\name.txt')).toBe('file\\name.txt');
+        expect(unescapePath('path\\to\\file.txt')).toBe('path\\to\\file.txt');
+    });
+    it('should handle escaped backslashes in unescaping', () => {
+        // Should correctly unescape when there are escaped backslashes
+        expect(unescapePath('path\\\\\\ file.txt')).toBe('path\\\\ file.txt');
+        expect(unescapePath('path\\\\\\\\\\ file.txt')).toBe('path\\\\\\\\ file.txt');
+        expect(unescapePath('file\\\\\\(test\\).txt')).toBe('file\\\\(test).txt');
+    });
+});
+describe('isSubpath', () => {
+    it('should return true for a direct subpath', () => {
+        expect(isSubpath('/a/b', '/a/b/c')).toBe(true);
+    });
+    it('should return true for the same path', () => {
+        expect(isSubpath('/a/b', '/a/b')).toBe(true);
+    });
+    it('should return false for a parent path', () => {
+        expect(isSubpath('/a/b/c', '/a/b')).toBe(false);
+    });
+    it('should return false for a completely different path', () => {
+        expect(isSubpath('/a/b', '/x/y')).toBe(false);
+    });
+    it('should handle relative paths', () => {
+        expect(isSubpath('a/b', 'a/b/c')).toBe(true);
+        expect(isSubpath('a/b', 'a/c')).toBe(false);
+    });
+    it('should handle paths with ..', () => {
+        expect(isSubpath('/a/b', '/a/b/../b/c')).toBe(true);
+        expect(isSubpath('/a/b', '/a/c/../b')).toBe(true);
+    });
+    it('should handle root paths', () => {
+        expect(isSubpath('/', '/a')).toBe(true);
+        expect(isSubpath('/a', '/')).toBe(false);
+    });
+    it('should handle trailing slashes', () => {
+        expect(isSubpath('/a/b/', '/a/b/c')).toBe(true);
+        expect(isSubpath('/a/b', '/a/b/c/')).toBe(true);
+        expect(isSubpath('/a/b/', '/a/b/c/')).toBe(true);
+    });
+});
+describe('isSubpath on Windows', () => {
+    const originalPlatform = process.platform;
+    beforeAll(() => {
+        Object.defineProperty(process, 'platform', {
+            value: 'win32',
+        });
+    });
+    afterAll(() => {
+        Object.defineProperty(process, 'platform', {
+            value: originalPlatform,
+        });
+    });
+    it('should return true for a direct subpath on Windows', () => {
+        expect(isSubpath('C:\\Users\\Test', 'C:\\Users\\Test\\file.txt')).toBe(true);
+    });
+    it('should return true for the same path on Windows', () => {
+        expect(isSubpath('C:\\Users\\Test', 'C:\\Users\\Test')).toBe(true);
+    });
+    it('should return false for a parent path on Windows', () => {
+        expect(isSubpath('C:\\Users\\Test\\file.txt', 'C:\\Users\\Test')).toBe(false);
+    });
+    it('should return false for a different drive on Windows', () => {
+        expect(isSubpath('C:\\Users\\Test', 'D:\\Users\\Test')).toBe(false);
+    });
+    it('should be case-insensitive for drive letters on Windows', () => {
+        expect(isSubpath('c:\\Users\\Test', 'C:\\Users\\Test\\file.txt')).toBe(true);
+    });
+    it('should be case-insensitive for path components on Windows', () => {
+        expect(isSubpath('C:\\Users\\Test', 'c:\\users\\test\\file.txt')).toBe(true);
+    });
+    it('should handle mixed slashes on Windows', () => {
+        expect(isSubpath('C:/Users/Test', 'C:\\Users\\Test\\file.txt')).toBe(true);
+    });
+    it('should handle trailing slashes on Windows', () => {
+        expect(isSubpath('C:\\Users\\Test\\', 'C:\\Users\\Test\\file.txt')).toBe(true);
+    });
+    it('should handle relative paths correctly on Windows', () => {
+        expect(isSubpath('Users\\Test', 'Users\\Test\\file.txt')).toBe(true);
+        expect(isSubpath('Users\\Test\\file.txt', 'Users\\Test')).toBe(false);
+    });
+});
+//# sourceMappingURL=paths.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"paths.test.js","sourceRoot":"","sources":["../../../src/utils/paths.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEjE,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAC5B,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,CACpD,2CAA2C,CAC5C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC1D,MAAM,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,iFAAiF;QACjF,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACpE,uGAAuG;QACvG,MAAM,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACtE,0FAA0F;QAC1F,MAAM,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,iEAAiE;QACjE,MAAM,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACxE,+BAA+B;QAC/B,MAAM,CAAC,UAAU,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CACnD,8BAA8B,CAC/B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC5D,MAAM,CAAC,UAAU,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,UAAU,CAAC,+CAA+C,CAAC,CAAC,CAAC,IAAI,CACtE,6DAA6D,CAC9D,CAAC;QACF,MAAM,CAAC,UAAU,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CACvD,4CAA4C,CAC7C,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,UAAU,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAC/C,kEAAkE,CACnE,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,CAAC,YAAY,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CACnD,sBAAsB,CACvB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,CAAC,YAAY,CAAC,2CAA2C,CAAC,CAAC,CAAC,IAAI,CACpE,6BAA6B,CAC9B,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9D,MAAM,CAAC,YAAY,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CACjD,wBAAwB,CACzB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,CACJ,YAAY,CACV,kEAAkE,CACnE,CACF,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,SAAS,GAAG;YAChB,aAAa;YACb,aAAa;YACb,kBAAkB;YAClB,+CAA+C;YAC/C,gCAAgC;YAChC,wBAAwB;YACxB,sBAAsB;SACvB,CAAC;QAEF,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7B,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC9D,MAAM,CAAC,YAAY,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,+DAA+D;QAC/D,MAAM,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACtE,MAAM,CAAC,YAAY,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAClD,uBAAuB,CACxB,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAC5E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,MAAM,gBAAgB,GAAG,OAAO,CAAC,QAAQ,CAAC;IAE1C,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE;YACzC,KAAK,EAAE,OAAO;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,GAAG,EAAE;QACZ,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE;YACzC,KAAK,EAAE,gBAAgB;SACxB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE,2BAA2B,CAAC,CAAC,CAAC,IAAI,CACpE,IAAI,CACL,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,CAAC,SAAS,CAAC,2BAA2B,EAAE,iBAAiB,CAAC,CAAC,CAAC,IAAI,CACpE,KAAK,CACN,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE,2BAA2B,CAAC,CAAC,CAAC,IAAI,CACpE,IAAI,CACL,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,CAAC,SAAS,CAAC,iBAAiB,EAAE,2BAA2B,CAAC,CAAC,CAAC,IAAI,CACpE,IAAI,CACL,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,CAAC,SAAS,CAAC,eAAe,EAAE,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,SAAS,CAAC,mBAAmB,EAAE,2BAA2B,CAAC,CAAC,CAAC,IAAI,CACtE,IAAI,CACL,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,CAAC,SAAS,CAAC,aAAa,EAAE,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrE,MAAM,CAAC,SAAS,CAAC,uBAAuB,EAAE,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AsyncLocalStorage } from 'node:async_hooks';
+export declare const promptIdContext: AsyncLocalStorage<string>;

package/dist/src/utils/promptIdContext.js

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"promptIdContext.js","sourceRoot":"","sources":["../../../src/utils/promptIdContext.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,iBAAiB,EAAU,CAAC"}

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

@@ -0,0 +1,18 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { StructuredError } from '../core/turn.js';
+export interface ApiError {
+    error: {
+        code: number;
+        message: string;
+        status: string;
+        details: unknown[];
+    };
+}
+export declare function isApiError(error: unknown): error is ApiError;
+export declare function isStructuredError(error: unknown): error is StructuredError;
+export declare function isProQuotaExceededError(error: unknown): boolean;
+export declare function isGenericQuotaExceededError(error: unknown): boolean;

package/dist/src/utils/quotaErrorDetection.js

@@ -0,0 +1,65 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export function isApiError(error) {
+    return (typeof error === 'object' &&
+        error !== null &&
+        'error' in error &&
+        typeof error.error === 'object' &&
+        'message' in error.error);
+}
+export function isStructuredError(error) {
+    return (typeof error === 'object' &&
+        error !== null &&
+        'message' in error &&
+        typeof error.message === 'string');
+}
+export function isProQuotaExceededError(error) {
+    // Check for Pro quota exceeded errors by looking for the specific pattern
+    // This will match patterns like:
+    // - "Quota exceeded for quota metric 'Gemini 2.5 Pro Requests'"
+    // - "Quota exceeded for quota metric 'Gemini 2.5-preview Pro Requests'"
+    // We use string methods instead of regex to avoid ReDoS vulnerabilities
+    const checkMessage = (message) => message.includes("Quota exceeded for quota metric 'Gemini") &&
+        message.includes("Pro Requests'");
+    if (typeof error === 'string') {
+        return checkMessage(error);
+    }
+    if (isStructuredError(error)) {
+        return checkMessage(error.message);
+    }
+    if (isApiError(error)) {
+        return checkMessage(error.error.message);
+    }
+    // Check if it's a Gaxios error with response data
+    if (error && typeof error === 'object' && 'response' in error) {
+        const gaxiosError = error;
+        if (gaxiosError.response && gaxiosError.response.data) {
+            if (typeof gaxiosError.response.data === 'string') {
+                return checkMessage(gaxiosError.response.data);
+            }
+            if (typeof gaxiosError.response.data === 'object' &&
+                gaxiosError.response.data !== null &&
+                'error' in gaxiosError.response.data) {
+                const errorData = gaxiosError.response.data;
+                return checkMessage(errorData.error?.message || '');
+            }
+        }
+    }
+    return false;
+}
+export function isGenericQuotaExceededError(error) {
+    if (typeof error === 'string') {
+        return error.includes('Quota exceeded for quota metric');
+    }
+    if (isStructuredError(error)) {
+        return error.message.includes('Quota exceeded for quota metric');
+    }
+    if (isApiError(error)) {
+        return error.error.message.includes('Quota exceeded for quota metric');
+    }
+    return false;
+}
+//# sourceMappingURL=quotaErrorDetection.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"quotaErrorDetection.js","sourceRoot":"","sources":["../../../src/utils/quotaErrorDetection.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAaH,MAAM,UAAU,UAAU,CAAC,KAAc;IACvC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,OAAO,IAAI,KAAK;QAChB,OAAQ,KAAkB,CAAC,KAAK,KAAK,QAAQ;QAC7C,SAAS,IAAK,KAAkB,CAAC,KAAK,CACvC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAAc;IAC9C,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,SAAS,IAAI,KAAK;QAClB,OAAQ,KAAyB,CAAC,OAAO,KAAK,QAAQ,CACvD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,KAAc;IACpD,0EAA0E;IAC1E,iCAAiC;IACjC,gEAAgE;IAChE,wEAAwE;IACxE,wEAAwE;IAExE,MAAM,YAAY,GAAG,CAAC,OAAe,EAAW,EAAE,CAChD,OAAO,CAAC,QAAQ,CAAC,yCAAyC,CAAC;QAC3D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC;IAEpC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IAED,IAAI,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,YAAY,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC;IAED,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QACtB,OAAO,YAAY,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,kDAAkD;IAClD,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,UAAU,IAAI,KAAK,EAAE,CAAC;QAC9D,MAAM,WAAW,GAAG,KAInB,CAAC;QACF,IAAI,WAAW,CAAC,QAAQ,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtD,IAAI,OAAO,WAAW,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAClD,OAAO,YAAY,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACjD,CAAC;YACD,IACE,OAAO,WAAW,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ;gBAC7C,WAAW,CAAC,QAAQ,CAAC,IAAI,KAAK,IAAI;gBAClC,OAAO,IAAI,WAAW,CAAC,QAAQ,CAAC,IAAI,EACpC,CAAC;gBACD,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,CAAC,IAEtC,CAAC;gBACF,OAAO,YAAY,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,KAAc;IACxD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC,QAAQ,CAAC,iCAAiC,CAAC,CAAC;IAC3D,CAAC;IAED,IAAI,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,iCAAiC,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QACtB,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,iCAAiC,CAAC,CAAC;IACzE,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}

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

@@ -0,0 +1,30 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export interface HttpError extends Error {
+    status?: number;
+}
+export interface RetryOptions {
+    maxAttempts: number;
+    initialDelayMs: number;
+    maxDelayMs: number;
+    shouldRetry: (error: Error) => boolean;
+    onPersistent429?: (authType?: string, error?: unknown) => Promise<string | boolean | null>;
+    authType?: string;
+}
+/**
+ * Retries a function with exponential backoff and jitter.
+ * @param fn The asynchronous function to retry.
+ * @param options Optional retry configuration.
+ * @returns A promise that resolves with the result of the function if successful.
+ * @throws The last error encountered if all attempts fail.
+ */
+export declare function retryWithBackoff<T>(fn: () => Promise<T>, options?: Partial<RetryOptions>): Promise<T>;
+/**
+ * Extracts the HTTP status code from an error object.
+ * @param error The error object.
+ * @returns The HTTP status code, or undefined if not found.
+ */
+export declare function getErrorStatus(error: unknown): number | undefined;

package/dist/src/utils/retry.js

@@ -0,0 +1,276 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType } from '../core/contentGenerator.js';
+import { isProQuotaExceededError, isGenericQuotaExceededError, } from './quotaErrorDetection.js';
+const DEFAULT_RETRY_OPTIONS = {
+    maxAttempts: 5,
+    initialDelayMs: 5000,
+    maxDelayMs: 30000, // 30 seconds
+    shouldRetry: defaultShouldRetry,
+};
+/**
+ * Default predicate function to determine if a retry should be attempted.
+ * Retries on 429 (Too Many Requests) and 5xx server errors.
+ * @param error The error object.
+ * @returns True if the error is a transient error, false otherwise.
+ */
+function defaultShouldRetry(error) {
+    // Check for common transient error status codes either in message or a status property
+    if (error && typeof error.status === 'number') {
+        const status = error.status;
+        if (status === 429 || (status >= 500 && status < 600)) {
+            return true;
+        }
+    }
+    if (error instanceof Error && error.message) {
+        if (error.message.includes('429'))
+            return true;
+        if (error.message.match(/5\d{2}/))
+            return true;
+    }
+    return false;
+}
+/**
+ * Delays execution for a specified number of milliseconds.
+ * @param ms The number of milliseconds to delay.
+ * @returns A promise that resolves after the delay.
+ */
+function delay(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+/**
+ * Retries a function with exponential backoff and jitter.
+ * @param fn The asynchronous function to retry.
+ * @param options Optional retry configuration.
+ * @returns A promise that resolves with the result of the function if successful.
+ * @throws The last error encountered if all attempts fail.
+ */
+export async function retryWithBackoff(fn, options) {
+    const { maxAttempts, initialDelayMs, maxDelayMs, onPersistent429, authType, shouldRetry, } = {
+        ...DEFAULT_RETRY_OPTIONS,
+        ...options,
+    };
+    let attempt = 0;
+    let currentDelay = initialDelayMs;
+    let consecutive429Count = 0;
+    while (attempt < maxAttempts) {
+        attempt++;
+        try {
+            return await fn();
+        }
+        catch (error) {
+            const errorStatus = getErrorStatus(error);
+            // Check for Pro quota exceeded error first - immediate fallback for OAuth users
+            if (errorStatus === 429 &&
+                authType === AuthType.LOGIN_WITH_GOOGLE &&
+                isProQuotaExceededError(error) &&
+                onPersistent429) {
+                try {
+                    const fallbackModel = await onPersistent429(authType, error);
+                    if (fallbackModel !== false && fallbackModel !== null) {
+                        // Reset attempt counter and try with new model
+                        attempt = 0;
+                        consecutive429Count = 0;
+                        currentDelay = initialDelayMs;
+                        // With the model updated, we continue to the next attempt
+                        continue;
+                    }
+                    else {
+                        // Fallback handler returned null/false, meaning don't continue - stop retry process
+                        throw error;
+                    }
+                }
+                catch (fallbackError) {
+                    // If fallback fails, continue with original error
+                    console.warn('Fallback to Flash model failed:', fallbackError);
+                }
+            }
+            // Check for generic quota exceeded error (but not Pro, which was handled above) - immediate fallback for OAuth users
+            if (errorStatus === 429 &&
+                authType === AuthType.LOGIN_WITH_GOOGLE &&
+                !isProQuotaExceededError(error) &&
+                isGenericQuotaExceededError(error) &&
+                onPersistent429) {
+                try {
+                    const fallbackModel = await onPersistent429(authType, error);
+                    if (fallbackModel !== false && fallbackModel !== null) {
+                        // Reset attempt counter and try with new model
+                        attempt = 0;
+                        consecutive429Count = 0;
+                        currentDelay = initialDelayMs;
+                        // With the model updated, we continue to the next attempt
+                        continue;
+                    }
+                    else {
+                        // Fallback handler returned null/false, meaning don't continue - stop retry process
+                        throw error;
+                    }
+                }
+                catch (fallbackError) {
+                    // If fallback fails, continue with original error
+                    console.warn('Fallback to Flash model failed:', fallbackError);
+                }
+            }
+            // Track consecutive 429 errors
+            if (errorStatus === 429) {
+                consecutive429Count++;
+            }
+            else {
+                consecutive429Count = 0;
+            }
+            // If we have persistent 429s and a fallback callback for OAuth
+            if (consecutive429Count >= 2 &&
+                onPersistent429 &&
+                authType === AuthType.LOGIN_WITH_GOOGLE) {
+                try {
+                    const fallbackModel = await onPersistent429(authType, error);
+                    if (fallbackModel !== false && fallbackModel !== null) {
+                        // Reset attempt counter and try with new model
+                        attempt = 0;
+                        consecutive429Count = 0;
+                        currentDelay = initialDelayMs;
+                        // With the model updated, we continue to the next attempt
+                        continue;
+                    }
+                    else {
+                        // Fallback handler returned null/false, meaning don't continue - stop retry process
+                        throw error;
+                    }
+                }
+                catch (fallbackError) {
+                    // If fallback fails, continue with original error
+                    console.warn('Fallback to Flash model failed:', fallbackError);
+                }
+            }
+            // Check if we've exhausted retries or shouldn't retry
+            if (attempt >= maxAttempts || !shouldRetry(error)) {
+                throw error;
+            }
+            const { delayDurationMs, errorStatus: delayErrorStatus } = getDelayDurationAndStatus(error);
+            if (delayDurationMs > 0) {
+                // Respect Retry-After header if present and parsed
+                console.warn(`Attempt ${attempt} failed with status ${delayErrorStatus ?? 'unknown'}. Retrying after explicit delay of ${delayDurationMs}ms...`, error);
+                await delay(delayDurationMs);
+                // Reset currentDelay for next potential non-429 error, or if Retry-After is not present next time
+                currentDelay = initialDelayMs;
+            }
+            else {
+                // Fall back to exponential backoff with jitter
+                logRetryAttempt(attempt, error, errorStatus);
+                // Add jitter: +/- 30% of currentDelay
+                const jitter = currentDelay * 0.3 * (Math.random() * 2 - 1);
+                const delayWithJitter = Math.max(0, currentDelay + jitter);
+                await delay(delayWithJitter);
+                currentDelay = Math.min(maxDelayMs, currentDelay * 2);
+            }
+        }
+    }
+    // This line should theoretically be unreachable due to the throw in the catch block.
+    // Added for type safety and to satisfy the compiler that a promise is always returned.
+    throw new Error('Retry attempts exhausted');
+}
+/**
+ * Extracts the HTTP status code from an error object.
+ * @param error The error object.
+ * @returns The HTTP status code, or undefined if not found.
+ */
+export function getErrorStatus(error) {
+    if (typeof error === 'object' && error !== null) {
+        if ('status' in error && typeof error.status === 'number') {
+            return error.status;
+        }
+        // Check for error.response.status (common in axios errors)
+        if ('response' in error &&
+            typeof error.response === 'object' &&
+            error.response !== null) {
+            const response = error.response;
+            if ('status' in response && typeof response.status === 'number') {
+                return response.status;
+            }
+        }
+    }
+    return undefined;
+}
+/**
+ * Extracts the Retry-After delay from an error object's headers.
+ * @param error The error object.
+ * @returns The delay in milliseconds, or 0 if not found or invalid.
+ */
+function getRetryAfterDelayMs(error) {
+    if (typeof error === 'object' && error !== null) {
+        // Check for error.response.headers (common in axios errors)
+        if ('response' in error &&
+            typeof error.response === 'object' &&
+            error.response !== null) {
+            const response = error.response;
+            if ('headers' in response &&
+                typeof response.headers === 'object' &&
+                response.headers !== null) {
+                const headers = response.headers;
+                const retryAfterHeader = headers['retry-after'];
+                if (typeof retryAfterHeader === 'string') {
+                    const retryAfterSeconds = parseInt(retryAfterHeader, 10);
+                    if (!isNaN(retryAfterSeconds)) {
+                        return retryAfterSeconds * 1000;
+                    }
+                    // It might be an HTTP date
+                    const retryAfterDate = new Date(retryAfterHeader);
+                    if (!isNaN(retryAfterDate.getTime())) {
+                        return Math.max(0, retryAfterDate.getTime() - Date.now());
+                    }
+                }
+            }
+        }
+    }
+    return 0;
+}
+/**
+ * Determines the delay duration based on the error, prioritizing Retry-After header.
+ * @param error The error object.
+ * @returns An object containing the delay duration in milliseconds and the error status.
+ */
+function getDelayDurationAndStatus(error) {
+    const errorStatus = getErrorStatus(error);
+    let delayDurationMs = 0;
+    if (errorStatus === 429) {
+        delayDurationMs = getRetryAfterDelayMs(error);
+    }
+    return { delayDurationMs, errorStatus };
+}
+/**
+ * Logs a message for a retry attempt when using exponential backoff.
+ * @param attempt The current attempt number.
+ * @param error The error that caused the retry.
+ * @param errorStatus The HTTP status code of the error, if available.
+ */
+function logRetryAttempt(attempt, error, errorStatus) {
+    let message = `Attempt ${attempt} failed. Retrying with backoff...`;
+    if (errorStatus) {
+        message = `Attempt ${attempt} failed with status ${errorStatus}. Retrying with backoff...`;
+    }
+    if (errorStatus === 429) {
+        console.warn(message, error);
+    }
+    else if (errorStatus && errorStatus >= 500 && errorStatus < 600) {
+        console.error(message, error);
+    }
+    else if (error instanceof Error) {
+        // Fallback for errors that might not have a status but have a message
+        if (error.message.includes('429')) {
+            console.warn(`Attempt ${attempt} failed with 429 error (no Retry-After header). Retrying with backoff...`, error);
+        }
+        else if (error.message.match(/5\d{2}/)) {
+            console.error(`Attempt ${attempt} failed with 5xx error. Retrying with backoff...`, error);
+        }
+        else {
+            console.warn(message, error); // Default to warn for other errors
+        }
+    }
+    else {
+        console.warn(message, error); // Default to warn if error type is unknown
+    }
+}
+//# sourceMappingURL=retry.js.map