@office-ai/aioncli-core 0.26.0 → 0.30.1

package/dist/src/agents/browser/mcpToolWrapper.js

@@ -0,0 +1,358 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { DeclarativeTool, BaseToolInvocation, Kind, } from '../../tools/tools.js';
+import { debugLogger } from '../../utils/debugLogger.js';
+/**
+ * Tool invocation that dispatches to BrowserManager's isolated MCP client.
+ */
+class McpToolInvocation extends BaseToolInvocation {
+    browserManager;
+    toolName;
+    constructor(browserManager, toolName, params, messageBus) {
+        super(params, messageBus, toolName, toolName);
+        this.browserManager = browserManager;
+        this.toolName = toolName;
+    }
+    getDescription() {
+        return `Calling MCP tool: ${this.toolName}`;
+    }
+    async getConfirmationDetails(_abortSignal) {
+        if (!this.messageBus) {
+            return false;
+        }
+        return {
+            type: 'mcp',
+            title: `Confirm MCP Tool: ${this.toolName}`,
+            serverName: 'browser-agent',
+            toolName: this.toolName,
+            toolDisplayName: this.toolName,
+            onConfirm: async (outcome) => {
+                await this.publishPolicyUpdate(outcome);
+            },
+        };
+    }
+    getPolicyUpdateOptions(_outcome) {
+        return {
+            mcpName: 'browser-agent',
+        };
+    }
+    async execute(signal) {
+        try {
+            const callToolPromise = this.browserManager.callTool(this.toolName, this.params, signal);
+            const result = await callToolPromise;
+            // Extract text content from MCP response
+            let textContent = '';
+            if (result.content && Array.isArray(result.content)) {
+                textContent = result.content
+                    .filter((c) => c.type === 'text' && c.text)
+                    .map((c) => c.text)
+                    .join('\n');
+            }
+            // Post-process to add contextual hints for common error patterns
+            const processedContent = postProcessToolResult(this.toolName, textContent);
+            if (result.isError) {
+                return {
+                    llmContent: `Error: ${processedContent}`,
+                    returnDisplay: `Error: ${processedContent}`,
+                    error: { message: textContent },
+                };
+            }
+            return {
+                llmContent: processedContent || 'Tool executed successfully.',
+                returnDisplay: processedContent || 'Tool executed successfully.',
+            };
+        }
+        catch (error) {
+            const errorMsg = error instanceof Error ? error.message : String(error);
+            // Chrome connection errors are fatal — re-throw to terminate the agent
+            // immediately instead of returning a result the LLM would retry.
+            if (errorMsg.includes('Could not connect to Chrome')) {
+                throw error;
+            }
+            debugLogger.error(`MCP tool ${this.toolName} failed: ${errorMsg}`);
+            return {
+                llmContent: `Error: ${errorMsg}`,
+                returnDisplay: `Error: ${errorMsg}`,
+                error: { message: errorMsg },
+            };
+        }
+    }
+}
+/**
+ * Composite tool invocation that types a full string by calling press_key
+ * for each character internally, avoiding N model round-trips.
+ */
+class TypeTextInvocation extends BaseToolInvocation {
+    browserManager;
+    text;
+    submitKey;
+    constructor(browserManager, text, submitKey, messageBus) {
+        super({ text, submitKey }, messageBus, 'type_text', 'type_text');
+        this.browserManager = browserManager;
+        this.text = text;
+        this.submitKey = submitKey;
+    }
+    getDescription() {
+        const preview = `"${this.text.substring(0, 50)}${this.text.length > 50 ? '...' : ''}"`;
+        return this.submitKey
+            ? `type_text: ${preview} + ${this.submitKey}`
+            : `type_text: ${preview}`;
+    }
+    async getConfirmationDetails(_abortSignal) {
+        if (!this.messageBus) {
+            return false;
+        }
+        return {
+            type: 'mcp',
+            title: `Confirm Tool: type_text`,
+            serverName: 'browser-agent',
+            toolName: 'type_text',
+            toolDisplayName: 'type_text',
+            onConfirm: async (outcome) => {
+                await this.publishPolicyUpdate(outcome);
+            },
+        };
+    }
+    getPolicyUpdateOptions(_outcome) {
+        return {
+            mcpName: 'browser-agent',
+        };
+    }
+    async execute(signal) {
+        try {
+            if (signal.aborted) {
+                return {
+                    llmContent: 'Error: Operation cancelled before typing started.',
+                    returnDisplay: 'Operation cancelled before typing started.',
+                    error: { message: 'Operation cancelled' },
+                };
+            }
+            await this.typeCharByChar(signal);
+            // Optionally press a submit key (Enter, Tab, etc.) after typing
+            if (this.submitKey && !signal.aborted) {
+                const keyResult = await this.browserManager.callTool('press_key', { key: this.submitKey }, signal);
+                if (keyResult.isError) {
+                    const errText = this.extractErrorText(keyResult);
+                    debugLogger.warn(`type_text: submitKey("${this.submitKey}") failed: ${errText}`);
+                }
+            }
+            const summary = this.submitKey
+                ? `Successfully typed "${this.text}" and pressed ${this.submitKey}`
+                : `Successfully typed "${this.text}"`;
+            return {
+                llmContent: summary,
+                returnDisplay: summary,
+            };
+        }
+        catch (error) {
+            const errorMsg = error instanceof Error ? error.message : String(error);
+            // Chrome connection errors are fatal
+            if (errorMsg.includes('Could not connect to Chrome')) {
+                throw error;
+            }
+            debugLogger.error(`type_text failed: ${errorMsg}`);
+            return {
+                llmContent: `Error: ${errorMsg}`,
+                returnDisplay: `Error: ${errorMsg}`,
+                error: { message: errorMsg },
+            };
+        }
+    }
+    /** Types each character via individual press_key MCP calls. */
+    async typeCharByChar(signal) {
+        const chars = [...this.text]; // Handle Unicode correctly
+        for (const char of chars) {
+            if (signal.aborted)
+                return;
+            // Map special characters to key names
+            const key = char === ' ' ? 'Space' : char;
+            const result = await this.browserManager.callTool('press_key', { key }, signal);
+            if (result.isError) {
+                debugLogger.warn(`type_text: press_key("${key}") failed: ${this.extractErrorText(result)}`);
+            }
+        }
+    }
+    /** Extract error text from an MCP tool result. */
+    extractErrorText(result) {
+        return (result.content
+            ?.filter((c) => c.type === 'text' && c.text)
+            .map((c) => c.text)
+            .join('\n') || 'Unknown error');
+    }
+}
+/**
+ * DeclarativeTool wrapper for an MCP tool.
+ */
+class McpDeclarativeTool extends DeclarativeTool {
+    browserManager;
+    constructor(browserManager, name, description, parameterSchema, messageBus) {
+        super(name, name, description, Kind.Other, parameterSchema, messageBus, 
+        /* isOutputMarkdown */ true, 
+        /* canUpdateOutput */ false);
+        this.browserManager = browserManager;
+    }
+    build(params) {
+        return new McpToolInvocation(this.browserManager, this.name, params, this.messageBus);
+    }
+}
+/**
+ * DeclarativeTool for the custom type_text composite tool.
+ */
+class TypeTextDeclarativeTool extends DeclarativeTool {
+    browserManager;
+    constructor(browserManager, messageBus) {
+        super('type_text', 'type_text', 'Types a full text string into the currently focused element. ' +
+            'Much faster than calling press_key for each character individually. ' +
+            'Use this to enter text into form fields, search boxes, spreadsheet cells, or any focused input. ' +
+            'The element must already be focused (e.g., after a click). ' +
+            'Use submitKey to press a key after typing (e.g., submitKey="Enter" to submit a form or confirm a value, submitKey="Tab" to move to the next field).', Kind.Other, {
+            type: 'object',
+            properties: {
+                text: {
+                    type: 'string',
+                    description: 'The text to type into the focused element.',
+                },
+                submitKey: {
+                    type: 'string',
+                    description: 'Optional key to press after typing (e.g., "Enter", "Tab", "Escape"). ' +
+                        'Useful for submitting form fields or moving to the next cell in a spreadsheet.',
+                },
+            },
+            required: ['text'],
+        }, messageBus, 
+        /* isOutputMarkdown */ true, 
+        /* canUpdateOutput */ false);
+        this.browserManager = browserManager;
+    }
+    build(params) {
+        const submitKey = 
+        // eslint-disable-next-line no-restricted-syntax
+        typeof params['submitKey'] === 'string' && params['submitKey']
+            ? params['submitKey']
+            : undefined;
+        return new TypeTextInvocation(this.browserManager, String(params['text'] ?? ''), submitKey, this.messageBus);
+    }
+}
+/**
+ * Creates DeclarativeTool instances from dynamically discovered MCP tools,
+ * plus custom composite tools (like type_text).
+ *
+ * These tools are registered in the browser agent's isolated ToolRegistry,
+ * NOT in the main agent's registry.
+ *
+ * Tool definitions are fetched dynamically from the MCP server at runtime.
+ *
+ * @param browserManager The browser manager with isolated MCP client
+ * @param messageBus Message bus for tool invocations
+ * @returns Array of DeclarativeTools that dispatch to the isolated MCP client
+ */
+export async function createMcpDeclarativeTools(browserManager, messageBus) {
+    // Get dynamically discovered tools from the MCP server
+    const mcpTools = await browserManager.getDiscoveredTools();
+    debugLogger.log(`Creating ${mcpTools.length} declarative tools for browser agent`);
+    const tools = mcpTools.map((mcpTool) => {
+        const schema = convertMcpToolToFunctionDeclaration(mcpTool);
+        // Augment description with uid-context hints
+        const augmentedDescription = augmentToolDescription(mcpTool.name, mcpTool.description ?? '');
+        return new McpDeclarativeTool(browserManager, mcpTool.name, augmentedDescription, schema.parametersJsonSchema, messageBus);
+    });
+    // Add custom composite tools
+    tools.push(new TypeTextDeclarativeTool(browserManager, messageBus));
+    debugLogger.log(`Total tools registered: ${tools.length} (${mcpTools.length} MCP + 1 custom)`);
+    return tools;
+}
+/**
+ * Converts MCP tool definition to Gemini FunctionDeclaration.
+ */
+function convertMcpToolToFunctionDeclaration(mcpTool) {
+    // MCP tool inputSchema is a JSON Schema object
+    // We pass it directly as parametersJsonSchema
+    return {
+        name: mcpTool.name,
+        description: mcpTool.description ?? '',
+        parametersJsonSchema: mcpTool.inputSchema ?? {
+            type: 'object',
+            properties: {},
+        },
+    };
+}
+/**
+ * Augments MCP tool descriptions with usage guidance.
+ * Adds semantic hints and usage rules directly in tool descriptions
+ * so the model makes correct tool choices without system prompt overhead.
+ *
+ * Actual chrome-devtools-mcp tools:
+ *   Input: click, drag, fill, fill_form, handle_dialog, hover, press_key, upload_file
+ *   Navigation: close_page, list_pages, navigate_page, new_page, select_page, wait_for
+ *   Emulation: emulate, resize_page
+ *   Performance: performance_analyze_insight, performance_start_trace, performance_stop_trace
+ *   Network: get_network_request, list_network_requests
+ *   Debugging: evaluate_script, get_console_message, list_console_messages, take_screenshot, take_snapshot
+ *   Vision (--experimental-vision): click_at, analyze_screenshot
+ */
+function augmentToolDescription(toolName, description) {
+    // More-specific keys MUST come before shorter keys to prevent
+    // partial matching from short-circuiting (e.g., fill_form before fill).
+    const hints = {
+        fill_form: ' Fills multiple standard HTML form fields at once. Same limitations as fill — does not work on canvas/custom widgets.',
+        fill: ' Fills standard HTML form fields (<input>, <textarea>, <select>) by uid. Does NOT work on custom/canvas-based widgets (e.g., Google Sheets cells, Notion blocks). If fill times out or fails, click the element first then use press_key with individual characters instead.',
+        click_at: ' Clicks at exact pixel coordinates (x, y). Use when you have specific coordinates for visual elements.',
+        click: ' Use the element uid from the accessibility tree snapshot (e.g., uid="87_4"). UIDs are invalidated after this action — call take_snapshot before using another uid.',
+        hover: ' Use the element uid from the accessibility tree snapshot to hover over elements.',
+        take_snapshot: ' Returns the accessibility tree with uid values for each element. Call this FIRST to see available elements, and AFTER every state-changing action (click, fill, press_key) before using any uid.',
+        navigate_page: ' Navigate to the specified URL. Call take_snapshot after to see the new page.',
+        new_page: ' Opens a new page/tab with the specified URL. Call take_snapshot after to see the new page.',
+        press_key: ' Press a SINGLE keyboard key (e.g., "Enter", "Tab", "Escape", "ArrowDown", "a", "8"). ONLY accepts one key name — do NOT pass multi-character strings like "Hello" or "A1\\nEnter". To type text, use type_text instead of calling press_key for each character.',
+    };
+    // Check for partial matches — order matters! More-specific keys first.
+    for (const [key, hint] of Object.entries(hints)) {
+        if (toolName.toLowerCase().includes(key)) {
+            return description + hint;
+        }
+    }
+    return description;
+}
+/**
+ * Post-processes tool results to add contextual hints for common error patterns.
+ * This helps the agent recover from overlay blocking, element not found, etc.
+ * Also strips embedded snapshots to prevent token bloat.
+ */
+export function postProcessToolResult(toolName, result) {
+    // Strip embedded snapshots to prevent token bloat (except for take_snapshot,
+    // whose accessibility tree the model needs for uid-based interactions).
+    let processedResult = result;
+    if (toolName !== 'take_snapshot' &&
+        result.includes('## Latest page snapshot')) {
+        const parts = result.split('## Latest page snapshot');
+        processedResult = parts[0].trim();
+        if (parts[1]) {
+            debugLogger.log('Stripped embedded snapshot from tool response');
+        }
+    }
+    // Detect overlay/interactable issues
+    const overlayPatterns = [
+        'not interactable',
+        'obscured',
+        'intercept',
+        'blocked',
+        'element is not visible',
+        'element not found',
+    ];
+    const isOverlayIssue = overlayPatterns.some((pattern) => processedResult.toLowerCase().includes(pattern));
+    if (isOverlayIssue && (toolName === 'click' || toolName.includes('click'))) {
+        return (processedResult +
+            '\n\n⚠️ This action may have been blocked by an overlay, popup, or tooltip. ' +
+            'Look for close/dismiss buttons (×, Close, "Got it", "Accept") in the accessibility tree and click them first.');
+    }
+    // Detect stale element references
+    if (processedResult.toLowerCase().includes('stale') ||
+        processedResult.toLowerCase().includes('detached')) {
+        return (processedResult +
+            '\n\n⚠️ The element reference is stale. Call take_snapshot to get fresh element uids.');
+    }
+    return processedResult;
+}
+//# sourceMappingURL=mcpToolWrapper.js.map

package/dist/src/agents/browser/mcpToolWrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcpToolWrapper.js","sourceRoot":"","sources":["../../../../src/agents/browser/mcpToolWrapper.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAeH,OAAO,EAEL,eAAe,EACf,kBAAkB,EAClB,IAAI,GAKL,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD;;GAEG;AACH,MAAM,iBAAkB,SAAQ,kBAG/B;IAEsB;IACA;IAFrB,YACqB,cAA8B,EAC9B,QAAgB,EACnC,MAA+B,EAC/B,UAAsB;QAEtB,KAAK,CAAC,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAL3B,mBAAc,GAAd,cAAc,CAAgB;QAC9B,aAAQ,GAAR,QAAQ,CAAQ;IAKrC,CAAC;IAED,cAAc;QACZ,OAAO,qBAAqB,IAAI,CAAC,QAAQ,EAAE,CAAC;IAC9C,CAAC;IAEkB,KAAK,CAAC,sBAAsB,CAC7C,YAAyB;QAEzB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO;YACL,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,qBAAqB,IAAI,CAAC,QAAQ,EAAE;YAC3C,UAAU,EAAE,eAAe;YAC3B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,eAAe,EAAE,IAAI,CAAC,QAAQ;YAC9B,SAAS,EAAE,KAAK,EAAE,OAAgC,EAAE,EAAE;gBACpD,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAC1C,CAAC;SACF,CAAC;IACJ,CAAC;IAEQ,sBAAsB,CAC7B,QAAiC;QAEjC,OAAO;YACL,OAAO,EAAE,eAAe;SACzB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAClD,IAAI,CAAC,QAAQ,EACb,IAAI,CAAC,MAAM,EACX,MAAM,CACP,CAAC;YAEF,MAAM,MAAM,GAAsB,MAAM,eAAe,CAAC;YAExD,yCAAyC;YACzC,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,MAAM,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACpD,WAAW,GAAG,MAAM,CAAC,OAAO;qBACzB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC;qBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;qBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,CAAC;YAED,iEAAiE;YACjE,MAAM,gBAAgB,GAAG,qBAAqB,CAC5C,IAAI,CAAC,QAAQ,EACb,WAAW,CACZ,CAAC;YAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,OAAO;oBACL,UAAU,EAAE,UAAU,gBAAgB,EAAE;oBACxC,aAAa,EAAE,UAAU,gBAAgB,EAAE;oBAC3C,KAAK,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE;iBAChC,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,UAAU,EAAE,gBAAgB,IAAI,6BAA6B;gBAC7D,aAAa,EAAE,gBAAgB,IAAI,6BAA6B;aACjE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAExE,uEAAuE;YACvE,iEAAiE;YACjE,IAAI,QAAQ,CAAC,QAAQ,CAAC,6BAA6B,CAAC,EAAE,CAAC;gBACrD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,WAAW,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,QAAQ,YAAY,QAAQ,EAAE,CAAC,CAAC;YACnE,OAAO;gBACL,UAAU,EAAE,UAAU,QAAQ,EAAE;gBAChC,aAAa,EAAE,UAAU,QAAQ,EAAE;gBACnC,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC7B,CAAC;QACJ,CAAC;IACH,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,kBAAmB,SAAQ,kBAGhC;IAEoB;IACA;IACA;IAHnB,YACmB,cAA8B,EAC9B,IAAY,EACZ,SAA6B,EAC9C,UAAsB;QAEtB,KAAK,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;QALhD,mBAAc,GAAd,cAAc,CAAgB;QAC9B,SAAI,GAAJ,IAAI,CAAQ;QACZ,cAAS,GAAT,SAAS,CAAoB;IAIhD,CAAC;IAED,cAAc;QACZ,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC;QACvF,OAAO,IAAI,CAAC,SAAS;YACnB,CAAC,CAAC,cAAc,OAAO,MAAM,IAAI,CAAC,SAAS,EAAE;YAC7C,CAAC,CAAC,cAAc,OAAO,EAAE,CAAC;IAC9B,CAAC;IAEkB,KAAK,CAAC,sBAAsB,CAC7C,YAAyB;QAEzB,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO;YACL,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,yBAAyB;YAChC,UAAU,EAAE,eAAe;YAC3B,QAAQ,EAAE,WAAW;YACrB,eAAe,EAAE,WAAW;YAC5B,SAAS,EAAE,KAAK,EAAE,OAAgC,EAAE,EAAE;gBACpD,MAAM,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;YAC1C,CAAC;SACF,CAAC;IACJ,CAAC;IAEQ,sBAAsB,CAC7B,QAAiC;QAEjC,OAAO;YACL,OAAO,EAAE,eAAe;SACzB,CAAC;IACJ,CAAC;IAEQ,KAAK,CAAC,OAAO,CAAC,MAAmB;QACxC,IAAI,CAAC;YACH,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,OAAO;oBACL,UAAU,EAAE,mDAAmD;oBAC/D,aAAa,EAAE,4CAA4C;oBAC3D,KAAK,EAAE,EAAE,OAAO,EAAE,qBAAqB,EAAE;iBAC1C,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAElC,gEAAgE;YAChE,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACtC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAClD,WAAW,EACX,EAAE,GAAG,EAAE,IAAI,CAAC,SAAS,EAAE,EACvB,MAAM,CACP,CAAC;gBACF,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;oBACtB,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;oBACjD,WAAW,CAAC,IAAI,CACd,yBAAyB,IAAI,CAAC,SAAS,cAAc,OAAO,EAAE,CAC/D,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS;gBAC5B,CAAC,CAAC,uBAAuB,IAAI,CAAC,IAAI,iBAAiB,IAAI,CAAC,SAAS,EAAE;gBACnE,CAAC,CAAC,uBAAuB,IAAI,CAAC,IAAI,GAAG,CAAC;YAExC,OAAO;gBACL,UAAU,EAAE,OAAO;gBACnB,aAAa,EAAE,OAAO;aACvB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAExE,qCAAqC;YACrC,IAAI,QAAQ,CAAC,QAAQ,CAAC,6BAA6B,CAAC,EAAE,CAAC;gBACrD,MAAM,KAAK,CAAC;YACd,CAAC;YAED,WAAW,CAAC,KAAK,CAAC,qBAAqB,QAAQ,EAAE,CAAC,CAAC;YACnD,OAAO;gBACL,UAAU,EAAE,UAAU,QAAQ,EAAE;gBAChC,aAAa,EAAE,UAAU,QAAQ,EAAE;gBACnC,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC7B,CAAC;QACJ,CAAC;IACH,CAAC;IAED,+DAA+D;IACvD,KAAK,CAAC,cAAc,CAAC,MAAmB;QAC9C,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,2BAA2B;QACzD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,MAAM,CAAC,OAAO;gBAAE,OAAO;YAE3B,sCAAsC;YACtC,MAAM,GAAG,GAAG,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAC/C,WAAW,EACX,EAAE,GAAG,EAAE,EACP,MAAM,CACP,CAAC;YAEF,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CACd,yBAAyB,GAAG,cAAc,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAC1E,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,kDAAkD;IAC1C,gBAAgB,CAAC,MAAyB;QAChD,OAAO,CACL,MAAM,CAAC,OAAO;YACZ,EAAE,MAAM,CACN,CAAC,CAAkC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CACpE;aACA,GAAG,CAAC,CAAC,CAAkC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aACnD,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,CACjC,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,kBAAmB,SAAQ,eAGhC;IAEsB;IADrB,YACqB,cAA8B,EACjD,IAAY,EACZ,WAAmB,EACnB,eAAwB,EACxB,UAAsB;QAEtB,KAAK,CACH,IAAI,EACJ,IAAI,EACJ,WAAW,EACX,IAAI,CAAC,KAAK,EACV,eAAe,EACf,UAAU;QACV,sBAAsB,CAAC,IAAI;QAC3B,qBAAqB,CAAC,KAAK,CAC5B,CAAC;QAfiB,mBAAc,GAAd,cAAc,CAAgB;IAgBnD,CAAC;IAED,KAAK,CACH,MAA+B;QAE/B,OAAO,IAAI,iBAAiB,CAC1B,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,IAAI,EACT,MAAM,EACN,IAAI,CAAC,UAAU,CAChB,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,uBAAwB,SAAQ,eAGrC;IAEoB;IADnB,YACmB,cAA8B,EAC/C,UAAsB;QAEtB,KAAK,CACH,WAAW,EACX,WAAW,EACX,+DAA+D;YAC7D,sEAAsE;YACtE,kGAAkG;YAClG,6DAA6D;YAC7D,qJAAqJ,EACvJ,IAAI,CAAC,KAAK,EACV;YACE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE;oBACJ,IAAI,EAAE,QAAQ;oBACd,WAAW,EAAE,4CAA4C;iBAC1D;gBACD,SAAS,EAAE;oBACT,IAAI,EAAE,QAAQ;oBACd,WAAW,EACT,uEAAuE;wBACvE,gFAAgF;iBACnF;aACF;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB,EACD,UAAU;QACV,sBAAsB,CAAC,IAAI;QAC3B,qBAAqB,CAAC,KAAK,CAC5B,CAAC;QA/Be,mBAAc,GAAd,cAAc,CAAgB;IAgCjD,CAAC;IAED,KAAK,CACH,MAA+B;QAE/B,MAAM,SAAS;QACb,gDAAgD;QAChD,OAAO,MAAM,CAAC,WAAW,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,WAAW,CAAC;YAC5D,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC;YACrB,CAAC,CAAC,SAAS,CAAC;QAChB,OAAO,IAAI,kBAAkB,CAC3B,IAAI,CAAC,cAAc,EACnB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,EAC5B,SAAS,EACT,IAAI,CAAC,UAAU,CAChB,CAAC;IACJ,CAAC;CACF;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,cAA8B,EAC9B,UAAsB;IAEtB,uDAAuD;IACvD,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,kBAAkB,EAAE,CAAC;IAE3D,WAAW,CAAC,GAAG,CACb,YAAY,QAAQ,CAAC,MAAM,sCAAsC,CAClE,CAAC;IAEF,MAAM,KAAK,GACT,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;QACvB,MAAM,MAAM,GAAG,mCAAmC,CAAC,OAAO,CAAC,CAAC;QAC5D,6CAA6C;QAC7C,MAAM,oBAAoB,GAAG,sBAAsB,CACjD,OAAO,CAAC,IAAI,EACZ,OAAO,CAAC,WAAW,IAAI,EAAE,CAC1B,CAAC;QACF,OAAO,IAAI,kBAAkB,CAC3B,cAAc,EACd,OAAO,CAAC,IAAI,EACZ,oBAAoB,EACpB,MAAM,CAAC,oBAAoB,EAC3B,UAAU,CACX,CAAC;IACJ,CAAC,CAAC,CAAC;IAEL,6BAA6B;IAC7B,KAAK,CAAC,IAAI,CAAC,IAAI,uBAAuB,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC,CAAC;IAEpE,WAAW,CAAC,GAAG,CACb,2BAA2B,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM,kBAAkB,CAC9E,CAAC;IAEF,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,mCAAmC,CAC1C,OAAgB;IAEhB,+CAA+C;IAC/C,8CAA8C;IAC9C,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,EAAE;QACtC,oBAAoB,EAAE,OAAO,CAAC,WAAW,IAAI;YAC3C,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE;SACf;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,SAAS,sBAAsB,CAAC,QAAgB,EAAE,WAAmB;IACnE,8DAA8D;IAC9D,wEAAwE;IACxE,MAAM,KAAK,GAA2B;QACpC,SAAS,EACP,uHAAuH;QACzH,IAAI,EAAE,8QAA8Q;QACpR,QAAQ,EACN,wGAAwG;QAC1G,KAAK,EACH,qKAAqK;QACvK,KAAK,EACH,mFAAmF;QACrF,aAAa,EACX,mMAAmM;QACrM,aAAa,EACX,+EAA+E;QACjF,QAAQ,EACN,6FAA6F;QAC/F,SAAS,EACP,kQAAkQ;KACrQ,CAAC;IAEF,uEAAuE;IACvE,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAChD,IAAI,QAAQ,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACzC,OAAO,WAAW,GAAG,IAAI,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CACnC,QAAgB,EAChB,MAAc;IAEd,6EAA6E;IAC7E,wEAAwE;IACxE,IAAI,eAAe,GAAG,MAAM,CAAC;IAE7B,IACE,QAAQ,KAAK,eAAe;QAC5B,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAC1C,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACtD,eAAe,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YACb,WAAW,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,qCAAqC;IACrC,MAAM,eAAe,GAAG;QACtB,kBAAkB;QAClB,UAAU;QACV,WAAW;QACX,SAAS;QACT,wBAAwB;QACxB,mBAAmB;KACpB,CAAC;IAEF,MAAM,cAAc,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CACtD,eAAe,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAChD,CAAC;IAEF,IAAI,cAAc,IAAI,CAAC,QAAQ,KAAK,OAAO,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;QAC3E,OAAO,CACL,eAAe;YACf,6EAA6E;YAC7E,+GAA+G,CAChH,CAAC;IACJ,CAAC;IAED,kCAAkC;IAClC,IACE,eAAe,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;QAC/C,eAAe,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,EAClD,CAAC;QACD,OAAO,CACL,eAAe;YACf,sFAAsF,CACvF,CAAC;IACJ,CAAC;IAED,OAAO,eAAe,CAAC;AACzB,CAAC"}

package/dist/src/agents/browser/mcpToolWrapper.test.d.ts

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

package/dist/src/agents/browser/mcpToolWrapper.test.js

@@ -0,0 +1,126 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { createMcpDeclarativeTools } from './mcpToolWrapper.js';
+describe('mcpToolWrapper', () => {
+    let mockBrowserManager;
+    let mockMessageBus;
+    let mockMcpTools;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        // Setup mock MCP tools discovered from server
+        mockMcpTools = [
+            {
+                name: 'take_snapshot',
+                description: 'Take a snapshot of the page accessibility tree',
+                inputSchema: {
+                    type: 'object',
+                    properties: {
+                        verbose: { type: 'boolean', description: 'Include details' },
+                    },
+                },
+            },
+            {
+                name: 'click',
+                description: 'Click on an element by uid',
+                inputSchema: {
+                    type: 'object',
+                    properties: {
+                        uid: { type: 'string', description: 'Element uid' },
+                    },
+                    required: ['uid'],
+                },
+            },
+        ];
+        // Setup mock browser manager
+        mockBrowserManager = {
+            getDiscoveredTools: vi.fn().mockResolvedValue(mockMcpTools),
+            callTool: vi.fn().mockResolvedValue({
+                content: [{ type: 'text', text: 'Tool result' }],
+            }),
+        };
+        // Setup mock message bus
+        mockMessageBus = {
+            publish: vi.fn().mockResolvedValue(undefined),
+            subscribe: vi.fn(),
+            unsubscribe: vi.fn(),
+        };
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('createMcpDeclarativeTools', () => {
+        it('should create declarative tools from discovered MCP tools', async () => {
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            expect(tools).toHaveLength(3);
+            expect(tools[0].name).toBe('take_snapshot');
+            expect(tools[1].name).toBe('click');
+            expect(tools[2].name).toBe('type_text');
+        });
+        it('should return tools with correct description', async () => {
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            // Descriptions include augmented hints, so we check they contain the original
+            expect(tools[0].description).toContain('Take a snapshot of the page accessibility tree');
+            expect(tools[1].description).toContain('Click on an element by uid');
+        });
+        it('should return tools with proper FunctionDeclaration schema', async () => {
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            const schema = tools[0].schema;
+            expect(schema.name).toBe('take_snapshot');
+            expect(schema.parametersJsonSchema).toBeDefined();
+        });
+    });
+    describe('McpDeclarativeTool.build', () => {
+        it('should create invocation that can be executed', async () => {
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            const invocation = tools[0].build({ verbose: true });
+            expect(invocation).toBeDefined();
+            expect(invocation.params).toEqual({ verbose: true });
+        });
+        it('should return invocation with correct description', async () => {
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            const invocation = tools[0].build({});
+            expect(invocation.getDescription()).toContain('take_snapshot');
+        });
+    });
+    describe('McpToolInvocation.execute', () => {
+        it('should call browserManager.callTool with correct params', async () => {
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            const invocation = tools[1].build({ uid: 'elem-123' });
+            await invocation.execute(new AbortController().signal);
+            expect(mockBrowserManager.callTool).toHaveBeenCalledWith('click', {
+                uid: 'elem-123',
+            }, expect.any(AbortSignal));
+        });
+        it('should return success result from MCP tool', async () => {
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            const invocation = tools[0].build({ verbose: true });
+            const result = await invocation.execute(new AbortController().signal);
+            expect(result.llmContent).toBe('Tool result');
+            expect(result.error).toBeUndefined();
+        });
+        it('should handle MCP tool errors', async () => {
+            vi.mocked(mockBrowserManager.callTool).mockResolvedValue({
+                content: [{ type: 'text', text: 'Element not found' }],
+                isError: true,
+            });
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            const invocation = tools[1].build({ uid: 'invalid' });
+            const result = await invocation.execute(new AbortController().signal);
+            expect(result.error).toBeDefined();
+            expect(result.error?.message).toBe('Element not found');
+        });
+        it('should handle exceptions during tool call', async () => {
+            vi.mocked(mockBrowserManager.callTool).mockRejectedValue(new Error('Connection lost'));
+            const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+            const invocation = tools[0].build({});
+            const result = await invocation.execute(new AbortController().signal);
+            expect(result.error).toBeDefined();
+            expect(result.error?.message).toBe('Connection lost');
+        });
+    });
+});
+//# sourceMappingURL=mcpToolWrapper.test.js.map

package/dist/src/agents/browser/mcpToolWrapper.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcpToolWrapper.test.js","sourceRoot":"","sources":["../../../../src/agents/browser/mcpToolWrapper.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAKhE,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,kBAAkC,CAAC;IACvC,IAAI,cAA0B,CAAC;IAC/B,IAAI,YAAuB,CAAC;IAE5B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,8CAA8C;QAC9C,YAAY,GAAG;YACb;gBACE,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,gDAAgD;gBAC7D,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,OAAO,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE;qBAC7D;iBACF;aACF;YACD;gBACE,IAAI,EAAE,OAAO;gBACb,WAAW,EAAE,4BAA4B;gBACzC,WAAW,EAAE;oBACX,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,aAAa,EAAE;qBACpD;oBACD,QAAQ,EAAE,CAAC,KAAK,CAAC;iBAClB;aACF;SACF,CAAC;QAEF,6BAA6B;QAC7B,kBAAkB,GAAG;YACnB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,YAAY,CAAC;YAC3D,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAClC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;aAC5B,CAAC;SACK,CAAC;QAE/B,yBAAyB;QACzB,cAAc,GAAG;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC7C,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;SACI,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,8EAA8E;YAC9E,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,SAAS,CACpC,gDAAgD,CACjD,CAAC;YACF,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,WAAW,EAAE,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAErD,MAAM,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;YACjC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAEtC,MAAM,CAAC,UAAU,CAAC,cAAc,EAAE,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,CAAC;YACvD,MAAM,UAAU,CAAC,OAAO,CAAC,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC;YAEvD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CACtD,OAAO,EACP;gBACE,GAAG,EAAE,UAAU;aAChB,EACD,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CACxB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YACrD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC;gBACvD,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC;gBACtD,OAAO,EAAE,IAAI;aACO,CAAC,CAAC;YAExB,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CACtD,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAC7B,CAAC;YAEF,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;YAEF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC,CAAC;YAEtE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/browser/mcpToolWrapperConfirmation.test.d.ts

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

package/dist/src/agents/browser/mcpToolWrapperConfirmation.test.js

@@ -0,0 +1,59 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { createMcpDeclarativeTools } from './mcpToolWrapper.js';
+import { MessageBusType } from '../../confirmation-bus/types.js';
+import { ToolConfirmationOutcome, } from '../../tools/tools.js';
+import { makeFakeConfig } from '../../test-utils/config.js';
+describe('mcpToolWrapper Confirmation', () => {
+    let mockBrowserManager;
+    let mockMessageBus;
+    beforeEach(() => {
+        makeFakeConfig(); // ensure config module is loaded
+        mockBrowserManager = {
+            getDiscoveredTools: vi
+                .fn()
+                .mockResolvedValue([
+                { name: 'test_tool', description: 'desc', inputSchema: {} },
+            ]),
+            callTool: vi.fn(),
+        };
+        mockMessageBus = {
+            publish: vi.fn().mockResolvedValue(undefined),
+            subscribe: vi.fn(),
+            unsubscribe: vi.fn(),
+        };
+    });
+    it('getConfirmationDetails returns specific MCP details', async () => {
+        const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+        const invocation = tools[0].build({});
+        const details = await invocation.getConfirmationDetails(new AbortController().signal);
+        expect(details).toEqual(expect.objectContaining({
+            type: 'mcp',
+            serverName: 'browser-agent',
+            toolName: 'test_tool',
+        }));
+        // Verify onConfirm publishes policy update
+        const outcome = ToolConfirmationOutcome.ProceedAlways;
+        if (details && typeof details === 'object' && 'onConfirm' in details) {
+            await details.onConfirm(outcome);
+        }
+        expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+            type: MessageBusType.UPDATE_POLICY,
+            mcpName: 'browser-agent',
+            persist: false,
+        }));
+    });
+    it('getPolicyUpdateOptions returns correct options', async () => {
+        const tools = await createMcpDeclarativeTools(mockBrowserManager, mockMessageBus);
+        const invocation = tools[0].build({});
+        const options = invocation.getPolicyUpdateOptions(ToolConfirmationOutcome.ProceedAlways);
+        expect(options).toEqual({
+            mcpName: 'browser-agent',
+        });
+    });
+});
+//# sourceMappingURL=mcpToolWrapperConfirmation.test.js.map

package/dist/src/agents/browser/mcpToolWrapperConfirmation.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcpToolWrapperConfirmation.test.js","sourceRoot":"","sources":["../../../../src/agents/browser/mcpToolWrapperConfirmation.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAGhE,OAAO,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AACjE,OAAO,EACL,uBAAuB,GAGxB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAW5D,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,kBAAkC,CAAC;IACvC,IAAI,cAA0B,CAAC;IAE/B,UAAU,CAAC,GAAG,EAAE;QACd,cAAc,EAAE,CAAC,CAAC,iCAAiC;QACnD,kBAAkB,GAAG;YACnB,kBAAkB,EAAE,EAAE;iBACnB,EAAE,EAAE;iBACJ,iBAAiB,CAAC;gBACjB,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,EAAE;aAC5D,CAAC;YACJ,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SACW,CAAC;QAE/B,cAAc,GAAG;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;YAC7C,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;SACI,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;QACF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAoC,CAAC;QAEzE,MAAM,OAAO,GAAG,MAAM,UAAU,CAAC,sBAAsB,CACrD,IAAI,eAAe,EAAE,CAAC,MAAM,CAC7B,CAAC;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CACrB,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE,KAAK;YACX,UAAU,EAAE,eAAe;YAC3B,QAAQ,EAAE,WAAW;SACtB,CAAC,CACH,CAAC;QAEF,2CAA2C;QAC3C,MAAM,OAAO,GAAG,uBAAuB,CAAC,aAAa,CAAC;QAEtD,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,WAAW,IAAI,OAAO,EAAE,CAAC;YACrE,MAAM,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;QAED,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE,cAAc,CAAC,aAAa;YAClC,OAAO,EAAE,eAAe;YACxB,OAAO,EAAE,KAAK;SACf,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,KAAK,GAAG,MAAM,yBAAyB,CAC3C,kBAAkB,EAClB,cAAc,CACf,CAAC;QACF,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAoC,CAAC;QAEzE,MAAM,OAAO,GAAG,UAAU,CAAC,sBAAsB,CAC/C,uBAAuB,CAAC,aAAa,CACtC,CAAC;QAEF,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;YACtB,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/agents/browser/modelAvailability.d.ts

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * @fileoverview Model configuration for browser agent.
+ *
+ * Provides the default visual agent model and utilities for resolving
+ * the configured model.
+ */
+import type { Config } from '../../config/config.js';
+/**
+ * Default model for the visual agent (Computer Use capable).
+ */
+export declare const VISUAL_AGENT_MODEL = "gemini-2.5-computer-use-preview-10-2025";
+/**
+ * Gets the visual agent model from config, falling back to default.
+ *
+ * @param config Runtime configuration
+ * @returns The model to use for visual agent
+ */
+export declare function getVisualAgentModel(config: Config): string;

package/dist/src/agents/browser/modelAvailability.js

@@ -0,0 +1,23 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { debugLogger } from '../../utils/debugLogger.js';
+/**
+ * Default model for the visual agent (Computer Use capable).
+ */
+export const VISUAL_AGENT_MODEL = 'gemini-2.5-computer-use-preview-10-2025';
+/**
+ * Gets the visual agent model from config, falling back to default.
+ *
+ * @param config Runtime configuration
+ * @returns The model to use for visual agent
+ */
+export function getVisualAgentModel(config) {
+    const browserConfig = config.getBrowserAgentConfig();
+    const model = browserConfig.customConfig.visualModel ?? VISUAL_AGENT_MODEL;
+    debugLogger.log(`Visual agent model: ${model}`);
+    return model;
+}
+//# sourceMappingURL=modelAvailability.js.map

package/dist/src/agents/browser/modelAvailability.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"modelAvailability.js","sourceRoot":"","sources":["../../../../src/agents/browser/modelAvailability.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAUH,OAAO,EAAE,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEzD;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,yCAAyC,CAAC;AAE5E;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAAc;IAChD,MAAM,aAAa,GAAG,MAAM,CAAC,qBAAqB,EAAE,CAAC;IACrD,MAAM,KAAK,GAAG,aAAa,CAAC,YAAY,CAAC,WAAW,IAAI,kBAAkB,CAAC;IAE3E,WAAW,CAAC,GAAG,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC;IAChD,OAAO,KAAK,CAAC;AACf,CAAC"}