@cydm/pie 1.0.51 → 2.1.1

package/dist/chunks/{chunk-KDPAGHYY.js → chunk-TJ2F4FL5.js}

@@ -1,4 +1,8 @@
 import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
+import {
+  getBuiltinDir,
+  getConfigDir
+} from "./chunk-5353DEEC.js";
 import {
   classifyShellCommand,
   createCodeIntelTool,
@@ -8,129 +12,16 @@ import {
   createSharedWebSearchTool,
   interpretShellExit,
   requestInteraction
-} from "./chunk-A644H2XX.js";
+} from "./chunk-LU7RKALH.js";
 import {
   Type
-} from "./chunk-S2J3SZZX.js";
-
-// src/config.ts
-import { existsSync, mkdirSync, readFileSync, renameSync } from "fs";
-import { homedir } from "os";
-import { dirname, join } from "path";
-import { fileURLToPath } from "url";
-var __filename = fileURLToPath(import.meta.url);
-var __dirname = dirname(__filename);
-var isBunRuntime = !!process.versions.bun;
-function getPackageDir() {
-  const envDir = process.env.PIE_PACKAGE_DIR;
-  if (envDir) {
-    if (envDir === "~") return homedir();
-    if (envDir.startsWith("~/")) return homedir() + envDir.slice(1);
-    return envDir;
-  }
-  let dir = __dirname;
-  while (dir !== dirname(dir)) {
-    if (existsSync(join(dir, "package.json"))) {
-      return dir;
-    }
-    dir = dirname(dir);
-  }
-  return __dirname;
-}
-function getThemesDir() {
-  const packageDir = getPackageDir();
-  const srcOrDist = existsSync(join(packageDir, "src")) ? "src" : "dist";
-  return join(packageDir, srcOrDist, "theme");
-}
-function getBuiltinDir() {
-  const packageDir = getPackageDir();
-  const distBuiltinDir = join(packageDir, "dist", "builtin");
-  if (existsSync(distBuiltinDir)) {
-    return distBuiltinDir;
-  }
-  return join(packageDir, "builtin");
-}
-function getPackageJsonPath() {
-  return join(getPackageDir(), "package.json");
-}
-var pkg = JSON.parse(readFileSync(getPackageJsonPath(), "utf-8"));
-var CONFIG_DIR_NAME = ".pie";
-var VERSION = pkg.version;
-var ENV_AGENT_DIR = "PIE_CODING_AGENT_DIR";
-function getConfigDir() {
-  const envDir = process.env[ENV_AGENT_DIR];
-  if (envDir) {
-    if (envDir === "~") return homedir();
-    if (envDir.startsWith("~/")) return homedir() + envDir.slice(1);
-    return envDir;
-  }
-  return join(homedir(), CONFIG_DIR_NAME);
-}
-function getAgentDir() {
-  return getConfigDir();
-}
-function getCustomThemesDir() {
-  return join(getConfigDir(), "themes");
-}
-function getSettingsPath() {
-  return join(getConfigDir(), "settings.json");
-}
-function getProjectSettingsPath(cwd) {
-  return join(cwd, CONFIG_DIR_NAME, "settings.json");
-}
-function getSessionsDir() {
-  return join(getConfigDir(), "sessions");
-}
-function getLogsDir() {
-  return join(getConfigDir(), "logs");
-}
-function getRuntimeLogPath() {
-  return join(getLogsDir(), "pie-cli.log");
-}
-function migrateConfigFromAgentDir() {
-  const oldAgentDir = join(homedir(), CONFIG_DIR_NAME, "agent");
-  const newConfigDir = getConfigDir();
-  if (!existsSync(oldAgentDir)) {
-    return;
-  }
-  if (!existsSync(newConfigDir)) {
-    mkdirSync(newConfigDir, { recursive: true });
-  }
-  const filesToMigrate = [
-    "keybindings.json",
-    "settings.json",
-    "pie-debug.log"
-  ];
-  for (const file of filesToMigrate) {
-    const oldPath = join(oldAgentDir, file);
-    const newPath = file === "pie-debug.log" ? getRuntimeLogPath() : join(newConfigDir, file);
-    if (existsSync(oldPath) && !existsSync(newPath)) {
-      try {
-        if (file === "pie-debug.log") {
-          mkdirSync(getLogsDir(), { recursive: true });
-        }
-        renameSync(oldPath, newPath);
-        console.log(`[Config] Migrated ${file} to ${newPath}`);
-      } catch (e) {
-      }
-    }
-  }
-  const oldThemesDir = join(oldAgentDir, "themes");
-  const newThemesDir = getCustomThemesDir();
-  if (existsSync(oldThemesDir) && !existsSync(newThemesDir)) {
-    try {
-      renameSync(oldThemesDir, newThemesDir);
-      console.log(`[Config] Migrated themes to ${newThemesDir}`);
-    } catch (e) {
-    }
-  }
-}
+} from "./chunk-XXGYUYXI.js";
 
 // src/capabilities/bash/index.ts
 import { spawn } from "child_process";
-import { closeSync, createWriteStream, existsSync as existsSync2, openSync, statSync, unlinkSync, writeFileSync } from "fs";
+import { closeSync, createWriteStream, existsSync, openSync, statSync, unlinkSync, writeFileSync } from "fs";
 import { tmpdir } from "os";
-import { join as join2, resolve as resolve2 } from "path";
+import { join, resolve } from "path";
 import { createHash, randomBytes } from "crypto";
 
 // src/capabilities/bash/truncate.ts
@@ -273,6 +164,7 @@ var bashSchema = Type.Object({
 });
 var DEFAULT_TIMEOUT_SECONDS = 120;
 var MAX_GENERIC_TIMEOUT_SECONDS = 120;
+var OUTPUT_DRAIN_TIMEOUT_MS = 1e3;
 function resolveTimeoutPolicy(requestedTimeout, commandClass) {
   const requestedSeconds = requestedTimeout ?? DEFAULT_TIMEOUT_SECONDS;
   if (requestedSeconds > MAX_GENERIC_TIMEOUT_SECONDS && commandClass !== "package_manager" && commandClass !== "test_or_build") {
@@ -302,12 +194,15 @@ function validateBashCommand(command) {
     throw new Error("Invalid bash command. The command argument must be only a real shell command, not tool-call markup.");
   }
 }
-function buildEffectiveCommand(command, options) {
+function buildEffectiveCommand(command, options, shell) {
   const prefix = options?.commandPrefix?.trim();
   if (!prefix) {
     return command;
   }
   validateBashCommand(prefix);
+  if (shell === "powershell") {
+    return `& { ${prefix} }; if (-not $?) { exit 1 }; ${command}`;
+  }
   return `${prefix} && ${command}`;
 }
 function resolveShell(shell) {
@@ -318,7 +213,8 @@ function resolveShell(shell) {
 }
 function buildSpawnCommand(shell, command) {
   if (shell === "powershell") {
-    return { file: process.platform === "win32" ? "powershell.exe" : "pwsh", args: ["-NoProfile", "-NonInteractive", "-Command", command], shell: false };
+    const utf8Command = `[Console]::OutputEncoding = New-Object System.Text.UTF8Encoding; $OutputEncoding = [Console]::OutputEncoding; ${command}`;
+    return { file: process.platform === "win32" ? "powershell.exe" : "pwsh", args: ["-NoProfile", "-NonInteractive", "-Command", utf8Command], shell: false };
   }
   if (shell === "cmd") {
     return { file: "cmd.exe", args: ["/d", "/s", "/c", command], shell: false };
@@ -341,7 +237,7 @@ async function requestHighRiskConfirmation(request) {
   return response.type === "confirm" && response.confirmed === true;
 }
 function validateWorkingDirectory(cwd) {
-  if (!existsSync2(cwd)) {
+  if (!existsSync(cwd)) {
     throw new Error(`Invalid bash cwd: ${cwd} does not exist. Use the current workspace path or pass an existing directory in cwd.`);
   }
   let stat;
@@ -356,11 +252,11 @@ function validateWorkingDirectory(cwd) {
 }
 function normalizeWorkingDirectory(cwd, defaultCwd) {
   const requestedCwd = typeof cwd === "string" && cwd.trim().length > 0 ? cwd.trim() : void 0;
-  return resolve2(requestedCwd ?? defaultCwd ?? process.cwd());
+  return resolve(requestedCwd ?? defaultCwd ?? process.cwd());
 }
 function getTempFilePath() {
   const id = randomBytes(8).toString("hex");
-  return join2(tmpdir(), `pie-bash-${id}.log`);
+  return join(tmpdir(), `pie-bash-${id}.log`);
 }
 var retainedOutputArtifacts = /* @__PURE__ */ new Set();
 function getErrorCode(error) {
@@ -370,15 +266,15 @@ function getErrorCode(error) {
 }
 function createPrivateOutputArtifact() {
   for (let attempt = 0; attempt < 10; attempt++) {
-    const path3 = getTempFilePath();
+    const path4 = getTempFilePath();
     try {
-      const fd = openSync(path3, "wx", 384);
-      retainedOutputArtifacts.add(path3);
+      const fd = openSync(path4, "wx", 384);
+      retainedOutputArtifacts.add(path4);
       try {
-        return { path: path3, stream: createWriteStream(path3, { fd, autoClose: true }) };
+        return { path: path4, stream: createWriteStream(path4, { fd, autoClose: true }) };
       } catch (error) {
         closeSync(fd);
-        removeOutputArtifact(path3);
+        removeOutputArtifact(path4);
         throw error;
       }
     } catch (error) {
@@ -387,34 +283,34 @@ function createPrivateOutputArtifact() {
   }
   throw new Error("Unable to allocate a private bash output artifact.");
 }
-function removeOutputArtifact(path3) {
-  if (!path3) return;
+function removeOutputArtifact(path4) {
+  if (!path4) return;
   try {
-    unlinkSync(path3);
+    unlinkSync(path4);
   } catch (error) {
     if (getErrorCode(error) !== "ENOENT") throw error;
   }
-  retainedOutputArtifacts.delete(path3);
+  retainedOutputArtifacts.delete(path4);
 }
-function discardOutputArtifact(path3, stream) {
+function discardOutputArtifact(path4, stream) {
   if (stream && !stream.closed) {
     stream.once("close", () => {
       try {
-        removeOutputArtifact(path3);
+        removeOutputArtifact(path4);
       } catch {
       }
     });
     stream.destroy();
   }
   try {
-    removeOutputArtifact(path3);
+    removeOutputArtifact(path4);
   } catch {
   }
 }
 function cleanupBashOutputArtifacts() {
-  for (const path3 of [...retainedOutputArtifacts]) {
+  for (const path4 of [...retainedOutputArtifacts]) {
     try {
-      removeOutputArtifact(path3);
+      removeOutputArtifact(path4);
     } catch {
     }
   }
@@ -422,9 +318,13 @@ function cleanupBashOutputArtifacts() {
 process.once("exit", cleanupBashOutputArtifacts);
 function finishTempFileStream(stream) {
   if (!stream || stream.closed) return Promise.resolve();
-  return new Promise((resolve3, reject) => {
+  return new Promise((resolve2, reject) => {
     let done = false;
+    const timer = setTimeout(() => {
+      stream.destroy(new Error("Timed out flushing shell output."));
+    }, OUTPUT_DRAIN_TIMEOUT_MS);
     const cleanup = () => {
+      clearTimeout(timer);
       stream.off("close", onClose);
       stream.off("error", onError);
     };
@@ -432,7 +332,7 @@ function finishTempFileStream(stream) {
       if (done) return;
       done = true;
       cleanup();
-      resolve3();
+      resolve2();
     };
     const onError = (error) => {
       if (done) return;
@@ -457,7 +357,7 @@ function buildBashRecoveryHint(command, output) {
   if (/\bnpm ERR!|\bnpm error\b/i.test(output) && /\b(?:ECONNRESET|ETIMEDOUT|ENOTFOUND|EAI_AGAIN|fetch failed|socket hang up|registry|proxy)\b/i.test(output)) {
     return "npm could not reach the registry. Treat this as an environment/network/proxy issue, not a source-code failure; retry npm install/npm ci when network is available or check npm proxy/registry settings.";
   }
-  if (/\bnpm\s+run\b/.test(command) && (/(?:^|\n)\s*(?:sh|bash|zsh):\s+[\w@./-]+:\s+command not found\b/i.test(output) || /(?:^|\r?\n)\s*'[^']+'\s+is not recognized as an internal or external command\b/i.test(output))) {
+  if (/\bnpm\s+run\b/.test(command) && (/(?:^|\n)\s*(?:sh|bash|zsh):\s+[\w@./-]+:\s+command not found\b/i.test(output) || /(?:^|\r?\n)\s*'[^']+'\s+is not recognized as an internal or external command\b/i.test(output) || /\u4e0d\u662f\u5185\u90e8\u6216\u5916\u90e8\u547d\u4ee4/.test(output))) {
     return "This npm script could not find a local executable. Check whether dependencies were installed successfully in cwd, whether node_modules exists, and whether a prior npm install/npm ci failed before treating this as a code failure.";
   }
   return void 0;
@@ -506,8 +406,9 @@ function killProcessTree(pid) {
     try {
       spawn("taskkill", ["/F", "/T", "/PID", String(pid)], {
         stdio: "ignore",
-        detached: true
-      });
+        windowsHide: true
+      }).on("error", () => {
+      }).unref();
     } catch {
     }
   } else {
@@ -542,7 +443,7 @@ function createBashTool(defaultCwd, options) {
     execute: async (_toolCallId, { command, timeout, cwd, shell }, signal, onUpdate) => {
       validateBashCommand(command);
       const resolvedShell = resolveShell(shell);
-      const effectiveCommand = buildEffectiveCommand(command, options);
+      const effectiveCommand = buildEffectiveCommand(command, options, resolvedShell);
       const workingDir = normalizeWorkingDirectory(cwd, defaultCwd);
       validateWorkingDirectory(workingDir);
       const risk = classifyShellCommand(effectiveCommand, resolvedShell, {
@@ -581,7 +482,7 @@ function createBashTool(defaultCwd, options) {
       }
       const timeoutPolicy = resolveTimeoutPolicy(timeout, risk.commandClass);
       const timeoutMs = timeoutPolicy.effectiveSeconds * 1e3;
-      return new Promise((resolve3, reject) => {
+      return new Promise((resolve2, reject) => {
         if (signal?.aborted) {
           reject(new Error("Operation aborted"));
           return;
@@ -605,25 +506,42 @@ function createBashTool(defaultCwd, options) {
           headBytes += prefix.length;
         };
         let timedOut = false;
+        let processExited = false;
+        let exitCode = null;
+        let exitSignal = null;
+        let drainTimer;
         const spawnCommand = buildSpawnCommand(resolvedShell, effectiveCommand);
         const child = (options?.spawnProcess ?? spawn)(spawnCommand.file, spawnCommand.args, {
           shell: spawnCommand.shell,
           cwd: workingDir,
           env: process.env,
-          detached: process.platform !== "win32"
+          detached: process.platform !== "win32",
+          windowsHide: true
         });
+        const closePipes = () => {
+          child.stdin?.destroy();
+          child.stdout?.destroy();
+          child.stderr?.destroy();
+        };
         const timeoutId = setTimeout(() => {
+          if (settled) return;
           timedOut = true;
-          if (child.pid) {
+          if (child.pid && !processExited) {
             killProcessTree(child.pid);
           }
+          closePipes();
+          child.unref?.();
+          void finishExecution(exitCode, exitSignal, true);
         }, timeoutMs);
         const failExecution = (error) => {
           if (settled) return;
           settled = true;
           clearTimeout(timeoutId);
+          if (drainTimer) clearTimeout(drainTimer);
           signal?.removeEventListener("abort", onAbort);
-          if (child.pid) killProcessTree(child.pid);
+          if (child.pid && !processExited) killProcessTree(child.pid);
+          closePipes();
+          child.unref?.();
           discardOutputArtifact(tempFilePath, tempFileStream);
           reject(error);
         };
@@ -715,11 +633,13 @@ function createBashTool(defaultCwd, options) {
             chunksBytes -= removed.length;
           }
         });
-        child.on("close", async (code, sig) => {
+        const finishExecution = async (code, sig, stdioIncomplete = false) => {
+          if (settled) return;
+          settled = true;
           try {
             clearTimeout(timeoutId);
+            if (drainTimer) clearTimeout(drainTimer);
             signal?.removeEventListener("abort", onAbort);
-            if (settled) return;
             const fullBuffer = Buffer.concat(chunks);
             let fullOutput = fullBuffer.toString("utf-8");
             if (totalBytes > fullBuffer.length) {
@@ -736,10 +656,10 @@ ${fullOutput}`;
             }
             let outputText = truncation.content || "(no output)";
             if (truncation.truncated && !tempFilePath) {
-              const path3 = getTempFilePath();
-              const fd = openSync(path3, "wx", 384);
-              tempFilePath = path3;
-              retainedOutputArtifacts.add(path3);
+              const path4 = getTempFilePath();
+              const fd = openSync(path4, "wx", 384);
+              tempFilePath = path4;
+              retainedOutputArtifacts.add(path4);
               try {
                 writeFileSync(fd, fullOutput, "utf-8");
               } finally {
@@ -751,6 +671,7 @@ ${fullOutput}`;
               exitCode: code,
               signal: sig?.toString() ?? null,
               timedOut,
+              ...stdioIncomplete ? { stdioIncomplete: true } : {},
               shell: resolvedShell,
               commandClass: risk.commandClass,
               riskWarnings: risk.riskWarnings,
@@ -806,6 +727,9 @@ ${commandNonzeroHint}`;
 [Command timed out after ${timeoutPolicy.effectiveSeconds} seconds]`;
               outputText += "\n[Timeout hint: split custom searches or service polls into smaller bounded steps with progress output, inspect partial results/logs, then make a decision. Avoid repeating long waits. Reserve long timeouts for installs, builds, and tests.]";
             }
+            if (stdioIncomplete && !timedOut) {
+              outputText += "\n\n[Output capture ended after the command exited because a descendant kept stdio open. Captured output may be incomplete.]";
+            }
             if (timeoutPolicy.capped) {
               outputText += `
 
@@ -826,13 +750,26 @@ ${commandNonzeroHint}`;
               outputText += "\n\n[Repeat hint: this exact command already returned the same result in this session. If it proves the task, stop; otherwise change the command or inspect a different file/range.]";
             }
             const content = [{ type: "text", text: outputText }];
-            settled = true;
-            resolve3({ content, details, isError: outcome.toolFailure });
+            resolve2({ content, details, isError: outcome.toolFailure });
           } catch (error) {
             discardOutputArtifact(tempFilePath, tempFileStream);
-            settled = true;
             reject(error);
           }
+        };
+        child.on("exit", (code, sig) => {
+          processExited = true;
+          exitCode = code;
+          exitSignal = sig;
+          if (settled) return;
+          drainTimer = setTimeout(() => {
+            closePipes();
+            child.unref?.();
+            void finishExecution(code, sig, true);
+          }, OUTPUT_DRAIN_TIMEOUT_MS);
+        });
+        child.on("close", (code, sig) => {
+          processExited = true;
+          void finishExecution(code, sig);
         });
         child.on("error", (err) => {
           failExecution(new Error(`Failed to start bash command in ${workingDir}: ${err.message}. Check that the command is executable and cwd exists.`));
@@ -845,7 +782,7 @@ ${commandNonzeroHint}`;
 // src/capabilities/code-intel.ts
 import * as fs from "node:fs";
 import * as path from "node:path";
-import { fileURLToPath as fileURLToPath2, pathToFileURL } from "node:url";
+import { fileURLToPath, pathToFileURL } from "node:url";
 import { Worker } from "node:worker_threads";
 var WORKER_TIMEOUT_MS = 3e4;
 var MAX_ACTIVE_CODE_INTEL_WORKERS = 1;
@@ -873,8 +810,8 @@ function acquireWorkerSlot(deadline, signal) {
   if (codeIntelWorkerWaiters.length >= MAX_QUEUED_CODE_INTEL_WORKERS) {
     return Promise.reject(new Error("TypeScript code intelligence queue is full"));
   }
-  return new Promise((resolve3, reject) => {
-    const waiter = { resolve: () => resolve3(releaseWorkerSlot), reject, signal };
+  return new Promise((resolve2, reject) => {
+    const waiter = { resolve: () => resolve2(releaseWorkerSlot), reject, signal };
     waiter.onAbort = () => {
       const index = codeIntelWorkerWaiters.indexOf(waiter);
       if (index >= 0) codeIntelWorkerWaiters.splice(index, 1);
@@ -893,7 +830,7 @@ function acquireWorkerSlot(deadline, signal) {
   });
 }
 function resolveWorkerPath() {
-  const moduleDirectory = path.dirname(fileURLToPath2(import.meta.url));
+  const moduleDirectory = path.dirname(fileURLToPath(import.meta.url));
   const candidates = [
     path.join(moduleDirectory, "code-intel-worker.cjs"),
     path.join(moduleDirectory, "..", "code-intel-worker.cjs"),
@@ -904,7 +841,7 @@ function resolveWorkerPath() {
   return workerPath;
 }
 function runWorker(request, timeoutMs, signal) {
-  return new Promise((resolve3, reject) => {
+  return new Promise((resolve2, reject) => {
     if (signal?.aborted) {
       reject(new Error("Operation aborted"));
       return;
@@ -929,7 +866,7 @@ function runWorker(request, timeoutMs, signal) {
           }
         }
         if (error) reject(error);
-        else resolve3(result);
+        else resolve2(result);
       })();
     };
     const onAbort = () => finish(new Error("Operation aborted"));
@@ -1017,21 +954,33 @@ function createCodeIntelTool2(options) {
 // src/capabilities/rendered-page-reader.ts
 import { spawn as spawn2 } from "node:child_process";
 import { createRequire } from "node:module";
+import * as fs2 from "node:fs";
+import * as os from "node:os";
+import * as path2 from "node:path";
 var require2 = createRequire(import.meta.url);
-function resolvePlaywrightCli() {
-  return require2.resolve("@playwright/cli/playwright-cli.js");
+function resolveAxiCli() {
+  const runtime = path2.join(getBuiltinDir(), "skills", "chrome-devtools-axi", "chrome-devtools-axi-runtime.js");
+  if (!fs2.existsSync(runtime)) throw new Error("Pie's packaged browser runtime is missing. Rebuild or reinstall Pie.");
+  return runtime;
+}
+function resolveMcpScript() {
+  const packageJson = require2.resolve("chrome-devtools-mcp/package.json");
+  return path2.join(path2.dirname(packageJson), "build", "src", "bin", "chrome-devtools-mcp.js");
 }
-function runPlaywright(args, timeoutMs = 6e4) {
-  return new Promise((resolve3, reject) => {
-    const child = spawn2(process.execPath, [resolvePlaywrightCli(), ...args], {
+function runAxi(args, options = {}) {
+  const timeoutMs = options.timeoutMs ?? 6e4;
+  return new Promise((resolve2, reject) => {
+    const child = spawn2(process.execPath, [resolveAxiCli(), ...args], {
       stdio: ["ignore", "pipe", "pipe"],
-      shell: false
+      shell: false,
+      env: options.env,
+      cwd: options.cwd
     });
     let stdout = "";
     let stderr = "";
     const timer = setTimeout(() => {
       child.kill("SIGTERM");
-      reject(new Error(`Playwright CLI timed out after ${Math.ceil(timeoutMs / 1e3)} seconds`));
+      reject(new Error(`chrome-devtools-axi timed out after ${Math.ceil(timeoutMs / 1e3)} seconds`));
     }, timeoutMs);
     child.stdout.setEncoding("utf8");
     child.stderr.setEncoding("utf8");
@@ -1047,24 +996,40 @@ function runPlaywright(args, timeoutMs = 6e4) {
     });
     child.on("close", (code) => {
       clearTimeout(timer);
-      if (code === 0) resolve3(stdout.trim());
-      else reject(new Error((stderr || stdout || `Playwright CLI exited with ${code}`).trim()));
+      if (code === 0) resolve2(stdout.trim());
+      else reject(new Error((stderr || stdout || `chrome-devtools-axi exited with ${code}`).trim()));
     });
   });
 }
 function linesOf(output, max = 50) {
-  return output.split(/\r?\n/).map((line) => line.trim()).filter(Boolean).slice(0, max);
+  if (/^(console|network)_failed:/i.test(output.trim())) return [];
+  return output.split(/\r?\n/).map((line) => line.trim()).filter(Boolean).filter((line) => !/^(console:|network:|##\s|Showing \d|\(\d+ args\)$|<no .+ found>$|no \w+ found\.$)/i.test(line)).slice(0, max);
 }
 function parseScreenshotPath(output) {
-  const match = output.match(/\(([^)]+\.png)\)/i) || output.match(/([^\s]+\.png)/i);
+  if (output.startsWith("screenshot_failed:")) return void 0;
+  const match = output.match(/(?:^|\n)screenshot:\s*(\S+)/i) || output.match(/([^\s]+\.(?:png|jpe?g|webp))/i);
   return match?.[1];
 }
+function unwrapJsonValue(payload) {
+  let value = payload;
+  for (let depth = 0; depth < 4; depth += 1) {
+    if (typeof value !== "string") break;
+    const text = value.trim();
+    if (!text.startsWith('"') && !text.startsWith("{") && !text.startsWith("[")) break;
+    try {
+      value = JSON.parse(text);
+    } catch {
+      break;
+    }
+  }
+  return value;
+}
 function parsePageInfo(raw) {
   const trimmed = raw.trim();
   if (!trimmed) return {};
+  const payload = trimmed.replace(/^result:\s*/i, "");
   try {
-    const decoded = trimmed.startsWith('"') ? JSON.parse(trimmed) : trimmed;
-    const parsed = typeof decoded === "string" ? JSON.parse(decoded) : decoded;
+    const parsed = unwrapJsonValue(payload);
     if (!parsed || typeof parsed !== "object") return {};
     return {
       finalUrl: typeof parsed.href === "string" ? parsed.href : void 0,
@@ -1075,23 +1040,53 @@ function parsePageInfo(raw) {
     return {};
   }
 }
+function stripHelpHints(output) {
+  return output.replace(/(?:^|\n)help\[\d+\]:[\s\S]*$/i, "").trim();
+}
+function parseSnapshot(output) {
+  const match = output.match(/(?:^|\n)snapshot:\n([\s\S]*)$/i);
+  return stripHelpHints(match ? match[1] : output);
+}
+function cleanStaleScreenshotDirs(retentionMs = 60 * 60 * 1e3) {
+  try {
+    const tmp = os.tmpdir();
+    const cutoff = Date.now() - retentionMs;
+    for (const entry of fs2.readdirSync(tmp)) {
+      if (!entry.startsWith("pie-axi-shot-")) continue;
+      const dir = path2.join(tmp, entry);
+      try {
+        if (fs2.statSync(dir).mtimeMs < cutoff) fs2.rmSync(dir, { recursive: true, force: true });
+      } catch {
+      }
+    }
+  } catch {
+  }
+}
 function createCliRenderedPageReader() {
   return async ({ url }) => {
     const session = `pie-web-research-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
+    const env = { ...process.env, CHROME_DEVTOOLS_AXI_SESSION: session };
+    delete env.CHROME_DEVTOOLS_AXI_PORT;
+    if (!env.CHROME_DEVTOOLS_AXI_MCP_PATH?.trim()) env.CHROME_DEVTOOLS_AXI_MCP_PATH = resolveMcpScript();
+    if (!env.CHROME_DEVTOOLS_MCP_NO_UPDATE_CHECKS) env.CHROME_DEVTOOLS_MCP_NO_UPDATE_CHECKS = "1";
+    cleanStaleScreenshotDirs();
+    const screenshotDir = fs2.mkdtempSync(path2.join(os.tmpdir(), "pie-axi-shot-"));
     try {
-      await runPlaywright([`-s=${session}`, "open", "about:blank"]);
-      await runPlaywright([`-s=${session}`, "goto", url]);
-      const [snapshot, pageInfoRaw, screenshotOutput, consoleOutput, requestsOutput] = await Promise.all([
-        runPlaywright([`-s=${session}`, "snapshot"]),
-        runPlaywright([`-s=${session}`, "--raw", "eval", "() => JSON.stringify({ title: document.title, href: location.href, text: document.body ? document.body.innerText : '' })"]),
-        runPlaywright([`-s=${session}`, "--raw", "screenshot"]).catch((error) => `screenshot_failed: ${error instanceof Error ? error.message : String(error)}`),
-        runPlaywright([`-s=${session}`, "--raw", "console"]).catch((error) => `console_failed: ${error instanceof Error ? error.message : String(error)}`),
-        runPlaywright([`-s=${session}`, "--raw", "requests"]).catch((error) => `requests_failed: ${error instanceof Error ? error.message : String(error)}`)
+      await runAxi(["open", `data:text/html,<title>pie-warm-${session}</title>`], { env });
+      await runAxi(["open", url], { env });
+      const [snapshotOutput, pageInfoRaw, screenshotOutput, consoleOutput, networkOutput] = await Promise.all([
+        runAxi(["snapshot"], { env }),
+        // --full: eval output is truncated at ~8KB by default, which would
+        // break JSON parsing and drop title/finalUrl/text on real pages.
+        runAxi(["eval", "() => JSON.stringify({ title: document.title, href: location.href, text: document.body ? document.body.innerText : '' })", "--full"], { env }),
+        runAxi(["screenshot", path2.join(screenshotDir, "page.png")], { env }).catch((error) => `screenshot_failed: ${error instanceof Error ? error.message : String(error)}`),
+        runAxi(["console"], { env }).catch((error) => `console_failed: ${error instanceof Error ? error.message : String(error)}`),
+        runAxi(["network"], { env }).catch((error) => `network_failed: ${error instanceof Error ? error.message : String(error)}`)
       ]);
-      const pageInfo = parsePageInfo(pageInfoRaw);
-      const text = pageInfo.text?.trim() || snapshot.trim();
+      const pageInfo = parsePageInfo(stripHelpHints(pageInfoRaw));
+      const text = pageInfo.text?.trim() || parseSnapshot(snapshotOutput);
       if (!text) {
-        throw new Error("Playwright snapshot returned no text.");
+        throw new Error("chrome-devtools-axi snapshot returned no text.");
       }
       return {
         url,
@@ -1100,18 +1095,18 @@ function createCliRenderedPageReader() {
         text,
         citationAnchors: [{ url: pageInfo.finalUrl || url, title: pageInfo.title, startOffset: 0, endOffset: Math.min(120, text.length) }],
         screenshotPath: parseScreenshotPath(screenshotOutput),
-        consoleMessages: linesOf(consoleOutput),
-        networkRequests: linesOf(requestsOutput)
+        consoleMessages: linesOf(stripHelpHints(consoleOutput)),
+        networkRequests: linesOf(stripHelpHints(networkOutput))
       };
     } finally {
-      runPlaywright([`-s=${session}`, "close"], 1e4).catch(() => void 0);
+      runAxi(["stop"], { env, timeoutMs: 1e4 }).catch(() => void 0);
     }
   };
 }
 
 // src/capabilities/web-research-health.ts
-import fs2 from "node:fs";
-import path2 from "node:path";
+import fs3 from "node:fs";
+import path3 from "node:path";
 function isRecord(value) {
   return !!value && typeof value === "object" && !Array.isArray(value);
 }
@@ -1133,11 +1128,11 @@ function parseHealthEntry(value) {
   };
 }
 function createCliWebResearchHealthStore(configDir) {
-  const filePath = path2.join(configDir, "web-research-health.json");
+  const filePath = path3.join(configDir, "web-research-health.json");
   return {
     loadProviderHealth: () => {
       try {
-        const parsed = JSON.parse(fs2.readFileSync(filePath, "utf8"));
+        const parsed = JSON.parse(fs3.readFileSync(filePath, "utf8"));
         if (!Array.isArray(parsed)) return [];
         return parsed.map(parseHealthEntry).filter((entry) => !!entry);
       } catch {
@@ -1146,8 +1141,8 @@ function createCliWebResearchHealthStore(configDir) {
     },
     saveProviderHealth: (entries) => {
       try {
-        fs2.mkdirSync(path2.dirname(filePath), { recursive: true });
-        fs2.writeFileSync(filePath, JSON.stringify(entries, null, 2));
+        fs3.mkdirSync(path3.dirname(filePath), { recursive: true });
+        fs3.writeFileSync(filePath, JSON.stringify(entries, null, 2));
       } catch {
       }
     }
@@ -1213,22 +1208,10 @@ function createCliHostCapabilities(cwd, options = {}) {
 }
 
 export {
-  getPackageDir,
-  getThemesDir,
-  getBuiltinDir,
-  VERSION,
-  getConfigDir,
-  getAgentDir,
-  getCustomThemesDir,
-  getSettingsPath,
-  getProjectSettingsPath,
-  getSessionsDir,
-  getLogsDir,
-  getRuntimeLogPath,
-  migrateConfigFromAgentDir,
   createCliWebResearchHealthStore,
   truncateHeadTail,
   formatSize,
   cleanupBashOutputArtifacts,
+  createBashTool,
   createCliHostCapabilities
 };