@mastra/stagehand 0.3.2 → 0.3.3

package/dist/index.js

@@ -230,14 +230,17 @@ 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("stagehand_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,
@@ -441,6 +444,7 @@ var StagehandBrowser = class extends MastraBrowser {
 	stagehandConfig;
 	/** Debounce timers per thread for tab change reconnection */
 	tabChangeDebounceTimers = /* @__PURE__ */ new Map();
+	pendingCloseReasons = /* @__PURE__ */ new Map();
 	constructor(config = {}) {
 		super(config);
 		this.id = `stagehand-${Date.now()}`;
@@ -520,6 +524,7 @@ var StagehandBrowser = class extends MastraBrowser {
 		return stagehand;
 	}
 	async doLaunch() {
+		this.pendingCloseReasons.clear();
 		const scope = this.getScope();
 		this.threadManager.setCreateStagehand(() => this.createStagehandInstance());
 		if (scope === "thread") return;
@@ -547,6 +552,7 @@ var StagehandBrowser = class extends MastraBrowser {
 		const handleDisconnect = () => {
 			if (disconnectHandled) return;
 			disconnectHandled = true;
+			this.rememberClosedBrowserState(stagehand, "user", threadId);
 			onDisconnect();
 		};
 		try {
@@ -587,9 +593,30 @@ var StagehandBrowser = class extends MastraBrowser {
 		this.patchExitType();
 	}
 	async closeThreadSession(threadId) {
+		const stagehand = this.threadManager.getExistingManagerForThread(threadId);
+		if (stagehand) this.rememberClosedBrowserState(stagehand, "agent", threadId);
 		await super.closeThreadSession(threadId);
 		this.patchExitType();
 	}
+	browserStateKey(threadId) {
+		return threadId ?? this.getCurrentThread() ?? DEFAULT_THREAD_ID;
+	}
+	markBrowserCloseReason(reason, threadId) {
+		this.pendingCloseReasons.set(this.browserStateKey(threadId), reason);
+	}
+	getCloseReason(threadId) {
+		return this.pendingCloseReasons.get(this.browserStateKey(threadId)) ?? this.pendingCloseReasons.get(DEFAULT_THREAD_ID);
+	}
+	rememberClosedBrowserState(stagehand, reason, threadId) {
+		const state = this.getBrowserStateFromStagehand(stagehand, 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;
+	}
 	patchExitType() {
 		if (!this.config.profile) return;
 		patchProfileExitType(this.config.profile, this.logger);
@@ -946,9 +973,9 @@ var StagehandBrowser = class extends MastraBrowser {
 			if (scope === "thread" && effectiveThreadId) {
 				const stagehand = this.threadManager.getExistingManagerForThread(effectiveThreadId);
 				if (!stagehand) return null;
-				return this.getBrowserStateFromStagehand(stagehand);
+				return this.getBrowserStateFromStagehand(stagehand, effectiveThreadId);
 			}
-			return this.getBrowserStateFromStagehand(this.sharedManager);
+			return this.getBrowserStateFromStagehand(this.sharedManager, effectiveThreadId);
 		} catch {
 			return null;
 		}
@@ -960,23 +987,26 @@ var StagehandBrowser = class extends MastraBrowser {
 	getBrowserStateForThread(threadId) {
 		const effectiveThreadId = threadId ?? this.getCurrentThread() ?? DEFAULT_THREAD_ID;
 		const stagehand = this.threadManager.getExistingManagerForThread(effectiveThreadId);
-		return this.getBrowserStateFromStagehand(stagehand);
+		return this.getBrowserStateFromStagehand(stagehand, effectiveThreadId);
 	}
 	/**
 	* Get browser state from a specific Stagehand instance.
 	*/
-	getBrowserStateFromStagehand(stagehand) {
+	getBrowserStateFromStagehand(stagehand, threadId) {
 		if (!stagehand?.context) return null;
 		try {
 			const pages = stagehand.context.pages();
 			const activePage = stagehand.context.activePage();
 			let activeIndex = 0;
+			const tabs = pages.map((page, index) => {
+				if (page === activePage) activeIndex = index;
+				return { url: page.url() };
+			});
+			const closeReason = this.getCloseReason(threadId);
 			return {
-				tabs: pages.map((page, index) => {
-					if (page === activePage) activeIndex = index;
-					return { url: page.url() };
-				}),
-				activeTabIndex: activeIndex
+				tabs,
+				activeTabIndex: activeIndex,
+				...closeReason ? { closeReason } : {}
 			};
 		} catch {
 			return null;