@mastra/agent-browser 0.3.0 → 0.3.1

package/dist/index.d.cts

@@ -474,6 +474,8 @@ declare class AgentBrowser extends MastraBrowser {
     private defaultTimeout;
     /** Pending PID lookups — awaited in disconnect handlers to avoid racing. */
     private pidLookups;
+    private readonly pendingCloseReasons;
+    private readonly activeUrlChangeSources;
     /** Thread manager - narrowed type from base class */
     protected threadManager: AgentBrowserThreadManager;
     private browserConfig;
@@ -499,6 +501,7 @@ declare class AgentBrowser extends MastraBrowser {
      */
     protected setupCloseListenerForSharedScope(manager: BrowserManager): void;
     protected doClose(): Promise<void>;
+    closeThreadSession(threadId: string): Promise<void>;
     /**
      * Check if the browser is still alive by verifying the page is connected.
      * Called by base class ensureReady() to detect externally closed browsers.
@@ -509,6 +512,12 @@ declare class AgentBrowser extends MastraBrowser {
      * Returns 16 flat tools for browser automation.
      */
     getTools(): Record<string, Tool<any, any>>;
+    private browserStateKey;
+    markBrowserCloseReason(reason: 'agent' | 'user' | 'process_restart' | 'error', threadId?: string): void;
+    private markActiveUrlChangeSource;
+    private getCloseReason;
+    private getActiveUrlChangeSource;
+    private rememberClosedBrowserState;
     /**
      * Get the page for the current thread.
      * Uses thread scope if enabled, otherwise returns the shared page.

package/dist/index.d.ts

@@ -474,6 +474,8 @@ declare class AgentBrowser extends MastraBrowser {
     private defaultTimeout;
     /** Pending PID lookups — awaited in disconnect handlers to avoid racing. */
     private pidLookups;
+    private readonly pendingCloseReasons;
+    private readonly activeUrlChangeSources;
     /** Thread manager - narrowed type from base class */
     protected threadManager: AgentBrowserThreadManager;
     private browserConfig;
@@ -499,6 +501,7 @@ declare class AgentBrowser extends MastraBrowser {
      */
     protected setupCloseListenerForSharedScope(manager: BrowserManager): void;
     protected doClose(): Promise<void>;
+    closeThreadSession(threadId: string): Promise<void>;
     /**
      * Check if the browser is still alive by verifying the page is connected.
      * Called by base class ensureReady() to detect externally closed browsers.
@@ -509,6 +512,12 @@ declare class AgentBrowser extends MastraBrowser {
      * Returns 16 flat tools for browser automation.
      */
     getTools(): Record<string, Tool<any, any>>;
+    private browserStateKey;
+    markBrowserCloseReason(reason: 'agent' | 'user' | 'process_restart' | 'error', threadId?: string): void;
+    private markActiveUrlChangeSource;
+    private getCloseReason;
+    private getActiveUrlChangeSource;
+    private rememberClosedBrowserState;
     /**
      * Get the page for the current thread.
      * Uses thread scope if enabled, otherwise returns the shared page.

package/dist/index.js

@@ -314,13 +314,16 @@ function createCloseTool(browser) {
     inputSchema: closeInputSchema,
     execute: async (_input, { agent }) => {
       const threadId = agent?.threadId;
+      browser.setCurrentThread(threadId);
       if (browser.getScope() !== "shared") {
         if (!threadId) {
           throw new Error("browser_close requires agent.threadId when browser scope is not shared");
         }
+        browser.markBrowserCloseReason("agent", threadId);
         await browser.closeThreadSession(threadId);
         return { success: true, hint: "Thread's browser session closed. A new session will be created on next use." };
       }
+      browser.markBrowserCloseReason("agent");
       await browser.close();
       return { success: true, hint: "Browser closed. It will be re-launched automatically on next use." };
     }
@@ -571,6 +574,8 @@ var AgentBrowser = class extends MastraBrowser {
   defaultTimeout = 3e4;
   /** Pending PID lookups — awaited in disconnect handlers to avoid racing. */
   pidLookups = /* @__PURE__ */ new Set();
+  pendingCloseReasons = /* @__PURE__ */ new Map();
+  activeUrlChangeSources = /* @__PURE__ */ new Map();
   browserConfig;
   constructor(config = {}) {
     super(config);
@@ -639,6 +644,8 @@ var AgentBrowser = class extends MastraBrowser {
   // Lifecycle
   // ---------------------------------------------------------------------------
   async doLaunch() {
+    this.pendingCloseReasons.clear();
+    this.activeUrlChangeSources.clear();
     const scope = this.threadManager.getScope();
     if (scope === "thread") {
       this.sharedManager = new BrowserManager();
@@ -675,6 +682,7 @@ var AgentBrowser = class extends MastraBrowser {
       const handleDisconnect = () => {
         if (disconnectHandled) return;
         disconnectHandled = true;
+        this.rememberClosedBrowserState(manager, "user");
         void pidLookup.catch(() => void 0).then(() => this.handleBrowserDisconnected());
       };
       const context = manager.getContext();
@@ -704,6 +712,14 @@ var AgentBrowser = class extends MastraBrowser {
     }
     this.sharedManager = null;
   }
+  async closeThreadSession(threadId) {
+    const manager = this.threadManager.getExistingManagerForThread(threadId);
+    if (manager) {
+      const state = this.getBrowserStateForManager(manager, threadId);
+      if (state) this.threadManager.updateBrowserState(threadId, state);
+    }
+    await super.closeThreadSession(threadId);
+  }
   /**
    * Check if the browser is still alive by verifying the page is connected.
    * Called by base class ensureReady() to detect externally closed browsers.
@@ -754,6 +770,32 @@ var AgentBrowser = class extends MastraBrowser {
   // ---------------------------------------------------------------------------
   // Helpers
   // ---------------------------------------------------------------------------
+  browserStateKey(threadId) {
+    return threadId ?? this.getCurrentThread() ?? DEFAULT_THREAD_ID;
+  }
+  markBrowserCloseReason(reason, threadId) {
+    this.pendingCloseReasons.set(this.browserStateKey(threadId), reason);
+  }
+  markActiveUrlChangeSource(source, url, threadId) {
+    this.activeUrlChangeSources.set(this.browserStateKey(threadId), { url, source });
+  }
+  getCloseReason(threadId) {
+    return this.pendingCloseReasons.get(this.browserStateKey(threadId)) ?? this.pendingCloseReasons.get(DEFAULT_THREAD_ID);
+  }
+  getActiveUrlChangeSource(activeUrl, threadId) {
+    const entry = this.activeUrlChangeSources.get(this.browserStateKey(threadId));
+    return entry && entry.url === activeUrl ? entry.source : void 0;
+  }
+  rememberClosedBrowserState(manager, reason, threadId) {
+    const state = this.getBrowserStateForManager(manager, threadId);
+    if (!state || state.tabs.length === 0) return;
+    const closedState = { ...state, closeReason: this.getCloseReason(threadId) ?? reason };
+    if (threadId) {
+      this.threadManager.updateBrowserState(threadId, closedState);
+    } else {
+      this.lastBrowserState = closedState;
+    }
+  }
   /**
    * Get the page for the current thread.
    * Uses thread scope if enabled, otherwise returns the shared page.
@@ -796,6 +838,7 @@ var AgentBrowser = class extends MastraBrowser {
       const handleDisconnect = () => {
         if (disconnectHandled) return;
         disconnectHandled = true;
+        this.rememberClosedBrowserState(manager, "user", threadId);
         void pidLookup.catch(() => void 0).then(() => this.handleThreadBrowserDisconnected(threadId));
       };
       const context = manager.getContext();
@@ -927,12 +970,12 @@ var AgentBrowser = class extends MastraBrowser {
    * Get the current browser state (all tabs and active tab index).
    */
   async getBrowserState(threadId) {
-    if (!this.isBrowserRunning()) {
+    if (!this.isBrowserRunning(threadId)) {
       return null;
     }
     try {
       const manager = await this.getManagerForThread(threadId);
-      return this.getBrowserStateForManager(manager);
+      return this.getBrowserStateForManager(manager, threadId);
     } catch {
       return null;
     }
@@ -945,22 +988,32 @@ var AgentBrowser = class extends MastraBrowser {
     const effectiveThreadId = threadId ?? this.getCurrentThread() ?? DEFAULT_THREAD_ID;
     const manager = this.threadManager.getExistingManagerForThread(effectiveThreadId);
     if (!manager) return null;
-    return this.getBrowserStateForManager(manager);
+    return this.getBrowserStateForManager(manager, effectiveThreadId);
   }
   /**
    * Get browser state from a specific manager instance.
    */
-  getBrowserStateForManager(manager) {
+  getBrowserStateForManager(manager, threadId) {
     try {
+      const stateKey = this.browserStateKey(threadId);
       const pages = manager.getPages();
       const activeIndex = manager.getActiveIndex();
       const tabs = pages.map((page) => ({
         url: page.url()
       }));
-      return {
+      const activeUrl = tabs[activeIndex]?.url;
+      const previousState = this.threadManager.getSavedBrowserState(stateKey) ?? this.lastBrowserState;
+      const previousUrl = previousState?.tabs[previousState.activeTabIndex]?.url;
+      const activeUrlChangeSource = this.getActiveUrlChangeSource(activeUrl, stateKey) ?? (previousUrl && activeUrl !== previousUrl ? "user" : void 0);
+      const state = {
         tabs,
-        activeTabIndex: activeIndex
+        activeTabIndex: activeIndex,
+        ...this.getCloseReason(stateKey) ? { closeReason: this.getCloseReason(stateKey) } : {},
+        ...activeUrlChangeSource ? { activeUrlChangeSource } : {}
       };
+      this.threadManager.updateBrowserState(stateKey, state);
+      this.lastBrowserState = state;
+      return state;
     } catch {
       return null;
     }
@@ -1015,9 +1068,11 @@ var AgentBrowser = class extends MastraBrowser {
         timeout: input.timeout ?? this.defaultTimeout,
         waitUntil: input.waitUntil ?? "domcontentloaded"
       });
+      const url = page.url();
+      this.markActiveUrlChangeSource("agent", url, threadId);
       return {
         success: true,
-        url: page.url(),
+        url,
         title: await page.title(),
         hint: "Take a snapshot to see interactive elements and get refs."
       };
@@ -1304,9 +1359,11 @@ var AgentBrowser = class extends MastraBrowser {
     try {
       const page = await this.getPage(threadId);
       await page.goBack({ timeout: this.defaultTimeout });
+      const url = page.url();
+      this.markActiveUrlChangeSource("agent", url, threadId);
       return {
         success: true,
-        url: page.url(),
+        url,
         title: await page.title(),
         hint: "Take a new snapshot to see the previous page."
       };
@@ -1437,6 +1494,7 @@ var AgentBrowser = class extends MastraBrowser {
           if (input.url) {
             const page = await this.getPage(threadId);
             await page.goto(input.url);
+            this.markActiveUrlChangeSource("agent", page.url(), threadId);
           }
           this.updateSessionBrowserState(threadId);
           return {
@@ -1457,6 +1515,7 @@ var AgentBrowser = class extends MastraBrowser {
           await this.reconnectScreencastForThread(threadId, "tab switch");
           const page = browser.getPage();
           const pageUrl = page.url();
+          this.markActiveUrlChangeSource("agent", pageUrl, threadId);
           const streamKey = this.getStreamKey(threadId);
           const stream = this.activeScreencastStreams.get(streamKey);
           if (pageUrl && stream?.isActive()) {