pi-agent-browser-native 0.2.24 → 0.2.26

package/extensions/agent-browser/lib/runtime.ts

@@ -47,10 +47,22 @@ const LAUNCH_SCOPED_FLAG_DEFINITIONS = [
 		flag: "--init-script",
 		reason: "registers page init scripts before the upstream browser session is launched",
 	},
+	{
+		flag: "--device",
+		reason: "selects the provider device for the upstream launch",
+	},
 	{
 		flag: "--profile",
 		reason: "selects Chrome profile state for the upstream launch",
 	},
+	{
+		flag: "--provider",
+		reason: "selects the upstream browser provider for the launch",
+	},
+	{
+		flag: "-p",
+		reason: "selects the upstream browser provider for the launch",
+	},
 	{
 		flag: "--session-name",
 		reason: "selects upstream saved auth/session state for the launch",
@@ -92,7 +104,7 @@ const BROWSER_PROMPT_PATTERNS = [
 	/\b(?:browse|click|fill|login|navigate|open|visit)\b.*\b(?:https?:\/\/\S+|page|site|tab|url|web(?:site| page)?)\b/i,
 ];
 const INSPECTION_FLAGS = new Set(["--help", "-h", "--version", "-V"]);
-const SENSITIVE_VALUE_FLAGS = new Set(["--headers", "--password", "--proxy"]);
+const SENSITIVE_VALUE_FLAGS = new Set(["--body", "--headers", "--password", "--proxy"]);
 const GLOBAL_VALUE_FLAGS_ALLOWING_DASH_VALUE = new Set(["--args"]);
 const GLOBAL_BOOLEAN_FLAGS_WITH_OPTIONAL_VALUES = new Set([
 	"--allow-file-access",
@@ -115,7 +127,7 @@ const SENSITIVE_QUERY_PARAM_PATTERN =
 const SENSITIVE_FIELD_NAME_PATTERN =
 	/^(?:access(?:_|-)?token|api(?:_|-)?key|auth(?:orization)?|bearer|client(?:_|-)?secret|cookie|id(?:_|-)?token|pass(?:word)?|proxy(?:_|-)?authorization|refresh(?:_|-)?token|secret|session(?:_|-)?id|set(?:_|-)?cookie|sig(?:nature)?|token|x(?:_|-)?api(?:_|-)?key)$/i;
 
-const GLOBAL_FLAGS_WITH_VALUES = new Set([
+const VALUE_FLAGS = new Set([
 	"--session",
 	"--cdp",
 	"--config",
@@ -146,6 +158,29 @@ const GLOBAL_FLAGS_WITH_VALUES = new Set([
 	"--confirm-actions",
 	"--max-output",
 	"--model",
+	"--baseline",
+	"--body",
+	"--categories",
+	"--curl",
+	"--depth",
+	"-d",
+	"--domain",
+	"--expires",
+	"--filter",
+	"--fn",
+	"--label",
+	"--load",
+	"--name",
+	"--path",
+	"--resource-type",
+	"--sameSite",
+	"--selector",
+	"-s",
+	"--text",
+	"--timeout",
+	"--url",
+	"--username",
+	"--password",
 ]);
 const DEFAULT_HEADLESS_COMPAT_USER_AGENT_BY_PLATFORM: Partial<Record<NodeJS.Platform, string>> = {
 	darwin: "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/146.0.0.0 Safari/537.36",
@@ -440,7 +475,7 @@ export function redactInvocationArgs(args: string[]): string[] {
 			continue;
 		}
 
-		redacted.push(redactUrlToken(token));
+		redacted.push(redactSensitiveText(redactUrlToken(token)));
 	}
 
 	const commandStartIndex = findCommandStartIndex(args);
@@ -452,6 +487,20 @@ export function redactInvocationArgs(args: string[]): string[] {
 		}
 	}
 
+	if (commandStartIndex !== undefined && args[commandStartIndex] === "cookies" && args[commandStartIndex + 1] === "set" && redacted[commandStartIndex + 3] !== undefined) {
+		redacted[commandStartIndex + 3] = "[REDACTED]";
+	}
+
+	if (
+		commandStartIndex !== undefined
+		&& args[commandStartIndex] === "storage"
+		&& ["local", "session"].includes(args[commandStartIndex + 1] ?? "")
+		&& args[commandStartIndex + 2] === "set"
+		&& redacted[commandStartIndex + 4] !== undefined
+	) {
+		redacted[commandStartIndex + 4] = "[REDACTED]";
+	}
+
 	return redacted;
 }
 
@@ -467,6 +516,10 @@ export function isPlainTextInspectionArgs(args: string[]): boolean {
 	return args.some((token) => INSPECTION_FLAGS.has(token));
 }
 
+function isStatelessInspectionCommand(commandInfo: CommandInfo): boolean {
+	return commandInfo.command === "skills" && ["list", "get", "path"].includes(commandInfo.subcommand ?? "");
+}
+
 export function hasUsableBraveApiKey(apiKey: string | null | undefined = process.env[BRAVE_API_KEY_ENV]): boolean {
 	return typeof apiKey === "string" && apiKey.trim().length > 0;
 }
@@ -596,9 +649,14 @@ export function restoreManagedSessionStateFromBranch(
 
 		const messageIsError = typeof message.isError === "boolean" ? message.isError : undefined;
 		const exitCode = typeof details.exitCode === "number" ? details.exitCode : undefined;
-		const succeeded = messageIsError === undefined ? exitCode === undefined || exitCode === 0 : !messageIsError;
+		const outcome = typeof details.managedSessionOutcome === "object" && details.managedSessionOutcome !== null ? details.managedSessionOutcome as Record<string, unknown> : undefined;
+		const outcomeStatus = typeof outcome?.status === "string" ? outcome.status : undefined;
+		const outcomeCurrentSessionName = typeof outcome?.currentSessionName === "string" ? outcome.currentSessionName : undefined;
+		const outcomeActiveAfter = outcome?.activeAfter === true;
+		const outcomeRepresentsActiveCurrentSession = outcomeActiveAfter && outcomeCurrentSessionName === managedSessionName && (outcomeStatus === "created" || outcomeStatus === "replaced" || outcomeStatus === "unchanged");
+		const succeeded = outcomeRepresentsActiveCurrentSession ? true : messageIsError === undefined ? exitCode === undefined || exitCode === 0 : !messageIsError;
 		const command = typeof details.command === "string" ? details.command : parseCommandInfo(args).command;
-		if (succeeded && sessionMode === "fresh") {
+		if ((succeeded || outcomeRepresentsActiveCurrentSession) && sessionMode === "fresh") {
 			freshSessionOrdinal += 1;
 		}
 		const staleCompletion = succeeded && command !== "close" && restoreRank < activeRestoreRank;
@@ -693,7 +751,7 @@ function getInvalidValueFlagDetails(args: string[]): InvalidValueFlagDetails | u
 			continue;
 		}
 		const normalizedToken = token.split("=", 1)[0] ?? token;
-		if (!GLOBAL_FLAGS_WITH_VALUES.has(normalizedToken)) {
+		if (!VALUE_FLAGS.has(normalizedToken)) {
 			continue;
 		}
 		if (token.includes("=")) {
@@ -861,10 +919,23 @@ export function extractExplicitSessionName(args: string[]): string | undefined {
 	return undefined;
 }
 
+function hasLaunchScopedFlagToken(args: string[], flag: string): boolean {
+	const commandStartIndex = findCommandStartIndex(args);
+	const command = commandStartIndex === undefined ? undefined : args[commandStartIndex];
+	return args.some((token, index) => {
+		if (token !== flag && !token.startsWith(`${flag}=`)) return false;
+		if (flag === "--auto-connect") return isBooleanFlagEnabled(args, flag);
+		if (flag === "--state" && command === "wait" && commandStartIndex !== undefined && index > commandStartIndex) {
+			return false;
+		}
+		return true;
+	});
+}
+
 export function getStartupScopedFlags(args: string[]): string[] {
 	return LAUNCH_SCOPED_FLAG_DEFINITIONS
 		.map((definition) => definition.flag)
-		.filter((flag) => flag === "--auto-connect" ? isBooleanFlagEnabled(args, flag) : hasFlagToken(args, flag));
+		.filter((flag) => hasLaunchScopedFlagToken(args, flag));
 }
 
 export function hasLaunchScopedTabCorrectionFlag(args: string[]): boolean {
@@ -923,6 +994,7 @@ export function buildExecutionPlan(
 	const startupScopedFlags = getStartupScopedFlags(args);
 	const plainTextInspection = isPlainTextInspectionArgs(args);
 	const commandInfo = parseCommandInfo(args);
+	const statelessInspection = plainTextInspection || isStatelessInspectionCommand(commandInfo);
 	const effectiveArgs = plainTextInspection ? [...args] : args.includes("--json") ? [] : ["--json"];
 	if (invalidValueFlag) {
 		return {
@@ -956,7 +1028,7 @@ export function buildExecutionPlan(
 	let usedImplicitSession = false;
 	let validationError: string | undefined;
 
-	if (!explicitSessionName && options.sessionMode === "auto") {
+	if (!explicitSessionName && options.sessionMode === "auto" && !statelessInspection) {
 		if (options.managedSessionActive && startupScopedFlags.length > 0) {
 			recoveryHint = {
 				exampleArgs: args,
@@ -975,7 +1047,7 @@ export function buildExecutionPlan(
 			sessionName = options.managedSessionName;
 			usedImplicitSession = true;
 		}
-	} else if (shouldCreateFreshManagedSession) {
+	} else if (shouldCreateFreshManagedSession && !statelessInspection) {
 		effectiveArgs.push("--session", options.freshSessionName);
 		managedSessionName = options.freshSessionName;
 		sessionName = options.freshSessionName;
@@ -1080,7 +1152,7 @@ function findCommandStartIndex(args: string[]): number | undefined {
 		}
 		if (token.startsWith("-")) {
 			const normalizedToken = token.split("=", 1)[0] ?? token;
-			if (GLOBAL_FLAGS_WITH_VALUES.has(normalizedToken) && !token.includes("=")) {
+			if (VALUE_FLAGS.has(normalizedToken) && !token.includes("=")) {
 				index += 1;
 			} else if (
 				GLOBAL_BOOLEAN_FLAGS_WITH_OPTIONAL_VALUES.has(normalizedToken) &&

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-agent-browser-native",
-  "version": "0.2.24",
+  "version": "0.2.26",
   "description": "pi extension that exposes agent-browser as a native tool for browser automation",
   "type": "module",
   "author": "Mitch Fultz (https://github.com/fitchmultz)",
@@ -40,6 +40,7 @@
     "docs/COMMAND_REFERENCE.md",
     "docs/RELEASE.md",
     "docs/REQUIREMENTS.md",
+    "docs/SUPPORT_MATRIX.md",
     "docs/TOOL_CONTRACT.md"
   ],
   "pi": {
@@ -66,9 +67,10 @@
   "scripts": {
     "docs": "node ./scripts/project.mjs docs",
     "doctor": "node ./scripts/doctor.mjs",
+    "benchmark:agent-browser": "node ./scripts/agent-browser-efficiency-benchmark.mjs",
     "test": "tsx --test test/**/*.test.ts",
     "verify": "node ./scripts/project.mjs verify",
-    "prepublishOnly": "npm run verify && npm pack --dry-run"
+    "prepublishOnly": "npm run verify -- release && npm pack --dry-run"
   },
   "packageManager": "npm@11.14.0"
 }