stagehand 4.4.0a0.dev1660__tar.gz → 4.4.0a0.dev1661__tar.gz

{stagehand-4.4.0a0.dev1660 → stagehand-4.4.0a0.dev1661}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stagehand
-Version: 4.4.0a0.dev1660
+Version: 4.4.0a0.dev1661
 Summary: Browserbase's SDK for building browser agents
 Keywords: ai,agent,ai-agent,browser,browser-agent,browser-automation,computer-use,headless,playwright,cdp,llm,automation,web-scraping,testing
 Author: Browserbase

{stagehand-4.4.0a0.dev1660 → stagehand-4.4.0a0.dev1661}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "stagehand"
-version = "4.4.0a0.dev1660"
+version = "4.4.0a0.dev1661"
 description = "Browserbase's SDK for building browser agents"
 readme = "README.md"
 requires-python = ">=3.11"

{stagehand-4.4.0a0.dev1660 → stagehand-4.4.0a0.dev1661}/src/stagehand/_extension/service-worker.js

@@ -9206,7 +9206,7 @@ function isRecord$1(value) {
 }
 var package_default$1 = {
 	name: "@browserbasehq/stagehand-protocol",
-	version: "3.1.1-alpha-d59afde37e995b165a41f754a12a446f11d471a2",
+	version: "3.1.1-alpha-44d99dfbdd24baaf06986492792348582266d8f0",
 	"private": true,
 	type: "module",
 	exports: {
@@ -56394,7 +56394,7 @@ var WebTracerProvider = class extends BasicTracerProvider {
 };
 var package_default = {
 	name: "@browserbasehq/stagehand-extension",
-	version: "1.3.0-alpha-d59afde37e995b165a41f754a12a446f11d471a2",
+	version: "1.3.0-alpha-44d99dfbdd24baaf06986492792348582266d8f0",
 	"private": true,
 	type: "module",
 	scripts: {
@@ -56601,7 +56601,7 @@ var FrameRegistry = class {
 	* Record that a frame attached. If `parentId` is null and `frameId` differs from the current
 	* root, this is a root swap and we rename the root id.
 	*
-	* IMPORTANT: The emitter's `sessionId` is the **owner** for the new/attached frame.
+	* The provided `sessionId` is the selected **owner** for the new/attached frame.
 	*/
 	onFrameAttached(frameId, parentId, sessionId) {
 		if (!parentId && frameId !== this.rootFrameId) {
@@ -58449,7 +58449,7 @@ var Page = class Page {
 	}
 	/**
 	* Parent/child session emitted a `frameAttached`.
-	* Topology update + ownership stamped to **emitting session**.
+	* Topology update + ownership stamped to the session selected by Context.
 	*/
 	onFrameAttached(frameId, parentId, session) {
 		this.ensureOrdinal(frameId);
@@ -58497,7 +58497,7 @@ var Page = class Page {
 	* An OOPIF child session whose **main** frame id equals the parent iframe’s frameId
 	* has been attached; adopt the session into this Page and seed ownership for its subtree.
 	*/
-	adoptOopifSession(childSession, childMainFrameId) {
+	adoptOopifSession(childSession, childMainFrameId, onFrameTreeSeeded) {
 		if (this.disposed) return;
 		const previous = childSession.id ? this.sessions.get(childSession.id) : void 0;
 		if (childSession.id) this.sessions.set(childSession.id, childSession);
@@ -58512,9 +58512,6 @@ var Page = class Page {
 		childSession.on("Page.frameNavigated", (evt) => {
 			this.onFrameNavigated(evt.frame, childSession);
 		});
-		childSession.on("Page.frameAttached", (evt) => {
-			this.onFrameAttached(evt.frameId, evt.parentFrameId ?? null, childSession);
-		});
 		childSession.on("Page.frameDetached", (evt) => {
 			this.onFrameDetached(evt.frameId, evt.reason ?? "remove");
 		});
@@ -58532,6 +58529,7 @@ var Page = class Page {
 				if (!this.disposed && this.sessions.get(childSession.id ?? "") === childSession) {
 					this.registry.seedFromFrameTree(childSession.id ?? "child", frameTree);
 					this.reconcilePendingWebMCPTools();
+					onFrameTreeSeeded?.();
 				}
 			} catch {}
 		})();
@@ -61574,6 +61572,8 @@ var BrowserContext = class BrowserContext {
 		this.sessionOwnerPage = /* @__PURE__ */ new Map();
 		this.frameOwnerPage = /* @__PURE__ */ new Map();
 		this.pendingOopifByMainFrame = /* @__PURE__ */ new Map();
+		this.pendingOopifParentByMainFrame = /* @__PURE__ */ new Map();
+		this.adoptedOopifSessionByFrame = /* @__PURE__ */ new Map();
 		this.createdAtByTarget = /* @__PURE__ */ new Map();
 		this.typeByTarget = /* @__PURE__ */ new Map();
 		this.pendingCreatedTargetUrl = /* @__PURE__ */ new Map();
@@ -61899,6 +61899,8 @@ var BrowserContext = class BrowserContext {
 		this.sessionOwnerPage.clear();
 		this.frameOwnerPage.clear();
 		this.pendingOopifByMainFrame.clear();
+		this.pendingOopifParentByMainFrame.clear();
+		this.adoptedOopifSessionByFrame.clear();
 		this.createdAtByTarget.clear();
 		this.typeByTarget.clear();
 		this.pendingCreatedTargetUrl.clear();
@@ -62103,34 +62105,91 @@ var BrowserContext = class BrowserContext {
 				this.pendingCreatedTargetUrl.delete(info.targetId);
 				page.seedCurrentUrl(pendingSeedUrl ?? info.url ?? "");
 				this.installFrameEventBridges(sessionId, page);
+				this.adoptPendingOopifsByParent();
 				await this.applyInitScriptsToPage(page, { seedOnly: scriptsInstalled });
 				return;
 			}
 			try {
 				const { frameTree } = await session.send("Page.getFrameTree");
 				const childMainId = frameTree.frame.id;
-				let owner = this.frameOwnerPage.get(childMainId);
-				if (!owner) {
-					for (const p of this.pagesByTarget.values()) if ((function find(n) {
-						if (n.frame.id === childMainId) return true;
-						for (const c of n.childFrames ?? []) if (find(c)) return true;
-						return false;
-					})(p.asProtocolFrameTree(p.mainFrameId()))) {
-						owner = p;
-						break;
-					}
-				}
+				const parentFrameId = frameTree.frame.parentId;
+				const owner = this.findPageOwningFrame(childMainId) ?? (parentFrameId ? this.findPageOwningFrame(parentFrameId) : void 0);
 				if (owner) {
-					owner.adoptOopifSession(session, childMainId);
-					this.sessionOwnerPage.set(sessionId, owner);
-					this.installFrameEventBridges(sessionId, owner);
-					executionContexts.waitForMainWorld(session, childMainId).catch(() => {});
-				} else this.pendingOopifByMainFrame.set(childMainId, sessionId);
+					this.adoptOopif(owner, session, childMainId);
+					this.adoptPendingOopifsByParent();
+				} else {
+					this.pendingOopifByMainFrame.set(childMainId, sessionId);
+					if (parentFrameId) this.pendingOopifParentByMainFrame.set(childMainId, parentFrameId);
+				}
 			} catch {}
 		} finally {
 			await resume();
 		}
 	}
+	/**
+	* Page whose frame graph contains `frameId`, including adopted OOPIF roots.
+	*/
+	findPageOwningFrame(frameId) {
+		const known = this.frameOwnerPage.get(frameId);
+		if (known) {
+			if (this.pagesByTarget.get(known.targetId()) === known && known.registry.frames.has(frameId)) return known;
+			this.frameOwnerPage.delete(frameId);
+		}
+		for (const p of this.pagesByTarget.values()) if (p.listAllFrameIds().includes(frameId)) return p;
+	}
+	adoptOopif(owner, session, childMainId) {
+		owner.adoptOopifSession(session, childMainId, () => {
+			this.adoptPendingOopifsByParent();
+		});
+		this.adoptedOopifSessionByFrame.set(childMainId, session.id);
+		this.sessionOwnerPage.set(session.id, owner);
+		this.frameOwnerPage.set(childMainId, owner);
+		this.installFrameEventBridges(session.id, owner);
+		executionContexts.waitForMainWorld(session, childMainId).catch(() => {});
+	}
+	removePendingOopif(frameId) {
+		this.pendingOopifByMainFrame.delete(frameId);
+		this.pendingOopifParentByMainFrame.delete(frameId);
+	}
+	removePendingOopifsForPage(page) {
+		const pageFrameIds = /* @__PURE__ */ new Set([page.mainFrameId(), ...page.registry.frames.keys()]);
+		for (const [frameId, owner] of this.frameOwnerPage) if (owner === page) pageFrameIds.add(frameId);
+		let removed;
+		do {
+			removed = false;
+			for (const [frameId] of this.pendingOopifByMainFrame) {
+				const parentId = this.pendingOopifParentByMainFrame.get(frameId);
+				if (!pageFrameIds.has(frameId) && (!parentId || !pageFrameIds.has(parentId))) continue;
+				pageFrameIds.add(frameId);
+				this.removePendingOopif(frameId);
+				removed = true;
+			}
+		} while (removed);
+	}
+	/**
+	* An OOPIF that already existed when we attached (e.g. reconnecting to a live
+	* browser, or a nested OOPIF) never emits `frameAttached` on its parent, so the
+	* pending map would hold it forever. Adopt it once the page owning its parent is known.
+	*/
+	adoptPendingOopifsByParent() {
+		let adopted = true;
+		while (adopted) {
+			adopted = false;
+			for (const [childMainId, parentFrameId] of Array.from(this.pendingOopifParentByMainFrame.entries())) {
+				const childSessionId = this.pendingOopifByMainFrame.get(childMainId);
+				const child = childSessionId ? this.conn.getSession(childSessionId) : void 0;
+				if (!child) {
+					this.removePendingOopif(childMainId);
+					continue;
+				}
+				const owner = this.findPageOwningFrame(parentFrameId);
+				if (!owner) continue;
+				this.removePendingOopif(childMainId);
+				this.adoptOopif(owner, child, childMainId);
+				adopted = true;
+			}
+		}
+	}
 	async closePopupIfBlockedByDomainPolicy(info, source) {
 		if (!this.domainPolicy || !isTopLevelPage(info)) return false;
 		if (!info.openerId && !info.openerFrameId) return false;
@@ -62179,13 +62238,16 @@ var BrowserContext = class BrowserContext {
 	* - Drop any staged child for this session.
 	*/
 	onDetachedFromTarget(sessionId, targetId) {
+		const closingPage = targetId ? this.pagesByTarget.get(targetId) : void 0;
+		if (closingPage) this.removePendingOopifsForPage(closingPage);
 		const owner = this.sessionOwnerPage.get(sessionId);
 		if (owner) {
 			owner.detachOopifSession(sessionId);
 			this.sessionOwnerPage.delete(sessionId);
 		}
 		if (targetId && this.pagesByTarget.has(targetId)) this.cleanupByTarget(targetId);
-		for (const [fid, sid] of Array.from(this.pendingOopifByMainFrame.entries())) if (sid === sessionId) this.pendingOopifByMainFrame.delete(fid);
+		for (const [fid, sid] of Array.from(this.pendingOopifByMainFrame.entries())) if (sid === sessionId) this.removePendingOopif(fid);
+		for (const [fid, sid] of Array.from(this.adoptedOopifSessionByFrame.entries())) if (sid === sessionId) this.adoptedOopifSessionByFrame.delete(fid);
 		this._targetSessionListeners.delete(sessionId);
 		const session = this.conn.getSession(sessionId);
 		if (session) this.uninstallDomainPolicyHandler(session);
@@ -62199,15 +62261,12 @@ var BrowserContext = class BrowserContext {
 		this.recordPageCreationFailure(targetId, /* @__PURE__ */ new Error(`Target closed before page registration (${targetId})`), "Target closed before page registration");
 		const page = this.pagesByTarget.get(targetId);
 		if (!page) return;
+		this.removePendingOopifsForPage(page);
 		page.dispose();
 		const mainId = page.mainFrameId();
 		this.mainFrameToTarget.delete(mainId);
-		this.frameOwnerPage.delete(mainId);
+		for (const [frameId, owner] of this.frameOwnerPage) if (owner === page) this.frameOwnerPage.delete(frameId);
 		for (const [sid, p] of Array.from(this.sessionOwnerPage.entries())) if (p === page) this.sessionOwnerPage.delete(sid);
-		for (const [fid] of Array.from(this.pendingOopifByMainFrame.entries())) {
-			const owner = this.frameOwnerPage.get(fid);
-			if (!owner || owner === page) this.pendingOopifByMainFrame.delete(fid);
-		}
 		this.pagesByTarget.delete(targetId);
 		this.createdAtByTarget.delete(targetId);
 		this.typeByTarget.delete(targetId);
@@ -62220,23 +62279,21 @@ var BrowserContext = class BrowserContext {
 	}
 	/**
 	* Wire Page-domain frame events for a session into the owning Page & mappings.
-	* We forward the *emitting session* with every event so Page can stamp ownership precisely.
+	* Attachments preserve live adopted OOPIF ownership while updating topology.
 	*/
 	installFrameEventBridges(sessionId, owner) {
 		const session = this.conn.getSession(sessionId);
 		if (!session) return;
 		session.on("Page.frameAttached", (evt) => {
 			const { frameId, parentFrameId } = evt;
-			owner.onFrameAttached(frameId, parentFrameId ?? null, session);
+			const adoptedSessionId = this.adoptedOopifSessionByFrame.get(frameId);
+			const adoptedSession = adoptedSessionId ? this.conn.getSession(adoptedSessionId) : void 0;
+			owner.onFrameAttached(frameId, parentFrameId ?? null, adoptedSession ?? session);
 			const pendingChildSessionId = this.pendingOopifByMainFrame.get(frameId);
 			if (pendingChildSessionId) {
 				const child = this.conn.getSession(pendingChildSessionId);
-				if (child) {
-					owner.adoptOopifSession(child, frameId);
-					this.sessionOwnerPage.set(child.id, owner);
-					this.installFrameEventBridges(pendingChildSessionId, owner);
-				}
-				this.pendingOopifByMainFrame.delete(frameId);
+				this.removePendingOopif(frameId);
+				if (child) this.adoptOopif(owner, child, frameId);
 			}
 			this.frameOwnerPage.set(frameId, owner);
 			if (!parentFrameId) {
@@ -62245,6 +62302,7 @@ var BrowserContext = class BrowserContext {
 				if (topTargetId) this.mainFrameToTarget.set(newRoot, topTargetId);
 				this.frameOwnerPage.set(newRoot, owner);
 			}
+			this.adoptPendingOopifsByParent();
 		});
 		session.on("Page.frameDetached", (evt) => {
 			owner.onFrameDetached(evt.frameId, evt.reason ?? "remove");

{stagehand-4.4.0a0.dev1660 → stagehand-4.4.0a0.dev1661}/src/stagehand/_generated/protocol_version.py

@@ -1,4 +1,4 @@
 """Generated from packages/protocol/package.json. Do not edit."""
 
 PROTOCOL_SEMVER_PATTERN = '^(0|[1-9]\\d*)\\.(0|[1-9]\\d*)\\.(0|[1-9]\\d*)(?:-((?:0|[1-9]\\d*|\\d*[A-Za-z-][0-9A-Za-z-]*)(?:\\.(?:0|[1-9]\\d*|\\d*[A-Za-z-][0-9A-Za-z-]*))*))?(?:\\+([0-9A-Za-z-]+(?:\\.[0-9A-Za-z-]+)*))?$'
-STAGEHAND_PROTOCOL_VERSION = '3.1.1-alpha-d59afde37e995b165a41f754a12a446f11d471a2'
+STAGEHAND_PROTOCOL_VERSION = '3.1.1-alpha-44d99dfbdd24baaf06986492792348582266d8f0'