@vybestack/llxprt-code-core 0.1.12

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

@@ -0,0 +1,146 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { SchemaValidator } from './schemaValidator.js';
+import { Type } from '@google/genai';
+describe('SchemaValidator', () => {
+    describe('validate', () => {
+        it('should return null for valid data', () => {
+            const schema = {
+                type: Type.OBJECT,
+                properties: {
+                    name: { type: Type.STRING },
+                    age: { type: Type.NUMBER },
+                },
+                required: ['name'],
+            };
+            const validData = { name: 'John', age: 30 };
+            expect(SchemaValidator.validate(schema, validData)).toBeNull();
+        });
+        it('should return error for invalid data', () => {
+            const schema = {
+                type: Type.OBJECT,
+                properties: {
+                    name: { type: Type.STRING },
+                    age: { type: Type.NUMBER },
+                },
+                required: ['name'],
+            };
+            const invalidData = { age: 30 }; // missing required 'name'
+            const error = SchemaValidator.validate(schema, invalidData);
+            expect(error).toContain('must have required property');
+        });
+        it('should handle numeric validation properties', () => {
+            const schema = {
+                type: Type.OBJECT,
+                properties: {
+                    items: {
+                        type: Type.ARRAY,
+                        minItems: '2', // @google/genai Schema expects strings for numeric properties
+                        maxItems: '5',
+                    },
+                    text: {
+                        type: Type.STRING,
+                        minLength: '3',
+                        maxLength: '10',
+                    },
+                    count: {
+                        type: Type.NUMBER,
+                        minimum: 0,
+                        maximum: 100,
+                    },
+                },
+            };
+            // Valid data
+            const validData = {
+                items: [1, 2, 3],
+                text: 'hello',
+                count: 50,
+            };
+            expect(SchemaValidator.validate(schema, validData)).toBeNull();
+            // Too few items
+            const tooFewItems = {
+                items: [1],
+                text: 'hello',
+                count: 50,
+            };
+            const error1 = SchemaValidator.validate(schema, tooFewItems);
+            expect(error1).toContain('must NOT have fewer than 2 items');
+            // Text too short
+            const textTooShort = {
+                items: [1, 2],
+                text: 'hi',
+                count: 50,
+            };
+            const error2 = SchemaValidator.validate(schema, textTooShort);
+            expect(error2).toContain('must NOT have fewer than 3 characters');
+        });
+        it('should handle nested schemas with anyOf, allOf, oneOf', () => {
+            const schema = {
+                type: Type.OBJECT,
+                properties: {
+                    value: {
+                        anyOf: [{ type: Type.STRING }, { type: Type.NUMBER }],
+                    },
+                },
+            };
+            expect(SchemaValidator.validate(schema, { value: 'test' })).toBeNull();
+            expect(SchemaValidator.validate(schema, { value: 123 })).toBeNull();
+            const error = SchemaValidator.validate(schema, { value: true });
+            expect(error).toContain('must match a schema in anyOf');
+        });
+        it('should handle arrays with item schemas', () => {
+            const schema = {
+                type: Type.ARRAY,
+                items: {
+                    type: Type.OBJECT,
+                    properties: {
+                        id: { type: Type.NUMBER },
+                        name: { type: Type.STRING },
+                    },
+                    required: ['id'],
+                },
+            };
+            const validData = [
+                { id: 1, name: 'Item 1' },
+                { id: 2, name: 'Item 2' },
+            ];
+            expect(SchemaValidator.validate(schema, validData)).toBeNull();
+            const invalidData = [
+                { id: 1, name: 'Item 1' },
+                { name: 'Item 2' }, // missing required 'id'
+            ];
+            const error = SchemaValidator.validate(schema, invalidData);
+            expect(error).toContain('must have required property');
+        });
+        it('should handle additionalProperties', () => {
+            const schema = {
+                type: Type.OBJECT,
+                properties: {
+                    name: { type: Type.STRING },
+                },
+                additionalProperties: { type: Type.NUMBER },
+            };
+            const validData = { name: 'test', extra: 123 };
+            expect(SchemaValidator.validate(schema, validData)).toBeNull();
+            const invalidData = { name: 'test', extra: 'not a number' };
+            const error = SchemaValidator.validate(schema, invalidData);
+            expect(error).toContain('must be number');
+        });
+        it('should convert UPPERCASE type enums to lowercase', () => {
+            const schema = {
+                type: Type.OBJECT,
+                properties: {
+                    name: { type: Type.STRING },
+                    count: { type: Type.NUMBER },
+                },
+            };
+            const validData = { name: 'test', count: 5 };
+            expect(SchemaValidator.validate(schema, validData)).toBeNull();
+        });
+    });
+});
+//# sourceMappingURL=schemaValidator.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"schemaValidator.test.js","sourceRoot":"","sources":["../../../src/utils/schemaValidator.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAErC,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;oBAC3B,GAAG,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;iBAC3B;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB,CAAC;YAEF,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC;YAC5C,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;oBAC3B,GAAG,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;iBAC3B;gBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;aACnB,CAAC;YAEF,MAAM,WAAW,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,0BAA0B;YAC3D,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,UAAU,EAAE;oBACV,KAAK,EAAE;wBACL,IAAI,EAAE,IAAI,CAAC,KAAK;wBAChB,QAAQ,EAAE,GAAG,EAAE,8DAA8D;wBAC7E,QAAQ,EAAE,GAAG;qBACd;oBACD,IAAI,EAAE;wBACJ,IAAI,EAAE,IAAI,CAAC,MAAM;wBACjB,SAAS,EAAE,GAAG;wBACd,SAAS,EAAE,IAAI;qBAChB;oBACD,KAAK,EAAE;wBACL,IAAI,EAAE,IAAI,CAAC,MAAM;wBACjB,OAAO,EAAE,CAAC;wBACV,OAAO,EAAE,GAAG;qBACb;iBACF;aACF,CAAC;YAEF,aAAa;YACb,MAAM,SAAS,GAAG;gBAChB,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;gBAChB,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE;aACV,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAE/D,gBAAgB;YAChB,MAAM,WAAW,GAAG;gBAClB,KAAK,EAAE,CAAC,CAAC,CAAC;gBACV,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,EAAE;aACV,CAAC;YACF,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kCAAkC,CAAC,CAAC;YAE7D,iBAAiB;YACjB,MAAM,YAAY,GAAG;gBACnB,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;gBACb,IAAI,EAAE,IAAI;gBACV,KAAK,EAAE,EAAE;aACV,CAAC;YACF,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uCAAuC,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,UAAU,EAAE;oBACV,KAAK,EAAE;wBACL,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;qBACtD;iBACF;aACF,CAAC;YAEF,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YACvE,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAEpE,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAChE,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,8BAA8B,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,IAAI,CAAC,KAAK;gBAChB,KAAK,EAAE;oBACL,IAAI,EAAE,IAAI,CAAC,MAAM;oBACjB,UAAU,EAAE;wBACV,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;wBACzB,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;qBAC5B;oBACD,QAAQ,EAAE,CAAC,IAAI,CAAC;iBACjB;aACF,CAAC;YAEF,MAAM,SAAS,GAAG;gBAChB,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC1B,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAE/D,MAAM,WAAW,GAAG;gBAClB,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE;gBACzB,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,wBAAwB;aAC7C,CAAC;YACF,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;iBAC5B;gBACD,oBAAoB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;aAC5C,CAAC;YAEF,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;YAC/C,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;YAE/D,MAAM,WAAW,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;YAC5D,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,MAAM,GAAG;gBACb,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,UAAU,EAAE;oBACV,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;oBAC3B,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE;iBAC7B;aACF,CAAC;YAEF,MAAM,SAAS,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YAC7C,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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/summarizer.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ToolResult } from '../tools/tools.js';
+import { GeminiClient } from '../core/client.js';
+/**
+ * A function that summarizes the result of a tool execution.
+ *
+ * @param result The result of the tool execution.
+ * @returns The summary of the result.
+ */
+export type Summarizer = (result: ToolResult, geminiClient: GeminiClient, abortSignal: AbortSignal) => Promise<string>;
+/**
+ * The default summarizer for tool results.
+ *
+ * @param result The result of the tool execution.
+ * @param geminiClient The Gemini client to use for summarization.
+ * @param abortSignal The abort signal to use for summarization.
+ * @returns The summary of the result.
+ */
+export declare const defaultSummarizer: Summarizer;
+export declare const llmSummarizer: Summarizer;
+export declare function summarizeToolOutput(textToSummarize: string, geminiClient: GeminiClient, abortSignal: AbortSignal, maxOutputTokens?: number): Promise<string>;

package/dist/src/utils/summarizer.js

@@ -0,0 +1,80 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js';
+/**
+ * The default summarizer for tool results.
+ *
+ * @param result The result of the tool execution.
+ * @param geminiClient The Gemini client to use for summarization.
+ * @param abortSignal The abort signal to use for summarization.
+ * @returns The summary of the result.
+ */
+export const defaultSummarizer = (result, _geminiClient, _abortSignal) => Promise.resolve(JSON.stringify(result.llmContent));
+// TODO: Move both these functions to utils
+function partToString(part) {
+    if (!part) {
+        return '';
+    }
+    if (typeof part === 'string') {
+        return part;
+    }
+    if (Array.isArray(part)) {
+        return part.map(partToString).join('');
+    }
+    if ('text' in part) {
+        return part.text ?? '';
+    }
+    return '';
+}
+function getResponseText(response) {
+    if (response.candidates && response.candidates.length > 0) {
+        const candidate = response.candidates[0];
+        if (candidate.content &&
+            candidate.content.parts &&
+            candidate.content.parts.length > 0) {
+            return candidate.content.parts
+                .filter((part) => part.text)
+                .map((part) => part.text)
+                .join('');
+        }
+    }
+    return null;
+}
+const SUMMARIZE_TOOL_OUTPUT_PROMPT = `Summarize the following tool output to be a maximum of {maxOutputTokens} tokens. The summary should be concise and capture the main points of the tool output.
+
+The summarization should be done based on the content that is provided. Here are the basic rules to follow:
+1. If the text is a directory listing or any output that is structural, use the history of the conversation to understand the context. Using this context try to understand what information we need from the tool output and return that as a response.
+2. If the text is text content and there is nothing structural that we need, summarize the text.
+3. If the text is the output of a shell command, use the history of the conversation to understand the context. Using this context try to understand what information we need from the tool output and return a summarization along with the stack trace of any error within the <error></error> tags. The stack trace should be complete and not truncated. If there are warnings, you should include them in the summary within <warning></warning> tags.
+
+
+Text to summarize:
+"{textToSummarize}"
+
+Return the summary string which should first contain an overall summarization of text followed by the full stack trace of errors and warnings in the tool output.
+`;
+export const llmSummarizer = (result, geminiClient, abortSignal) => summarizeToolOutput(partToString(result.llmContent), geminiClient, abortSignal);
+export async function summarizeToolOutput(textToSummarize, geminiClient, abortSignal, maxOutputTokens = 2000) {
+    // There is going to be a slight difference here since we are comparing length of string with maxOutputTokens.
+    // This is meant to be a ballpark estimation of if we need to summarize the tool output.
+    if (!textToSummarize || textToSummarize.length < maxOutputTokens) {
+        return textToSummarize;
+    }
+    const prompt = SUMMARIZE_TOOL_OUTPUT_PROMPT.replace('{maxOutputTokens}', String(maxOutputTokens)).replace('{textToSummarize}', textToSummarize);
+    const contents = [{ role: 'user', parts: [{ text: prompt }] }];
+    const toolOutputSummarizerConfig = {
+        maxOutputTokens,
+    };
+    try {
+        const parsedResponse = (await geminiClient.generateContent(contents, toolOutputSummarizerConfig, abortSignal, DEFAULT_GEMINI_FLASH_MODEL));
+        return getResponseText(parsedResponse) || textToSummarize;
+    }
+    catch (error) {
+        console.error('Failed to summarize tool output.', error);
+        return textToSummarize;
+    }
+}
+//# sourceMappingURL=summarizer.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"summarizer.js","sourceRoot":"","sources":["../../../src/utils/summarizer.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AASH,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AAejE;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAe,CAC3C,MAAkB,EAClB,aAA2B,EAC3B,YAAyB,EACzB,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;AAExD,2CAA2C;AAC3C,SAAS,YAAY,CAAC,IAAmB;IACvC,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACzC,CAAC;IACD,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;QACnB,OAAO,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;IACzB,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,eAAe,CAAC,QAAiC;IACxD,IAAI,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACzC,IACE,SAAS,CAAC,OAAO;YACjB,SAAS,CAAC,OAAO,CAAC,KAAK;YACvB,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,EAClC,CAAC;YACD,OAAO,SAAS,CAAC,OAAO,CAAC,KAAK;iBAC3B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBAC3B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;iBACxB,IAAI,CAAC,EAAE,CAAC,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,4BAA4B,GAAG;;;;;;;;;;;;CAYpC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAe,CAAC,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,CAC7E,mBAAmB,CACjB,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,EAC/B,YAAY,EACZ,WAAW,CACZ,CAAC;AAEJ,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,eAAuB,EACvB,YAA0B,EAC1B,WAAwB,EACxB,kBAA0B,IAAI;IAE9B,8GAA8G;IAC9G,wFAAwF;IACxF,IAAI,CAAC,eAAe,IAAI,eAAe,CAAC,MAAM,GAAG,eAAe,EAAE,CAAC;QACjE,OAAO,eAAe,CAAC;IACzB,CAAC;IACD,MAAM,MAAM,GAAG,4BAA4B,CAAC,OAAO,CACjD,mBAAmB,EACnB,MAAM,CAAC,eAAe,CAAC,CACxB,CAAC,OAAO,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAC;IAEhD,MAAM,QAAQ,GAAc,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1E,MAAM,0BAA0B,GAA0B;QACxD,eAAe;KAChB,CAAC;IACF,IAAI,CAAC;QACH,MAAM,cAAc,GAAG,CAAC,MAAM,YAAY,CAAC,eAAe,CACxD,QAAQ,EACR,0BAA0B,EAC1B,WAAW,EACX,0BAA0B,CAC3B,CAAuC,CAAC;QACzC,OAAO,eAAe,CAAC,cAAc,CAAC,IAAI,eAAe,CAAC;IAC5D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;QACzD,OAAO,eAAe,CAAC;IACzB,CAAC;AACH,CAAC"}

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

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

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

@@ -0,0 +1,131 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { GeminiClient } from '../core/client.js';
+import { Config } from '../config/config.js';
+import { summarizeToolOutput, llmSummarizer, defaultSummarizer, } from './summarizer.js';
+// Mock GeminiClient and Config constructor
+vi.mock('../core/client.js');
+vi.mock('../config/config.js');
+describe('summarizers', () => {
+    let mockGeminiClient;
+    let MockConfig;
+    const abortSignal = new AbortController().signal;
+    beforeEach(() => {
+        MockConfig = vi.mocked(Config);
+        const mockConfigInstance = new MockConfig('test-api-key', 'gemini-pro', false, '.', false, undefined, false, undefined, undefined, undefined);
+        mockGeminiClient = new GeminiClient(mockConfigInstance);
+        mockGeminiClient.generateContent = vi.fn();
+        vi.spyOn(console, 'error').mockImplementation(() => { });
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+        console.error.mockRestore();
+    });
+    describe('summarizeToolOutput', () => {
+        it('should return original text if it is shorter than maxLength', async () => {
+            const shortText = 'This is a short text.';
+            const result = await summarizeToolOutput(shortText, mockGeminiClient, abortSignal, 2000);
+            expect(result).toBe(shortText);
+            expect(mockGeminiClient.generateContent).not.toHaveBeenCalled();
+        });
+        it('should return original text if it is empty', async () => {
+            const emptyText = '';
+            const result = await summarizeToolOutput(emptyText, mockGeminiClient, abortSignal, 2000);
+            expect(result).toBe(emptyText);
+            expect(mockGeminiClient.generateContent).not.toHaveBeenCalled();
+        });
+        it('should call generateContent if text is longer than maxLength', async () => {
+            const longText = 'This is a very long text.'.repeat(200);
+            const summary = 'This is a summary.';
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [{ content: { parts: [{ text: summary }] } }],
+            });
+            const result = await summarizeToolOutput(longText, mockGeminiClient, abortSignal, 2000);
+            expect(mockGeminiClient.generateContent).toHaveBeenCalledTimes(1);
+            expect(result).toBe(summary);
+        });
+        it('should return original text if generateContent throws an error', async () => {
+            const longText = 'This is a very long text.'.repeat(200);
+            const error = new Error('API Error');
+            mockGeminiClient.generateContent.mockRejectedValue(error);
+            const result = await summarizeToolOutput(longText, mockGeminiClient, abortSignal, 2000);
+            expect(mockGeminiClient.generateContent).toHaveBeenCalledTimes(1);
+            expect(result).toBe(longText);
+            expect(console.error).toHaveBeenCalledWith('Failed to summarize tool output.', error);
+        });
+        it('should construct the correct prompt for summarization', async () => {
+            const longText = 'This is a very long text.'.repeat(200);
+            const summary = 'This is a summary.';
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [{ content: { parts: [{ text: summary }] } }],
+            });
+            await summarizeToolOutput(longText, mockGeminiClient, abortSignal, 1000);
+            const expectedPrompt = `Summarize the following tool output to be a maximum of 1000 tokens. The summary should be concise and capture the main points of the tool output.
+
+The summarization should be done based on the content that is provided. Here are the basic rules to follow:
+1. If the text is a directory listing or any output that is structural, use the history of the conversation to understand the context. Using this context try to understand what information we need from the tool output and return that as a response.
+2. If the text is text content and there is nothing structural that we need, summarize the text.
+3. If the text is the output of a shell command, use the history of the conversation to understand the context. Using this context try to understand what information we need from the tool output and return a summarization along with the stack trace of any error within the <error></error> tags. The stack trace should be complete and not truncated. If there are warnings, you should include them in the summary within <warning></warning> tags.
+
+
+Text to summarize:
+"${longText}"
+
+Return the summary string which should first contain an overall summarization of text followed by the full stack trace of errors and warnings in the tool output.
+`;
+            const calledWith = mockGeminiClient.generateContent.mock
+                .calls[0];
+            const contents = calledWith[0];
+            expect(contents[0].parts[0].text).toBe(expectedPrompt);
+        });
+    });
+    describe('llmSummarizer', () => {
+        it('should summarize tool output using summarizeToolOutput', async () => {
+            const toolResult = {
+                llmContent: 'This is a very long text.'.repeat(200),
+                returnDisplay: '',
+            };
+            const summary = 'This is a summary.';
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [{ content: { parts: [{ text: summary }] } }],
+            });
+            const result = await llmSummarizer(toolResult, mockGeminiClient, abortSignal);
+            expect(mockGeminiClient.generateContent).toHaveBeenCalledTimes(1);
+            expect(result).toBe(summary);
+        });
+        it('should handle different llmContent types', async () => {
+            const longText = 'This is a very long text.'.repeat(200);
+            const toolResult = {
+                llmContent: [{ text: longText }],
+                returnDisplay: '',
+            };
+            const summary = 'This is a summary.';
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [{ content: { parts: [{ text: summary }] } }],
+            });
+            const result = await llmSummarizer(toolResult, mockGeminiClient, abortSignal);
+            expect(mockGeminiClient.generateContent).toHaveBeenCalledTimes(1);
+            const calledWith = mockGeminiClient.generateContent.mock
+                .calls[0];
+            const contents = calledWith[0];
+            expect(contents[0].parts[0].text).toContain(`"${longText}"`);
+            expect(result).toBe(summary);
+        });
+    });
+    describe('defaultSummarizer', () => {
+        it('should stringify the llmContent', async () => {
+            const toolResult = {
+                llmContent: { text: 'some data' },
+                returnDisplay: '',
+            };
+            const result = await defaultSummarizer(toolResult, mockGeminiClient, abortSignal);
+            expect(result).toBe(JSON.stringify({ text: 'some data' }));
+            expect(mockGeminiClient.generateContent).not.toHaveBeenCalled();
+        });
+    });
+});
+//# sourceMappingURL=summarizer.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"summarizer.test.js","sourceRoot":"","sources":["../../../src/utils/summarizer.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAQ,MAAM,QAAQ,CAAC;AAC/E,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EACL,mBAAmB,EACnB,aAAa,EACb,iBAAiB,GAClB,MAAM,iBAAiB,CAAC;AAGzB,2CAA2C;AAC3C,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAC7B,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,gBAA8B,CAAC;IACnC,IAAI,UAAgB,CAAC;IACrB,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,MAAM,kBAAkB,GAAG,IAAI,UAAU,CACvC,cAAc,EACd,YAAY,EACZ,KAAK,EACL,GAAG,EACH,KAAK,EACL,SAAS,EACT,KAAK,EACL,SAAS,EACT,SAAS,EACT,SAAS,CACV,CAAC;QAEF,gBAAgB,GAAG,IAAI,YAAY,CAAC,kBAAkB,CAAC,CAAC;QACvD,gBAAgB,CAAC,eAAwB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAErD,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QAClB,OAAO,CAAC,KAAc,CAAC,WAAW,EAAE,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,SAAS,GAAG,uBAAuB,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,mBAAmB,CACtC,SAAS,EACT,gBAAgB,EAChB,WAAW,EACX,IAAI,CACL,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/B,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,SAAS,GAAG,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,MAAM,mBAAmB,CACtC,SAAS,EACT,gBAAgB,EAChB,WAAW,EACX,IAAI,CACL,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/B,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG,2BAA2B,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzD,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACpC,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,mBAAmB,CACtC,QAAQ,EACR,gBAAgB,EAChB,WAAW,EACX,IAAI,CACL,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,QAAQ,GAAG,2BAA2B,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACpC,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAEpE,MAAM,MAAM,GAAG,MAAM,mBAAmB,CACtC,QAAQ,EACR,gBAAgB,EAChB,WAAW,EACX,IAAI,CACL,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACxC,kCAAkC,EAClC,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,QAAQ,GAAG,2BAA2B,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzD,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACpC,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,mBAAmB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;YAEzE,MAAM,cAAc,GAAG;;;;;;;;;GAS1B,QAAQ;;;CAGV,CAAC;YACI,MAAM,UAAU,GAAI,gBAAgB,CAAC,eAAwB,CAAC,IAAI;iBAC/D,KAAK,CAAC,CAAC,CAAC,CAAC;YACZ,MAAM,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,UAAU,GAAe;gBAC7B,UAAU,EAAE,2BAA2B,CAAC,MAAM,CAAC,GAAG,CAAC;gBACnD,aAAa,EAAE,EAAE;aAClB,CAAC;YACF,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACpC,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,UAAU,EACV,gBAAgB,EAChB,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,QAAQ,GAAG,2BAA2B,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzD,MAAM,UAAU,GAAe;gBAC7B,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;gBAChC,aAAa,EAAE,EAAE;aAClB,CAAC;YACF,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACpC,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,aAAa,CAChC,UAAU,EACV,gBAAgB,EAChB,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAClE,MAAM,UAAU,GAAI,gBAAgB,CAAC,eAAwB,CAAC,IAAI;iBAC/D,KAAK,CAAC,CAAC,CAAC,CAAC;YACZ,MAAM,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,IAAI,QAAQ,GAAG,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,UAAU,GAAe;gBAC7B,UAAU,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;gBACjC,aAAa,EAAE,EAAE;aAClB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,iBAAiB,CACpC,UAAU,EACV,gBAAgB,EAChB,WAAW,CACZ,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;YAC3D,MAAM,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,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_account.d.ts

@@ -0,0 +1,9 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare function cacheGoogleAccount(email: string): Promise<void>;
+export declare function getCachedGoogleAccount(): string | null;
+export declare function getLifetimeGoogleAccounts(): number;
+export declare function clearCachedGoogleAccount(): Promise<void>;

package/dist/src/utils/user_account.js

@@ -0,0 +1,99 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import path from 'node:path';
+import { promises as fsp, existsSync, readFileSync } from 'node:fs';
+import * as os from 'os';
+import { LLXPRT_DIR, GOOGLE_ACCOUNTS_FILENAME } from './paths.js';
+function getGoogleAccountsCachePath() {
+    return path.join(os.homedir(), LLXPRT_DIR, GOOGLE_ACCOUNTS_FILENAME);
+}
+async function readAccounts(filePath) {
+    try {
+        const content = await fsp.readFile(filePath, 'utf-8');
+        if (!content.trim()) {
+            return { active: null, old: [] };
+        }
+        return JSON.parse(content);
+    }
+    catch (error) {
+        if (error instanceof Error && 'code' in error && error.code === 'ENOENT') {
+            // File doesn't exist, which is fine.
+            return { active: null, old: [] };
+        }
+        // File is corrupted or not valid JSON, start with a fresh object.
+        console.debug('Could not parse accounts file, starting fresh.', error);
+        return { active: null, old: [] };
+    }
+}
+export async function cacheGoogleAccount(email) {
+    const filePath = getGoogleAccountsCachePath();
+    await fsp.mkdir(path.dirname(filePath), { recursive: true });
+    const accounts = await readAccounts(filePath);
+    if (accounts.active && accounts.active !== email) {
+        if (!accounts.old.includes(accounts.active)) {
+            accounts.old.push(accounts.active);
+        }
+    }
+    // If the new email was in the old list, remove it
+    accounts.old = accounts.old.filter((oldEmail) => oldEmail !== email);
+    accounts.active = email;
+    await fsp.writeFile(filePath, JSON.stringify(accounts, null, 2), 'utf-8');
+}
+export function getCachedGoogleAccount() {
+    try {
+        const filePath = getGoogleAccountsCachePath();
+        if (existsSync(filePath)) {
+            const content = readFileSync(filePath, 'utf-8').trim();
+            if (!content) {
+                return null;
+            }
+            const accounts = JSON.parse(content);
+            return accounts.active;
+        }
+        return null;
+    }
+    catch (error) {
+        console.debug('Error reading cached Google Account:', error);
+        return null;
+    }
+}
+export function getLifetimeGoogleAccounts() {
+    try {
+        const filePath = getGoogleAccountsCachePath();
+        if (!existsSync(filePath)) {
+            return 0;
+        }
+        const content = readFileSync(filePath, 'utf-8').trim();
+        if (!content) {
+            return 0;
+        }
+        const accounts = JSON.parse(content);
+        let count = accounts.old.length;
+        if (accounts.active) {
+            count++;
+        }
+        return count;
+    }
+    catch (error) {
+        console.debug('Error reading lifetime Google Accounts:', error);
+        return 0;
+    }
+}
+export async function clearCachedGoogleAccount() {
+    const filePath = getGoogleAccountsCachePath();
+    if (!existsSync(filePath)) {
+        return;
+    }
+    const accounts = await readAccounts(filePath);
+    if (accounts.active) {
+        if (!accounts.old.includes(accounts.active)) {
+            accounts.old.push(accounts.active);
+        }
+        accounts.active = null;
+    }
+    await fsp.writeFile(filePath, JSON.stringify(accounts, null, 2), 'utf-8');
+}
+//# sourceMappingURL=user_account.js.map