@terminai/core 0.21.0

package/dist/src/tools/web-search.js

@@ -0,0 +1,142 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { WEB_SEARCH_TOOL_NAME } from './tool-names.js';
+import { BaseDeclarativeTool, BaseToolInvocation, Kind } from './tools.js';
+import { ToolErrorType } from './tool-error.js';
+import { getErrorMessage } from '../utils/errors.js';
+import {} from '../config/config.js';
+import { getResponseText } from '../utils/partUtils.js';
+class WebSearchToolInvocation extends BaseToolInvocation {
+    config;
+    constructor(config, params, messageBus, _toolName, _toolDisplayName) {
+        super(params, messageBus, _toolName, _toolDisplayName);
+        this.config = config;
+    }
+    getDescription() {
+        return `Searching the web for: "${this.params.query}"`;
+    }
+    async execute(signal) {
+        const geminiClient = this.config.getGeminiClient();
+        try {
+            const response = await geminiClient.generateContent({ model: 'web-search' }, [{ role: 'user', parts: [{ text: this.params.query }] }], signal);
+            const responseText = getResponseText(response);
+            const groundingMetadata = response.candidates?.[0]?.groundingMetadata;
+            const sources = groundingMetadata?.groundingChunks;
+            const groundingSupports = groundingMetadata?.groundingSupports;
+            if (!responseText || !responseText.trim()) {
+                return {
+                    llmContent: `No search results or information found for query: "${this.params.query}"`,
+                    returnDisplay: 'No information found.',
+                };
+            }
+            let modifiedResponseText = responseText;
+            const sourceListFormatted = [];
+            if (sources && sources.length > 0) {
+                sources.forEach((source, index) => {
+                    const title = source.web?.title || 'Untitled';
+                    const uri = source.web?.uri || 'No URI';
+                    sourceListFormatted.push(`[${index + 1}] ${title} (${uri})`);
+                });
+                if (groundingSupports && groundingSupports.length > 0) {
+                    const insertions = [];
+                    groundingSupports.forEach((support) => {
+                        if (support.segment && support.groundingChunkIndices) {
+                            const citationMarker = support.groundingChunkIndices
+                                .map((chunkIndex) => `[${chunkIndex + 1}]`)
+                                .join('');
+                            insertions.push({
+                                index: support.segment.endIndex,
+                                marker: citationMarker,
+                            });
+                        }
+                    });
+                    // Sort insertions by index in descending order to avoid shifting subsequent indices
+                    insertions.sort((a, b) => b.index - a.index);
+                    // Use TextEncoder/TextDecoder since segment indices are UTF-8 byte positions
+                    const encoder = new TextEncoder();
+                    const responseBytes = encoder.encode(modifiedResponseText);
+                    const parts = [];
+                    let lastIndex = responseBytes.length;
+                    for (const ins of insertions) {
+                        const pos = Math.min(ins.index, lastIndex);
+                        parts.unshift(responseBytes.subarray(pos, lastIndex));
+                        parts.unshift(encoder.encode(ins.marker));
+                        lastIndex = pos;
+                    }
+                    parts.unshift(responseBytes.subarray(0, lastIndex));
+                    // Concatenate all parts into a single buffer
+                    const totalLength = parts.reduce((sum, part) => sum + part.length, 0);
+                    const finalBytes = new Uint8Array(totalLength);
+                    let offset = 0;
+                    for (const part of parts) {
+                        finalBytes.set(part, offset);
+                        offset += part.length;
+                    }
+                    modifiedResponseText = new TextDecoder().decode(finalBytes);
+                }
+                if (sourceListFormatted.length > 0) {
+                    modifiedResponseText +=
+                        '\n\nSources:\n' + sourceListFormatted.join('\n');
+                }
+            }
+            return {
+                llmContent: `Web search results for "${this.params.query}":\n\n${modifiedResponseText}`,
+                returnDisplay: `Search results for "${this.params.query}" returned.`,
+                sources,
+            };
+        }
+        catch (error) {
+            const errorMessage = `Error during web search for query "${this.params.query}": ${getErrorMessage(error)}`;
+            console.error(errorMessage, error);
+            return {
+                llmContent: `Error: ${errorMessage}`,
+                returnDisplay: `Error performing web search.`,
+                error: {
+                    message: errorMessage,
+                    type: ToolErrorType.WEB_SEARCH_FAILED,
+                },
+            };
+        }
+    }
+}
+/**
+ * A tool to perform web searches using Google Search via the Gemini API.
+ */
+export class WebSearchTool extends BaseDeclarativeTool {
+    config;
+    static Name = WEB_SEARCH_TOOL_NAME;
+    constructor(config, messageBus) {
+        super(WebSearchTool.Name, 'GoogleSearch', 'Performs a web search using Google Search (via the Gemini API) and returns the results. This tool is useful for finding information on the internet based on a query.', Kind.Search, {
+            type: 'object',
+            properties: {
+                query: {
+                    type: 'string',
+                    description: 'The search query to find information on the web.',
+                },
+            },
+            required: ['query'],
+        }, true, // isOutputMarkdown
+        false, // canUpdateOutput
+        messageBus);
+        this.config = config;
+    }
+    /**
+     * Validates the parameters for the WebSearchTool.
+     * @param params The parameters to validate
+     * @returns An error message string if validation fails, null if valid
+     */
+    validateToolParamValues(params) {
+        if (!params.query || params.query.trim() === '') {
+            return "The 'query' parameter cannot be empty.";
+        }
+        return null;
+    }
+    createInvocation(params, messageBus, _toolName, _toolDisplayName) {
+        return new WebSearchToolInvocation(this.config, params, messageBus, _toolName, _toolDisplayName);
+    }
+}
+//# sourceMappingURL=web-search.js.map

package/dist/src/tools/web-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"web-search.js","sourceRoot":"","sources":["../../../src/tools/web-search.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAGvD,OAAO,EAAE,mBAAmB,EAAE,kBAAkB,EAAE,IAAI,EAAE,MAAM,YAAY,CAAC;AAC3E,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAe,MAAM,qBAAqB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AA4CxD,MAAM,uBAAwB,SAAQ,kBAGrC;IAEoB;IADnB,YACmB,MAAc,EAC/B,MAA2B,EAC3B,UAAuB,EACvB,SAAkB,EAClB,gBAAyB;QAEzB,KAAK,CAAC,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC;QANtC,WAAM,GAAN,MAAM,CAAQ;IAOjC,CAAC;IAEQ,cAAc;QACrB,OAAO,2BAA2B,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;QAEnD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,YAAY,CAAC,eAAe,CACjD,EAAE,KAAK,EAAE,YAAY,EAAE,EACvB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,EAAE,CAAC,EACxD,MAAM,CACP,CAAC;YAEF,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;YAC/C,MAAM,iBAAiB,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC;YACtE,MAAM,OAAO,GAAG,iBAAiB,EAAE,eAEtB,CAAC;YACd,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,iBAEhC,CAAC;YAEd,IAAI,CAAC,YAAY,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC1C,OAAO;oBACL,UAAU,EAAE,sDAAsD,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG;oBACtF,aAAa,EAAE,uBAAuB;iBACvC,CAAC;YACJ,CAAC;YAED,IAAI,oBAAoB,GAAG,YAAY,CAAC;YACxC,MAAM,mBAAmB,GAAa,EAAE,CAAC;YAEzC,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,OAAO,CAAC,OAAO,CAAC,CAAC,MAA0B,EAAE,KAAa,EAAE,EAAE;oBAC5D,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,EAAE,KAAK,IAAI,UAAU,CAAC;oBAC9C,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,QAAQ,CAAC;oBACxC,mBAAmB,CAAC,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,KAAK,KAAK,KAAK,GAAG,GAAG,CAAC,CAAC;gBAC/D,CAAC,CAAC,CAAC;gBAEH,IAAI,iBAAiB,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtD,MAAM,UAAU,GAA6C,EAAE,CAAC;oBAChE,iBAAiB,CAAC,OAAO,CAAC,CAAC,OAA6B,EAAE,EAAE;wBAC1D,IAAI,OAAO,CAAC,OAAO,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;4BACrD,MAAM,cAAc,GAAG,OAAO,CAAC,qBAAqB;iCACjD,GAAG,CAAC,CAAC,UAAkB,EAAE,EAAE,CAAC,IAAI,UAAU,GAAG,CAAC,GAAG,CAAC;iCAClD,IAAI,CAAC,EAAE,CAAC,CAAC;4BACZ,UAAU,CAAC,IAAI,CAAC;gCACd,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ;gCAC/B,MAAM,EAAE,cAAc;6BACvB,CAAC,CAAC;wBACL,CAAC;oBACH,CAAC,CAAC,CAAC;oBAEH,oFAAoF;oBACpF,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;oBAE7C,6EAA6E;oBAC7E,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;oBAClC,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;oBAC3D,MAAM,KAAK,GAAiB,EAAE,CAAC;oBAC/B,IAAI,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC;oBACrC,KAAK,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;wBAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;wBAC3C,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC;wBACtD,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;wBAC1C,SAAS,GAAG,GAAG,CAAC;oBAClB,CAAC;oBACD,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC;oBAEpD,6CAA6C;oBAC7C,MAAM,WAAW,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;oBACtE,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;oBAC/C,IAAI,MAAM,GAAG,CAAC,CAAC;oBACf,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;wBAC7B,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC;oBACxB,CAAC;oBACD,oBAAoB,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBAC9D,CAAC;gBAED,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACnC,oBAAoB;wBAClB,gBAAgB,GAAG,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;YAED,OAAO;gBACL,UAAU,EAAE,2BAA2B,IAAI,CAAC,MAAM,CAAC,KAAK,SAAS,oBAAoB,EAAE;gBACvF,aAAa,EAAE,uBAAuB,IAAI,CAAC,MAAM,CAAC,KAAK,aAAa;gBACpE,OAAO;aACR,CAAC;QACJ,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,MAAM,YAAY,GAAG,sCACnB,IAAI,CAAC,MAAM,CAAC,KACd,MAAM,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;YACnC,OAAO;gBACL,UAAU,EAAE,UAAU,YAAY,EAAE;gBACpC,aAAa,EAAE,8BAA8B;gBAC7C,KAAK,EAAE;oBACL,OAAO,EAAE,YAAY;oBACrB,IAAI,EAAE,aAAa,CAAC,iBAAiB;iBACtC;aACF,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,aAAc,SAAQ,mBAGlC;IAIoB;IAHnB,MAAM,CAAU,IAAI,GAAG,oBAAoB,CAAC;IAE5C,YACmB,MAAc,EAC/B,UAAuB;QAEvB,KAAK,CACH,aAAa,CAAC,IAAI,EAClB,cAAc,EACd,uKAAuK,EACvK,IAAI,CAAC,MAAM,EACX;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,KAAK,EAAE;oBACL,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,kDAAkD;iBAChE;aACF;YACD,QAAQ,EAAE,CAAC,OAAO,CAAC;SACpB,EACD,IAAI,EAAE,mBAAmB;QACzB,KAAK,EAAE,kBAAkB;QACzB,UAAU,CACX,CAAC;QArBe,WAAM,GAAN,MAAM,CAAQ;IAsBjC,CAAC;IAED;;;;OAIG;IACgB,uBAAuB,CACxC,MAA2B;QAE3B,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAChD,OAAO,wCAAwC,CAAC;QAClD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAES,gBAAgB,CACxB,MAA2B,EAC3B,UAAuB,EACvB,SAAkB,EAClB,gBAAyB;QAEzB,OAAO,IAAI,uBAAuB,CAChC,IAAI,CAAC,MAAM,EACX,MAAM,EACN,UAAU,EACV,SAAS,EACT,gBAAgB,CACjB,CAAC;IACJ,CAAC"}

package/dist/src/tools/web-search.test.d.ts

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

package/dist/src/tools/web-search.test.js

@@ -0,0 +1,242 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { WebSearchTool } from './web-search.js';
+import { GeminiClient } from '../core/client.js';
+import { ToolErrorType } from './tool-error.js';
+// Mock GeminiClient and Config constructor
+vi.mock('../core/client.js');
+vi.mock('../config/config.js');
+describe('WebSearchTool', () => {
+    const abortSignal = new AbortController().signal;
+    let mockGeminiClient;
+    let tool;
+    beforeEach(() => {
+        const mockConfigInstance = {
+            getGeminiClient: () => mockGeminiClient,
+            getProxy: () => undefined,
+            generationConfigService: {
+                getResolvedConfig: vi.fn().mockImplementation(({ model }) => ({
+                    model,
+                    sdkConfig: {},
+                })),
+            },
+        };
+        mockGeminiClient = new GeminiClient(mockConfigInstance);
+        tool = new WebSearchTool(mockConfigInstance);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('build', () => {
+        it('should return an invocation for a valid query', () => {
+            const params = { query: 'test query' };
+            const invocation = tool.build(params);
+            expect(invocation).toBeDefined();
+            expect(invocation.params).toEqual(params);
+        });
+        it('should throw an error for an empty query', () => {
+            const params = { query: '' };
+            expect(() => tool.build(params)).toThrow("The 'query' parameter cannot be empty.");
+        });
+        it('should throw an error for a query with only whitespace', () => {
+            const params = { query: '   ' };
+            expect(() => tool.build(params)).toThrow("The 'query' parameter cannot be empty.");
+        });
+    });
+    describe('getDescription', () => {
+        it('should return a description of the search', () => {
+            const params = { query: 'test query' };
+            const invocation = tool.build(params);
+            expect(invocation.getDescription()).toBe('Searching the web for: "test query"');
+        });
+    });
+    describe('execute', () => {
+        it('should return search results for a successful query', async () => {
+            const params = { query: 'successful query' };
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [
+                    {
+                        content: {
+                            role: 'model',
+                            parts: [{ text: 'Here are your results.' }],
+                        },
+                    },
+                ],
+            });
+            const invocation = tool.build(params);
+            const result = await invocation.execute(abortSignal);
+            expect(result.llmContent).toBe('Web search results for "successful query":\n\nHere are your results.');
+            expect(result.returnDisplay).toBe('Search results for "successful query" returned.');
+            expect(result.sources).toBeUndefined();
+        });
+        it('should handle no search results found', async () => {
+            const params = { query: 'no results query' };
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [
+                    {
+                        content: {
+                            role: 'model',
+                            parts: [{ text: '' }],
+                        },
+                    },
+                ],
+            });
+            const invocation = tool.build(params);
+            const result = await invocation.execute(abortSignal);
+            expect(result.llmContent).toBe('No search results or information found for query: "no results query"');
+            expect(result.returnDisplay).toBe('No information found.');
+        });
+        it('should return a WEB_SEARCH_FAILED error on failure', async () => {
+            const params = { query: 'error query' };
+            const testError = new Error('API Failure');
+            mockGeminiClient.generateContent.mockRejectedValue(testError);
+            const invocation = tool.build(params);
+            const result = await invocation.execute(abortSignal);
+            expect(result.error?.type).toBe(ToolErrorType.WEB_SEARCH_FAILED);
+            expect(result.llmContent).toContain('Error:');
+            expect(result.llmContent).toContain('API Failure');
+            expect(result.returnDisplay).toBe('Error performing web search.');
+        });
+        it('should correctly format results with sources and citations', async () => {
+            const params = { query: 'grounding query' };
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [
+                    {
+                        content: {
+                            role: 'model',
+                            parts: [{ text: 'This is a test response.' }],
+                        },
+                        groundingMetadata: {
+                            groundingChunks: [
+                                { web: { uri: 'https://example.com', title: 'Example Site' } },
+                                { web: { uri: 'https://google.com', title: 'Google' } },
+                            ],
+                            groundingSupports: [
+                                {
+                                    segment: { startIndex: 5, endIndex: 14 },
+                                    groundingChunkIndices: [0],
+                                },
+                                {
+                                    segment: { startIndex: 15, endIndex: 24 },
+                                    groundingChunkIndices: [0, 1],
+                                },
+                            ],
+                        },
+                    },
+                ],
+            });
+            const invocation = tool.build(params);
+            const result = await invocation.execute(abortSignal);
+            const expectedLlmContent = `Web search results for "grounding query":
+
+This is a test[1] response.[1][2]
+
+Sources:
+[1] Example Site (https://example.com)
+[2] Google (https://google.com)`;
+            expect(result.llmContent).toBe(expectedLlmContent);
+            expect(result.returnDisplay).toBe('Search results for "grounding query" returned.');
+            expect(result.sources).toHaveLength(2);
+        });
+        it('should append sources when grounding supports are missing', async () => {
+            const params = { query: 'sources only query' };
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [
+                    {
+                        content: {
+                            role: 'model',
+                            parts: [{ text: 'Short response without citations.' }],
+                        },
+                        groundingMetadata: {
+                            groundingChunks: [
+                                { web: { uri: 'https://example.com', title: 'Example Site' } },
+                            ],
+                        },
+                    },
+                ],
+            });
+            const invocation = tool.build(params);
+            const result = await invocation.execute(abortSignal);
+            const expectedLlmContent = `Web search results for "sources only query":
+
+Short response without citations.
+
+Sources:
+[1] Example Site (https://example.com)`;
+            expect(result.llmContent).toBe(expectedLlmContent);
+            expect(result.sources).toHaveLength(1);
+        });
+        it('should insert markers at correct byte positions for multibyte text', async () => {
+            const params = { query: 'multibyte query' };
+            mockGeminiClient.generateContent.mockResolvedValue({
+                candidates: [
+                    {
+                        content: {
+                            role: 'model',
+                            parts: [{ text: 'こんにちは! Gemini CLI✨️' }],
+                        },
+                        groundingMetadata: {
+                            groundingChunks: [
+                                {
+                                    web: {
+                                        title: 'Japanese Greeting',
+                                        uri: 'https://example.test/japanese-greeting',
+                                    },
+                                },
+                                {
+                                    web: {
+                                        title: 'google-gemini/gemini-cli',
+                                        uri: 'https://github.com/google-gemini/gemini-cli',
+                                    },
+                                },
+                                {
+                                    web: {
+                                        title: 'Gemini CLI: your open-source AI agent',
+                                        uri: 'https://blog.google/technology/developers/introducing-gemini-cli-open-source-ai-agent/',
+                                    },
+                                },
+                            ],
+                            groundingSupports: [
+                                {
+                                    segment: {
+                                        // Byte range of "こんにちは!" (utf-8 encoded)
+                                        startIndex: 0,
+                                        endIndex: 16,
+                                    },
+                                    groundingChunkIndices: [0],
+                                },
+                                {
+                                    segment: {
+                                        // Byte range of "Gemini CLI✨️" (utf-8 encoded)
+                                        startIndex: 17,
+                                        endIndex: 33,
+                                    },
+                                    groundingChunkIndices: [1, 2],
+                                },
+                            ],
+                        },
+                    },
+                ],
+            });
+            const invocation = tool.build(params);
+            const result = await invocation.execute(abortSignal);
+            const expectedLlmContent = `Web search results for "multibyte query":
+
+こんにちは![1] Gemini CLI✨️[2][3]
+
+Sources:
+[1] Japanese Greeting (https://example.test/japanese-greeting)
+[2] google-gemini/gemini-cli (https://github.com/google-gemini/gemini-cli)
+[3] Gemini CLI: your open-source AI agent (https://blog.google/technology/developers/introducing-gemini-cli-open-source-ai-agent/)`;
+            expect(result.llmContent).toBe(expectedLlmContent);
+            expect(result.returnDisplay).toBe('Search results for "multibyte query" returned.');
+            expect(result.sources).toHaveLength(3);
+        });
+    });
+});
+//# sourceMappingURL=web-search.test.js.map

package/dist/src/tools/web-search.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"web-search.test.js","sourceRoot":"","sources":["../../../src/tools/web-search.test.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AAEzE,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,2CAA2C;AAC3C,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;AAC7B,EAAE,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AAE/B,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,WAAW,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;IACjD,IAAI,gBAA8B,CAAC;IACnC,IAAI,IAAmB,CAAC;IAExB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,kBAAkB,GAAG;YACzB,eAAe,EAAE,GAAG,EAAE,CAAC,gBAAgB;YACvC,QAAQ,EAAE,GAAG,EAAE,CAAC,SAAS;YACzB,uBAAuB,EAAE;gBACvB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;oBAC5D,KAAK;oBACL,SAAS,EAAE,EAAE;iBACd,CAAC,CAAC;aACJ;SACmB,CAAC;QACvB,gBAAgB,GAAG,IAAI,YAAY,CAAC,kBAAkB,CAAC,CAAC;QACxD,IAAI,GAAG,IAAI,aAAa,CAAC,kBAAkB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;YAC5D,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;YACjC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAClD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CACtC,wCAAwC,CACzC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;YACrD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CACtC,wCAAwC,CACzC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;YAC5D,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CACtC,qCAAqC,CACtC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;YACjE,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,CAAC;yBAC5C;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAC5B,sEAAsE,CACvE,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAC/B,iDAAiD,CAClD,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;YACjE,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;yBACtB;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAC5B,sEAAsE,CACvE,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;YAC7D,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;YAC1C,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAExE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAErD,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC;YAChE,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC;yBAC9C;wBACD,iBAAiB,EAAE;4BACjB,eAAe,EAAE;gCACf,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,qBAAqB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;gCAC9D,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,oBAAoB,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE;6BACxD;4BACD,iBAAiB,EAAE;gCACjB;oCACE,OAAO,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;oCACxC,qBAAqB,EAAE,CAAC,CAAC,CAAC;iCAC3B;gCACD;oCACE,OAAO,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;oCACzC,qBAAqB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;iCAC9B;6BACF;yBACF;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAErD,MAAM,kBAAkB,GAAG;;;;;;gCAMD,CAAC;YAE3B,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAC/B,gDAAgD,CACjD,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC;YACnE,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,mCAAmC,EAAE,CAAC;yBACvD;wBACD,iBAAiB,EAAE;4BACjB,eAAe,EAAE;gCACf,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,qBAAqB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;6BAC/D;yBACF;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAErD,MAAM,kBAAkB,GAAG;;;;;uCAKM,CAAC;YAElC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,MAAM,GAAwB,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC;YAChE,gBAAgB,CAAC,eAAwB,CAAC,iBAAiB,CAAC;gBAC3D,UAAU,EAAE;oBACV;wBACE,OAAO,EAAE;4BACP,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,qBAAqB,EAAE,CAAC;yBACzC;wBACD,iBAAiB,EAAE;4BACjB,eAAe,EAAE;gCACf;oCACE,GAAG,EAAE;wCACH,KAAK,EAAE,mBAAmB;wCAC1B,GAAG,EAAE,wCAAwC;qCAC9C;iCACF;gCACD;oCACE,GAAG,EAAE;wCACH,KAAK,EAAE,0BAA0B;wCACjC,GAAG,EAAE,6CAA6C;qCACnD;iCACF;gCACD;oCACE,GAAG,EAAE;wCACH,KAAK,EAAE,uCAAuC;wCAC9C,GAAG,EAAE,wFAAwF;qCAC9F;iCACF;6BACF;4BACD,iBAAiB,EAAE;gCACjB;oCACE,OAAO,EAAE;wCACP,yCAAyC;wCACzC,UAAU,EAAE,CAAC;wCACb,QAAQ,EAAE,EAAE;qCACb;oCACD,qBAAqB,EAAE,CAAC,CAAC,CAAC;iCAC3B;gCACD;oCACE,OAAO,EAAE;wCACP,+CAA+C;wCAC/C,UAAU,EAAE,EAAE;wCACd,QAAQ,EAAE,EAAE;qCACb;oCACD,qBAAqB,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;iCAC9B;6BACF;yBACF;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAErD,MAAM,kBAAkB,GAAG;;;;;;;mIAOkG,CAAC;YAE9H,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAC/B,gDAAgD,CACjD,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/write-file.d.ts

@@ -0,0 +1,54 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * Portions Copyright 2025 TerminaI Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { ToolInvocation, ToolResult } from './tools.js';
+import { BaseDeclarativeTool } from './tools.js';
+import type { ModifiableDeclarativeTool, ModifyContext } from './modifiable-tool.js';
+import type { MessageBus } from '../confirmation-bus/message-bus.js';
+/**
+ * Parameters for the WriteFile tool
+ */
+export interface WriteFileToolParams {
+    /**
+     * The absolute path to the file to write to
+     */
+    file_path: string;
+    /**
+     * The content to write to the file
+     */
+    content: string;
+    /**
+     * Whether the proposed content was modified by the user.
+     */
+    modified_by_user?: boolean;
+    /**
+     * Initially proposed content.
+     */
+    ai_proposed_content?: string;
+}
+interface GetCorrectedFileContentResult {
+    originalContent: string;
+    correctedContent: string;
+    fileExists: boolean;
+    error?: {
+        message: string;
+        code?: string;
+    };
+}
+export declare function getCorrectedFileContent(config: Config, filePath: string, proposedContent: string, abortSignal: AbortSignal): Promise<GetCorrectedFileContentResult>;
+/**
+ * Implementation of the WriteFile tool logic
+ */
+export declare class WriteFileTool extends BaseDeclarativeTool<WriteFileToolParams, ToolResult> implements ModifiableDeclarativeTool<WriteFileToolParams> {
+    private readonly config;
+    static readonly Name = "write_to_file";
+    constructor(config: Config, messageBus?: MessageBus);
+    protected validateToolParamValues(params: WriteFileToolParams): string | null;
+    protected createInvocation(params: WriteFileToolParams): ToolInvocation<WriteFileToolParams, ToolResult>;
+    getModifyContext(abortSignal: AbortSignal): ModifyContext<WriteFileToolParams>;
+}
+export {};