@playwright/mcp 0.0.24 → 0.0.26

package/config.d.ts

@@ -28,6 +28,11 @@ export type Config = {
      */
     browserName?: 'chromium' | 'firefox' | 'webkit';
 
+    /**
+     * Keep the browser profile in memory, do not save it to disk.
+     */
+    isolated?: boolean;
+
     /**
      * Path to a user data directory for browser profile persistence.
      * Temporary directory is created by default.
@@ -89,6 +94,11 @@ export type Config = {
    */
   vision?: boolean;
 
+  /**
+   * Whether to save the Playwright trace of the session into the output directory.
+   */
+  saveTrace?: boolean;
+
   /**
    * The directory to save output files.
    */

package/lib/config.js

@@ -25,6 +25,7 @@ const defaultConfig = {
         launchOptions: {
             channel: 'chrome',
             headless: os.platform() === 'linux' && !process.env.DISPLAY,
+            chromiumSandbox: true,
         },
         contextOptions: {
             viewport: null,
@@ -34,11 +35,19 @@ const defaultConfig = {
         allowedOrigins: undefined,
         blockedOrigins: undefined,
     },
+    outputDir: path.join(os.tmpdir(), 'playwright-mcp-output', sanitizeForFilePath(new Date().toISOString())),
 };
-export async function resolveConfig(cliOptions) {
-    const config = await loadConfig(cliOptions.config);
+export async function resolveConfig(config) {
+    return mergeConfig(defaultConfig, config);
+}
+export async function resolveCLIConfig(cliOptions) {
+    const configInFile = await loadConfig(cliOptions.config);
     const cliOverrides = await configFromCLIOptions(cliOptions);
-    return mergeConfig(defaultConfig, mergeConfig(config, cliOverrides));
+    const result = mergeConfig(mergeConfig(defaultConfig, configInFile), cliOverrides);
+    // Derive artifact output directory from config.outputDir
+    if (result.saveTrace)
+        result.browser.launchOptions.tracesDir = path.join(result.outputDir, 'traces');
+    return result;
 }
 export async function configFromCLIOptions(cliOptions) {
     let browserName;
@@ -66,17 +75,51 @@ export async function configFromCLIOptions(cliOptions) {
             browserName = 'chromium';
             channel = 'chrome';
     }
+    // Launch options
     const launchOptions = {
         channel,
         executablePath: cliOptions.executablePath,
         headless: cliOptions.headless,
     };
-    if (browserName === 'chromium')
+    if (browserName === 'chromium') {
         launchOptions.cdpPort = await findFreePort();
-    const contextOptions = cliOptions.device ? devices[cliOptions.device] : undefined;
-    return {
+        if (!cliOptions.sandbox) {
+            // --no-sandbox was passed, disable the sandbox
+            launchOptions.chromiumSandbox = false;
+        }
+    }
+    if (cliOptions.proxyServer) {
+        launchOptions.proxy = {
+            server: cliOptions.proxyServer
+        };
+        if (cliOptions.proxyBypass)
+            launchOptions.proxy.bypass = cliOptions.proxyBypass;
+    }
+    // Context options
+    const contextOptions = cliOptions.device ? devices[cliOptions.device] : {};
+    if (cliOptions.storageState)
+        contextOptions.storageState = cliOptions.storageState;
+    if (cliOptions.userAgent)
+        contextOptions.userAgent = cliOptions.userAgent;
+    if (cliOptions.viewportSize) {
+        try {
+            const [width, height] = cliOptions.viewportSize.split(',').map(n => +n);
+            if (isNaN(width) || isNaN(height))
+                throw new Error('bad values');
+            contextOptions.viewport = { width, height };
+        }
+        catch (e) {
+            throw new Error('Invalid viewport size format: use "width,height", for example --viewport-size="800,600"');
+        }
+    }
+    if (cliOptions.ignoreHttpsErrors)
+        contextOptions.ignoreHTTPSErrors = true;
+    if (cliOptions.blockServiceWorkers)
+        contextOptions.serviceWorkers = 'block';
+    const result = {
         browser: {
             browserName,
+            isolated: cliOptions.isolated,
             userDataDir: cliOptions.userDataDir,
             launchOptions,
             contextOptions,
@@ -92,8 +135,14 @@ export async function configFromCLIOptions(cliOptions) {
             allowedOrigins: cliOptions.allowedOrigins,
             blockedOrigins: cliOptions.blockedOrigins,
         },
+        saveTrace: cliOptions.saveTrace,
         outputDir: cliOptions.outputDir,
     };
+    if (!cliOptions.imageResponses) {
+        // --no-image-responses was passed, disable image responses
+        result.noImageResponses = true;
+    }
+    return result;
 }
 async function findFreePort() {
     return new Promise((resolve, reject) => {
@@ -116,18 +165,17 @@ async function loadConfig(configFile) {
     }
 }
 export async function outputFile(config, name) {
-    const result = config.outputDir ?? os.tmpdir();
-    await fs.promises.mkdir(result, { recursive: true });
+    await fs.promises.mkdir(config.outputDir, { recursive: true });
     const fileName = sanitizeForFilePath(name);
-    return path.join(result, fileName);
+    return path.join(config.outputDir, fileName);
 }
 function pickDefined(obj) {
     return Object.fromEntries(Object.entries(obj ?? {}).filter(([_, v]) => v !== undefined));
 }
 function mergeConfig(base, overrides) {
     const browser = {
-        ...pickDefined(base.browser),
-        ...pickDefined(overrides.browser),
+        browserName: overrides.browser?.browserName ?? base.browser?.browserName ?? 'chromium',
+        isolated: overrides.browser?.isolated ?? base.browser?.isolated ?? false,
         launchOptions: {
             ...pickDefined(base.browser?.launchOptions),
             ...pickDefined(overrides.browser?.launchOptions),
@@ -137,6 +185,9 @@ function mergeConfig(base, overrides) {
             ...pickDefined(base.browser?.contextOptions),
             ...pickDefined(overrides.browser?.contextOptions),
         },
+        userDataDir: overrides.browser?.userDataDir ?? base.browser?.userDataDir,
+        cdpEndpoint: overrides.browser?.cdpEndpoint ?? base.browser?.cdpEndpoint,
+        remoteEndpoint: overrides.browser?.remoteEndpoint ?? base.browser?.remoteEndpoint,
     };
     if (browser.browserName !== 'chromium' && browser.launchOptions)
         delete browser.launchOptions.channel;
@@ -147,6 +198,6 @@ function mergeConfig(base, overrides) {
         network: {
             ...pickDefined(base.network),
             ...pickDefined(overrides.network),
-        },
+        }
     };
 }

package/lib/connection.js

@@ -17,9 +17,9 @@ import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
 import { zodToJsonSchema } from 'zod-to-json-schema';
 import { Context, packageJSON } from './context.js';
-import { snapshotTools, screenshotTools } from './tools.js';
+import { snapshotTools, visionTools } from './tools.js';
 export async function createConnection(config) {
-    const allTools = config.vision ? screenshotTools : snapshotTools;
+    const allTools = config.vision ? visionTools : snapshotTools;
     const tools = allTools.filter(tool => !config.capabilities || tool.capability === 'core' || config.capabilities.includes(tool.capability));
     const context = new Context(tools, config);
     const server = new Server({ name: 'Playwright', version: packageJSON.version }, {

package/lib/context.js

@@ -18,16 +18,14 @@ import url from 'node:url';
 import os from 'node:os';
 import path from 'node:path';
 import * as playwright from 'playwright';
-import { waitForCompletion } from './tools/utils.js';
+import { callOnPageNoTrace, waitForCompletion } from './tools/utils.js';
 import { ManualPromise } from './manualPromise.js';
 import { Tab } from './tab.js';
 import { outputFile } from './config.js';
 export class Context {
     tools;
     config;
-    _browser;
-    _browserContext;
-    _createBrowserContextPromise;
+    _browserContextPromise;
     _tabs = [];
     _currentTab;
     _modalStates = [];
@@ -65,7 +63,7 @@ export class Context {
         return this._currentTab;
     }
     async newTab() {
-        const browserContext = await this._ensureBrowserContext();
+        const { browserContext } = await this._ensureBrowserContext();
         const page = await browserContext.newPage();
         this._currentTab = this._tabs.find(t => t.page === page);
         return this._currentTab;
@@ -75,9 +73,9 @@ export class Context {
         await this._currentTab.page.bringToFront();
     }
     async ensureTab() {
-        const context = await this._ensureBrowserContext();
+        const { browserContext } = await this._ensureBrowserContext();
         if (!this._currentTab)
-            await context.newPage();
+            await browserContext.newPage();
         return this._currentTab;
     }
     async listTabsMarkdown() {
@@ -86,7 +84,7 @@ export class Context {
         const lines = ['### Open tabs'];
         for (let i = 0; i < this._tabs.length; i++) {
             const tab = this._tabs[i];
-            const title = await tab.page.title();
+            const title = await tab.title();
             const url = tab.page.url();
             const current = tab === this._currentTab ? ' (current)' : '';
             lines.push(`- ${i + 1}:${current} [${title}] (${url})`);
@@ -118,7 +116,7 @@ export class Context {
         let actionResult;
         try {
             if (waitForNetwork)
-                actionResult = await waitForCompletion(this, tab.page, async () => racingAction?.()) ?? undefined;
+                actionResult = await waitForCompletion(this, tab, async () => racingAction?.()) ?? undefined;
             else
                 actionResult = await racingAction?.() ?? undefined;
         }
@@ -155,7 +153,7 @@ ${code.join('\n')}
             result.push(await this.listTabsMarkdown(), '');
         if (this.tabs().length > 1)
             result.push('### Current tab');
-        result.push(`- Page URL: ${tab.page.url()}`, `- Page Title: ${await tab.page.title()}`);
+        result.push(`- Page URL: ${tab.page.url()}`, `- Page Title: ${await tab.title()}`);
         if (captureSnapshot && tab.hasSnapshot())
             result.push(tab.snapshotOrDie().text());
         const content = actionResult?.content ?? [];
@@ -170,10 +168,13 @@ ${code.join('\n')}
         };
     }
     async waitForTimeout(time) {
-        if (this._currentTab && !this._javaScriptBlocked())
-            await this._currentTab.page.evaluate(() => new Promise(f => setTimeout(f, 1000)));
-        else
+        if (!this._currentTab || this._javaScriptBlocked()) {
             await new Promise(f => setTimeout(f, time));
+            return;
+        }
+        await callOnPageNoTrace(this._currentTab.page, page => {
+            return page.evaluate(() => new Promise(f => setTimeout(f, 1000)));
+        });
     }
     async _raceAgainstModalDialogs(action) {
         this._pendingAction = {
@@ -226,20 +227,21 @@ ${code.join('\n')}
         this._tabs.splice(index, 1);
         if (this._currentTab === tab)
             this._currentTab = this._tabs[Math.min(index, this._tabs.length - 1)];
-        if (this._browserContext && !this._tabs.length)
+        if (!this._tabs.length)
             void this.close();
     }
     async close() {
-        if (!this._browserContext)
+        if (!this._browserContextPromise)
             return;
-        const browserContext = this._browserContext;
-        const browser = this._browser;
-        this._createBrowserContextPromise = undefined;
-        this._browserContext = undefined;
-        this._browser = undefined;
-        await browserContext?.close().then(async () => {
-            await browser?.close();
-        }).catch(() => { });
+        const promise = this._browserContextPromise;
+        this._browserContextPromise = undefined;
+        await promise.then(async ({ browserContext, browser }) => {
+            if (this.config.saveTrace)
+                await browserContext.tracing.stop();
+            await browserContext.close().then(async () => {
+                await browser?.close();
+            }).catch(() => { });
+        });
     }
     async _setupRequestInterception(context) {
         if (this.config.network?.allowedOrigins?.length) {
@@ -252,28 +254,32 @@ ${code.join('\n')}
                 await context.route(`*://${origin}/**`, route => route.abort('blockedbyclient'));
         }
     }
-    async _ensureBrowserContext() {
-        if (!this._browserContext) {
-            const context = await this._createBrowserContext();
-            this._browser = context.browser;
-            this._browserContext = context.browserContext;
-            await this._setupRequestInterception(this._browserContext);
-            for (const page of this._browserContext.pages())
-                this._onPageCreated(page);
-            this._browserContext.on('page', page => this._onPageCreated(page));
+    _ensureBrowserContext() {
+        if (!this._browserContextPromise) {
+            this._browserContextPromise = this._setupBrowserContext();
+            this._browserContextPromise.catch(() => {
+                this._browserContextPromise = undefined;
+            });
         }
-        return this._browserContext;
+        return this._browserContextPromise;
     }
-    async _createBrowserContext() {
-        if (!this._createBrowserContextPromise) {
-            this._createBrowserContextPromise = this._innerCreateBrowserContext();
-            void this._createBrowserContextPromise.catch(() => {
-                this._createBrowserContextPromise = undefined;
+    async _setupBrowserContext() {
+        const { browser, browserContext } = await this._createBrowserContext();
+        await this._setupRequestInterception(browserContext);
+        for (const page of browserContext.pages())
+            this._onPageCreated(page);
+        browserContext.on('page', page => this._onPageCreated(page));
+        if (this.config.saveTrace) {
+            await browserContext.tracing.start({
+                name: 'trace',
+                screenshots: false,
+                snapshots: true,
+                sources: false,
             });
         }
-        return this._createBrowserContextPromise;
+        return { browser, browserContext };
     }
-    async _innerCreateBrowserContext() {
+    async _createBrowserContext() {
         if (this.config.browser?.remoteEndpoint) {
             const url = new URL(this.config.browser?.remoteEndpoint);
             if (this.config.browser.browserName)
@@ -286,19 +292,35 @@ ${code.join('\n')}
         }
         if (this.config.browser?.cdpEndpoint) {
             const browser = await playwright.chromium.connectOverCDP(this.config.browser.cdpEndpoint);
-            const browserContext = browser.contexts()[0];
+            const browserContext = this.config.browser.isolated ? await browser.newContext() : browser.contexts()[0];
             return { browser, browserContext };
         }
-        const browserContext = await launchPersistentContext(this.config.browser);
-        return { browserContext };
+        return this.config.browser?.isolated ?
+            await createIsolatedContext(this.config.browser) :
+            await launchPersistentContext(this.config.browser);
     }
 }
-async function launchPersistentContext(browserConfig) {
+async function createIsolatedContext(browserConfig) {
     try {
         const browserName = browserConfig?.browserName ?? 'chromium';
-        const userDataDir = browserConfig?.userDataDir ?? await createUserDataDir({ ...browserConfig, browserName });
         const browserType = playwright[browserName];
-        return await browserType.launchPersistentContext(userDataDir, { ...browserConfig?.launchOptions, ...browserConfig?.contextOptions });
+        const browser = await browserType.launch(browserConfig.launchOptions);
+        const browserContext = await browser.newContext(browserConfig.contextOptions);
+        return { browser, browserContext };
+    }
+    catch (error) {
+        if (error.message.includes('Executable doesn\'t exist'))
+            throw new Error(`Browser specified in your config is not installed. Either install it (likely) or change the config.`);
+        throw error;
+    }
+}
+async function launchPersistentContext(browserConfig) {
+    try {
+        const browserName = browserConfig.browserName ?? 'chromium';
+        const userDataDir = browserConfig.userDataDir ?? await createUserDataDir({ ...browserConfig, browserName });
+        const browserType = playwright[browserName];
+        const browserContext = await browserType.launchPersistentContext(userDataDir, { ...browserConfig.launchOptions, ...browserConfig.contextOptions });
+        return { browserContext };
     }
     catch (error) {
         if (error.message.includes('Executable doesn\'t exist'))
@@ -316,12 +338,9 @@ async function createUserDataDir(browserConfig) {
         cacheDirectory = process.env.LOCALAPPDATA || path.join(os.homedir(), 'AppData', 'Local');
     else
         throw new Error('Unsupported platform: ' + process.platform);
-    const result = path.join(cacheDirectory, 'ms-playwright', `mcp-${browserConfig?.launchOptions?.channel ?? browserConfig?.browserName}-profile`);
+    const result = path.join(cacheDirectory, 'ms-playwright', `mcp-${browserConfig.launchOptions?.channel ?? browserConfig?.browserName}-profile`);
     await fs.promises.mkdir(result, { recursive: true });
     return result;
 }
-export async function generateLocator(locator) {
-    return locator._generateLocatorString();
-}
 const __filename = url.fileURLToPath(import.meta.url);
 export const packageJSON = JSON.parse(fs.readFileSync(path.join(path.dirname(__filename), '..', 'package.json'), 'utf8'));

package/lib/index.js

@@ -14,6 +14,8 @@
  * limitations under the License.
  */
 import { createConnection as createConnectionImpl } from './connection.js';
-export async function createConnection(config = {}) {
+import { resolveConfig } from './config.js';
+export async function createConnection(userConfig = {}) {
+    const config = await resolveConfig(userConfig);
     return createConnectionImpl(config);
 }

package/lib/pageSnapshot.js

@@ -13,6 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+import { callOnPageNoTrace } from './tools/utils.js';
 export class PageSnapshot {
     _page;
     _text;
@@ -28,11 +29,14 @@ export class PageSnapshot {
         return this._text;
     }
     async _build() {
-        const yamlDocument = await this._page._snapshotForAI();
+        // FIXME: Rountrip evaluate to ensure _snapshotForAI works.
+        // This probably broke once we moved off locator snapshots
+        await this._page.evaluate(() => 1);
+        const snapshot = await callOnPageNoTrace(this._page, page => page._snapshotForAI());
         this._text = [
             `- Page Snapshot`,
             '```yaml',
-            yamlDocument.toString({ indentSeq: false }).trim(),
+            snapshot,
             '```',
         ].join('\n');
     }

package/lib/program.js

@@ -15,34 +15,53 @@
  */
 import { program } from 'commander';
 import { startHttpTransport, startStdioTransport } from './transport.js';
-import { resolveConfig } from './config.js';
+import { resolveCLIConfig } from './config.js';
+// @ts-ignore
+import { startTraceViewerServer } from 'playwright-core/lib/server';
 import { packageJSON } from './context.js';
 program
     .version('Version ' + packageJSON.version)
     .name(packageJSON.name)
-    .option('--browser <browser>', 'Browser or chrome channel to use, possible values: chrome, firefox, webkit, msedge.')
-    .option('--caps <caps>', 'Comma-separated list of capabilities to enable, possible values: tabs, pdf, history, wait, files, install. Default is all.')
+    .option('--allowed-origins <origins>', 'semicolon-separated list of origins to allow the browser to request. Default is to allow all.', semicolonSeparatedList)
+    .option('--blocked-origins <origins>', 'semicolon-separated list of origins to block the browser from requesting. Blocklist is evaluated before allowlist. If used without the allowlist, requests not matching the blocklist are still allowed.', semicolonSeparatedList)
+    .option('--block-service-workers', 'block service workers')
+    .option('--browser <browser>', 'browser or chrome channel to use, possible values: chrome, firefox, webkit, msedge.')
+    .option('--caps <caps>', 'comma-separated list of capabilities to enable, possible values: tabs, pdf, history, wait, files, install. Default is all.')
     .option('--cdp-endpoint <endpoint>', 'CDP endpoint to connect to.')
-    .option('--executable-path <path>', 'Path to the browser executable.')
-    .option('--headless', 'Run browser in headless mode, headed by default')
-    .option('--device <device>', 'Device to emulate, for example: "iPhone 15"')
-    .option('--user-data-dir <path>', 'Path to the user data directory')
-    .option('--port <port>', 'Port to listen on for SSE transport.')
-    .option('--host <host>', 'Host to bind server to. Default is localhost. Use 0.0.0.0 to bind to all interfaces.')
-    .option('--allowed-origins <origins>', 'Semicolon-separated list of origins to allow the browser to request. Default is to allow all.', semicolonSeparatedList)
-    .option('--blocked-origins <origins>', 'Semicolon-separated list of origins to block the browser from requesting. Blocklist is evaluated before allowlist. If used without the allowlist, requests not matching the blocklist are still allowed.', semicolonSeparatedList)
+    .option('--config <path>', 'path to the configuration file.')
+    .option('--device <device>', 'device to emulate, for example: "iPhone 15"')
+    .option('--executable-path <path>', 'path to the browser executable.')
+    .option('--headless', 'run browser in headless mode, headed by default')
+    .option('--host <host>', 'host to bind server to. Default is localhost. Use 0.0.0.0 to bind to all interfaces.')
+    .option('--ignore-https-errors', 'ignore https errors')
+    .option('--isolated', 'keep the browser profile in memory, do not save it to disk.')
+    .option('--no-image-responses', 'do not send image responses to the client.')
+    .option('--no-sandbox', 'disable the sandbox for all process types that are normally sandboxed.')
+    .option('--output-dir <path>', 'path to the directory for output files.')
+    .option('--port <port>', 'port to listen on for SSE transport.')
+    .option('--proxy-bypass <bypass>', 'comma-separated domains to bypass proxy, for example ".com,chromium.org,.domain.com"')
+    .option('--proxy-server <proxy>', 'specify proxy server, for example "http://myproxy:3128" or "socks5://myproxy:8080"')
+    .option('--save-trace', 'Whether to save the Playwright Trace of the session into the output directory.')
+    .option('--storage-state <path>', 'path to the storage state file for isolated sessions.')
+    .option('--user-agent <ua string>', 'specify user agent string')
+    .option('--user-data-dir <path>', 'path to the user data directory. If not specified, a temporary directory will be created.')
+    .option('--viewport-size <size>', 'specify browser viewport size in pixels, for example "1280, 720"')
     .option('--vision', 'Run server that uses screenshots (Aria snapshots are used by default)')
-    .option('--no-image-responses', 'Do not send image responses to the client.')
-    .option('--output-dir <path>', 'Path to the directory for output files.')
-    .option('--config <path>', 'Path to the configuration file.')
     .action(async (options) => {
-    const config = await resolveConfig(options);
+    const config = await resolveCLIConfig(options);
     const connectionList = [];
     setupExitWatchdog(connectionList);
     if (options.port)
         startHttpTransport(config, +options.port, options.host, connectionList);
     else
         await startStdioTransport(config, connectionList);
+    if (config.saveTrace) {
+        const server = await startTraceViewerServer();
+        const urlPrefix = server.urlPrefix('human-readable');
+        const url = urlPrefix + '/trace/index.html?trace=' + config.browser.launchOptions.tracesDir + '/trace.json';
+        // eslint-disable-next-line no-console
+        console.error('\nTrace viewer listening on ' + url);
+    }
 });
 function setupExitWatchdog(connectionList) {
     const handleExit = async () => {

package/lib/tab.js

@@ -14,6 +14,7 @@
  * limitations under the License.
  */
 import { PageSnapshot } from './pageSnapshot.js';
+import { callOnPageNoTrace } from './tools/utils.js';
 export class Tab {
     context;
     page;
@@ -51,9 +52,15 @@ export class Tab {
         this._clearCollectedArtifacts();
         this._onPageClose(this);
     }
+    async title() {
+        return await callOnPageNoTrace(this.page, page => page.title());
+    }
+    async waitForLoadState(state, options) {
+        await callOnPageNoTrace(this.page, page => page.waitForLoadState(state, options).catch(() => { }));
+    }
     async navigate(url) {
         this._clearCollectedArtifacts();
-        const downloadEvent = this.page.waitForEvent('download').catch(() => { });
+        const downloadEvent = callOnPageNoTrace(this.page, page => page.waitForEvent('download').catch(() => { }));
         try {
             await this.page.goto(url, { waitUntil: 'domcontentloaded' });
         }
@@ -72,7 +79,7 @@ export class Tab {
                 throw e;
         }
         // Cap load event to 5 seconds, the page is operational at this point.
-        await this.page.waitForLoadState('load', { timeout: 5000 }).catch(() => { });
+        await this.waitForLoadState('load', { timeout: 5000 });
     }
     hasSnapshot() {
         return !!this._snapshot;

package/lib/tools/common.js

@@ -15,45 +15,6 @@
  */
 import { z } from 'zod';
 import { defineTool } from './tool.js';
-const wait = captureSnapshot => defineTool({
-    capability: 'wait',
-    schema: {
-        name: 'browser_wait_for',
-        title: 'Wait for',
-        description: 'Wait for text to appear or disappear or a specified time to pass',
-        inputSchema: z.object({
-            time: z.number().optional().describe('The time to wait in seconds'),
-            text: z.string().optional().describe('The text to wait for'),
-            textGone: z.string().optional().describe('The text to wait for to disappear'),
-        }),
-        type: 'readOnly',
-    },
-    handle: async (context, params) => {
-        if (!params.text && !params.textGone && !params.time)
-            throw new Error('Either time, text or textGone must be provided');
-        const code = [];
-        if (params.time) {
-            code.push(`await new Promise(f => setTimeout(f, ${params.time} * 1000));`);
-            await new Promise(f => setTimeout(f, Math.min(10000, params.time * 1000)));
-        }
-        const tab = context.currentTabOrDie();
-        const locator = params.text ? tab.page.getByText(params.text).first() : undefined;
-        const goneLocator = params.textGone ? tab.page.getByText(params.textGone).first() : undefined;
-        if (goneLocator) {
-            code.push(`await page.getByText(${JSON.stringify(params.textGone)}).first().waitFor({ state: 'hidden' });`);
-            await goneLocator.waitFor({ state: 'hidden' });
-        }
-        if (locator) {
-            code.push(`await page.getByText(${JSON.stringify(params.text)}).first().waitFor({ state: 'visible' });`);
-            await locator.waitFor({ state: 'visible' });
-        }
-        return {
-            code,
-            captureSnapshot,
-            waitForNetwork: false,
-        };
-    },
-});
 const close = defineTool({
     capability: 'core',
     schema: {
@@ -66,7 +27,7 @@ const close = defineTool({
     handle: async (context) => {
         await context.close();
         return {
-            code: [`// Internal to close the page`],
+            code: [`await page.close()`],
             captureSnapshot: false,
             waitForNetwork: false,
         };
@@ -103,6 +64,5 @@ const resize = captureSnapshot => defineTool({
 });
 export default (captureSnapshot) => [
     close,
-    wait(captureSnapshot),
     resize(captureSnapshot)
 ];