@vybestack/llxprt-code-tools 0.10.0-nightly.260613.1adad3b34

package/dist/src/tools/google-web-fetch.js

@@ -0,0 +1,417 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BaseDeclarativeTool, BaseToolInvocation, Kind, ToolConfirmationOutcome, } from './tools.js';
+import { GOOGLE_WEB_FETCH_TOOL } from '../types/tool-names.js';
+import { ToolErrorType } from '../types/tool-error.js';
+import { getErrorMessage } from '../utils/errors.js';
+import { fetchWithTimeout, isPrivateIp } from '../utils/fetch.js';
+import { convert } from 'html-to-text';
+import { debugLogger as logger } from '../utils/debugLogger.js';
+const URL_FETCH_TIMEOUT_MS = 10000;
+const MAX_CONTENT_LENGTH = 100000;
+/**
+ * Parses a prompt to extract valid URLs and identify malformed ones.
+ * Strips common trailing punctuation from tokens before URL validation.
+ */
+export function parsePrompt(text) {
+    const tokens = text.split(/\s+/);
+    const validUrls = [];
+    const errors = [];
+    const stripTrailingPunctuation = (input) => {
+        let endIndex = input.length;
+        // eslint-disable-next-line sonarjs/too-many-break-or-continue-in-loop -- Existing structure is intentionally preserved; refactoring this boundary is outside the lint slice.
+        while (endIndex > 0) {
+            const lastChar = input[endIndex - 1];
+            if (
+            // eslint-disable-next-line sonarjs/expression-complexity -- Existing structure is intentionally preserved; refactoring this boundary is outside the lint slice.
+            lastChar === '.' ||
+                lastChar === ',' ||
+                lastChar === ';' ||
+                lastChar === ':' ||
+                lastChar === '!' ||
+                lastChar === '?') {
+                endIndex--;
+                continue;
+            }
+            break;
+        }
+        return input.slice(0, endIndex);
+    };
+    for (const token of tokens) {
+        if (!token)
+            continue;
+        // Heuristic to check if the token appears to contain URL-like chars.
+        if (token.includes('://')) {
+            // Strip common trailing punctuation (period, comma, semicolon, colon, etc.)
+            // This handles natural language like "Check https://example.com."
+            const cleaned = stripTrailingPunctuation(token);
+            try {
+                // Validate with new URL()
+                const url = new URL(cleaned);
+                // Allowlist protocols
+                // eslint-disable-next-line sonarjs/nested-control-flow -- Existing structure is intentionally preserved; refactoring this boundary is outside the lint slice.
+                if (['http:', 'https:'].includes(url.protocol)) {
+                    validUrls.push(url.href);
+                }
+                else {
+                    errors.push(`Unsupported protocol in URL: "${cleaned}". Only http and https are supported.`);
+                }
+            }
+            catch {
+                // Malformed URL according to WHATWG standard.
+                errors.push(`Malformed URL detected: "${cleaned}".`);
+            }
+        }
+    }
+    return { validUrls, errors };
+}
+function getResponseText(response) {
+    const parts = response.candidates?.[0]?.content?.parts;
+    if (!parts) {
+        return undefined;
+    }
+    const textSegments = parts
+        .map((part) => part.text)
+        .filter((text) => typeof text === 'string');
+    return textSegments.length > 0 ? textSegments.join('') : undefined;
+}
+class GoogleWebFetchToolInvocation extends BaseToolInvocation {
+    host;
+    constructor(host, params, messageBus, toolName, displayName) {
+        super(params, messageBus, toolName, displayName);
+        this.host = host;
+    }
+    getToolName() {
+        return GoogleWebFetchTool.Name;
+    }
+    async executeFallback(_signal) {
+        const { validUrls: urls } = parsePrompt(this.params.prompt);
+        // For now, we only support one URL for fallback
+        let url = urls[0];
+        // Convert GitHub blob URL to raw URL
+        // Convert GitHub blob URL to raw URL
+        try {
+            const urlObj = new URL(url);
+            if (urlObj.hostname === 'github.com' && url.includes('/blob/')) {
+                url = url
+                    .replace('github.com', 'raw.githubusercontent.com')
+                    .replace('/blob/', '/');
+            }
+        }
+        catch {
+            // Invalid URL; fetchWithTimeout will handle the error.
+        }
+        try {
+            const response = await fetchWithTimeout(url, URL_FETCH_TIMEOUT_MS, _signal);
+            if (!response.ok) {
+                throw new Error(`Request failed with status code ${response.status} ${response.statusText}`);
+            }
+            const rawContent = await response.text();
+            // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: get() returns string | null, empty string should fall through
+            const contentType = response.headers.get('content-type') || '';
+            let textContent;
+            // Only use html-to-text if content type is HTML, or if no content type is provided (assume HTML)
+            if (contentType.toLowerCase().includes('text/html') ||
+                contentType === '') {
+                textContent = convert(rawContent, {
+                    wordwrap: false,
+                    selectors: [
+                        { selector: 'a', options: { ignoreHref: true } },
+                        { selector: 'img', format: 'skip' },
+                    ],
+                });
+            }
+            else {
+                // For other content types (text/plain, application/json, etc.), use raw text
+                textContent = rawContent;
+            }
+            textContent = textContent.substring(0, MAX_CONTENT_LENGTH);
+            // For private URLs in llxprt, we return raw content without AI processing
+            // This preserves our multi-provider approach
+            return {
+                llmContent: textContent,
+                returnDisplay: `Content for ${url} fetched directly.`,
+            };
+        }
+        catch (e) {
+            const error = e;
+            const errorMessage = `Error during fallback fetch for ${url}: ${error.message}`;
+            return {
+                llmContent: `Error: ${errorMessage}`,
+                returnDisplay: `Error: ${errorMessage}`,
+                error: {
+                    message: errorMessage,
+                    type: ToolErrorType.WEB_FETCH_FALLBACK_FAILED,
+                },
+            };
+        }
+    }
+    getDescription() {
+        const displayPrompt = this.params.prompt.length > 100
+            ? this.params.prompt.substring(0, 97) + '...'
+            : this.params.prompt;
+        return `Processing URLs and instructions from prompt: "${displayPrompt}"`;
+    }
+    async shouldConfirmExecute(abortSignal) {
+        // Try policy/message bus first when available.
+        if (this.messageBus) {
+            const decision = await this.getMessageBusDecision(abortSignal);
+            if (decision === 'ALLOW') {
+                return false;
+            }
+            if (decision === 'DENY') {
+                throw new Error('Tool execution denied by policy.');
+            }
+            // if 'ASK_USER', fall through to legacy confirmation UI
+        }
+        if (this.host.getApprovalMode() === 'auto') {
+            return false;
+        }
+        // Perform GitHub URL conversion here to differentiate between user-provided
+        // URL and the actual URL to be fetched.
+        const { validUrls } = parsePrompt(this.params.prompt);
+        const urls = validUrls.map((url) => {
+            try {
+                const urlObj = new URL(url);
+                if (urlObj.hostname === 'github.com' && url.includes('/blob/')) {
+                    return url
+                        .replace('github.com', 'raw.githubusercontent.com')
+                        .replace('/blob/', '/');
+                }
+            }
+            catch {
+                // Invalid URL; return as-is for error handling downstream.
+            }
+            return url;
+        });
+        const confirmationDetails = {
+            type: 'info',
+            title: `Confirm Web Fetch`,
+            prompt: this.params.prompt,
+            urls,
+            onConfirm: async (outcome) => {
+                if (outcome === ToolConfirmationOutcome.ProceedAlways) {
+                    // No need to publish a policy update as the default policy for
+                    // AUTO_EDIT already reflects always approving web-fetch.
+                    this.host.setApprovalMode('auto');
+                }
+                else {
+                    await this.publishPolicyUpdate(outcome);
+                }
+            },
+        };
+        return confirmationDetails;
+    }
+    async execute(signal) {
+        const userPrompt = this.params.prompt;
+        const { validUrls: urls } = parsePrompt(userPrompt);
+        const url = urls[0];
+        if (isPrivateIp(url))
+            return this.executePrivateUrlFallback(signal, url);
+        const serverToolsProvider = this.getServerToolsProvider();
+        if (serverToolsProvider instanceof Error) {
+            return this.createProviderErrorResult(serverToolsProvider.message);
+        }
+        try {
+            const response = await this.invokeWebFetch(serverToolsProvider, userPrompt, signal);
+            return await this.formatServerResponse(response, userPrompt, signal, url);
+        }
+        catch (error) {
+            return this.createProcessingErrorResult(error);
+        }
+    }
+    async executePrivateUrlFallback(signal, url) {
+        const errorMessage = 'Private/local URLs cannot be processed with AI analysis. Processing content directly.';
+        const result = await this.executeFallback(signal);
+        return {
+            ...result,
+            llmContent: `${errorMessage}\n\nContent from ${url}:\n\n${result.llmContent}`,
+        };
+    }
+    getServerToolsProvider() {
+        const serverToolsProvider = this.host.getServerToolsProvider?.();
+        if (!serverToolsProvider) {
+            return new Error(this.host.hasProviderManager?.() === false ? 'provider' : 'gemini');
+        }
+        const supportedTools = serverToolsProvider.getServerTools();
+        if (!supportedTools.includes(GOOGLE_WEB_FETCH_TOOL)) {
+            return new Error('unsupported');
+        }
+        return serverToolsProvider;
+    }
+    createProviderErrorResult(message) {
+        if (message === 'provider')
+            return this.createMissingProviderResult();
+        if (message === 'gemini')
+            return this.createMissingGeminiResult();
+        return this.createUnsupportedServerToolResult();
+    }
+    createMissingProviderResult() {
+        return {
+            llmContent: 'Web fetch requires a provider. Please use --provider gemini with authentication.',
+            returnDisplay: 'Web fetch requires a provider.',
+            error: {
+                message: 'No provider manager available',
+                type: ToolErrorType.WEB_FETCH_PROCESSING_ERROR,
+            },
+        };
+    }
+    createMissingGeminiResult() {
+        return {
+            llmContent: 'Web fetch requires Gemini provider to be configured. Please ensure Gemini is available with authentication.',
+            returnDisplay: 'Web fetch requires Gemini provider.',
+            error: {
+                message: 'No server tools provider available',
+                type: ToolErrorType.WEB_FETCH_PROCESSING_ERROR,
+            },
+        };
+    }
+    createUnsupportedServerToolResult() {
+        return {
+            llmContent: 'Web fetch is not available. The server tools provider does not support web fetch.',
+            returnDisplay: 'Web fetch not available.',
+            error: {
+                message: 'Server tools provider does not support web_fetch',
+                type: ToolErrorType.WEB_FETCH_PROCESSING_ERROR,
+            },
+        };
+    }
+    async invokeWebFetch(serverToolsProvider, userPrompt, signal) {
+        const response = await serverToolsProvider.invokeServerTool('web_fetch', { prompt: userPrompt }, { signal });
+        logger.log(`Full response for prompt "${userPrompt.substring(0, 50)}...":`, JSON.stringify(response, null, 2));
+        return response;
+    }
+    async formatServerResponse(response, userPrompt, signal, url) {
+        const typedResponse = response;
+        // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: getResponseText returns string | null | undefined, empty string should fall through
+        let responseText = getResponseText(typedResponse) || '';
+        const groundingMetadata = typedResponse.candidates?.[0]?.groundingMetadata;
+        const sources = groundingMetadata?.groundingChunks;
+        const groundingSupports = groundingMetadata?.groundingSupports;
+        if (this.hasProcessingError(typedResponse, responseText, sources)) {
+            if (isPrivateIp(url))
+                return this.executeFallback(signal);
+            return {
+                llmContent: 'No content found or URL retrieval failed.',
+                returnDisplay: 'No content found.',
+            };
+        }
+        responseText = this.addSourcesToResponse(responseText, sources, groundingSupports);
+        logger.log(`Formatted tool response for prompt "${userPrompt}:\n\n":`, responseText);
+        return {
+            llmContent: responseText,
+            returnDisplay: responseText,
+        };
+    }
+    hasProcessingError(response, responseText, sources) {
+        const urlMetadata = response.candidates?.[0]?.urlContextMetadata?.urlMetadata;
+        if (urlMetadata !== undefined && urlMetadata.length > 0) {
+            return urlMetadata.every((metadata) => 
+            // eslint-disable-next-line sonarjs/different-types-comparison -- Generated Gemini enum typing is narrower than runtime URL retrieval statuses; preserve the original success-status comparison.
+            metadata.urlRetrievalStatus !== 'URL_RETRIEVAL_STATUS_SUCCESS');
+        }
+        if (responseText.trim())
+            return false;
+        return sources === undefined || sources.length === 0;
+    }
+    addSourcesToResponse(responseText, sources, groundingSupports) {
+        if (sources === undefined || sources.length === 0)
+            return responseText;
+        let updatedText = this.insertCitationMarkers(responseText, groundingSupports);
+        const sourceListFormatted = this.formatSources(sources);
+        if (sourceListFormatted.length > 0) {
+            updatedText += `\n\nSources:\n${sourceListFormatted.join('\n')}`;
+        }
+        return updatedText;
+    }
+    formatSources(sources) {
+        return sources.map((source, index) => {
+            // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: title is optional string, empty string should fall through
+            const title = source.web?.title || 'Untitled';
+            // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: uri is optional string, empty string should fall through
+            const uri = source.web?.uri || 'Unknown URI';
+            return `[${index + 1}] ${title} (${uri})`;
+        });
+    }
+    insertCitationMarkers(responseText, groundingSupports) {
+        if (groundingSupports === undefined || groundingSupports.length === 0) {
+            return responseText;
+        }
+        const insertions = this.getCitationInsertions(groundingSupports);
+        const responseChars = responseText.split('');
+        for (const insertion of insertions) {
+            responseChars.splice(insertion.index, 0, insertion.marker);
+        }
+        return responseChars.join('');
+    }
+    getCitationInsertions(groundingSupports) {
+        const insertions = [];
+        for (const support of groundingSupports) {
+            if (!support.segment || !support.groundingChunkIndices)
+                continue;
+            const citationMarker = support.groundingChunkIndices
+                .map((chunkIndex) => `[${chunkIndex + 1}]`)
+                .join('');
+            insertions.push({
+                index: support.segment.endIndex,
+                marker: citationMarker,
+            });
+        }
+        return insertions.sort((a, b) => b.index - a.index);
+    }
+    createProcessingErrorResult(error) {
+        const errorMessage = `Error during web fetch: ${getErrorMessage(error)}`;
+        logger.error(errorMessage, error);
+        return {
+            llmContent: `Error: ${errorMessage}`,
+            returnDisplay: `Error: ${errorMessage}`,
+            error: {
+                message: errorMessage,
+                type: ToolErrorType.WEB_FETCH_PROCESSING_ERROR,
+            },
+        };
+    }
+}
+/**
+ * Implementation of the WebFetch tool logic
+ */
+export class GoogleWebFetchTool extends BaseDeclarativeTool {
+    host;
+    static Name = 'google_web_fetch';
+    constructor(host, messageBus) {
+        super(GoogleWebFetchTool.Name, 'GoogleWebFetch', "Processes content from URL(s), including local and private network addresses (e.g., localhost), embedded in a prompt. Include up to 20 URLs and instructions (e.g., summarize, extract specific data) directly in the 'prompt' parameter.", Kind.Fetch, {
+            properties: {
+                prompt: {
+                    description: 'A comprehensive prompt that includes the URL(s) (up to 20) to fetch and specific instructions on how to process their content (e.g., "Summarize https://example.com/article and extract key points from https://another.com/data"). All URLs to be fetched must be valid and complete, starting with "http://" or "https://", and be fully-formed with a valid hostname (e.g., a domain name like "example.com" or an IP address). For example, "https://example.com" is valid, but "example.com" is not.',
+                    type: 'string',
+                },
+            },
+            required: ['prompt'],
+            type: 'object',
+        }, false, // output is not markdown
+        false, // output cannot be updated
+        messageBus);
+        this.host = host;
+        // Proxy is now set globally in Config constructor with error handling
+    }
+    validateToolParamValues(params) {
+        if (!params.prompt || params.prompt.trim() === '') {
+            return "The 'prompt' parameter cannot be empty and must contain URL(s) and instructions.";
+        }
+        const { validUrls, errors } = parsePrompt(params.prompt);
+        if (errors.length > 0) {
+            return `Error(s) in prompt URLs:\n- ${errors.join('\n- ')}`;
+        }
+        if (validUrls.length === 0) {
+            return "The 'prompt' must contain at least one valid URL (starting with http:// or https://).";
+        }
+        return null;
+    }
+    createInvocation(params, messageBus, toolName, displayName) {
+        return new GoogleWebFetchToolInvocation(this.host, params, messageBus, toolName ?? this.name, displayName ?? this.displayName);
+    }
+}
+//# sourceMappingURL=google-web-fetch.js.map

package/dist/src/tools/google-web-fetch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-web-fetch.js","sourceRoot":"","sources":["../../../src/tools/google-web-fetch.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,IAAI,EAEJ,uBAAuB,GAGxB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAE/D,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,OAAO,EAAE,WAAW,IAAI,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAEhE,MAAM,oBAAoB,GAAG,KAAK,CAAC;AACnC,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAElC;;;GAGG;AACH,MAAM,UAAU,WAAW,CAAC,IAAY;IAItC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjC,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,MAAM,MAAM,GAAa,EAAE,CAAC;IAE5B,MAAM,wBAAwB,GAAG,CAAC,KAAa,EAAU,EAAE;QACzD,IAAI,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,6KAA6K;QAC7K,OAAO,QAAQ,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;YACrC;YACE,gKAAgK;YAChK,QAAQ,KAAK,GAAG;gBAChB,QAAQ,KAAK,GAAG;gBAChB,QAAQ,KAAK,GAAG;gBAChB,QAAQ,KAAK,GAAG;gBAChB,QAAQ,KAAK,GAAG;gBAChB,QAAQ,KAAK,GAAG,EAChB,CAAC;gBACD,QAAQ,EAAE,CAAC;gBACX,SAAS;YACX,CAAC;YACD,MAAM;QACR,CAAC;QACD,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IAClC,CAAC,CAAC;IAEF,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK;YAAE,SAAS;QAErB,qEAAqE;QACrE,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,4EAA4E;YAC5E,kEAAkE;YAClE,MAAM,OAAO,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;YAEhD,IAAI,CAAC;gBACH,0BAA0B;gBAC1B,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;gBAE7B,sBAAsB;gBACtB,8JAA8J;gBAC9J,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC/C,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,IAAI,CACT,iCAAiC,OAAO,uCAAuC,CAChF,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,8CAA8C;gBAC9C,MAAM,CAAC,IAAI,CAAC,4BAA4B,OAAO,IAAI,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;AAC/B,CAAC;AAgCD,SAAS,eAAe,CACtB,QAAiC;IAEjC,MAAM,KAAK,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC;IACvD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,KAAK;SACvB,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;IAC9D,OAAO,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACrE,CAAC;AAYD,MAAM,4BAA6B,SAAQ,kBAG1C;IAEoB;IADnB,YACmB,IAAe,EAChC,MAAgC,EAChC,UAA2B,EAC3B,QAAiB,EACjB,WAAoB;QAEpB,KAAK,CAAC,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;QANhC,SAAI,GAAJ,IAAI,CAAW;IAOlC,CAAC;IAEQ,WAAW;QAClB,OAAO,kBAAkB,CAAC,IAAI,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,eAAe,CAAC,OAAoB;QAChD,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC5D,gDAAgD;QAChD,IAAI,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAElB,qCAAqC;QACrC,qCAAqC;QACrC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;YAC5B,IAAI,MAAM,CAAC,QAAQ,KAAK,YAAY,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC/D,GAAG,GAAG,GAAG;qBACN,OAAO,CAAC,YAAY,EAAE,2BAA2B,CAAC;qBAClD,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,uDAAuD;QACzD,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CACrC,GAAG,EACH,oBAAoB,EACpB,OAAO,CACR,CAAC;YACF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CACb,mCAAmC,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAC5E,CAAC;YACJ,CAAC;YACD,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACzC,uKAAuK;YACvK,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAC/D,IAAI,WAAmB,CAAC;YAExB,iGAAiG;YACjG,IACE,WAAW,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC;gBAC/C,WAAW,KAAK,EAAE,EAClB,CAAC;gBACD,WAAW,GAAG,OAAO,CAAC,UAAU,EAAE;oBAChC,QAAQ,EAAE,KAAK;oBACf,SAAS,EAAE;wBACT,EAAE,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE;wBAChD,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE;qBACpC;iBACF,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,6EAA6E;gBAC7E,WAAW,GAAG,UAAU,CAAC;YAC3B,CAAC;YAED,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,kBAAkB,CAAC,CAAC;YAE3D,0EAA0E;YAC1E,6CAA6C;YAC7C,OAAO;gBACL,UAAU,EAAE,WAAW;gBACvB,aAAa,EAAE,eAAe,GAAG,oBAAoB;aACtD,CAAC;QACJ,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,CAAU,CAAC;YACzB,MAAM,YAAY,GAAG,mCAAmC,GAAG,KAAK,KAAK,CAAC,OAAO,EAAE,CAAC;YAChF,OAAO;gBACL,UAAU,EAAE,UAAU,YAAY,EAAE;gBACpC,aAAa,EAAE,UAAU,YAAY,EAAE;gBACvC,KAAK,EAAE;oBACL,OAAO,EAAE,YAAY;oBACrB,IAAI,EAAE,aAAa,CAAC,yBAAyB;iBAC9C;aACF,CAAC;QACJ,CAAC;IACH,CAAC;IAEQ,cAAc;QACrB,MAAM,aAAa,GACjB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,GAAG;YAC7B,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK;YAC7C,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QACzB,OAAO,kDAAkD,aAAa,GAAG,CAAC;IAC5E,CAAC;IAEQ,KAAK,CAAC,oBAAoB,CACjC,WAAwB;QAExB,+CAA+C;QAC/C,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,WAAW,CAAC,CAAC;YAC/D,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;gBACzB,OAAO,KAAK,CAAC;YACf,CAAC;YACD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtD,CAAC;YACD,wDAAwD;QAC1D,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,KAAK,MAAM,EAAE,CAAC;YAC3C,OAAO,KAAK,CAAC;QACf,CAAC;QAED,4EAA4E;QAC5E,wCAAwC;QACxC,MAAM,EAAE,SAAS,EAAE,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACjC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;gBAC5B,IAAI,MAAM,CAAC,QAAQ,KAAK,YAAY,IAAI,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC/D,OAAO,GAAG;yBACP,OAAO,CAAC,YAAY,EAAE,2BAA2B,CAAC;yBAClD,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,2DAA2D;YAC7D,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,MAAM,mBAAmB,GAAgC;YACvD,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,mBAAmB;YAC1B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;YAC1B,IAAI;YACJ,SAAS,EAAE,KAAK,EAAE,OAAgC,EAAE,EAAE;gBACpD,IAAI,OAAO,KAAK,uBAAuB,CAAC,aAAa,EAAE,CAAC;oBACtD,+DAA+D;oBAC/D,yDAAyD;oBACzD,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;gBACpC,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;SACF,CAAC;QACF,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QACtC,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;QACpD,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAEpB,IAAI,WAAW,CAAC,GAAG,CAAC;YAAE,OAAO,IAAI,CAAC,yBAAyB,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAEzE,MAAM,mBAAmB,GAAG,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC1D,IAAI,mBAAmB,YAAY,KAAK,EAAE,CAAC;YACzC,OAAO,IAAI,CAAC,yBAAyB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QACrE,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,cAAc,CACxC,mBAAmB,EACnB,UAAU,EACV,MAAM,CACP,CAAC;YACF,OAAO,MAAM,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;QAC5E,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,yBAAyB,CACrC,MAAmB,EACnB,GAAW;QAEX,MAAM,YAAY,GAChB,uFAAuF,CAAC;QAC1F,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAClD,OAAO;YACL,GAAG,MAAM;YACT,UAAU,EAAE,GAAG,YAAY,oBAAoB,GAAG,QAAQ,MAAM,CAAC,UAAU,EAAE;SAC9E,CAAC;IACJ,CAAC;IAEO,sBAAsB;QAC5B,MAAM,mBAAmB,GAAG,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,CAAC;QAEjE,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACzB,OAAO,IAAI,KAAK,CACd,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,KAAK,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CACnE,CAAC;QACJ,CAAC;QAED,MAAM,cAAc,GAAG,mBAAmB,CAAC,cAAc,EAAE,CAAC;QAC5D,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE,CAAC;YACpD,OAAO,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;QAClC,CAAC;QAED,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAEO,yBAAyB,CAAC,OAAe;QAC/C,IAAI,OAAO,KAAK,UAAU;YAAE,OAAO,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACtE,IAAI,OAAO,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC,yBAAyB,EAAE,CAAC;QAClE,OAAO,IAAI,CAAC,iCAAiC,EAAE,CAAC;IAClD,CAAC;IAEO,2BAA2B;QACjC,OAAO;YACL,UAAU,EACR,kFAAkF;YACpF,aAAa,EAAE,gCAAgC;YAC/C,KAAK,EAAE;gBACL,OAAO,EAAE,+BAA+B;gBACxC,IAAI,EAAE,aAAa,CAAC,0BAA0B;aAC/C;SACF,CAAC;IACJ,CAAC;IAEO,yBAAyB;QAC/B,OAAO;YACL,UAAU,EACR,6GAA6G;YAC/G,aAAa,EAAE,qCAAqC;YACpD,KAAK,EAAE;gBACL,OAAO,EAAE,oCAAoC;gBAC7C,IAAI,EAAE,aAAa,CAAC,0BAA0B;aAC/C;SACF,CAAC;IACJ,CAAC;IAEO,iCAAiC;QACvC,OAAO;YACL,UAAU,EACR,mFAAmF;YACrF,aAAa,EAAE,0BAA0B;YACzC,KAAK,EAAE;gBACL,OAAO,EAAE,kDAAkD;gBAC3D,IAAI,EAAE,aAAa,CAAC,0BAA0B;aAC/C;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,mBAAwC,EACxC,UAAkB,EAClB,MAAmB;QAEnB,MAAM,QAAQ,GAAG,MAAM,mBAAmB,CAAC,gBAAgB,CACzD,WAAW,EACX,EAAE,MAAM,EAAE,UAAU,EAAE,EACtB,EAAE,MAAM,EAAE,CACX,CAAC;QACF,MAAM,CAAC,GAAG,CACR,6BAA6B,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,EAC/D,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAClC,CAAC;QACF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,QAAiB,EACjB,UAAkB,EAClB,MAAmB,EACnB,GAAW;QAEX,MAAM,aAAa,GAAG,QAAmC,CAAC;QAC1D,6LAA6L;QAC7L,IAAI,YAAY,GAAG,eAAe,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QACxD,MAAM,iBAAiB,GAAG,aAAa,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,iBAAiB,CAAC;QAC3E,MAAM,OAAO,GAAG,iBAAiB,EAAE,eAEtB,CAAC;QACd,MAAM,iBAAiB,GAAG,iBAAiB,EAAE,iBAEhC,CAAC;QAEd,IAAI,IAAI,CAAC,kBAAkB,CAAC,aAAa,EAAE,YAAY,EAAE,OAAO,CAAC,EAAE,CAAC;YAClE,IAAI,WAAW,CAAC,GAAG,CAAC;gBAAE,OAAO,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAC1D,OAAO;gBACL,UAAU,EAAE,2CAA2C;gBACvD,aAAa,EAAE,mBAAmB;aACnC,CAAC;QACJ,CAAC;QAED,YAAY,GAAG,IAAI,CAAC,oBAAoB,CACtC,YAAY,EACZ,OAAO,EACP,iBAAiB,CAClB,CAAC;QACF,MAAM,CAAC,GAAG,CACR,uCAAuC,UAAU,SAAS,EAC1D,YAAY,CACb,CAAC;QAEF,OAAO;YACL,UAAU,EAAE,YAAY;YACxB,aAAa,EAAE,YAAY;SAC5B,CAAC;IACJ,CAAC;IAEO,kBAAkB,CACxB,QAAiC,EACjC,YAAoB,EACpB,OAA8B;QAE9B,MAAM,WAAW,GACf,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,WAAW,CAAC;QAC5D,IAAI,WAAW,KAAK,SAAS,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,OAAO,WAAW,CAAC,KAAK,CACtB,CAAC,QAAqB,EAAE,EAAE;YACxB,gMAAgM;YAChM,QAAQ,CAAC,kBAAkB,KAAK,8BAA8B,CACjE,CAAC;QACJ,CAAC;QAED,IAAI,YAAY,CAAC,IAAI,EAAE;YAAE,OAAO,KAAK,CAAC;QACtC,OAAO,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC;IACvD,CAAC;IAEO,oBAAoB,CAC1B,YAAoB,EACpB,OAA8B,EAC9B,iBAA0C;QAE1C,IAAI,OAAO,KAAK,SAAS,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,YAAY,CAAC;QAEvE,IAAI,WAAW,GAAG,IAAI,CAAC,qBAAqB,CAC1C,YAAY,EACZ,iBAAiB,CAClB,CAAC;QACF,MAAM,mBAAmB,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACxD,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,WAAW,IAAI,iBAAiB,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QACnE,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,aAAa,CAAC,OAA6B;QACjD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAA0B,EAAE,KAAa,EAAE,EAAE;YAC/D,oKAAoK;YACpK,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,EAAE,KAAK,IAAI,UAAU,CAAC;YAC9C,kKAAkK;YAClK,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,aAAa,CAAC;YAC7C,OAAO,IAAI,KAAK,GAAG,CAAC,KAAK,KAAK,KAAK,GAAG,GAAG,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,qBAAqB,CAC3B,YAAoB,EACpB,iBAA0C;QAE1C,IAAI,iBAAiB,KAAK,SAAS,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtE,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;QACjE,MAAM,aAAa,GAAG,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC7C,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,aAAa,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC;IAEO,qBAAqB,CAC3B,iBAAyC;QAEzC,MAAM,UAAU,GAA6C,EAAE,CAAC;QAChE,KAAK,MAAM,OAAO,IAAI,iBAAiB,EAAE,CAAC;YACxC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,qBAAqB;gBAAE,SAAS;YACjE,MAAM,cAAc,GAAG,OAAO,CAAC,qBAAqB;iBACjD,GAAG,CAAC,CAAC,UAAkB,EAAE,EAAE,CAAC,IAAI,UAAU,GAAG,CAAC,GAAG,CAAC;iBAClD,IAAI,CAAC,EAAE,CAAC,CAAC;YACZ,UAAU,CAAC,IAAI,CAAC;gBACd,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,QAAQ;gBAC/B,MAAM,EAAE,cAAc;aACvB,CAAC,CAAC;QACL,CAAC;QACD,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC;IAEO,2BAA2B,CAAC,KAAc;QAChD,MAAM,YAAY,GAAG,2BAA2B,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;QACzE,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;QAClC,OAAO;YACL,UAAU,EAAE,UAAU,YAAY,EAAE;YACpC,aAAa,EAAE,UAAU,YAAY,EAAE;YACvC,KAAK,EAAE;gBACL,OAAO,EAAE,YAAY;gBACrB,IAAI,EAAE,aAAa,CAAC,0BAA0B;aAC/C;SACF,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,kBAAmB,SAAQ,mBAGvC;IAIoB;IAHnB,MAAM,CAAU,IAAI,GAAW,kBAAkB,CAAC;IAElD,YACmB,IAAe,EAChC,UAA2B;QAE3B,KAAK,CACH,kBAAkB,CAAC,IAAI,EACvB,gBAAgB,EAChB,2OAA2O,EAC3O,IAAI,CAAC,KAAK,EACV;YACE,UAAU,EAAE;gBACV,MAAM,EAAE;oBACN,WAAW,EACT,2eAA2e;oBAC7e,IAAI,EAAE,QAAQ;iBACf;aACF;YACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;YACpB,IAAI,EAAE,QAAQ;SACf,EACD,KAAK,EAAE,yBAAyB;QAChC,KAAK,EAAE,2BAA2B;QAClC,UAAU,CACX,CAAC;QAtBe,SAAI,GAAJ,IAAI,CAAW;QAuBhC,sEAAsE;IACxE,CAAC;IAEkB,uBAAuB,CACxC,MAAgC;QAEhC,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YAClD,OAAO,kFAAkF,CAAC;QAC5F,CAAC;QAED,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEzD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtB,OAAO,+BAA+B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9D,CAAC;QAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO,uFAAuF,CAAC;QACjG,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAES,gBAAgB,CACxB,MAAgC,EAChC,UAA2B,EAC3B,QAAiB,EACjB,WAAoB;QAEpB,OAAO,IAAI,4BAA4B,CACrC,IAAI,CAAC,IAAI,EACT,MAAM,EACN,UAAU,EACV,QAAQ,IAAI,IAAI,CAAC,IAAI,EACrB,WAAW,IAAI,IAAI,CAAC,WAAW,CAChC,CAAC;IACJ,CAAC"}

package/dist/src/tools/google-web-search-invocation.d.ts

@@ -0,0 +1,53 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { GroundingMetadata } from '@google/genai';
+import type { IWebSearchService } from '../interfaces/index.js';
+import { BaseToolInvocation, type ToolResult } from './tools.js';
+export interface GroundingChunkWeb {
+    uri?: string;
+    title?: string;
+}
+export interface GroundingChunkItem {
+    web?: GroundingChunkWeb;
+}
+export interface GroundingSupportSegment {
+    startIndex: number;
+    endIndex: number;
+    text?: string;
+}
+export interface GroundingSupportItem {
+    segment?: GroundingSupportSegment;
+    groundingChunkIndices?: number[];
+    confidenceScores?: number[];
+}
+export interface WebSearchToolParams {
+    query: string;
+}
+export interface WebSearchToolResult extends ToolResult {
+    sources?: GroundingMetadata extends {
+        groundingChunks: GroundingChunkItem[];
+    } ? GroundingMetadata['groundingChunks'] : GroundingChunkItem[];
+}
+export declare class GoogleWebSearchToolInvocation extends BaseToolInvocation<WebSearchToolParams, WebSearchToolResult> {
+    private readonly webSearchService;
+    constructor(webSearchService: IWebSearchService, params: WebSearchToolParams, messageBus?: import('../interfaces/index.js').IToolMessageBus);
+    getToolName(): string;
+    getDescription(): string;
+    execute(signal: AbortSignal): Promise<WebSearchToolResult>;
+    private validateQuery;
+    private createEmptyQueryResult;
+    private getServerToolsProvider;
+    private createProviderErrorResult;
+    private createMissingProviderResult;
+    private createMissingGeminiResult;
+    private createUnsupportedServerToolResult;
+    private formatSearchResponse;
+    private addSourcesToResponse;
+    private formatSources;
+    private insertCitationMarkers;
+    private getCitationInsertions;
+    private createSearchErrorResult;
+}

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

@@ -0,0 +1,180 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { GOOGLE_WEB_SEARCH_TOOL } from '../types/tool-names.js';
+import { ToolErrorType } from '../types/tool-error.js';
+import { getErrorMessage } from '../utils/errors.js';
+import { BaseToolInvocation } from './tools.js';
+export class GoogleWebSearchToolInvocation extends BaseToolInvocation {
+    webSearchService;
+    constructor(webSearchService, params, messageBus) {
+        super(params, messageBus);
+        this.webSearchService = webSearchService;
+    }
+    getToolName() {
+        return GOOGLE_WEB_SEARCH_TOOL;
+    }
+    getDescription() {
+        return `Searching the web for: "${this.params.query}"`;
+    }
+    async execute(signal) {
+        const invalidQueryResult = this.validateQuery();
+        if (invalidQueryResult)
+            return invalidQueryResult;
+        try {
+            const serverToolsProvider = this.getServerToolsProvider();
+            if (serverToolsProvider instanceof Error) {
+                return this.createProviderErrorResult(serverToolsProvider.message);
+            }
+            const searchQuery = this.params.query.trim();
+            if (searchQuery === '')
+                return this.createEmptyQueryResult();
+            const response = await serverToolsProvider.invokeServerTool('web_search', { query: searchQuery }, { signal });
+            return this.formatSearchResponse(response);
+        }
+        catch (error) {
+            return this.createSearchErrorResult(error);
+        }
+    }
+    validateQuery() {
+        if (!this.params.query ||
+            this.params.query.trim() === '' ||
+            this.params.query === 'undefined') {
+            return {
+                llmContent: `Error: Web search requires a valid search query. Please provide a specific search query.`,
+                returnDisplay: `Error: No valid search query provided.`,
+            };
+        }
+        return undefined;
+    }
+    createEmptyQueryResult() {
+        return {
+            llmContent: `Error: Cannot perform web search without a query.`,
+            returnDisplay: `Error: Empty search query.`,
+        };
+    }
+    getServerToolsProvider() {
+        const serverToolsProvider = this.webSearchService.getServerToolsProvider();
+        if (!serverToolsProvider)
+            return new Error('gemini');
+        const serverTools = serverToolsProvider.getServerTools();
+        if (!serverTools.includes('web_search'))
+            return new Error('unsupported');
+        return serverToolsProvider;
+    }
+    createProviderErrorResult(message) {
+        if (message === 'provider')
+            return this.createMissingProviderResult();
+        if (message === 'gemini')
+            return this.createMissingGeminiResult();
+        return this.createUnsupportedServerToolResult();
+    }
+    createMissingProviderResult() {
+        return {
+            llmContent: `Web search requires a provider. Please use --provider gemini with authentication.`,
+            returnDisplay: 'Web search requires a provider.',
+        };
+    }
+    createMissingGeminiResult() {
+        return {
+            llmContent: `Web search requires Gemini provider to be configured. Please ensure Gemini is available with authentication.`,
+            returnDisplay: 'Web search requires Gemini provider.',
+        };
+    }
+    createUnsupportedServerToolResult() {
+        return {
+            llmContent: `Web search is not available. The server tools provider does not support web search.`,
+            returnDisplay: `Web search not available.`,
+        };
+    }
+    formatSearchResponse(response) {
+        const geminiResponse = response;
+        const responseText = getResponseText(geminiResponse);
+        const groundingMetadata = geminiResponse.candidates?.[0]?.groundingMetadata;
+        const sources = groundingMetadata?.groundingChunks;
+        const groundingSupports = groundingMetadata?.groundingSupports;
+        if (!responseText?.trim()) {
+            return {
+                llmContent: `No search results or information found for query: "${this.params.query}"`,
+                returnDisplay: 'No information found.',
+            };
+        }
+        const modifiedResponseText = this.addSourcesToResponse(responseText, sources, groundingSupports);
+        return {
+            llmContent: `Web search results for "${this.params.query}":\n\n${modifiedResponseText}`,
+            returnDisplay: `Search results for "${this.params.query}" returned.`,
+            sources,
+        };
+    }
+    addSourcesToResponse(responseText, sources, groundingSupports) {
+        if (sources === undefined || sources.length === 0)
+            return responseText;
+        let modifiedResponseText = this.insertCitationMarkers(responseText, groundingSupports);
+        const sourceListFormatted = this.formatSources(sources);
+        if (sourceListFormatted.length > 0) {
+            modifiedResponseText += '\n\nSources:\n' + sourceListFormatted.join('\n');
+        }
+        return modifiedResponseText;
+    }
+    formatSources(sources) {
+        return sources.map((source, index) => {
+            // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: empty title should show 'Untitled'
+            const title = source.web?.title || 'Untitled';
+            // eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- intentional falsy coalescing: empty URI should show 'No URI'
+            const uri = source.web?.uri || 'No URI';
+            return `[${index + 1}] ${title} (${uri})`;
+        });
+    }
+    insertCitationMarkers(responseText, groundingSupports) {
+        if (groundingSupports === undefined || groundingSupports.length === 0) {
+            return responseText;
+        }
+        const textBuffer = Buffer.from(responseText, 'utf-8');
+        const insertions = this.getCitationInsertions(groundingSupports, textBuffer);
+        const responseChars = responseText.split('');
+        for (const insertion of insertions) {
+            responseChars.splice(insertion.index, 0, insertion.marker);
+        }
+        return responseChars.join('');
+    }
+    getCitationInsertions(groundingSupports, textBuffer) {
+        const insertions = [];
+        for (const support of groundingSupports) {
+            if (!support.segment || !support.groundingChunkIndices)
+                continue;
+            const citationMarker = support.groundingChunkIndices
+                .map((chunkIndex) => `[${chunkIndex + 1}]`)
+                .join('');
+            const charIndex = textBuffer
+                .subarray(0, support.segment.endIndex)
+                .toString('utf-8').length;
+            insertions.push({ index: charIndex, marker: citationMarker });
+        }
+        return insertions.sort((a, b) => b.index - a.index);
+    }
+    createSearchErrorResult(error) {
+        const queryDisplay = this.params.query || 'undefined';
+        const errorMessage = `Error during web search for query "${queryDisplay}": ${getErrorMessage(error)}`;
+        return {
+            llmContent: `Error: ${errorMessage}`,
+            returnDisplay: `Error performing web search.`,
+            error: {
+                type: ToolErrorType.WEB_SEARCH_FAILED,
+                message: `Web search failed: ${getErrorMessage(error)}`,
+            },
+        };
+    }
+}
+function getResponseText(response) {
+    const parts = response.candidates?.[0]?.content?.parts;
+    if (!parts)
+        return undefined;
+    const textSegments = parts
+        .filter((part) => part.thought !== true && part.thoughtSignature === undefined)
+        .map((part) => part.text)
+        .filter((text) => typeof text === 'string');
+    return textSegments.length > 0 ? textSegments.join('') : undefined;
+}
+//# sourceMappingURL=google-web-search-invocation.js.map