@browserbasehq/stagehand 3.0.7-alpha-e822f5a8898df9eb48ca32c321025f0c74b638f0 → 3.0.7-alpha-a890f16fa3a752f308f858e5ab9c9a0faf6b3b34

package/dist/index.js

@@ -179,7 +179,7 @@ var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")])
 var STAGEHAND_VERSION;
 var init_version = __esm({
   "lib/version.ts"() {
-    STAGEHAND_VERSION = "3.0.7-alpha-e822f5a8898df9eb48ca32c321025f0c74b638f0";
+    STAGEHAND_VERSION = "3.0.7-alpha-a890f16fa3a752f308f858e5ab9c9a0faf6b3b34";
   }
 });
 
@@ -988,6 +988,115 @@ function absoluteXPathForBackendNode(session, backendNodeId) {
     }
   });
 }
+function isCborStackError(message) {
+  return message.includes("CBOR: stack limit exceeded");
+}
+function shouldExpandNode(node) {
+  var _a4, _b, _c;
+  const declaredChildren = (_a4 = node.childNodeCount) != null ? _a4 : 0;
+  const realizedChildren = (_c = (_b = node.children) == null ? void 0 : _b.length) != null ? _c : 0;
+  return declaredChildren > realizedChildren;
+}
+function mergeDomNodes(target, source) {
+  var _a4, _b, _c, _d2;
+  target.childNodeCount = (_a4 = source.childNodeCount) != null ? _a4 : target.childNodeCount;
+  target.children = (_b = source.children) != null ? _b : target.children;
+  target.shadowRoots = (_c = source.shadowRoots) != null ? _c : target.shadowRoots;
+  target.contentDocument = (_d2 = source.contentDocument) != null ? _d2 : target.contentDocument;
+}
+function collectDomTraversalTargets(node) {
+  const targets = [];
+  if (node.children) targets.push(...node.children);
+  if (node.shadowRoots) targets.push(...node.shadowRoots);
+  if (node.contentDocument) targets.push(node.contentDocument);
+  return targets;
+}
+function hydrateDomTree(session, root, pierce) {
+  return __async(this, null, function* () {
+    var _a4, _b;
+    const stack = [root];
+    const expandedNodeIds = /* @__PURE__ */ new Set();
+    const expandedBackendIds = /* @__PURE__ */ new Set();
+    while (stack.length) {
+      const node = stack.pop();
+      const nodeId = typeof node.nodeId === "number" && node.nodeId > 0 ? node.nodeId : void 0;
+      const backendId = typeof node.backendNodeId === "number" && node.backendNodeId > 0 ? node.backendNodeId : void 0;
+      const seenByNode = nodeId ? expandedNodeIds.has(nodeId) : false;
+      const seenByBackend = !nodeId && backendId ? expandedBackendIds.has(backendId) : false;
+      if (seenByNode || seenByBackend) continue;
+      if (nodeId) expandedNodeIds.add(nodeId);
+      else if (backendId) expandedBackendIds.add(backendId);
+      const needsExpansion = shouldExpandNode(node);
+      if (needsExpansion && (nodeId || backendId)) {
+        const describeParamsBase = nodeId ? { nodeId } : { backendNodeId: backendId };
+        let expanded = false;
+        for (const depth of DESCRIBE_DEPTH_ATTEMPTS) {
+          try {
+            const described = yield session.send(
+              "DOM.describeNode",
+              __spreadProps(__spreadValues({}, describeParamsBase), {
+                depth,
+                pierce
+              })
+            );
+            mergeDomNodes(node, described.node);
+            if (!nodeId && described.node.nodeId && described.node.nodeId > 0) {
+              node.nodeId = described.node.nodeId;
+              expandedNodeIds.add(described.node.nodeId);
+            }
+            expanded = true;
+            break;
+          } catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            if (isCborStackError(message)) {
+              continue;
+            }
+            const identifier = (_a4 = nodeId != null ? nodeId : backendId) != null ? _a4 : "unknown";
+            throw new StagehandDomProcessError(
+              `Failed to expand DOM node ${identifier}: ${String(err)}`
+            );
+          }
+        }
+        if (!expanded) {
+          const identifier = (_b = nodeId != null ? nodeId : backendId) != null ? _b : "unknown";
+          throw new StagehandDomProcessError(
+            `Unable to expand DOM node ${identifier} after describeNode depth retries`
+          );
+        }
+      }
+      for (const child of collectDomTraversalTargets(node)) {
+        stack.push(child);
+      }
+    }
+  });
+}
+function getDomTreeWithFallback(session, pierce) {
+  return __async(this, null, function* () {
+    let lastCborMessage = "";
+    for (const depth of DOM_DEPTH_ATTEMPTS) {
+      try {
+        const { root } = yield session.send(
+          "DOM.getDocument",
+          { depth, pierce }
+        );
+        if (depth !== -1) {
+          yield hydrateDomTree(session, root, pierce);
+        }
+        return root;
+      } catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        if (isCborStackError(message)) {
+          lastCborMessage = message;
+          continue;
+        }
+        throw err;
+      }
+    }
+    throw new StagehandDomProcessError(
+      lastCborMessage ? `CDP DOM.getDocument failed after adaptive depth retries: ${lastCborMessage}` : "CDP DOM.getDocument failed after adaptive depth retries."
+    );
+  });
+}
 function captureHybridSnapshot(page, options) {
   return __async(this, null, function* () {
     var _a4, _b, _c, _d2, _e, _f, _g, _h, _i, _j, _k, _l;
@@ -1106,10 +1215,7 @@ function captureHybridSnapshot(page, options) {
         var _a5, _b2;
         yield session.send("DOM.enable").catch(() => {
         });
-        const { root } = yield session.send(
-          "DOM.getDocument",
-          { depth: -1, pierce: pierce2 }
-        );
+        const root = yield getDomTreeWithFallback(session, pierce2);
         const absByBe = /* @__PURE__ */ new Map();
         const tagByBe = /* @__PURE__ */ new Map();
         const scrollByBe = /* @__PURE__ */ new Map();
@@ -1466,10 +1572,7 @@ function domMapsForSession(session, frameId, pierce, encode = (fid, be) => `${fi
     var _a4, _b;
     yield session.send("DOM.enable").catch(() => {
     });
-    const { root } = yield session.send(
-      "DOM.getDocument",
-      { depth: -1, pierce }
-    );
+    const root = yield getDomTreeWithFallback(session, pierce);
     let startNode = root;
     if (attemptOwnerLookup) {
       try {
@@ -1966,12 +2069,14 @@ function parentSession(page, parentByFrame, frameId) {
   }
   return page.getSessionForFrame(parentId);
 }
-var IFRAME_STEP_RE;
+var DOM_DEPTH_ATTEMPTS, DESCRIBE_DEPTH_ATTEMPTS, IFRAME_STEP_RE;
 var init_snapshot = __esm({
   "lib/v3/understudy/a11y/snapshot.ts"() {
     init_executionContextRegistry();
     init_logger();
     init_sdkErrors();
+    DOM_DEPTH_ATTEMPTS = [-1, 256, 128, 64, 32, 16, 8, 4, 2, 1];
+    DESCRIBE_DEPTH_ATTEMPTS = [-1, 64, 32, 16, 8, 4, 2, 1];
     IFRAME_STEP_RE = /^iframe(?:\[\d+])?$/i;
   }
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@browserbasehq/stagehand",
-  "version": "3.0.7-alpha-e822f5a8898df9eb48ca32c321025f0c74b638f0",
+  "version": "3.0.7-alpha-a890f16fa3a752f308f858e5ab9c9a0faf6b3b34",
   "description": "An AI web browsing framework focused on simplicity and extensibility.",
   "main": "./dist/index.js",
   "module": "./dist/index.js",