@didim365/agent-cli-core 0.1.0

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

@@ -0,0 +1,184 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { calculateRequestTokenCount, estimateLlmTokenCount, } from './tokenCalculation.js';
+describe('calculateRequestTokenCount', () => {
+    const mockContentGenerator = {
+        countTokens: vi.fn(),
+    };
+    const model = 'gemini-pro';
+    it('should use countTokens API for media requests (images/files)', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockResolvedValue({
+            totalTokens: 100,
+        });
+        const request = [{ inlineData: { mimeType: 'image/png', data: 'data' } }];
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        expect(count).toBe(100);
+        expect(mockContentGenerator.countTokens).toHaveBeenCalled();
+    });
+    it('should estimate tokens locally for tool calls', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockClear();
+        const request = [{ functionCall: { name: 'foo', args: { bar: 'baz' } } }];
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        // Estimation logic: JSON.stringify(part).length / 4
+        // JSON: {"functionCall":{"name":"foo","args":{"bar":"baz"}}}
+        // Length: ~53 chars. 53 / 4 = 13.25 -> 13.
+        expect(count).toBeGreaterThan(0);
+        expect(mockContentGenerator.countTokens).not.toHaveBeenCalled();
+    });
+    it('should estimate tokens locally for simple ASCII text', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockClear();
+        // 12 chars. 12 * 0.25 = 3 tokens.
+        const request = 'Hello world!';
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        expect(count).toBe(3);
+        expect(mockContentGenerator.countTokens).not.toHaveBeenCalled();
+    });
+    it('should estimate tokens locally for CJK text with higher weight', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockClear();
+        // 2 chars. 2 * 1.3 = 2.6 -> floor(2.6) = 2.
+        // Old logic would be 2/4 = 0.5 -> 0.
+        const request = '你好';
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        expect(count).toBeGreaterThanOrEqual(2);
+        expect(mockContentGenerator.countTokens).not.toHaveBeenCalled();
+    });
+    it('should handle mixed content', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockClear();
+        // 'Hi': 2 * 0.25 = 0.5
+        // '你好': 2 * 1.3 = 2.6
+        // Total: 3.1 -> 3
+        const request = 'Hi你好';
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        expect(count).toBe(3);
+        expect(mockContentGenerator.countTokens).not.toHaveBeenCalled();
+    });
+    it('should handle empty text', async () => {
+        const request = '';
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        expect(count).toBe(0);
+    });
+    it('should fallback to local estimation when countTokens API fails', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockRejectedValue(new Error('API error'));
+        const request = [
+            { text: 'Hello' },
+            { inlineData: { mimeType: 'image/png', data: 'data' } },
+        ];
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        // Should fallback to estimation:
+        // 'Hello': 5 chars * 0.25 = 1.25
+        // inlineData: 3000
+        // Total: 3001.25 -> 3001
+        expect(count).toBe(3001);
+        expect(mockContentGenerator.countTokens).toHaveBeenCalled();
+    });
+    it('should use fixed estimate for images in fallback', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockRejectedValue(new Error('API error'));
+        const request = [
+            { inlineData: { mimeType: 'image/png', data: 'large_data' } },
+        ];
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        expect(count).toBe(3000);
+    });
+    it('should use countTokens API for PDF requests', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockResolvedValue({
+            totalTokens: 5160,
+        });
+        const request = [
+            { inlineData: { mimeType: 'application/pdf', data: 'pdf_data' } },
+        ];
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        expect(count).toBe(5160);
+        expect(mockContentGenerator.countTokens).toHaveBeenCalled();
+    });
+    it('should use fixed estimate for PDFs in fallback', async () => {
+        vi.mocked(mockContentGenerator.countTokens).mockRejectedValue(new Error('API error'));
+        const request = [
+            { inlineData: { mimeType: 'application/pdf', data: 'large_pdf_data' } },
+        ];
+        const count = await calculateRequestTokenCount(request, mockContentGenerator, model);
+        // PDF estimate: 25800 tokens (~100 pages at 258 tokens/page)
+        expect(count).toBe(25800);
+    });
+});
+// =================================================================
+// 2.5.1 Provider-independent token estimation (LlmContent)
+// =================================================================
+describe('estimateLlmTokenCount', () => {
+    it('should estimate tokens for text content', () => {
+        const contents = [{ type: 'text', text: 'Hello world!' }];
+        // 12 ASCII chars * 0.25 = 3
+        expect(estimateLlmTokenCount(contents)).toBe(3);
+    });
+    it('should estimate tokens for CJK text', () => {
+        const contents = [{ type: 'text', text: '你好' }];
+        // 2 non-ASCII chars * 1.3 = 2.6 → floor = 2
+        expect(estimateLlmTokenCount(contents)).toBeGreaterThanOrEqual(2);
+    });
+    it('should estimate tokens for image content', () => {
+        const contents = [
+            {
+                type: 'image',
+                source: { type: 'base64', mediaType: 'image/png', data: 'abc123' },
+            },
+        ];
+        expect(estimateLlmTokenCount(contents)).toBe(3000);
+    });
+    it('should estimate tokens for PDF image content', () => {
+        const contents = [
+            {
+                type: 'image',
+                source: {
+                    type: 'base64',
+                    mediaType: 'application/pdf',
+                    data: 'pdf_data',
+                },
+            },
+        ];
+        expect(estimateLlmTokenCount(contents)).toBe(25800);
+    });
+    it('should estimate tokens for tool_call content', () => {
+        const contents = [
+            {
+                type: 'tool_call',
+                id: 'call-1',
+                name: 'read_file',
+                arguments: { path: '/tmp/test.txt' },
+            },
+        ];
+        // JSON.stringify length / 4
+        expect(estimateLlmTokenCount(contents)).toBeGreaterThan(0);
+    });
+    it('should estimate tokens for tool_result content', () => {
+        const contents = [
+            {
+                type: 'tool_result',
+                toolCallId: 'call-1',
+                content: 'file contents here',
+            },
+        ];
+        expect(estimateLlmTokenCount(contents)).toBeGreaterThan(0);
+    });
+    it('should estimate tokens for thought content', () => {
+        const contents = [
+            { type: 'thought', thought: 'I should think about this carefully' },
+        ];
+        expect(estimateLlmTokenCount(contents)).toBeGreaterThan(0);
+    });
+    it('should sum tokens across multiple contents', () => {
+        const contents = [
+            { type: 'text', text: 'Hello' },
+            { type: 'text', text: 'World' },
+        ];
+        // 5 + 5 = 10 ASCII chars * 0.25 = 2.5 → floor = 2
+        const result = estimateLlmTokenCount(contents);
+        expect(result).toBe(2);
+    });
+    it('should return 0 for empty array', () => {
+        expect(estimateLlmTokenCount([])).toBe(0);
+    });
+});
+//# sourceMappingURL=tokenCalculation.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tokenCalculation.test.js","sourceRoot":"","sources":["../../../src/utils/tokenCalculation.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EACL,0BAA0B,EAC1B,qBAAqB,GACtB,MAAM,uBAAuB,CAAC;AAI/B,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,MAAM,oBAAoB,GAAG;QAC3B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;KACU,CAAC;IAEjC,MAAM,KAAK,GAAG,YAAY,CAAC;IAE3B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAAC;YAC5D,WAAW,EAAE,GAAG;SACjB,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,CAAC,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAE1E,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxB,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,CAAC;QACxD,MAAM,OAAO,GAAG,CAAC,EAAE,YAAY,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QAE1E,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,oDAAoD;QACpD,6DAA6D;QAC7D,2CAA2C;QAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,CAAC;QACxD,kCAAkC;QAClC,MAAM,OAAO,GAAG,cAAc,CAAC;QAE/B,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,CAAC;QACxD,4CAA4C;QAC5C,qCAAqC;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC;QAErB,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,SAAS,EAAE,CAAC;QACxD,uBAAuB;QACvB,sBAAsB;QACtB,kBAAkB;QAClB,MAAM,OAAO,GAAG,MAAM,CAAC;QAEvB,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,OAAO,GAAG,EAAE,CAAC;QACnB,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAC3D,IAAI,KAAK,CAAC,WAAW,CAAC,CACvB,CAAC;QACF,MAAM,OAAO,GAAG;YACd,EAAE,IAAI,EAAE,OAAO,EAAE;YACjB,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;SACxD,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,iCAAiC;QACjC,iCAAiC;QACjC,mBAAmB;QACnB,yBAAyB;QACzB,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAC3D,IAAI,KAAK,CAAC,WAAW,CAAC,CACvB,CAAC;QACF,MAAM,OAAO,GAAG;YACd,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE;SAC9D,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAAC;YAC5D,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;QACH,MAAM,OAAO,GAAG;YACd,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,iBAAiB,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE;SAClE,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzB,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,iBAAiB,CAC3D,IAAI,KAAK,CAAC,WAAW,CAAC,CACvB,CAAC;QACF,MAAM,OAAO,GAAG;YACd,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,iBAAiB,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE;SACxE,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,0BAA0B,CAC5C,OAAO,EACP,oBAAoB,EACpB,KAAK,CACN,CAAC;QAEF,6DAA6D;QAC7D,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,oEAAoE;AACpE,2DAA2D;AAC3D,oEAAoE;AAEpE,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,QAAQ,GAAiB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;QAExE,4BAA4B;QAC5B,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,QAAQ,GAAiB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAE9D,4CAA4C;QAC5C,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,QAAQ,GAAiB;YAC7B;gBACE,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE;aACnE;SACF,CAAC;QAEF,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAiB;YAC7B;gBACE,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,SAAS,EAAE,iBAAiB;oBAC5B,IAAI,EAAE,UAAU;iBACjB;aACF;SACF,CAAC;QAEF,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAiB;YAC7B;gBACE,IAAI,EAAE,WAAW;gBACjB,EAAE,EAAE,QAAQ;gBACZ,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE;aACrC;SACF,CAAC;QAEF,4BAA4B;QAC5B,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,QAAQ,GAAiB;YAC7B;gBACE,IAAI,EAAE,aAAa;gBACnB,UAAU,EAAE,QAAQ;gBACpB,OAAO,EAAE,oBAAoB;aAC9B;SACF,CAAC;QAEF,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,QAAQ,GAAiB;YAC7B,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,qCAAqC,EAAE;SACpE,CAAC;QAEF,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,QAAQ,GAAiB;YAC7B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;YAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;SAChC,CAAC;QAEF,kDAAkD;QAClD,MAAM,MAAM,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/utils/tool-utils.d.ts

@@ -0,0 +1,28 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { AnyDeclarativeTool, AnyToolInvocation } from '../index.js';
+/**
+ * Generates a suggestion string for a tool name that was not found in the registry.
+ * It finds the closest matches based on Levenshtein distance.
+ * @param unknownToolName The tool name that was not found.
+ * @param allToolNames The list of all available tool names.
+ * @param topN The number of suggestions to return. Defaults to 3.
+ * @returns A suggestion string like " Did you mean 'tool'?" or " Did you mean one of: 'tool1', 'tool2'?", or an empty string if no suggestions are found.
+ */
+export declare function getToolSuggestion(unknownToolName: string, allToolNames: string[], topN?: number): string;
+/**
+ * Checks if a tool invocation matches any of a list of patterns.
+ *
+ * @param toolOrToolName The tool object or the name of the tool being invoked.
+ * @param invocation The invocation object for the tool or the command invoked.
+ * @param patterns A list of patterns to match against.
+ *   Patterns can be:
+ *   - A tool name (e.g., "ReadFileTool") to match any invocation of that tool.
+ *   - A tool name with a prefix (e.g., "ShellTool(git status)") to match
+ *     invocations where the arguments start with that prefix.
+ * @returns True if the invocation matches any pattern, false otherwise.
+ */
+export declare function doesToolInvocationMatch(toolOrToolName: AnyDeclarativeTool | string, invocation: AnyToolInvocation | string, patterns: string[]): boolean;

package/dist/src/utils/tool-utils.js

@@ -0,0 +1,96 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isTool } from '../index.js';
+import { SHELL_TOOL_NAMES } from './shell-utils.js';
+import levenshtein from 'fast-levenshtein';
+/**
+ * Generates a suggestion string for a tool name that was not found in the registry.
+ * It finds the closest matches based on Levenshtein distance.
+ * @param unknownToolName The tool name that was not found.
+ * @param allToolNames The list of all available tool names.
+ * @param topN The number of suggestions to return. Defaults to 3.
+ * @returns A suggestion string like " Did you mean 'tool'?" or " Did you mean one of: 'tool1', 'tool2'?", or an empty string if no suggestions are found.
+ */
+export function getToolSuggestion(unknownToolName, allToolNames, topN = 3) {
+    const matches = allToolNames.map((toolName) => ({
+        name: toolName,
+        distance: levenshtein.get(unknownToolName, toolName),
+    }));
+    matches.sort((a, b) => a.distance - b.distance);
+    const topNResults = matches.slice(0, topN);
+    if (topNResults.length === 0) {
+        return '';
+    }
+    const suggestedNames = topNResults
+        .map((match) => `"${match.name}"`)
+        .join(', ');
+    if (topNResults.length > 1) {
+        return ` Did you mean one of: ${suggestedNames}?`;
+    }
+    else {
+        return ` Did you mean ${suggestedNames}?`;
+    }
+}
+/**
+ * Checks if a tool invocation matches any of a list of patterns.
+ *
+ * @param toolOrToolName The tool object or the name of the tool being invoked.
+ * @param invocation The invocation object for the tool or the command invoked.
+ * @param patterns A list of patterns to match against.
+ *   Patterns can be:
+ *   - A tool name (e.g., "ReadFileTool") to match any invocation of that tool.
+ *   - A tool name with a prefix (e.g., "ShellTool(git status)") to match
+ *     invocations where the arguments start with that prefix.
+ * @returns True if the invocation matches any pattern, false otherwise.
+ */
+export function doesToolInvocationMatch(toolOrToolName, invocation, patterns) {
+    let toolNames;
+    if (isTool(toolOrToolName)) {
+        toolNames = [toolOrToolName.name, toolOrToolName.constructor.name];
+    }
+    else {
+        toolNames = [toolOrToolName];
+    }
+    if (toolNames.some((name) => SHELL_TOOL_NAMES.includes(name))) {
+        toolNames = [...new Set([...toolNames, ...SHELL_TOOL_NAMES])];
+    }
+    for (const pattern of patterns) {
+        const openParen = pattern.indexOf('(');
+        if (openParen === -1) {
+            // No arguments, just a tool name
+            if (toolNames.includes(pattern)) {
+                return true;
+            }
+            continue;
+        }
+        const patternToolName = pattern.substring(0, openParen);
+        if (!toolNames.includes(patternToolName)) {
+            continue;
+        }
+        if (!pattern.endsWith(')')) {
+            continue;
+        }
+        const argPattern = pattern.substring(openParen + 1, pattern.length - 1);
+        let command;
+        if (typeof invocation === 'string') {
+            command = invocation;
+        }
+        else {
+            if (!('command' in invocation.params)) {
+                // This invocation has no command - nothing to check.
+                continue;
+            }
+            command = String(invocation.params.command);
+        }
+        if (toolNames.some((name) => SHELL_TOOL_NAMES.includes(name))) {
+            if (command === argPattern || command.startsWith(argPattern + ' ')) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+//# sourceMappingURL=tool-utils.js.map

package/dist/src/utils/tool-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-utils.js","sourceRoot":"","sources":["../../../src/utils/tool-utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,WAAW,MAAM,kBAAkB,CAAC;AAE3C;;;;;;;GAOG;AACH,MAAM,UAAU,iBAAiB,CAC/B,eAAuB,EACvB,YAAsB,EACtB,IAAI,GAAG,CAAC;IAER,MAAM,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAC9C,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,WAAW,CAAC,GAAG,CAAC,eAAe,EAAE,QAAQ,CAAC;KACrD,CAAC,CAAC,CAAC;IAEJ,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;IAEhD,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAE3C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,cAAc,GAAG,WAAW;SAC/B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,IAAI,GAAG,CAAC;SACjC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,OAAO,yBAAyB,cAAc,GAAG,CAAC;IACpD,CAAC;SAAM,CAAC;QACN,OAAO,iBAAiB,cAAc,GAAG,CAAC;IAC5C,CAAC;AACH,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,uBAAuB,CACrC,cAA2C,EAC3C,UAAsC,EACtC,QAAkB;IAElB,IAAI,SAAmB,CAAC;IACxB,IAAI,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC;QAC3B,SAAS,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,cAAc,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACrE,CAAC;SAAM,CAAC;QACN,SAAS,GAAG,CAAC,cAAc,CAAC,CAAC;IAC/B,CAAC;IAED,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QAC9D,SAAS,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,SAAS,EAAE,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAEvC,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;YACrB,iCAAiC;YACjC,IAAI,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAChC,OAAO,IAAI,CAAC;YACd,CAAC;YACD,SAAS;QACX,CAAC;QAED,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACxD,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YACzC,SAAS;QACX,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,SAAS;QACX,CAAC;QAED,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,SAAS,GAAG,CAAC,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAExE,IAAI,OAAe,CAAC;QACpB,IAAI,OAAO,UAAU,KAAK,QAAQ,EAAE,CAAC;YACnC,OAAO,GAAG,UAAU,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,CAAC,SAAS,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBACtC,qDAAqD;gBACrD,SAAS;YACX,CAAC;YACD,OAAO,GAAG,MAAM,CAAE,UAAU,CAAC,MAA8B,CAAC,OAAO,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;YAC9D,IAAI,OAAO,KAAK,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,UAAU,GAAG,GAAG,CAAC,EAAE,CAAC;gBACnE,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/src/utils/tool-utils.test.d.ts

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

package/dist/src/utils/tool-utils.test.js

@@ -0,0 +1,84 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { expect, describe, it } from 'vitest';
+import { doesToolInvocationMatch, getToolSuggestion } from './tool-utils.js';
+import { ReadFileTool } from '../tools/read-file.js';
+import { createMockMessageBus } from '../test-utils/mock-message-bus.js';
+describe('getToolSuggestion', () => {
+    it('should suggest the top N closest tool names for a typo', () => {
+        const allToolNames = ['list_files', 'read_file', 'write_file'];
+        // Test that the right tool is selected, with only 1 result, for typos
+        const misspelledTool = getToolSuggestion('list_fils', allToolNames, 1);
+        expect(misspelledTool).toBe(' Did you mean "list_files"?');
+        // Test that the right tool is selected, with only 1 result, for prefixes
+        const prefixedTool = getToolSuggestion('github.list_files', allToolNames, 1);
+        expect(prefixedTool).toBe(' Did you mean "list_files"?');
+        // Test that the right tool is first
+        const suggestionMultiple = getToolSuggestion('list_fils', allToolNames);
+        expect(suggestionMultiple).toBe(' Did you mean one of: "list_files", "read_file", "write_file"?');
+    });
+});
+describe('doesToolInvocationMatch', () => {
+    it('should not match a partial command prefix', () => {
+        const invocation = {
+            params: { command: 'git commitsomething' },
+        };
+        const patterns = ['ShellTool(git commit)'];
+        const result = doesToolInvocationMatch('run_shell_command', invocation, patterns);
+        expect(result).toBe(false);
+    });
+    it('should match an exact command', () => {
+        const invocation = {
+            params: { command: 'git status' },
+        };
+        const patterns = ['ShellTool(git status)'];
+        const result = doesToolInvocationMatch('run_shell_command', invocation, patterns);
+        expect(result).toBe(true);
+    });
+    it('should match a command with an alias', () => {
+        const invocation = {
+            params: { command: 'wc -l' },
+        };
+        const patterns = ['ShellTool(wc)'];
+        const result = doesToolInvocationMatch('ShellTool', invocation, patterns);
+        expect(result).toBe(true);
+    });
+    it('should match a command that is a prefix', () => {
+        const invocation = {
+            params: { command: 'git status -v' },
+        };
+        const patterns = ['ShellTool(git status)'];
+        const result = doesToolInvocationMatch('run_shell_command', invocation, patterns);
+        expect(result).toBe(true);
+    });
+    describe('for non-shell tools', () => {
+        const readFileTool = new ReadFileTool({}, createMockMessageBus());
+        const invocation = {
+            params: { file: 'test.txt' },
+        };
+        it('should match by tool name', () => {
+            const patterns = ['read_file'];
+            const result = doesToolInvocationMatch(readFileTool, invocation, patterns);
+            expect(result).toBe(true);
+        });
+        it('should match by tool class name', () => {
+            const patterns = ['ReadFileTool'];
+            const result = doesToolInvocationMatch(readFileTool, invocation, patterns);
+            expect(result).toBe(true);
+        });
+        it('should not match if neither name is in the patterns', () => {
+            const patterns = ['some_other_tool', 'AnotherToolClass'];
+            const result = doesToolInvocationMatch(readFileTool, invocation, patterns);
+            expect(result).toBe(false);
+        });
+        it('should match by tool name when passed as a string', () => {
+            const patterns = ['read_file'];
+            const result = doesToolInvocationMatch('read_file', invocation, patterns);
+            expect(result).toBe(true);
+        });
+    });
+});
+//# sourceMappingURL=tool-utils.test.js.map

package/dist/src/utils/tool-utils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-utils.test.js","sourceRoot":"","sources":["../../../src/utils/tool-utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAE7E,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,oBAAoB,EAAE,MAAM,mCAAmC,CAAC;AAEzE,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,YAAY,GAAG,CAAC,YAAY,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAE/D,sEAAsE;QACtE,MAAM,cAAc,GAAG,iBAAiB,CAAC,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;QACvE,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAE3D,yEAAyE;QACzE,MAAM,YAAY,GAAG,iBAAiB,CACpC,mBAAmB,EACnB,YAAY,EACZ,CAAC,CACF,CAAC;QACF,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAEzD,oCAAoC;QACpC,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;QACxE,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAC7B,gEAAgE,CACjE,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,OAAO,EAAE,qBAAqB,EAAE;SACtB,CAAC;QACvB,MAAM,QAAQ,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,uBAAuB,CACpC,mBAAmB,EACnB,UAAU,EACV,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,OAAO,EAAE,YAAY,EAAE;SACb,CAAC;QACvB,MAAM,QAAQ,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,uBAAuB,CACpC,mBAAmB,EACnB,UAAU,EACV,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SACR,CAAC;QACvB,MAAM,QAAQ,GAAG,CAAC,eAAe,CAAC,CAAC;QACnC,MAAM,MAAM,GAAG,uBAAuB,CAAC,WAAW,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;QAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;SAChB,CAAC;QACvB,MAAM,QAAQ,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,uBAAuB,CACpC,mBAAmB,EACnB,UAAU,EACV,QAAQ,CACT,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,EAAY,EAAE,oBAAoB,EAAE,CAAC,CAAC;QAC5E,MAAM,UAAU,GAAG;YACjB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE;SACR,CAAC;QAEvB,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,uBAAuB,CACpC,YAAY,EACZ,UAAU,EACV,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAG,CAAC,cAAc,CAAC,CAAC;YAClC,MAAM,MAAM,GAAG,uBAAuB,CACpC,YAAY,EACZ,UAAU,EACV,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,CAAC,iBAAiB,EAAE,kBAAkB,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,uBAAuB,CACpC,YAAY,EACZ,UAAU,EACV,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,QAAQ,GAAG,CAAC,WAAW,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,uBAAuB,CAAC,WAAW,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AsyncLocalStorage } from 'node:async_hooks';
+/**
+ * Contextual information for a tool call execution.
+ */
+export interface ToolCallContext {
+    /** The unique ID of the tool call. */
+    callId: string;
+    /** The ID of the scheduler managing the execution. */
+    schedulerId: string;
+    /** The ID of the parent tool call, if this is a nested execution (e.g., in a subagent). */
+    parentCallId?: string;
+}
+/**
+ * AsyncLocalStorage instance for tool call context.
+ */
+export declare const toolCallContext: AsyncLocalStorage<ToolCallContext>;
+/**
+ * Runs a function within a tool call context.
+ *
+ * @param context The context to set.
+ * @param fn The function to run.
+ * @returns The result of the function.
+ */
+export declare function runWithToolCallContext<T>(context: ToolCallContext, fn: () => T): T;
+/**
+ * Retrieves the current tool call context.
+ *
+ * @returns The current ToolCallContext, or undefined if not in a context.
+ */
+export declare function getToolCallContext(): ToolCallContext | undefined;

package/dist/src/utils/toolCallContext.js

@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AsyncLocalStorage } from 'node:async_hooks';
+/**
+ * AsyncLocalStorage instance for tool call context.
+ */
+export const toolCallContext = new AsyncLocalStorage();
+/**
+ * Runs a function within a tool call context.
+ *
+ * @param context The context to set.
+ * @param fn The function to run.
+ * @returns The result of the function.
+ */
+export function runWithToolCallContext(context, fn) {
+    return toolCallContext.run(context, fn);
+}
+/**
+ * Retrieves the current tool call context.
+ *
+ * @returns The current ToolCallContext, or undefined if not in a context.
+ */
+export function getToolCallContext() {
+    return toolCallContext.getStore();
+}
+//# sourceMappingURL=toolCallContext.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"toolCallContext.js","sourceRoot":"","sources":["../../../src/utils/toolCallContext.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAcrD;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,iBAAiB,EAAmB,CAAC;AAExE;;;;;;GAMG;AACH,MAAM,UAAU,sBAAsB,CACpC,OAAwB,EACxB,EAAW;IAEX,OAAO,eAAe,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;AAC1C,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,kBAAkB;IAChC,OAAO,eAAe,CAAC,QAAQ,EAAE,CAAC;AACpC,CAAC"}

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

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

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

@@ -0,0 +1,68 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { runWithToolCallContext, getToolCallContext, } from './toolCallContext.js';
+describe('toolCallContext', () => {
+    it('should store and retrieve tool call context', () => {
+        const context = {
+            callId: 'test-call-id',
+            schedulerId: 'test-scheduler-id',
+        };
+        runWithToolCallContext(context, () => {
+            const storedContext = getToolCallContext();
+            expect(storedContext).toEqual(context);
+        });
+    });
+    it('should return undefined when no context is set', () => {
+        expect(getToolCallContext()).toBeUndefined();
+    });
+    it('should support nested contexts', () => {
+        const parentContext = {
+            callId: 'parent-call-id',
+            schedulerId: 'parent-scheduler-id',
+        };
+        const childContext = {
+            callId: 'child-call-id',
+            schedulerId: 'child-scheduler-id',
+            parentCallId: 'parent-call-id',
+        };
+        runWithToolCallContext(parentContext, () => {
+            expect(getToolCallContext()).toEqual(parentContext);
+            runWithToolCallContext(childContext, () => {
+                expect(getToolCallContext()).toEqual(childContext);
+            });
+            expect(getToolCallContext()).toEqual(parentContext);
+        });
+    });
+    it('should maintain isolation between parallel executions', async () => {
+        const context1 = {
+            callId: 'call-1',
+            schedulerId: 'scheduler-1',
+        };
+        const context2 = {
+            callId: 'call-2',
+            schedulerId: 'scheduler-2',
+        };
+        const promise1 = new Promise((resolve) => {
+            runWithToolCallContext(context1, () => {
+                setTimeout(() => {
+                    expect(getToolCallContext()).toEqual(context1);
+                    resolve();
+                }, 10);
+            });
+        });
+        const promise2 = new Promise((resolve) => {
+            runWithToolCallContext(context2, () => {
+                setTimeout(() => {
+                    expect(getToolCallContext()).toEqual(context2);
+                    resolve();
+                }, 5);
+            });
+        });
+        await Promise.all([promise1, promise2]);
+    });
+});
+//# sourceMappingURL=toolCallContext.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"toolCallContext.test.js","sourceRoot":"","sources":["../../../src/utils/toolCallContext.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,sBAAsB,EACtB,kBAAkB,GACnB,MAAM,sBAAsB,CAAC;AAE9B,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,OAAO,GAAG;YACd,MAAM,EAAE,cAAc;YACtB,WAAW,EAAE,mBAAmB;SACjC,CAAC;QAEF,sBAAsB,CAAC,OAAO,EAAE,GAAG,EAAE;YACnC,MAAM,aAAa,GAAG,kBAAkB,EAAE,CAAC;YAC3C,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,aAAa,GAAG;YACpB,MAAM,EAAE,gBAAgB;YACxB,WAAW,EAAE,qBAAqB;SACnC,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,MAAM,EAAE,eAAe;YACvB,WAAW,EAAE,oBAAoB;YACjC,YAAY,EAAE,gBAAgB;SAC/B,CAAC;QAEF,sBAAsB,CAAC,aAAa,EAAE,GAAG,EAAE;YACzC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAEpD,sBAAsB,CAAC,YAAY,EAAE,GAAG,EAAE;gBACxC,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE,QAAQ;YAChB,WAAW,EAAE,aAAa;SAC3B,CAAC;QAEF,MAAM,QAAQ,GAAG;YACf,MAAM,EAAE,QAAQ;YAChB,WAAW,EAAE,aAAa;SAC3B,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAC7C,sBAAsB,CAAC,QAAQ,EAAE,GAAG,EAAE;gBACpC,UAAU,CAAC,GAAG,EAAE;oBACd,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;oBAC/C,OAAO,EAAE,CAAC;gBACZ,CAAC,EAAE,EAAE,CAAC,CAAC;YACT,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAC7C,sBAAsB,CAAC,QAAQ,EAAE,GAAG,EAAE;gBACpC,UAAU,CAAC,GAAG,EAAE;oBACd,MAAM,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;oBAC/C,OAAO,EAAE,CAAC;gBACZ,CAAC,EAAE,CAAC,CAAC,CAAC;YACR,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,20 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare class UserAccountManager {
+    private getGoogleAccountsCachePath;
+    /**
+     * Parses and validates the string content of an accounts file.
+     * @param content The raw string content from the file.
+     * @returns A valid UserAccounts object.
+     */
+    private parseAndValidateAccounts;
+    private readAccountsSync;
+    private readAccounts;
+    cacheGoogleAccount(email: string): Promise<void>;
+    getCachedGoogleAccount(): string | null;
+    getLifetimeGoogleAccounts(): number;
+    clearCachedGoogleAccount(): Promise<void>;
+}