@mastra/agent-browser 0.2.2 → 0.3.0-alpha.1

package/dist/index.d.cts

@@ -1,4 +1,4 @@
-import { BrowserConfig as BrowserConfig$1, ThreadManager, ThreadManagerConfig, ThreadSession, MastraBrowser, BrowserToolError, BrowserState, BrowserTabState, ScreencastOptions, ScreencastStream, MouseEventParams, KeyboardEventParams } from '@mastra/core/browser';
+import { BrowserConfig as BrowserConfig$1, ThreadManager, ThreadManagerConfig, ThreadSession, BrowserState, MastraBrowser, BrowserToolError, BrowserTabState, ScreencastOptions, ScreencastStream, MouseEventParams, KeyboardEventParams } from '@mastra/core/browser';
 import { Tool } from '@mastra/core/tools';
 import { BrowserManager } from 'agent-browser';
 import { Page } from 'playwright-core';
@@ -54,6 +54,12 @@ declare const clickInputSchema: z.ZodObject<{
         Meta: "Meta";
         Shift: "Shift";
     }>>>;
+    waitUntil: z.ZodOptional<z.ZodEnum<{
+        load: "load";
+        domcontentloaded: "domcontentloaded";
+        networkidle: "networkidle";
+    }>>;
+    timeout: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 type ClickInput = z.output<typeof clickInputSchema>;
 /**
@@ -77,6 +83,12 @@ declare const pressInputSchema: z.ZodObject<{
         Meta: "Meta";
         Shift: "Shift";
     }>>>;
+    waitUntil: z.ZodOptional<z.ZodEnum<{
+        load: "load";
+        domcontentloaded: "domcontentloaded";
+        networkidle: "networkidle";
+    }>>;
+    timeout: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 type PressInput = z.output<typeof pressInputSchema>;
 /**
@@ -87,6 +99,12 @@ declare const selectInputSchema: z.ZodObject<{
     value: z.ZodOptional<z.ZodString>;
     label: z.ZodOptional<z.ZodString>;
     index: z.ZodOptional<z.ZodNumber>;
+    waitUntil: z.ZodOptional<z.ZodEnum<{
+        load: "load";
+        domcontentloaded: "domcontentloaded";
+        networkidle: "networkidle";
+    }>>;
+    timeout: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 type SelectInput = z.output<typeof selectInputSchema>;
 /**
@@ -213,6 +231,12 @@ declare const browserSchemas: {
             Meta: "Meta";
             Shift: "Shift";
         }>>>;
+        waitUntil: z.ZodOptional<z.ZodEnum<{
+            load: "load";
+            domcontentloaded: "domcontentloaded";
+            networkidle: "networkidle";
+        }>>;
+        timeout: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>;
     readonly type: z.ZodObject<{
         ref: z.ZodString;
@@ -228,12 +252,24 @@ declare const browserSchemas: {
             Meta: "Meta";
             Shift: "Shift";
         }>>>;
+        waitUntil: z.ZodOptional<z.ZodEnum<{
+            load: "load";
+            domcontentloaded: "domcontentloaded";
+            networkidle: "networkidle";
+        }>>;
+        timeout: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>;
     readonly select: z.ZodObject<{
         ref: z.ZodString;
         value: z.ZodOptional<z.ZodString>;
         label: z.ZodOptional<z.ZodString>;
         index: z.ZodOptional<z.ZodNumber>;
+        waitUntil: z.ZodOptional<z.ZodEnum<{
+            load: "load";
+            domcontentloaded: "domcontentloaded";
+            networkidle: "networkidle";
+        }>>;
+        timeout: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>;
     readonly scroll: z.ZodObject<{
         direction: z.ZodEnum<{
@@ -376,6 +412,11 @@ interface AgentBrowserThreadManagerConfig extends ThreadManagerConfig {
     /** Callback when a new browser manager is created for a thread */
     onBrowserCreated?: (manager: BrowserManager, threadId: string) => void;
 }
+/**
+ * Factory for custom thread managers (e.g. Firecrawl-hosted CDP per session).
+ * Defaults to {@link AgentBrowserThreadManager} when omitted.
+ */
+type CreateAgentBrowserThreadManager = (config: AgentBrowserThreadManagerConfig) => AgentBrowserThreadManager;
 /**
  * Thread manager implementation for AgentBrowser.
  *
@@ -384,9 +425,9 @@ interface AgentBrowserThreadManagerConfig extends ThreadManagerConfig {
  * - 'thread': Each thread gets a dedicated browser manager instance
  */
 declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
-    private readonly browserConfig;
+    protected readonly browserConfig: BrowserConfig;
     private readonly resolveCdpUrl?;
-    private readonly onBrowserCreated?;
+    protected readonly onBrowserCreated?: (manager: BrowserManager, threadId: string) => void;
     constructor(config: AgentBrowserThreadManagerConfig);
     /**
      * Get the page for a specific thread, creating session if needed.
@@ -399,7 +440,7 @@ declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
     /**
      * Restore browser state (multiple tabs) to a browser manager.
      */
-    private restoreBrowserState;
+    protected restoreBrowserState(manager: BrowserManager, state: BrowserState): Promise<void>;
     /**
      * Get the browser manager for a specific session.
      */
@@ -415,6 +456,10 @@ declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
     destroyAllSessions(): Promise<void>;
 }
 
+/** AgentBrowser accepts an optional thread-manager factory (see {@link CreateAgentBrowserThreadManager}). */
+type AgentBrowserConfig = BrowserConfig & {
+    createThreadManager?: CreateAgentBrowserThreadManager;
+};
 /**
  * AgentBrowser - Browser automation using agent-browser (vercel-labs/agent-browser)
  *
@@ -422,8 +467,8 @@ declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
  */
 declare class AgentBrowser extends MastraBrowser {
     readonly id: string;
-    readonly name = "AgentBrowser";
-    readonly provider = "vercel-labs/agent-browser";
+    readonly name: string;
+    readonly provider: string;
     /** Shared browser manager instance (for 'shared' scope) - narrowed type from base class */
     protected sharedManager: BrowserManager | null;
     private defaultTimeout;
@@ -432,7 +477,7 @@ declare class AgentBrowser extends MastraBrowser {
     /** Thread manager - narrowed type from base class */
     protected threadManager: AgentBrowserThreadManager;
     private browserConfig;
-    constructor(config?: BrowserConfig);
+    constructor(config?: AgentBrowserConfig);
     /**
      * Ensure browser is ready and thread session exists.
      * Creates a new page/context for the current thread if needed.
@@ -452,7 +497,7 @@ declare class AgentBrowser extends MastraBrowser {
      * Set up close event listeners for 'shared' scope browser.
      * This handles the case where the shared browser is closed externally.
      */
-    private setupCloseListenerForSharedScope;
+    protected setupCloseListenerForSharedScope(manager: BrowserManager): void;
     protected doClose(): Promise<void>;
     /**
      * Check if the browser is still alive by verifying the page is connected.
@@ -652,4 +697,4 @@ declare function getBrowserPid(manager: BrowserManager): Promise<number | undefi
  */
 declare function createAgentBrowserTools(browser: AgentBrowser): Record<string, Tool<any, any>>;
 
-export { AgentBrowser, BROWSER_TOOLS, type BackInput, type BrowserConfig, type BrowserToolName, type ClickInput, type CloseInput, type DialogInput, type DragInput, type EvaluateInput, type GotoInput, type HoverInput, type PressInput, type ScrollInput, type SelectInput, type SnapshotInput, type TabsInput, type TypeInput, type WaitInput, backInputSchema, browserSchemas, clickInputSchema, closeInputSchema, createAgentBrowserTools, dialogInputSchema, dragInputSchema, evaluateInputSchema, getBrowserPid, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };
+export { AgentBrowser, type AgentBrowserConfig, type AgentBrowserSession, AgentBrowserThreadManager, type AgentBrowserThreadManagerConfig, BROWSER_TOOLS, type BackInput, type BrowserConfig, type BrowserToolName, type ClickInput, type CloseInput, type CreateAgentBrowserThreadManager, type DialogInput, type DragInput, type EvaluateInput, type GotoInput, type HoverInput, type PressInput, type ScrollInput, type SelectInput, type SnapshotInput, type TabsInput, type TypeInput, type WaitInput, backInputSchema, browserSchemas, clickInputSchema, closeInputSchema, createAgentBrowserTools, dialogInputSchema, dragInputSchema, evaluateInputSchema, getBrowserPid, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { BrowserConfig as BrowserConfig$1, ThreadManager, ThreadManagerConfig, ThreadSession, MastraBrowser, BrowserToolError, BrowserState, BrowserTabState, ScreencastOptions, ScreencastStream, MouseEventParams, KeyboardEventParams } from '@mastra/core/browser';
+import { BrowserConfig as BrowserConfig$1, ThreadManager, ThreadManagerConfig, ThreadSession, BrowserState, MastraBrowser, BrowserToolError, BrowserTabState, ScreencastOptions, ScreencastStream, MouseEventParams, KeyboardEventParams } from '@mastra/core/browser';
 import { Tool } from '@mastra/core/tools';
 import { BrowserManager } from 'agent-browser';
 import { Page } from 'playwright-core';
@@ -54,6 +54,12 @@ declare const clickInputSchema: z.ZodObject<{
         Meta: "Meta";
         Shift: "Shift";
     }>>>;
+    waitUntil: z.ZodOptional<z.ZodEnum<{
+        load: "load";
+        domcontentloaded: "domcontentloaded";
+        networkidle: "networkidle";
+    }>>;
+    timeout: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 type ClickInput = z.output<typeof clickInputSchema>;
 /**
@@ -77,6 +83,12 @@ declare const pressInputSchema: z.ZodObject<{
         Meta: "Meta";
         Shift: "Shift";
     }>>>;
+    waitUntil: z.ZodOptional<z.ZodEnum<{
+        load: "load";
+        domcontentloaded: "domcontentloaded";
+        networkidle: "networkidle";
+    }>>;
+    timeout: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 type PressInput = z.output<typeof pressInputSchema>;
 /**
@@ -87,6 +99,12 @@ declare const selectInputSchema: z.ZodObject<{
     value: z.ZodOptional<z.ZodString>;
     label: z.ZodOptional<z.ZodString>;
     index: z.ZodOptional<z.ZodNumber>;
+    waitUntil: z.ZodOptional<z.ZodEnum<{
+        load: "load";
+        domcontentloaded: "domcontentloaded";
+        networkidle: "networkidle";
+    }>>;
+    timeout: z.ZodOptional<z.ZodNumber>;
 }, z.core.$strip>;
 type SelectInput = z.output<typeof selectInputSchema>;
 /**
@@ -213,6 +231,12 @@ declare const browserSchemas: {
             Meta: "Meta";
             Shift: "Shift";
         }>>>;
+        waitUntil: z.ZodOptional<z.ZodEnum<{
+            load: "load";
+            domcontentloaded: "domcontentloaded";
+            networkidle: "networkidle";
+        }>>;
+        timeout: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>;
     readonly type: z.ZodObject<{
         ref: z.ZodString;
@@ -228,12 +252,24 @@ declare const browserSchemas: {
             Meta: "Meta";
             Shift: "Shift";
         }>>>;
+        waitUntil: z.ZodOptional<z.ZodEnum<{
+            load: "load";
+            domcontentloaded: "domcontentloaded";
+            networkidle: "networkidle";
+        }>>;
+        timeout: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>;
     readonly select: z.ZodObject<{
         ref: z.ZodString;
         value: z.ZodOptional<z.ZodString>;
         label: z.ZodOptional<z.ZodString>;
         index: z.ZodOptional<z.ZodNumber>;
+        waitUntil: z.ZodOptional<z.ZodEnum<{
+            load: "load";
+            domcontentloaded: "domcontentloaded";
+            networkidle: "networkidle";
+        }>>;
+        timeout: z.ZodOptional<z.ZodNumber>;
     }, z.core.$strip>;
     readonly scroll: z.ZodObject<{
         direction: z.ZodEnum<{
@@ -376,6 +412,11 @@ interface AgentBrowserThreadManagerConfig extends ThreadManagerConfig {
     /** Callback when a new browser manager is created for a thread */
     onBrowserCreated?: (manager: BrowserManager, threadId: string) => void;
 }
+/**
+ * Factory for custom thread managers (e.g. Firecrawl-hosted CDP per session).
+ * Defaults to {@link AgentBrowserThreadManager} when omitted.
+ */
+type CreateAgentBrowserThreadManager = (config: AgentBrowserThreadManagerConfig) => AgentBrowserThreadManager;
 /**
  * Thread manager implementation for AgentBrowser.
  *
@@ -384,9 +425,9 @@ interface AgentBrowserThreadManagerConfig extends ThreadManagerConfig {
  * - 'thread': Each thread gets a dedicated browser manager instance
  */
 declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
-    private readonly browserConfig;
+    protected readonly browserConfig: BrowserConfig;
     private readonly resolveCdpUrl?;
-    private readonly onBrowserCreated?;
+    protected readonly onBrowserCreated?: (manager: BrowserManager, threadId: string) => void;
     constructor(config: AgentBrowserThreadManagerConfig);
     /**
      * Get the page for a specific thread, creating session if needed.
@@ -399,7 +440,7 @@ declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
     /**
      * Restore browser state (multiple tabs) to a browser manager.
      */
-    private restoreBrowserState;
+    protected restoreBrowserState(manager: BrowserManager, state: BrowserState): Promise<void>;
     /**
      * Get the browser manager for a specific session.
      */
@@ -415,6 +456,10 @@ declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
     destroyAllSessions(): Promise<void>;
 }
 
+/** AgentBrowser accepts an optional thread-manager factory (see {@link CreateAgentBrowserThreadManager}). */
+type AgentBrowserConfig = BrowserConfig & {
+    createThreadManager?: CreateAgentBrowserThreadManager;
+};
 /**
  * AgentBrowser - Browser automation using agent-browser (vercel-labs/agent-browser)
  *
@@ -422,8 +467,8 @@ declare class AgentBrowserThreadManager extends ThreadManager<BrowserManager> {
  */
 declare class AgentBrowser extends MastraBrowser {
     readonly id: string;
-    readonly name = "AgentBrowser";
-    readonly provider = "vercel-labs/agent-browser";
+    readonly name: string;
+    readonly provider: string;
     /** Shared browser manager instance (for 'shared' scope) - narrowed type from base class */
     protected sharedManager: BrowserManager | null;
     private defaultTimeout;
@@ -432,7 +477,7 @@ declare class AgentBrowser extends MastraBrowser {
     /** Thread manager - narrowed type from base class */
     protected threadManager: AgentBrowserThreadManager;
     private browserConfig;
-    constructor(config?: BrowserConfig);
+    constructor(config?: AgentBrowserConfig);
     /**
      * Ensure browser is ready and thread session exists.
      * Creates a new page/context for the current thread if needed.
@@ -452,7 +497,7 @@ declare class AgentBrowser extends MastraBrowser {
      * Set up close event listeners for 'shared' scope browser.
      * This handles the case where the shared browser is closed externally.
      */
-    private setupCloseListenerForSharedScope;
+    protected setupCloseListenerForSharedScope(manager: BrowserManager): void;
     protected doClose(): Promise<void>;
     /**
      * Check if the browser is still alive by verifying the page is connected.
@@ -652,4 +697,4 @@ declare function getBrowserPid(manager: BrowserManager): Promise<number | undefi
  */
 declare function createAgentBrowserTools(browser: AgentBrowser): Record<string, Tool<any, any>>;
 
-export { AgentBrowser, BROWSER_TOOLS, type BackInput, type BrowserConfig, type BrowserToolName, type ClickInput, type CloseInput, type DialogInput, type DragInput, type EvaluateInput, type GotoInput, type HoverInput, type PressInput, type ScrollInput, type SelectInput, type SnapshotInput, type TabsInput, type TypeInput, type WaitInput, backInputSchema, browserSchemas, clickInputSchema, closeInputSchema, createAgentBrowserTools, dialogInputSchema, dragInputSchema, evaluateInputSchema, getBrowserPid, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };
+export { AgentBrowser, type AgentBrowserConfig, type AgentBrowserSession, AgentBrowserThreadManager, type AgentBrowserThreadManagerConfig, BROWSER_TOOLS, type BackInput, type BrowserConfig, type BrowserToolName, type ClickInput, type CloseInput, type CreateAgentBrowserThreadManager, type DialogInput, type DragInput, type EvaluateInput, type GotoInput, type HoverInput, type PressInput, type ScrollInput, type SelectInput, type SnapshotInput, type TabsInput, type TypeInput, type WaitInput, backInputSchema, browserSchemas, clickInputSchema, closeInputSchema, createAgentBrowserTools, dialogInputSchema, dragInputSchema, evaluateInputSchema, getBrowserPid, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };

package/dist/index.js

@@ -1,4 +1,4 @@
-import { MastraBrowser, DEFAULT_THREAD_ID, ScreencastStreamImpl, ThreadManager } from '@mastra/core/browser';
+import { ThreadManager, MastraBrowser, DEFAULT_THREAD_ID, ScreencastStreamImpl } from '@mastra/core/browser';
 import { BrowserManager } from 'agent-browser';
 import { createTool } from '@mastra/core/tools';
 import { z } from 'zod';
@@ -139,7 +139,9 @@ var clickInputSchema = z.object({
   ref: z.string().describe("Element ref from snapshot (e.g., @e5)"),
   button: z.enum(["left", "right", "middle"]).optional().describe("Mouse button (default: left)"),
   clickCount: z.number().optional().describe("Number of clicks (default: 1, use 2 for double-click)"),
-  modifiers: z.array(z.enum(["Alt", "Control", "Meta", "Shift"])).optional().describe("Modifier keys to hold")
+  modifiers: z.array(z.enum(["Alt", "Control", "Meta", "Shift"])).optional().describe("Modifier keys to hold"),
+  waitUntil: z.enum(["load", "domcontentloaded", "networkidle"]).optional().describe("If the click triggers a navigation, wait for this page load state before returning"),
+  timeout: z.number().nonnegative().optional().describe("Timeout in milliseconds for the click and optional waitUntil")
 });
 var typeInputSchema = z.object({
   ref: z.string().describe("Element ref from snapshot"),
@@ -149,13 +151,17 @@ var typeInputSchema = z.object({
 });
 var pressInputSchema = z.object({
   key: z.string().describe("Key to press (e.g., Enter, Tab, Escape, Control+a)"),
-  modifiers: z.array(z.enum(["Alt", "Control", "Meta", "Shift"])).optional().describe("Modifier keys to hold")
+  modifiers: z.array(z.enum(["Alt", "Control", "Meta", "Shift"])).optional().describe("Modifier keys to hold"),
+  waitUntil: z.enum(["load", "domcontentloaded", "networkidle"]).optional().describe("If the key press triggers a navigation, wait for this page load state before returning"),
+  timeout: z.number().nonnegative().optional().describe("Timeout in milliseconds for the optional waitUntil")
 });
 var selectInputSchema = z.object({
   ref: z.string().describe("Select element ref from snapshot"),
   value: z.string().optional().describe("Option value to select"),
   label: z.string().optional().describe("Option label to select"),
-  index: z.number().int().min(0).optional().describe("Option index to select (0-based)")
+  index: z.number().int().min(0).optional().describe("Option index to select (0-based)"),
+  waitUntil: z.enum(["load", "domcontentloaded", "networkidle"]).optional().describe("If the selection triggers a navigation, wait for this page load state before returning"),
+  timeout: z.number().nonnegative().optional().describe("Timeout in milliseconds for the selection and optional waitUntil")
 }).superRefine((data, ctx) => {
   if (data.value === void 0 && data.label === void 0 && data.index === void 0) {
     ctx.addIssue({
@@ -291,7 +297,7 @@ function createBackTool(browser) {
 function createClickTool(browser) {
   return createTool({
     id: BROWSER_TOOLS.CLICK,
-    description: "Click an element using its ref from a snapshot. Use clickCount: 2 for double-click.",
+    description: "Click an element using its ref from a snapshot. Use clickCount: 2 for double-click. Pass waitUntil when the click triggers navigation so the page settles before the next snapshot.",
     inputSchema: clickInputSchema,
     execute: async (input, { agent }) => {
       const threadId = agent?.threadId;
@@ -388,7 +394,7 @@ function createHoverTool(browser) {
 function createPressTool(browser) {
   return createTool({
     id: BROWSER_TOOLS.PRESS,
-    description: "Press a keyboard key (e.g., Enter, Tab, Escape, Control+a).",
+    description: "Press a keyboard key (e.g., Enter, Tab, Escape, Control+a). Pass waitUntil when the keypress triggers navigation (e.g., Enter to submit a form) so the page settles before the next snapshot.",
     inputSchema: pressInputSchema,
     execute: async (input, { agent }) => {
       const threadId = agent?.threadId;
@@ -446,7 +452,7 @@ function createScrollTool(browser) {
 function createSelectTool(browser) {
   return createTool({
     id: BROWSER_TOOLS.SELECT,
-    description: "Select an option from a dropdown by value, label, or index.",
+    description: "Select an option from a dropdown by value, label, or index. Pass waitUntil when the selection triggers navigation so the page settles before the next snapshot.",
     inputSchema: selectInputSchema,
     execute: async (input, { agent }) => {
       const threadId = agent?.threadId;
@@ -574,7 +580,7 @@ var AgentBrowser = class extends MastraBrowser {
       this.defaultTimeout = config.timeout;
     }
     const effectiveScope = config.cdpUrl ? config.scope ?? "shared" : config.scope ?? "thread";
-    this.threadManager = new AgentBrowserThreadManager({
+    const threadManagerConfig = {
       scope: effectiveScope,
       browserConfig: { ...config, headless: this.headless },
       resolveCdpUrl: this.resolveCdpUrl.bind(this),
@@ -587,7 +593,9 @@ var AgentBrowser = class extends MastraBrowser {
       onBrowserCreated: (manager, threadId) => {
         this.setupCloseListenerForThread(manager, threadId);
       }
-    });
+    };
+    const createTm = config.createThreadManager ?? ((opts) => new AgentBrowserThreadManager(opts));
+    this.threadManager = createTm(threadManagerConfig);
   }
   // ---------------------------------------------------------------------------
   // Thread Scope (delegated to ThreadManager)
@@ -1087,12 +1095,15 @@ var AgentBrowser = class extends MastraBrowser {
           "Take a new snapshot to see the current page state and get fresh refs."
         );
       }
+      const timeout = input.timeout ?? this.defaultTimeout;
+      const navigation = input.waitUntil ? page.waitForNavigation({ waitUntil: input.waitUntil, timeout }) : void 0;
       await locator.click({
         button: input.button ?? "left",
         clickCount: input.clickCount ?? 1,
         modifiers: input.modifiers,
-        timeout: this.defaultTimeout
+        timeout
       });
+      await navigation;
       return {
         success: true,
         url: page.url(),
@@ -1161,7 +1172,10 @@ var AgentBrowser = class extends MastraBrowser {
   async press(input, threadId) {
     try {
       const page = await this.getPage(threadId);
+      const timeout = input.timeout ?? this.defaultTimeout;
+      const navigation = input.waitUntil ? page.waitForNavigation({ waitUntil: input.waitUntil, timeout }) : void 0;
       await page.keyboard.press(input.key);
+      await navigation;
       return {
         success: true,
         url: page.url(),
@@ -1189,9 +1203,10 @@ var AgentBrowser = class extends MastraBrowser {
       if (input.value) selectValue.value = input.value;
       if (input.label) selectValue.label = input.label;
       if (input.index !== void 0) selectValue.index = input.index;
-      const selected = await locator.selectOption(selectValue, {
-        timeout: this.defaultTimeout
-      });
+      const timeout = input.timeout ?? this.defaultTimeout;
+      const navigation = input.waitUntil ? page.waitForNavigation({ waitUntil: input.waitUntil, timeout }) : void 0;
+      const selected = await locator.selectOption(selectValue, { timeout });
+      await navigation;
       return {
         success: true,
         selected,
@@ -1684,6 +1699,6 @@ var AgentBrowser = class extends MastraBrowser {
   }
 };
 
-export { AgentBrowser, BROWSER_TOOLS, backInputSchema, browserSchemas, clickInputSchema, closeInputSchema, createAgentBrowserTools, dialogInputSchema, dragInputSchema, evaluateInputSchema, getBrowserPid, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };
+export { AgentBrowser, AgentBrowserThreadManager, BROWSER_TOOLS, backInputSchema, browserSchemas, clickInputSchema, closeInputSchema, createAgentBrowserTools, dialogInputSchema, dragInputSchema, evaluateInputSchema, getBrowserPid, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map