@mastra/agent-browser 0.1.0 → 0.2.0-alpha.0

package/dist/index.d.cts

@@ -283,11 +283,29 @@ declare const browserSchemas: {
     }, z.core.$strip>;
 };
 
+/**
+ * AgentBrowser-specific configuration extensions.
+ */
+interface AgentBrowserConfigExtensions {
+    /**
+     * Path to a Playwright storage state file (JSON) containing cookies and localStorage.
+     * This is a lighter-weight alternative to `profile` — it only persists
+     * authentication state, not the full browser profile.
+     *
+     * You can export storage state from a Playwright session and reuse it later.
+     *
+     * @example
+     * ```ts
+     * { storageState: './auth-state.json' }
+     * ```
+     */
+    storageState?: string;
+}
 /**
  * Configuration options for AgentBrowser.
- * Type alias for BaseBrowserConfig.
+ * Extends the base BrowserConfig with agent-browser specific options.
  */
-type BrowserConfig = BrowserConfig$1;
+type BrowserConfig = BrowserConfig$1 & AgentBrowserConfigExtensions;
 
 /**
  * AgentBrowserThreadManager - Thread scope management for AgentBrowser
@@ -365,6 +383,8 @@ declare class AgentBrowser extends MastraBrowser {
     /** Shared browser manager instance (for 'shared' scope) - narrowed type from base class */
     protected sharedManager: BrowserManager | null;
     private defaultTimeout;
+    /** Pending PID lookups — awaited in disconnect handlers to avoid racing. */
+    private pidLookups;
     /** Thread manager - narrowed type from base class */
     protected threadManager: AgentBrowserThreadManager;
     constructor(config?: BrowserConfig);
@@ -454,6 +474,14 @@ declare class AgentBrowser extends MastraBrowser {
      * Get the active tab index.
      */
     getActiveTabIndex(threadId?: string): Promise<number>;
+    /**
+     * Export the current browser session's storage state (cookies, localStorage) to a JSON file.
+     * This can later be loaded via the `storageState` config option to restore the session.
+     *
+     * @param path - File path to save the storage state JSON
+     * @param threadId - Optional thread ID (defaults to current thread)
+     */
+    exportStorageState(path: string, threadId?: string): Promise<void>;
     goto(input: GotoInput, threadId?: string): Promise<{
         success: true;
         url: string;
@@ -550,6 +578,17 @@ declare class AgentBrowser extends MastraBrowser {
     injectKeyboardEvent(event: KeyboardEventParams, threadId?: string): Promise<void>;
 }
 
+/**
+ * Get the browser process PID from a BrowserManager instance via CDP.
+ *
+ * Playwright doesn't expose the browser process PID directly, so we use CDP's
+ * SystemInfo.getProcessInfo to get it. This works for both regular browser
+ * launches and persistent contexts (profiles).
+ *
+ * Returns undefined if the PID can't be retrieved (e.g., browser not running).
+ */
+declare function getBrowserPid(manager: BrowserManager): Promise<number | undefined>;
+
 /**
  * Browser Tool Constants
  */
@@ -585,4 +624,4 @@ type BrowserToolName = (typeof BROWSER_TOOLS)[keyof typeof BROWSER_TOOLS];
  */
 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, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };
+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 };

package/dist/index.d.ts

@@ -283,11 +283,29 @@ declare const browserSchemas: {
     }, z.core.$strip>;
 };
 
+/**
+ * AgentBrowser-specific configuration extensions.
+ */
+interface AgentBrowserConfigExtensions {
+    /**
+     * Path to a Playwright storage state file (JSON) containing cookies and localStorage.
+     * This is a lighter-weight alternative to `profile` — it only persists
+     * authentication state, not the full browser profile.
+     *
+     * You can export storage state from a Playwright session and reuse it later.
+     *
+     * @example
+     * ```ts
+     * { storageState: './auth-state.json' }
+     * ```
+     */
+    storageState?: string;
+}
 /**
  * Configuration options for AgentBrowser.
- * Type alias for BaseBrowserConfig.
+ * Extends the base BrowserConfig with agent-browser specific options.
  */
-type BrowserConfig = BrowserConfig$1;
+type BrowserConfig = BrowserConfig$1 & AgentBrowserConfigExtensions;
 
 /**
  * AgentBrowserThreadManager - Thread scope management for AgentBrowser
@@ -365,6 +383,8 @@ declare class AgentBrowser extends MastraBrowser {
     /** Shared browser manager instance (for 'shared' scope) - narrowed type from base class */
     protected sharedManager: BrowserManager | null;
     private defaultTimeout;
+    /** Pending PID lookups — awaited in disconnect handlers to avoid racing. */
+    private pidLookups;
     /** Thread manager - narrowed type from base class */
     protected threadManager: AgentBrowserThreadManager;
     constructor(config?: BrowserConfig);
@@ -454,6 +474,14 @@ declare class AgentBrowser extends MastraBrowser {
      * Get the active tab index.
      */
     getActiveTabIndex(threadId?: string): Promise<number>;
+    /**
+     * Export the current browser session's storage state (cookies, localStorage) to a JSON file.
+     * This can later be loaded via the `storageState` config option to restore the session.
+     *
+     * @param path - File path to save the storage state JSON
+     * @param threadId - Optional thread ID (defaults to current thread)
+     */
+    exportStorageState(path: string, threadId?: string): Promise<void>;
     goto(input: GotoInput, threadId?: string): Promise<{
         success: true;
         url: string;
@@ -550,6 +578,17 @@ declare class AgentBrowser extends MastraBrowser {
     injectKeyboardEvent(event: KeyboardEventParams, threadId?: string): Promise<void>;
 }
 
+/**
+ * Get the browser process PID from a BrowserManager instance via CDP.
+ *
+ * Playwright doesn't expose the browser process PID directly, so we use CDP's
+ * SystemInfo.getProcessInfo to get it. This works for both regular browser
+ * launches and persistent contexts (profiles).
+ *
+ * Returns undefined if the PID can't be retrieved (e.g., browser not running).
+ */
+declare function getBrowserPid(manager: BrowserManager): Promise<number | undefined>;
+
 /**
  * Browser Tool Constants
  */
@@ -585,4 +624,4 @@ type BrowserToolName = (typeof BROWSER_TOOLS)[keyof typeof BROWSER_TOOLS];
  */
 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, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };
+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 };

package/dist/index.js

@@ -35,7 +35,10 @@ var AgentBrowserThreadManager = class extends ThreadManager {
       const manager = new BrowserManager();
       const launchOptions = {
         headless: this.browserConfig.headless ?? true,
-        viewport: this.browserConfig.viewport
+        viewport: this.browserConfig.viewport,
+        profile: this.browserConfig.profile,
+        executablePath: this.browserConfig.executablePath,
+        storageState: this.browserConfig.storageState
       };
       if (this.browserConfig.cdpUrl && this.resolveCdpUrl) {
         launchOptions.cdpUrl = await this.resolveCdpUrl(this.browserConfig.cdpUrl);
@@ -489,12 +492,36 @@ function createAgentBrowserTools(browser) {
   };
 }
 
+// src/utils.ts
+async function getBrowserPid(manager) {
+  try {
+    let browser = manager.getBrowser();
+    if (!browser) {
+      const ctx = manager.getContext();
+      browser = ctx?.browser?.() ?? null;
+    }
+    if (!browser) return void 0;
+    const cdp = await browser.newBrowserCDPSession();
+    try {
+      const info = await cdp.send("SystemInfo.getProcessInfo");
+      const browserProcess = info.processInfo?.find((p) => p.type === "browser");
+      return browserProcess?.id;
+    } finally {
+      await cdp.detach().catch(() => void 0);
+    }
+  } catch {
+    return void 0;
+  }
+}
+
 // src/agent-browser.ts
 var AgentBrowser = class extends MastraBrowser {
   id;
   name = "AgentBrowser";
   provider = "vercel-labs/agent-browser";
   defaultTimeout = 3e4;
+  /** Pending PID lookups — awaited in disconnect handlers to avoid racing. */
+  pidLookups = /* @__PURE__ */ new Set();
   constructor(config = {}) {
     super(config);
     this.id = `agent-browser-${Date.now()}`;
@@ -532,11 +559,11 @@ var AgentBrowser = class extends MastraBrowser {
     const scope = this.threadManager.getScope();
     const threadId = this.getCurrentThread();
     const existingSession = this.threadManager.hasSession(threadId);
-    if (scope === "thread" && threadId !== DEFAULT_THREAD_ID && !existingSession) {
+    if (scope === "thread" && !existingSession) {
       await this.getManagerForThread(threadId);
     }
     await super.ensureReady();
-    if (scope === "thread" && threadId !== DEFAULT_THREAD_ID && existingSession) {
+    if (scope === "thread" && existingSession) {
       await this.getManagerForThread(threadId);
     }
   }
@@ -569,7 +596,10 @@ var AgentBrowser = class extends MastraBrowser {
     const localConfig = this.config;
     const launchOptions = {
       headless: localConfig.headless ?? true,
-      viewport: localConfig.viewport
+      viewport: localConfig.viewport,
+      profile: localConfig.profile,
+      executablePath: localConfig.executablePath,
+      storageState: localConfig.storageState
     };
     if (localConfig.cdpUrl) {
       launchOptions.cdpUrl = await this.resolveCdpUrl(localConfig.cdpUrl);
@@ -584,11 +614,15 @@ var AgentBrowser = class extends MastraBrowser {
    */
   setupCloseListenerForSharedScope(manager) {
     try {
+      const pidLookup = getBrowserPid(manager).then((pid) => {
+        if (pid && this.sharedManager === manager) this.sharedBrowserPid = pid;
+      }).finally(() => this.pidLookups.delete(pidLookup));
+      this.pidLookups.add(pidLookup);
       let disconnectHandled = false;
       const handleDisconnect = () => {
         if (disconnectHandled) return;
         disconnectHandled = true;
-        this.handleBrowserDisconnected();
+        void pidLookup.catch(() => void 0).then(() => this.handleBrowserDisconnected());
       };
       const context = manager.getContext();
       if (context) {
@@ -607,6 +641,8 @@ var AgentBrowser = class extends MastraBrowser {
     }
   }
   async doClose() {
+    await Promise.allSettled([...this.pidLookups]);
+    this.pidLookups.clear();
     await this.threadManager.destroyAllSessions();
     this.setCurrentThread(void 0);
     const scope = this.threadManager.getScope();
@@ -690,11 +726,17 @@ var AgentBrowser = class extends MastraBrowser {
    */
   setupCloseListenerForThread(manager, threadId) {
     try {
+      const pidLookup = getBrowserPid(manager).then((pid) => {
+        if (pid && this.threadManager?.getExistingManagerForThread(threadId) === manager) {
+          this.threadBrowserPids.set(threadId, pid);
+        }
+      }).finally(() => this.pidLookups.delete(pidLookup));
+      this.pidLookups.add(pidLookup);
       let disconnectHandled = false;
       const handleDisconnect = () => {
         if (disconnectHandled) return;
         disconnectHandled = true;
-        this.handleThreadBrowserDisconnected(threadId);
+        void pidLookup.catch(() => void 0).then(() => this.handleThreadBrowserDisconnected(threadId));
       };
       const context = manager.getContext();
       if (context) {
@@ -884,6 +926,25 @@ var AgentBrowser = class extends MastraBrowser {
       return 0;
     }
   }
+  /**
+   * Export the current browser session's storage state (cookies, localStorage) to a JSON file.
+   * This can later be loaded via the `storageState` config option to restore the session.
+   *
+   * @param path - File path to save the storage state JSON
+   * @param threadId - Optional thread ID (defaults to current thread)
+   */
+  async exportStorageState(path, threadId) {
+    const effectiveThreadId = threadId ?? this.getCurrentThread();
+    const manager = this.threadManager.getExistingManagerForThread(effectiveThreadId);
+    if (!manager) {
+      throw new Error("No browser is running. Launch a browser first before exporting storage state.");
+    }
+    const context = manager.getContext();
+    if (!context) {
+      throw new Error("Browser context not available");
+    }
+    await context.storageState({ path });
+  }
   // ---------------------------------------------------------------------------
   // 1. browser_goto - Navigate to URL
   // ---------------------------------------------------------------------------
@@ -1552,6 +1613,6 @@ var AgentBrowser = class extends MastraBrowser {
   }
 };
 
-export { AgentBrowser, BROWSER_TOOLS, backInputSchema, browserSchemas, clickInputSchema, closeInputSchema, createAgentBrowserTools, dialogInputSchema, dragInputSchema, evaluateInputSchema, gotoInputSchema, hoverInputSchema, pressInputSchema, scrollInputSchema, selectInputSchema, snapshotInputSchema, tabsInputSchema, typeInputSchema, waitInputSchema };
+export { AgentBrowser, 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