pi-agent-browser-native 0.9.3 → 0.9.4

package/dist/extensions/agent-browser/lib/results/snapshot-segments.js

@@ -25,7 +25,14 @@ const SNAPSHOT_SIGNAL_ROLES = new Set([
     "tab",
     "textbox",
 ]);
-const SNAPSHOT_SEGMENT_ROOT_ROLES = new Set(["article", "dialog", "heading", "main", "menu", "region"]);
+const SNAPSHOT_SEGMENT_ROOT_ROLES = new Set([
+    "article",
+    "dialog",
+    "heading",
+    "main",
+    "menu",
+    "region",
+]);
 const SNAPSHOT_ROLE_PRIORITY = {
     article: 0,
     main: 1,
@@ -71,20 +78,23 @@ const SNAPSHOT_CHROME_SECTION_PATTERNS = [
 export function getSnapshotRolePriority(role) {
     return SNAPSHOT_ROLE_PRIORITY[role] ?? 50;
 }
+function parseSnapshotHeadingLevel(text) {
+    const level = text.match(/\blevel=(\d+)/)?.[1];
+    return level !== undefined && level.length > 0 ? Number(level) : undefined;
+}
 export function parseSnapshotLines(snapshot) {
     return snapshot
         .split("\n")
         .filter((line) => line.length > 0)
         .map((raw, index) => {
         const trimmed = raw.trimStart();
-        const depth = Math.floor(((raw.match(/^\s*/) ?? [""])[0].length ?? 0) / 2);
+        const depth = Math.floor((raw.match(/^\s*/)?.[0].length ?? 0) / 2);
         const role = trimmed.match(/^[-*]\s+([^\s"]+)/)?.[1] ?? "unknown";
         const name = normalizeWhitespace(trimmed.match(/"([^"]*)"/)?.[1] ?? "");
         const ref = trimmed.match(/\bref=([^,\]\s]+)/)?.[1];
-        const headingLevel = trimmed.match(/\blevel=(\d+)/)?.[1];
         return {
             depth,
-            headingLevel: headingLevel ? Number(headingLevel) : undefined,
+            headingLevel: parseSnapshotHeadingLevel(trimmed),
             index,
             name,
             raw,
@@ -100,16 +110,19 @@ export function isChromeSectionName(name) {
     return SNAPSHOT_CHROME_SECTION_PATTERNS.some((pattern) => pattern.test(name));
 }
 function isNoiseSnapshotLine(line) {
-    if (line.name.length > 0 && isNoiseName(line.name))
+    if (line.name.length > 0 && isNoiseName(line.name)) {
         return true;
+    }
     const loweredRaw = line.raw.toLowerCase();
     return loweredRaw.includes("promoted") || loweredRaw.includes("sponsored");
 }
 function isPotentialSegmentRootLine(line) {
-    if (!SNAPSHOT_SEGMENT_ROOT_ROLES.has(line.role))
+    if (!SNAPSHOT_SEGMENT_ROOT_ROLES.has(line.role)) {
         return false;
-    if (isNoiseSnapshotLine(line))
+    }
+    if (isNoiseSnapshotLine(line)) {
         return false;
+    }
     if (line.role === "heading") {
         return line.name.length > 0 && (line.headingLevel ?? 99) <= 3;
     }
@@ -118,9 +131,23 @@ function isPotentialSegmentRootLine(line) {
     }
     return true;
 }
+function getHeadingPriorityScore(level) {
+    switch (level) {
+        case 1:
+            return 40;
+        case 2:
+            return 22;
+        case 3:
+            return 12;
+        case undefined:
+            return 0;
+        default:
+            return 0;
+    }
+}
 function scoreSegment(segment) {
     const { root } = segment;
-    const distinctRefs = new Set(segment.lines.flatMap((line) => (line.ref ? [line.ref] : []))).size;
+    const distinctRefs = new Set(segment.lines.flatMap((line) => line.ref !== undefined && line.ref.length > 0 ? [line.ref] : [])).size;
     let score = 0;
     score += 120 - getSnapshotRolePriority(root.role) * 8;
     score += Math.min(distinctRefs, 16);
@@ -128,43 +155,44 @@ function scoreSegment(segment) {
     score -= Math.min(root.index, 60) / 3;
     score -= root.depth * 6;
     if (root.role === "heading") {
-        if (root.headingLevel === 1)
-            score += 40;
-        else if (root.headingLevel === 2)
-            score += 22;
-        else if (root.headingLevel === 3)
-            score += 12;
+        score += getHeadingPriorityScore(root.headingLevel);
     }
-    if (root.name.length > 0)
+    if (root.name.length > 0) {
         score += 10;
-    if (root.name.length <= 2)
+    }
+    if (root.name.length <= 2) {
         score -= 18;
-    if (isChromeSectionName(root.name))
+    }
+    if (isChromeSectionName(root.name)) {
         score -= 45;
-    if (isNoiseName(root.name))
+    }
+    if (isNoiseName(root.name)) {
         score -= 1000;
+    }
     return score;
 }
+function hasDuplicateSegmentAncestor(line, stack) {
+    const name = normalizeWhitespace(line.name.toLowerCase());
+    if (name.length === 0) {
+        return false;
+    }
+    for (let index = stack.length - 1; index >= 0; index -= 1) {
+        const ancestor = stack.at(index);
+        if (ancestor &&
+            normalizeWhitespace(ancestor.name.toLowerCase()) === name &&
+            SNAPSHOT_SEGMENT_ROOT_ROLES.has(ancestor.role)) {
+            return true;
+        }
+    }
+    return false;
+}
 export function buildSnapshotSegments(snapshotLines) {
     const roots = [];
     const stack = [];
     for (const line of snapshotLines) {
         stack.length = line.depth;
-        if (isPotentialSegmentRootLine(line)) {
-            const normalizedName = normalizeWhitespace(line.name.toLowerCase());
-            let duplicateAncestor;
-            for (let index = stack.length - 1; index >= 0; index -= 1) {
-                const ancestor = stack[index];
-                if (normalizedName.length > 0 &&
-                    normalizeWhitespace(ancestor.name.toLowerCase()) === normalizedName &&
-                    SNAPSHOT_SEGMENT_ROOT_ROLES.has(ancestor.role)) {
-                    duplicateAncestor = ancestor;
-                    break;
-                }
-            }
-            if (!duplicateAncestor) {
-                roots.push(line);
-            }
+        if (isPotentialSegmentRootLine(line) && !hasDuplicateSegmentAncestor(line, stack)) {
+            roots.push(line);
         }
         stack[line.depth] = line;
     }
@@ -185,58 +213,75 @@ export function buildSnapshotSegments(snapshotLines) {
             score: 0,
             startIndex: root.index,
         };
-        segment.score = scoreSegment(segment);
-        return segment;
+        return Object.assign({}, segment, { score: scoreSegment(segment) });
     });
 }
 export function choosePrimarySegment(segments) {
-    if (segments.length === 0)
+    if (segments.length === 0) {
         return undefined;
+    }
     return (segments.find((segment) => segment.root.role === "main" || segment.root.role === "article") ??
         segments.find((segment) => segment.root.role === "heading" && segment.root.headingLevel === 1) ??
         segments.find((segment) => segment.score >= 90) ??
-        [...segments].sort((left, right) => right.score - left.score || left.startIndex - right.startIndex)[0]);
+        [...segments].sort((left, right) => {
+            const difference = right.score - left.score;
+            return difference !== 0 && !Number.isNaN(difference)
+                ? difference
+                : left.startIndex - right.startIndex;
+        })[0]);
+}
+function isChromePreviewSegment(segment) {
+    return (isChromeSectionName(segment.root.name) ||
+        (segment.root.role === "heading" && segment.root.name.length <= 2));
 }
 export function chooseAdditionalSegments(segments, primary) {
-    if (!primary)
+    if (!primary) {
         return [];
+    }
     const seenNames = new Set([normalizeWhitespace(primary.root.name.toLowerCase())]);
     const rankedCandidates = segments
         .filter((segment) => segment !== primary && segment.score >= 45)
         .sort((left, right) => {
         const leftDistance = Math.abs(left.startIndex - primary.startIndex);
         const rightDistance = Math.abs(right.startIndex - primary.startIndex);
-        if (leftDistance !== rightDistance)
+        if (leftDistance !== rightDistance) {
             return leftDistance - rightDistance;
-        if (right.score !== left.score)
+        }
+        if (right.score !== left.score) {
             return right.score - left.score;
+        }
         return left.startIndex - right.startIndex;
     });
     const chosen = [];
     for (const segment of rankedCandidates) {
-        if (chosen.length >= SNAPSHOT_MAX_ADDITIONAL_SECTIONS)
+        if (chosen.length >= SNAPSHOT_MAX_ADDITIONAL_SECTIONS) {
             break;
-        if (isChromeSectionName(segment.root.name))
-            continue;
-        if (segment.root.role === "heading" && segment.root.name.length <= 2)
+        }
+        if (isChromePreviewSegment(segment)) {
             continue;
+        }
         const nameKey = normalizeWhitespace(segment.root.name.toLowerCase());
-        if (nameKey && seenNames.has(nameKey))
+        if (nameKey.length > 0 && seenNames.has(nameKey)) {
             continue;
+        }
         chosen.push(segment);
-        if (nameKey)
+        if (nameKey.length > 0) {
             seenNames.add(nameKey);
+        }
     }
     return chosen.sort((left, right) => left.startIndex - right.startIndex);
 }
 export function getMeaningfulSegmentLines(segment) {
     return segment.lines.filter((line) => {
-        if (isNoiseSnapshotLine(line))
+        if (isNoiseSnapshotLine(line)) {
             return false;
-        if (line.role === "generic" && !line.ref && line.name.length === 0)
+        }
+        if (line.role === "generic" && (line.ref ?? "") === "" && line.name.length === 0) {
             return false;
-        if (line.role === "link" && line.name.length === 0)
+        }
+        if (line.role === "link" && line.name.length === 0) {
             return false;
+        }
         return true;
     });
 }
@@ -245,6 +290,15 @@ function formatPreviewLine(line, baseDepth) {
     const stripChars = Math.min(leadingWhitespace, Math.max(0, baseDepth) * 2);
     return truncateText(line.raw.slice(stripChars), SNAPSHOT_LINE_MAX_CHARS);
 }
+function isRedundantPreviewLine(line, root) {
+    if (line === root) {
+        return false;
+    }
+    return ((root.role !== "heading" && line.depth - root.depth > 2) ||
+        (root.name.length > 0 &&
+            line.name === root.name &&
+            (line.role === "heading" || line.role === "link")));
+}
 export function buildSegmentPreview(segment, maxLines) {
     const meaningfulLines = getMeaningfulSegmentLines(segment);
     if (meaningfulLines.length === 0) {
@@ -255,23 +309,21 @@ export function buildSegmentPreview(segment, maxLines) {
     const seenPreviewKeys = new Set();
     const rootDepth = segment.root.depth;
     for (const line of meaningfulLines) {
-        if (previewLines.length >= maxLines)
+        if (previewLines.length >= maxLines) {
             break;
-        if (line !== segment.root) {
-            const relativeDepth = line.depth - rootDepth;
-            if (segment.root.role !== "heading" && relativeDepth > 2)
-                continue;
-            if (segment.root.name.length > 0 && line.name === segment.root.name && (line.role === "heading" || line.role === "link")) {
-                continue;
-            }
+        }
+        if (isRedundantPreviewLine(line, segment.root)) {
+            continue;
         }
         const key = `${line.role}:${line.name}:${line.ref ?? ""}:${line.depth}`;
-        if (seenPreviewKeys.has(key))
+        if (seenPreviewKeys.has(key)) {
             continue;
+        }
         seenPreviewKeys.add(key);
         previewLines.push(line);
-        if (line.ref)
+        if (line.ref !== undefined && line.ref.length > 0) {
             previewRefIds.add(line.ref);
+        }
     }
     return {
         omittedCount: Math.max(0, meaningfulLines.length - previewLines.length),
@@ -279,17 +331,22 @@ export function buildSegmentPreview(segment, maxLines) {
         lines: previewLines.map((line) => formatPreviewLine(line, rootDepth)),
     };
 }
+function hasSnapshotSignal(line) {
+    return (SNAPSHOT_SIGNAL_ROLES.has(line.role) || (line.ref ?? "").length > 0 || line.name.length > 0);
+}
 export function buildFallbackSnapshotOutline(snapshotLines) {
     const selected = new Set();
     for (let index = 0; index < snapshotLines.length && selected.size < 4; index += 1) {
-        if (!isNoiseSnapshotLine(snapshotLines[index]))
+        if (!isNoiseSnapshotLine(snapshotLines[index])) {
             selected.add(index);
+        }
     }
     for (let index = 0; index < snapshotLines.length && selected.size < SNAPSHOT_FALLBACK_PREVIEW_MAX_LINES; index += 1) {
         const line = snapshotLines[index];
-        if (isNoiseSnapshotLine(line))
+        if (isNoiseSnapshotLine(line)) {
             continue;
-        if (SNAPSHOT_SIGNAL_ROLES.has(line.role) || line.ref || line.name.length > 0) {
+        }
+        if (hasSnapshotSignal(line)) {
             selected.add(index);
         }
     }
@@ -299,25 +356,27 @@ export function buildFallbackSnapshotOutline(snapshotLines) {
         .map((index) => snapshotLines[index]);
     return {
         omittedCount: Math.max(0, snapshotLines.length - chosenLines.length),
-        refIds: chosenLines.flatMap((line) => (line.ref ? [line.ref] : [])),
+        refIds: chosenLines.flatMap((line) => line.ref !== undefined && line.ref.length > 0 ? [line.ref] : []),
         lines: chosenLines.map((line) => truncateText(line.raw, SNAPSHOT_LINE_MAX_CHARS)),
     };
 }
 export function buildRefLineOrderMap(snapshotLines) {
     const map = new Map();
     for (const line of snapshotLines) {
-        if (!line.ref || map.has(line.ref))
+        if (line.ref === undefined || line.ref.length === 0 || map.has(line.ref)) {
             continue;
+        }
         map.set(line.ref, line.index);
     }
     return map;
 }
 export function canUseStructuredSnapshotPreview(snapshotLines, refEntries) {
-    if (snapshotLines.length === 0)
+    if (snapshotLines.length === 0) {
         return false;
+    }
     const linesWithRecognizedRoles = snapshotLines.filter((line) => line.role !== "unknown").length;
     const linesWithNames = snapshotLines.filter((line) => line.name.length > 0).length;
-    const parsedRefIds = new Set(snapshotLines.flatMap((line) => (line.ref ? [line.ref] : [])));
+    const parsedRefIds = new Set(snapshotLines.flatMap((line) => line.ref !== undefined && line.ref.length > 0 ? [line.ref] : []));
     return (linesWithRecognizedRoles >= Math.min(snapshotLines.length, 3) ||
         linesWithNames >= Math.min(snapshotLines.length, 3) ||
         parsedRefIds.size >= Math.min(refEntries.length, 3));

package/dist/extensions/agent-browser/lib/results/snapshot-spill.js

@@ -1,5 +1,5 @@
-import { redactSensitiveValue } from "../runtime.js";
-import { writePersistentSessionArtifactFile, writeSecureTempFile } from "../temp.js";
+import { redactSensitiveValue } from "../runtime-redaction.js";
+import { writePersistentSessionArtifactFile, writeSecureTempFile, } from "../temp.js";
 import { buildEvictedSessionArtifactEntries, formatSessionArtifactRetentionSummary, mergeSessionArtifactManifest, } from "./artifact-manifest.js";
 const SNAPSHOT_SPILL_FILE_PREFIX = "pi-agent-browser-snapshot";
 export async function writeSnapshotSpillFile(data, persistentArtifactStore) {
@@ -9,14 +9,24 @@ export async function writeSnapshotSpillFile(data, persistentArtifactStore) {
         suffix: ".json",
     };
     if (persistentArtifactStore) {
-        const result = await writePersistentSessionArtifactFile({ ...options, store: persistentArtifactStore });
+        const result = await writePersistentSessionArtifactFile({
+            ...options,
+            store: persistentArtifactStore,
+        });
         return { ...result, storageScope: "persistent-session" };
     }
-    return { evictedArtifacts: [], path: await writeSecureTempFile(options), storageScope: "process-temp" };
+    return {
+        evictedArtifacts: [],
+        path: await writeSecureTempFile(options),
+        storageScope: "process-temp",
+    };
 }
 export function applySnapshotArtifactManifest(options) {
-    if (!options.fullOutputPath || !options.spill)
+    if (options.fullOutputPath === undefined ||
+        options.fullOutputPath.length === 0 ||
+        !options.spill) {
         return {};
+    }
     const nowMs = Date.now();
     const entries = [
         {
@@ -29,6 +39,15 @@ export function applySnapshotArtifactManifest(options) {
         },
         ...buildEvictedSessionArtifactEntries(options.spill.evictedArtifacts, nowMs),
     ];
-    const artifactManifest = mergeSessionArtifactManifest({ base: options.baseManifest, entries, nowMs });
-    return artifactManifest ? { artifactManifest, artifactRetentionSummary: formatSessionArtifactRetentionSummary(artifactManifest) } : {};
+    const artifactManifest = mergeSessionArtifactManifest({
+        base: options.baseManifest,
+        entries,
+        nowMs,
+    });
+    return artifactManifest
+        ? {
+            artifactManifest,
+            artifactRetentionSummary: formatSessionArtifactRetentionSummary(artifactManifest),
+        }
+        : {};
 }