yiyan-browser-agent 1.1.3 → 1.3.0

package/dist/index.js

@@ -1,8 +1,14 @@
 #!/usr/bin/env node
+var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
+  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
+}) : x)(function(x) {
+  if (typeof require !== "undefined") return require.apply(this, arguments);
+  throw Error('Dynamic require of "' + x + '" is not supported');
+});
 
 // src/agent.ts
-import path3 from "path";
-import os2 from "os";
+import path from "path";
+import os from "os";
 
 // src/types.ts
 var YiyanAgentError = class extends Error {
@@ -21,214 +27,74 @@ var DEFAULT_OPTIONS = {
 };
 var YIYAN_CHAT_URL = "https://yiyan.baidu.com/";
 
-// src/profile.ts
-import path from "path";
-import fs from "fs";
-import fsp from "fs/promises";
-function getChromeExecutablePath(platform) {
-  switch (platform) {
-    case "win32":
-      const programFiles = process.env.PROGRAMFILES || "C:\\Program Files";
-      return path.join(programFiles, "Google", "Chrome", "Application", "chrome.exe");
-    case "darwin":
-      return "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome";
-    case "linux":
-      return "/usr/bin/google-chrome";
-    default:
-      return null;
-  }
-}
-var PROFILE_TARGET_NAME = "chrome-profile";
-async function clearCopiedProfile(profileDir) {
-  const targetPath = path.join(profileDir, PROFILE_TARGET_NAME);
-  if (fs.existsSync(targetPath)) {
-    await fsp.rm(targetPath, { recursive: true, force: true });
-  }
-}
-function profileExists(profileDir) {
-  const targetPath = path.join(profileDir, PROFILE_TARGET_NAME);
-  return fs.existsSync(targetPath);
-}
-
 // src/browser.ts
-import { chromium } from "playwright-core";
-import http from "http";
-import { spawn, execSync } from "child_process";
-import path2 from "path";
-import os from "os";
-import fs2 from "fs";
-var CDP_PORT = 19222;
-var DEBUG_DIR = path2.join(os.homedir(), ".yiyan-browser-agent", "debug");
+import { chromium } from "playwright";
+import fs from "fs";
 function log(verbose, msg) {
   if (verbose) {
     process.stderr.write(`[yiyan-agent] ${msg}
 `);
   }
 }
-function ensureDebugDir() {
-  if (!fs2.existsSync(DEBUG_DIR)) {
-    fs2.mkdirSync(DEBUG_DIR, { recursive: true });
-  }
-}
-function killProcessOnPort(port) {
-  try {
-    if (process.platform === "win32") {
-      const output = execSync(
-        `netstat -ano | findstr :${port} | findstr LISTENING`,
-        { encoding: "utf-8", timeout: 5e3 }
-      ).trim();
-      if (output) {
-        const lines = output.split("\n").filter((l) => l.trim());
-        const pids = /* @__PURE__ */ new Set();
-        for (const line of lines) {
-          const parts = line.trim().split(/\s+/);
-          const pid = parts[parts.length - 1];
-          if (pid && /^\d+$/.test(pid)) pids.add(pid);
-        }
-        for (const pid of pids) {
-          try {
-            execSync(`taskkill /F /T /PID ${pid}`, { timeout: 5e3 });
-          } catch {
-          }
-        }
-      }
-    } else {
-      const output = execSync(
-        `lsof -ti :${port}`,
-        { encoding: "utf-8", timeout: 5e3 }
-      ).trim();
-      if (output) {
-        const pids = output.split("\n").filter((l) => l.trim());
-        for (const pid of pids) {
-          try {
-            process.kill(parseInt(pid, 10), "SIGKILL");
-          } catch {
-          }
-        }
-      }
-    }
-  } catch {
-  }
-}
-function killProcessTree(pid) {
-  try {
-    if (process.platform === "win32") {
-      execSync(`taskkill /F /T /PID ${pid}`, { timeout: 5e3 });
-    } else {
-      try {
-        process.kill(-pid, "SIGKILL");
-      } catch {
-        process.kill(pid, "SIGKILL");
-      }
-    }
-  } catch {
-  }
-}
-function waitForCDP(port, timeout = 15e3) {
-  return new Promise((resolve, reject) => {
-    const start = Date.now();
-    const check = () => {
-      http.get(`http://localhost:${port}/json/version`, (res) => {
-        let data = "";
-        res.on("data", (chunk) => data += chunk);
-        res.on("end", () => {
-          try {
-            resolve(JSON.parse(data));
-          } catch (e) {
-            reject(e);
-          }
-        });
-      }).on("error", () => {
-        if (Date.now() - start > timeout) reject(new Error("CDP connection timeout"));
-        else setTimeout(check, 500);
-      });
-    };
-    check();
-  });
-}
 async function launchBrowser(options) {
-  const { chromePath, profilePath, headless, timeout = 3e4, verbose = false } = options;
+  const { sessionDir, headless, timeout = 3e4, verbose = false } = options;
   try {
-    log(verbose, "\u6E05\u7406\u6B8B\u7559 Chrome \u8FDB\u7A0B...");
-    killProcessOnPort(CDP_PORT);
-    await new Promise((resolve) => setTimeout(resolve, 500));
-    const chromeArgs = [
-      `--remote-debugging-port=${CDP_PORT}`,
-      `--user-data-dir=${profilePath}`,
-      "--no-first-run",
-      "--no-default-browser-check",
-      "--disable-blink-features=AutomationControlled"
-    ];
-    if (headless) {
-      chromeArgs.push("--headless=new");
+    log(verbose, "\u542F\u52A8 Chromium \u6D4F\u89C8\u5668...");
+    if (!fs.existsSync(sessionDir)) {
+      fs.mkdirSync(sessionDir, { recursive: true });
     }
-    log(verbose, `\u542F\u52A8 Chrome: ${chromePath} ${headless ? "(headless)" : "(headed)"}`);
-    const chromeProcess = spawn(chromePath, chromeArgs, {
-      detached: true,
-      stdio: "ignore"
-    });
-    chromeProcess.unref();
-    const spawnError = new Promise((_, reject) => {
-      chromeProcess.on("error", (err) => {
-        reject(new YiyanAgentError("BROWSER_LAUNCH", `Failed to spawn Chrome: ${err.message}`));
-      });
+    const context = await chromium.launchPersistentContext(sessionDir, {
+      headless,
+      viewport: { width: 1280, height: 900 },
+      userAgent: [
+        "Mozilla/5.0 (X11; Linux x86_64)",
+        "AppleWebKit/537.36 (KHTML, like Gecko)",
+        "Chrome/124.0.0.0 Safari/537.36"
+      ].join(" "),
+      args: [
+        "--disable-blink-features=AutomationControlled",
+        "--no-first-run",
+        "--no-default-browser-check",
+        "--no-sandbox",
+        "--disable-setuid-sandbox",
+        "--disable-dev-shm-usage"
+      ],
+      ignoreDefaultArgs: ["--enable-automation"]
     });
-    log(verbose, "\u7B49\u5F85 CDP \u7AEF\u53E3\u5C31\u7EEA...");
-    await Promise.race([
-      waitForCDP(CDP_PORT, timeout),
-      spawnError
-    ]);
-    log(verbose, "\u901A\u8FC7 CDP \u8FDE\u63A5\u6D4F\u89C8\u5668...");
-    const cdpBrowser = await chromium.connectOverCDP(`http://localhost:${CDP_PORT}`);
-    const contexts = cdpBrowser.contexts();
-    const context = contexts.length > 0 ? contexts[0] : await cdpBrowser.newContext();
     context.setDefaultTimeout(timeout);
     const pages = context.pages();
     const page = pages.length > 0 ? pages[0] : await context.newPage();
+    await page.addInitScript(() => {
+      Object.defineProperty(navigator, "webdriver", { get: () => false });
+    });
     log(verbose, "\u6D4F\u89C8\u5668\u542F\u52A8\u5B8C\u6210");
-    return { browser: context, cdpBrowser, page, chromeProcess };
+    return { context, page };
   } catch (error) {
-    killProcessOnPort(CDP_PORT);
     throw new YiyanAgentError(
       "BROWSER_LAUNCH",
       `Failed to launch browser: ${error instanceof Error ? error.message : String(error)}`
     );
   }
 }
-async function closeBrowser(browser, chromeProcess, cdpBrowser, verbose) {
+async function closeBrowser(context, verbose) {
   log(!!verbose, "\u5173\u95ED\u6D4F\u89C8\u5668...");
-  if (cdpBrowser) {
-    try {
-      await cdpBrowser.close();
-    } catch {
-    }
-  }
   try {
-    const pages = browser.pages();
-    for (const page of pages) {
-      await page.close().catch(() => {
-      });
-    }
+    await context.close();
+    log(!!verbose, "\u6D4F\u89C8\u5668\u5DF2\u5173\u95ED");
   } catch {
   }
-  if (chromeProcess && chromeProcess.pid) {
-    killProcessTree(chromeProcess.pid);
-  }
-  killProcessOnPort(CDP_PORT);
-  log(!!verbose, "\u6D4F\u89C8\u5668\u5DF2\u5173\u95ED");
 }
 async function navigateToYiyan(page, verbose = false) {
   log(verbose, "\u5BFC\u822A\u5230\u6587\u5FC3\u4E00\u8A00\u804A\u5929\u9875\u9762...");
-  await page.goto(YIYAN_CHAT_URL, {
-    waitUntil: "networkidle",
-    timeout: 6e4
-  });
-  await page.waitForTimeout(5e3);
   try {
-    await page.waitForSelector('[contenteditable="true"]', { timeout: 15e3 });
-    log(verbose, "\u9875\u9762\u52A0\u8F7D\u5B8C\u6210\uFF0C\u8F93\u5165\u6846\u5DF2\u5C31\u7EEA");
-  } catch {
-    log(verbose, "\u8B66\u544A\uFF1A\u672A\u68C0\u6D4B\u5230\u8F93\u5165\u6846\uFF0C\u7EE7\u7EED\u6267\u884C");
+    await page.goto(YIYAN_CHAT_URL, {
+      waitUntil: "domcontentloaded",
+      timeout: 3e4
+    });
+    await page.waitForTimeout(1500);
+    log(verbose, "\u9875\u9762\u52A0\u8F7D\u5B8C\u6210");
+  } catch (err) {
+    log(verbose, `\u5BFC\u822A\u8B66\u544A: ${err instanceof Error ? err.message : String(err)}`);
   }
 }
 async function checkCaptcha(page, verbose = false) {
@@ -284,7 +150,7 @@ async function waitForUserAction(page, reason, verbose = false) {
   const maxWait = 18e4;
   const startTime = Date.now();
   while (Date.now() - startTime < maxWait) {
-    await new Promise((resolve) => setTimeout(resolve, 2e3));
+    await page.waitForTimeout(2e3);
     if (reason === "captcha") {
       const stillHasCaptcha = await checkCaptcha(page, false);
       if (!stillHasCaptcha) {
@@ -313,24 +179,38 @@ async function waitForUserAction(page, reason, verbose = false) {
   log(verbose, "\u26A0 \u7B49\u5F85\u7528\u6237\u64CD\u4F5C\u8D85\u65F6\uFF083\u5206\u949F\uFF09");
 }
 async function sendMessage(page, message, verbose = false, headful = false) {
-  const inputSelector = '[contenteditable="true"]';
+  const inputSelectors = [
+    '[contenteditable="true"][role="textbox"]',
+    '[contenteditable="true"]',
+    "textarea[placeholder]",
+    "textarea"
+  ];
   log(verbose, "\u7B49\u5F85\u8F93\u5165\u6846\u51FA\u73B0...");
-  const inputElement = await page.waitForSelector(inputSelector, { timeout: 15e3 });
+  let inputElement = null;
+  for (const sel of inputSelectors) {
+    try {
+      inputElement = await page.waitForSelector(sel, { timeout: 4e3, state: "visible" });
+      if (inputElement) break;
+    } catch {
+    }
+  }
   if (!inputElement) {
-    log(verbose, "\u26A0 \u8F93\u5165\u6846\u672A\u627E\u5230\uFF01");
-    throw new YiyanAgentError("NETWORK", "Input element not found on page");
+    throw new YiyanAgentError("NETWORK", "Cannot find the Yiyan chat input box");
   }
   log(verbose, "\u2713 \u8F93\u5165\u6846\u5DF2\u627E\u5230");
-  await inputElement.click();
-  await page.waitForTimeout(500);
+  await inputElement.click({ force: true });
+  await page.waitForTimeout(200);
+  await page.keyboard.press("Control+a");
+  await page.waitForTimeout(100);
+  await page.evaluate((element, content) => {
+    element.focus();
+    document.execCommand("selectAll", false, null);
+    document.execCommand("delete", false, null);
+    document.execCommand("insertText", false, content);
+    element.dispatchEvent(new InputEvent("input", { bubbles: true, data: content }));
+  }, inputElement, message);
   log(verbose, `\u8F93\u5165\u95EE\u9898: "${message}"`);
-  await page.keyboard.type(message, { delay: 30 });
-  const filledValue = await inputElement.innerText();
-  if (filledValue.includes(message)) {
-    log(verbose, "\u2713 \u95EE\u9898\u5DF2\u6210\u529F\u586B\u5165\u8F93\u5165\u6846");
-  } else {
-    log(verbose, `\u26A0 \u8F93\u5165\u6846\u5185\u5BB9: "${filledValue.substring(0, 50)}"`);
-  }
+  await page.waitForTimeout(400);
   log(verbose, "\u6309 Enter \u53D1\u9001\u6D88\u606F...");
   await page.keyboard.press("Enter");
   await page.waitForTimeout(3e3);
@@ -338,47 +218,32 @@ async function sendMessage(page, message, verbose = false, headful = false) {
     if (headful) {
       await waitForUserAction(page, "captcha", verbose);
       log(verbose, "\u9A8C\u8BC1\u7801\u5DF2\u901A\u8FC7\uFF0C\u91CD\u65B0\u53D1\u9001\u6D88\u606F...");
-      const inputEl = await page.waitForSelector(inputSelector, { timeout: 1e4 });
-      if (inputEl) {
-        await inputEl.click();
-        await page.waitForTimeout(300);
-        await page.keyboard.type(message, { delay: 30 });
-        await page.keyboard.press("Enter");
-        await page.waitForTimeout(3e3);
+      for (const sel of inputSelectors) {
+        try {
+          const inputEl = await page.waitForSelector(sel, { timeout: 1e4 });
+          if (inputEl) {
+            await inputEl.click({ force: true });
+            await page.waitForTimeout(300);
+            await page.evaluate((element, content) => {
+              element.focus();
+              document.execCommand("selectAll", false, null);
+              document.execCommand("delete", false, null);
+              document.execCommand("insertText", false, content);
+            }, inputEl, message);
+            await page.keyboard.press("Enter");
+            await page.waitForTimeout(3e3);
+            break;
+          }
+        } catch {
+        }
       }
     } else {
       throw new YiyanAgentError(
         "CAPTCHA",
-        'Yiyan detected automation and triggered a captcha. Use headed mode (default) to manually solve it, or run "login" first.'
+        "Yiyan detected automation and triggered a captcha. Use headed mode to manually solve it."
       );
     }
   }
-  const url = page.url();
-  log(verbose, `\u5F53\u524D URL: ${url}`);
-  if (url.includes("/chat/")) {
-    log(verbose, "\u2713 \u5DF2\u81EA\u52A8\u8FDB\u5165\u5BF9\u8BDD\u9875\u9762");
-  } else if (url === "https://yiyan.baidu.com/" || url === "https://yiyan.baidu.com") {
-    log(verbose, "\u4ECD\u5728\u6B22\u8FCE\u9875\uFF0C\u5C1D\u8BD5\u70B9\u51FB\u53D1\u9001\u6309\u94AE...");
-    try {
-      const sendBtn = page.locator('[class*="send__"]').first();
-      await sendBtn.click({ timeout: 5e3 });
-      await page.waitForTimeout(3e3);
-      log(verbose, "\u2713 \u5DF2\u70B9\u51FB\u53D1\u9001\u6309\u94AE");
-    } catch {
-      log(verbose, "\u26A0 \u70B9\u51FB\u53D1\u9001\u6309\u94AE\u5931\u8D25\uFF0C\u7EE7\u7EED\u7B49\u5F85\u56DE\u590D...");
-    }
-  } else {
-    log(verbose, `\u5F53\u524D\u9875\u9762: ${url}`);
-  }
-}
-async function saveDebugScreenshot(page, name, verbose) {
-  try {
-    ensureDebugDir();
-    const screenshotPath = path2.join(DEBUG_DIR, `${name}.png`);
-    await page.screenshot({ path: screenshotPath, fullPage: true });
-    log(verbose, `\u8C03\u8BD5\u622A\u56FE\u5DF2\u4FDD\u5B58: ${screenshotPath}`);
-  } catch {
-  }
 }
 async function waitForReply(page, timeout, verbose = false, headful = false) {
   const maxWait = Math.min(timeout, 6e4);
@@ -414,13 +279,13 @@ async function waitForReply(page, timeout, verbose = false, headful = false) {
     });
     if (state.hasAiReply && state.aiTextLen > 0) {
       if (!replyStarted) {
-        log(verbose, `\u2713 AI \u56DE\u590D\u5DF2\u5F00\u59CB\u751F\u6210\uFF08${state.aiTextLen}\u5B57\uFF09: ${state.aiPreview.substring(0, 60)}...`);
+        log(verbose, `\u2713 AI \u56DE\u590D\u5DF2\u5F00\u59CB\u751F\u6210\uFF08${state.aiTextLen}\u5B57\uFF09`);
         replyStarted = true;
       }
       if (state.aiTextLen === lastLen) {
         stableCount++;
         if (stableCount >= 3) {
-          log(verbose, `\u2713 AI \u56DE\u590D\u5DF2\u5B8C\u6210\uFF08${state.aiTextLen}\u5B57\uFF0C\u8FDE\u7EED ${stableCount} \u6B21\u957F\u5EA6\u4E0D\u53D8\uFF09`);
+          log(verbose, `\u2713 AI \u56DE\u590D\u5DF2\u5B8C\u6210\uFF08${state.aiTextLen}\u5B57\uFF09`);
           break;
         }
       } else {
@@ -428,134 +293,31 @@ async function waitForReply(page, timeout, verbose = false, headful = false) {
         lastLen = state.aiTextLen;
       }
     }
-    await page.waitForTimeout(1500);
+    await page.waitForTimeout(500);
   }
   if (!replyStarted) {
     if (await checkCaptcha(page, verbose)) {
       if (headful) {
         await waitForUserAction(page, "captcha", verbose);
-        const retryStart = Date.now();
-        const retryMaxWait = 3e4;
-        while (Date.now() - retryStart < retryMaxWait) {
-          const retryState = await page.evaluate(() => {
-            const answerBox = document.querySelector('[class*="answerBox"]');
-            if (answerBox) {
-              const text = answerBox.innerText?.trim() || "";
-              return { hasAiReply: text.length > 0, aiTextLen: text.length };
-            }
-            return { hasAiReply: false, aiTextLen: 0 };
-          });
-          if (retryState.hasAiReply) {
-            log(verbose, "\u2713 \u9A8C\u8BC1\u7801\u901A\u8FC7\u540E AI \u5DF2\u5F00\u59CB\u56DE\u590D");
-            replyStarted = true;
-            break;
-          }
-          await page.waitForTimeout(1500);
-        }
       } else {
         throw new YiyanAgentError(
           "CAPTCHA",
-          'Yiyan detected automation and triggered a captcha. Use --headful flag to manually solve it, or run "login" first.'
+          "Yiyan detected automation and triggered a captcha."
         );
       }
     } else if (headful) {
       await waitForUserAction(page, "no-reply", verbose);
-      const retryStart = Date.now();
-      const retryMaxWait = 6e4;
-      while (Date.now() - retryStart < retryMaxWait) {
-        const retryState = await page.evaluate(() => {
-          const answerBox = document.querySelector('[class*="answerBox"]');
-          if (answerBox) {
-            const text = answerBox.innerText?.trim() || "";
-            return { hasAiReply: text.length > 0, aiTextLen: text.length };
-          }
-          return { hasAiReply: false, aiTextLen: 0 };
-        });
-        if (retryState.hasAiReply) {
-          if (retryState.aiTextLen === lastLen) {
-            stableCount++;
-            if (stableCount >= 3) {
-              log(verbose, `\u2713 \u7528\u6237\u64CD\u4F5C\u540E AI \u56DE\u590D\u5DF2\u5B8C\u6210\uFF08${retryState.aiTextLen}\u5B57\uFF09`);
-              replyStarted = true;
-              break;
-            }
-          } else {
-            stableCount = 0;
-            lastLen = retryState.aiTextLen;
-          }
-        }
-        await page.waitForTimeout(1500);
-      }
     } else {
       throw new YiyanAgentError(
         "TIMEOUT",
-        'AI reply timeout in headless mode. Try running "yiyan-agent login" first, or use --headful flag.'
+        "AI reply timeout in headless mode. Try running login first, or use headed mode."
       );
     }
   }
-  await page.waitForTimeout(2e3);
-  const pageDump = await page.evaluate(() => {
-    const result = {};
-    const selectors = [
-      '[class*="answerBox"]',
-      '[class*="dialogueCardList"]',
-      '[class*="dialogue_card_item"]',
-      '[class*="chatViewer"]',
-      '[class*="flowBox"]',
-      '[class*="roleSystem"]',
-      '[class*="mdRenderContainer"]',
-      '[class*="agent-markdown"]',
-      '[class*="markdown"]',
-      '[class*="content"]',
-      '[class*="chat"]',
-      '[class*="message"]'
-    ];
-    for (const sel of selectors) {
-      result[`selector:${sel}`] = document.querySelectorAll(sel).length;
-    }
-    const answerBox = document.querySelector('[class*="answerBox"]');
-    if (answerBox) {
-      result["answerBox"] = {
-        textLen: answerBox.innerText?.trim().length || 0,
-        preview: (answerBox.innerText?.trim() || "").substring(0, 200),
-        classes: answerBox.className?.toString().substring(0, 100) || ""
-      };
-    }
-    result["bodyText"] = document.body.innerText?.substring(0, 1e3) || "";
-    const longTextEls = [];
-    const mainContent = document.querySelector('[class*="chatViewer"]') || document.querySelector('[class*="chat"]') || document.body;
-    for (const el of Array.from(mainContent.querySelectorAll("div, section, article"))) {
-      const text = el.innerText?.trim() || "";
-      if (text.length > 50 && text.length < 1e4) {
-        longTextEls.push({
-          tag: el.tagName,
-          classes: el.className?.toString().split(" ").slice(0, 3).join(" ") || "",
-          textLen: text.length,
-          preview: text.substring(0, 80)
-        });
-      }
-    }
-    result["longTextElements"] = longTextEls.sort((a, b) => b.textLen - a.textLen).slice(0, 10);
-    return result;
-  });
-  log(verbose, "=== \u9875\u9762\u8BCA\u65AD ===");
-  for (const [key, value] of Object.entries(pageDump)) {
-    if (key === "bodyText" || key === "longTextElements") continue;
-    log(verbose, `  ${key}: ${JSON.stringify(value)}`);
-  }
-  if (pageDump["longTextElements"] && pageDump["longTextElements"].length > 0) {
-    log(verbose, "  \u957F\u6587\u672C\u5143\u7D20 TOP 10:");
-    for (const el of pageDump["longTextElements"]) {
-      log(verbose, `    <${el.tag}> .${el.classes} (${el.textLen}\u5B57): ${el.preview}...`);
-    }
-  }
-  if (pageDump["bodyText"]) {
-    log(verbose, `  \u9875\u9762\u6587\u672C(\u524D1000\u5B57): ${pageDump["bodyText"].substring(0, 1e3)}`);
-  }
+  await page.waitForTimeout(1e3);
 }
 async function extractReply(page, question, verbose = false) {
   log(verbose, "\u63D0\u53D6 AI \u56DE\u590D\u5185\u5BB9...");
-  await saveDebugScreenshot(page, "before-extract", verbose);
   const reply = await page.evaluate((userQuestion) => {
     const debugInfo = [];
     const answerBox = document.querySelector('[class*="answerBox"]');
@@ -563,19 +325,12 @@ async function extractReply(page, question, verbose = false) {
     if (answerBox) {
       const clone = answerBox.cloneNode(true);
       const processItems = clone.querySelectorAll('[class*="processItem"], [class*="processContent"]');
-      for (const item of processItems) {
-        item.remove();
-      }
+      for (const item of processItems) item.remove();
       const toolMessages = clone.querySelectorAll('[class*="toolMessage"]');
-      for (const msg of toolMessages) {
-        msg.remove();
-      }
+      for (const msg of toolMessages) msg.remove();
       const thinkHeaders = clone.querySelectorAll('[class*="headerMask"], [class*="topHeader"]');
-      for (const h of thinkHeaders) {
-        h.remove();
-      }
+      for (const h of thinkHeaders) h.remove();
       const text = clone.innerText?.trim() || "";
-      debugInfo.push(`answerBox after removing thinking: ${text.length} chars`);
       const cleanedText = text.replace(/^参考\d+个网页\s*/, "").replace(/^深度思考已完成\s*/, "").replace(/^思考完成[:：]\s*/, "").replace(/^准备输出结果\s*/, "").trim();
       if (cleanedText.length > 0) {
         debugInfo.push(`method1 success: ${cleanedText.length} chars`);
@@ -587,12 +342,10 @@ async function extractReply(page, question, verbose = false) {
     for (const container of allMdContainers) {
       const parent = container.parentElement;
       const parentClass = parent?.className?.toString() || "";
-      if (parentClass.includes("toolMessage") || parentClass.includes("process")) {
-        continue;
-      }
+      if (parentClass.includes("toolMessage") || parentClass.includes("process")) continue;
       const text = container.innerText?.trim() || "";
       if (text.length > 10) {
-        debugInfo.push(`method2 success: ${text.length} chars (parent: ${parentClass.substring(0, 50)})`);
+        debugInfo.push(`method2 success: ${text.length} chars`);
         return JSON.stringify({ text, debug: debugInfo });
       }
     }
@@ -611,21 +364,6 @@ async function extractReply(page, question, verbose = false) {
         }
       }
     }
-    const flowBoxes = document.querySelectorAll('[class*="flowBox"]');
-    debugInfo.push(`flowBoxes count: ${flowBoxes.length}`);
-    for (let i = flowBoxes.length - 1; i >= 0; i--) {
-      const box = flowBoxes[i];
-      const boxClass = box.className?.toString() || "";
-      if (boxClass.includes("questionBox")) continue;
-      const clone = box.cloneNode(true);
-      const thinkEls = clone.querySelectorAll('[class*="processItem"], [class*="processContent"], [class*="toolMessage"]');
-      for (const el of thinkEls) el.remove();
-      const text = clone.innerText?.trim() || "";
-      if (text.length > 10) {
-        debugInfo.push(`method4 success: flowBox[${i}] ${text.length} chars`);
-        return JSON.stringify({ text, debug: debugInfo });
-      }
-    }
     const fullText = document.body.innerText;
     const lines = fullText.split("\n").map((l) => l.trim()).filter((l) => l.length > 0);
     const uiWords = /* @__PURE__ */ new Set([
@@ -642,27 +380,15 @@ async function extractReply(page, question, verbose = false) {
       "\u672A\u767B\u5F55",
       "\u767B\u5F55",
       "\u5185\u5BB9\u7531AI\u751F\u6210\uFF0C\u4EC5\u4F9B\u53C2\u8003\uFF0C\u8BF7\u4ED4\u7EC6\u7504\u522B",
-      "\u53C2\u8003",
-      "\u6DF1\u5EA6\u5206\u6790\u9700\u6C42\u5E76\u89E3\u7B54",
-      "\u901A\u7528\u5199\u4F5C",
-      "\u9605\u8BFB\u5206\u6790",
-      "\u7F51\u9875\u5DE5\u574A",
-      "\u667A\u80FD\u7FFB\u8BD1",
-      "\u4EE3\u7801\u7F16\u7A0B"
+      "\u53C2\u8003"
     ]);
     const thinkingKeywords = [
       "\u6DF1\u5EA6\u601D\u8003\u5DF2\u5B8C\u6210",
       "\u601D\u8003\u5B8C\u6210",
-      "\u51C6\u5907\u8F93\u51FA\u7ED3\u679C",
-      "picaole\u5728\u8BE2\u95EE",
-      "picaole\u95EE\u4E86\u4E00\u4E2A",
-      "picaole\u5728\u641C\u7D22",
-      "\u76F4\u63A5\u56DE\u7B54\u5373\u53EF",
-      "\u4E0D\u9700\u8981\u590D\u6742\u5206\u6790",
-      "\u6309\u7167\u56DE\u7B54\u65B9\u5F0F"
+      "\u51C6\u5907\u8F93\u51FA\u7ED3\u679C"
     ];
     const userQIdx = lines.findIndex((l) => l === userQuestion || l.includes(userQuestion));
-    debugInfo.push(`userQ idx in body text: ${userQIdx}, total lines: ${lines.length}`);
+    debugInfo.push(`userQ idx: ${userQIdx}, lines: ${lines.length}`);
     if (userQIdx >= 0) {
       const replyLines = [];
       for (let i = userQIdx + 1; i < lines.length; i++) {
@@ -671,34 +397,15 @@ async function extractReply(page, question, verbose = false) {
         if (thinkingKeywords.some((kw) => line.includes(kw))) continue;
         if (line.startsWith("\u641C\u7D22") || line.includes("\u7BC7\u8D44\u6599")) continue;
         if (line.match(/^参考\d+个网页/)) continue;
-        if (line.length > 0) {
-          replyLines.push(line);
-        }
+        if (line.length > 0) replyLines.push(line);
         if (line === "\u5FEB\u901F" || line === "\u66F4\u591A" || line.includes("\u5185\u5BB9\u7531AI\u751F\u6210")) break;
       }
       if (replyLines.length > 0) {
-        debugInfo.push(`method5 success: ${replyLines.length} lines`);
+        debugInfo.push(`method4 success: ${replyLines.length} lines`);
         return JSON.stringify({ text: replyLines.join("\n"), debug: debugInfo });
       }
     }
-    const allTextBlocks = [];
-    const blockElements = document.querySelectorAll("div, section, article, p");
-    for (const el of blockElements) {
-      const elText = el.innerText?.trim() || "";
-      if (elText.length > 50 && elText.length < 1e4) {
-        if (uiWords.has(elText)) continue;
-        if (thinkingKeywords.some((kw) => elText.includes(kw))) continue;
-        allTextBlocks.push({ text: elText, length: elText.length });
-      }
-    }
-    allTextBlocks.sort((a, b) => b.length - a.length);
-    if (allTextBlocks.length > 0) {
-      debugInfo.push(`method6 fallback: longest text block ${allTextBlocks[0].length} chars`);
-      return JSON.stringify({ text: allTextBlocks[0].text, debug: debugInfo });
-    }
-    debugInfo.push(`ALL METHODS FAILED`);
-    debugInfo.push(`body text length: ${fullText.length}`);
-    debugInfo.push(`body text first 500: ${fullText.substring(0, 500)}`);
+    debugInfo.push("ALL METHODS FAILED");
     return JSON.stringify({ text: "", debug: debugInfo });
   }, question);
   let parsed;
@@ -714,68 +421,39 @@ async function extractReply(page, question, verbose = false) {
     log(verbose, `\u63D0\u53D6\u6210\u529F\uFF0C\u56DE\u590D\u957F\u5EA6: ${parsed.text.length} \u5B57\u7B26`);
     return parsed.text;
   }
-  log(verbose, "\u63D0\u53D6\u5931\u8D25\uFF0C\u6240\u6709\u65B9\u6CD5\u5747\u672A\u627E\u5230\u56DE\u590D\u5185\u5BB9");
   throw new YiyanAgentError("TIMEOUT", "Failed to extract reply content");
 }
 
 // src/agent.ts
-var DEFAULT_PROFILE_BASE_DIR = path3.join(
-  os2.homedir(),
-  ".yiyan-browser-agent"
-);
+var DEFAULT_SESSION_DIR = path.join(os.homedir(), ".yiyan-browser-agent", "session");
 var YiyanAgent = class {
   options;
-  profileDir;
+  sessionDir;
   verbose;
   constructor(options) {
     this.options = {
       ...DEFAULT_OPTIONS,
       ...options
     };
-    this.profileDir = this.options.profileDir || DEFAULT_PROFILE_BASE_DIR;
+    this.sessionDir = this.options.profileDir || DEFAULT_SESSION_DIR;
     this.verbose = options?.verbose ?? false;
   }
   /**
    * 发送问题并获取答案
    * @param question 问题文本
-   * @param headful 是否使用有头浏览器（可见窗口），默认为 false（优先 headless，登录后无需弹窗）
-   * 
-   * 策略：先尝试 headless（不弹窗），如果失败（验证码/未登录）自动回退到有头模式
+   * @param headful 是否使用有头浏览器（可见窗口），默认为 false（headless）
    */
   async ask(question, headful = false) {
-    if (headful) {
-      return this.executeAsk(question, true);
-    }
-    try {
-      return await this.executeAsk(question, false);
-    } catch (error) {
-      if (error instanceof YiyanAgentError && (error.type === "CAPTCHA" || error.type === "TIMEOUT")) {
-        if (this.verbose) {
-          process.stderr.write("[yiyan-agent] headless \u6A21\u5F0F\u5931\u8D25\uFF0C\u81EA\u52A8\u5207\u6362\u5230\u6709\u5934\u6A21\u5F0F...\n");
-        }
-        return this.executeAsk(question, true);
-      }
-      throw error;
-    }
+    return this.executeAsk(question, headful);
   }
   /**
    * 执行单次问答
    */
   async executeAsk(question, headful) {
-    const platform = os2.platform();
-    const chromePath = this.options.chromePath || getChromeExecutablePath(platform);
-    if (!chromePath) {
-      throw new YiyanAgentError(
-        "BROWSER_LAUNCH",
-        `Unsupported platform: ${platform}`
-      );
-    }
-    const profilePath = path3.join(this.profileDir, "chrome-profile");
-    const { browser, cdpBrowser, page, chromeProcess } = await launchBrowser({
-      chromePath,
-      profilePath,
+    const { context, page } = await launchBrowser({
+      sessionDir: this.sessionDir,
       headless: !headful,
-      timeout: 3e4,
+      timeout: this.options.timeout,
       verbose: this.verbose
     });
     try {
@@ -796,7 +474,7 @@ var YiyanAgent = class {
       const reply = await extractReply(page, question, this.verbose);
       return reply;
     } finally {
-      await closeBrowser(browser, chromeProcess, cdpBrowser, this.verbose);
+      await closeBrowser(context, this.verbose);
     }
   }
   /**
@@ -804,66 +482,96 @@ var YiyanAgent = class {
    * 登录成功后，后续的 ask 调用将自动使用登录状态
    */
   async login() {
-    const platform = os2.platform();
-    const chromePath = this.options.chromePath || getChromeExecutablePath(platform);
-    if (!chromePath) {
-      throw new YiyanAgentError(
-        "BROWSER_LAUNCH",
-        `Unsupported platform: ${platform}`
-      );
-    }
-    const profilePath = path3.join(this.profileDir, "chrome-profile");
-    const { browser, cdpBrowser, page, chromeProcess } = await launchBrowser({
-      chromePath,
-      profilePath,
+    const { context, page } = await launchBrowser({
+      sessionDir: this.sessionDir,
       headless: false,
       timeout: 6e4,
       verbose: this.verbose
     });
     try {
-      await page.goto("https://yiyan.baidu.com/", {
-        waitUntil: "networkidle",
-        timeout: 6e4
-      });
-      await page.waitForTimeout(5e3);
-      process.stderr.write("\n[yiyan-agent] ============================================\n");
-      process.stderr.write("[yiyan-agent] \u8BF7\u5728\u6D4F\u89C8\u5668\u4E2D\u5B8C\u6210\u767B\u5F55\u6587\u5FC3\u4E00\u8A00\n");
-      process.stderr.write("[yiyan-agent] \u767B\u5F55\u6210\u529F\u540E\uFF0C\u56DE\u5230\u6B64\u5904\u6309 Enter \u952E\u7EE7\u7EED\n");
-      process.stderr.write("[yiyan-agent] ============================================\n\n");
-      const readline = await import("readline");
-      await new Promise((resolve) => {
-        const rl = readline.createInterface({
-          input: process.stdin,
-          output: process.stderr
-        });
-        rl.question("\u6309 Enter \u7EE7\u7EED...", () => {
-          rl.close();
-          resolve();
-        });
-      });
-      process.stderr.write("[yiyan-agent] \u2713 \u767B\u5F55\u5B8C\u6210\uFF0C\u6B63\u5728\u4FDD\u5B58\u767B\u5F55\u72B6\u6001...\n");
+      await navigateToYiyan(page, this.verbose);
+      process.stderr.write("\n[yiyan-agent] \u2554\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2557\n");
+      process.stderr.write("[yiyan-agent] \u2551  \u{1F510}  LOGIN REQUIRED                          \u2551\n");
+      process.stderr.write("[yiyan-agent] \u2551                                              \u2551\n");
+      process.stderr.write("[yiyan-agent] \u2551  1. Log in to Yiyan in the browser window     \u2551\n");
+      process.stderr.write("[yiyan-agent] \u2551  2. Return here and press ENTER to continue  \u2551\n");
+      process.stderr.write("[yiyan-agent] \u255A\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u2550\u255D\n\n");
+      await this.waitForEnter();
+      process.stderr.write("[yiyan-agent] \u2713 Login complete, saving session...\n");
       await new Promise((resolve) => setTimeout(resolve, 3e3));
     } finally {
-      await closeBrowser(browser, chromeProcess, cdpBrowser, this.verbose);
+      await closeBrowser(context, this.verbose);
     }
   }
-  sleep(ms) {
-    return new Promise((resolve) => setTimeout(resolve, ms));
+  /**
+   * 等待用户按 Enter
+   */
+  async waitForEnter() {
+    const readline = await import("readline");
+    return new Promise((resolve) => {
+      const rl = readline.createInterface({
+        input: process.stdin,
+        output: process.stderr
+      });
+      rl.question("Press ENTER to continue...", () => {
+        rl.close();
+        resolve();
+      });
+    });
   }
+  /**
+   * 清除保存的 session
+   */
   async reset() {
-    await clearCopiedProfile(this.profileDir);
+    const fs3 = await import("fs/promises");
+    try {
+      await fs3.rm(this.sessionDir, { recursive: true, force: true });
+    } catch {
+    }
   }
+  /**
+   * 检查状态
+   */
   status() {
+    const fs3 = __require("fs");
     return {
-      loggedIn: profileExists(this.profileDir),
-      profilePath: path3.join(this.profileDir, "chrome-profile")
+      loggedIn: fs3.existsSync(this.sessionDir),
+      sessionPath: this.sessionDir
     };
   }
 };
+
+// src/profile.ts
+import path2 from "path";
+import os2 from "os";
+import fs2 from "fs";
+var CONFIG_FILE_PATH = path2.join(os2.homedir(), ".yiyan-browser-agent", "config.json");
+var DEFAULT_SESSION_DIR2 = path2.join(os2.homedir(), ".yiyan-browser-agent", "session");
+function readConfig() {
+  try {
+    if (fs2.existsSync(CONFIG_FILE_PATH)) {
+      const content = fs2.readFileSync(CONFIG_FILE_PATH, "utf-8");
+      return JSON.parse(content);
+    }
+  } catch {
+  }
+  return {};
+}
+function saveConfig(config) {
+  const configDir = path2.dirname(CONFIG_FILE_PATH);
+  if (!fs2.existsSync(configDir)) {
+    fs2.mkdirSync(configDir, { recursive: true });
+  }
+  fs2.writeFileSync(CONFIG_FILE_PATH, JSON.stringify(config, null, 2));
+}
 export {
+  CONFIG_FILE_PATH,
   DEFAULT_OPTIONS,
+  DEFAULT_SESSION_DIR2 as DEFAULT_SESSION_DIR,
   YIYAN_CHAT_URL,
   YiyanAgent,
-  YiyanAgentError
+  YiyanAgentError,
+  readConfig,
+  saveConfig
 };
 //# sourceMappingURL=index.js.map