yiyan-browser-agent 1.3.3 → 1.4.0

package/dist/cli.js

@@ -30,9 +30,48 @@ var YIYAN_CHAT_URL = "https://yiyan.baidu.com/";
 // src/browser.ts
 import { chromium } from "playwright";
 import fs from "fs";
+var SELECTORS = {
+  // 输入框选择器（按优先级排序）
+  chatInput: [
+    "#chat-input",
+    'textarea[placeholder*="\u8F93\u5165"]',
+    "textarea[placeholder]",
+    "textarea",
+    '[contenteditable="true"][role="textbox"]',
+    '[contenteditable="true"]'
+  ],
+  // 发送按钮选择器
+  sendButton: [
+    'button[aria-label*="\u53D1\u9001"]',
+    'button[aria-label*="Send"]',
+    '[data-testid="send-button"]',
+    'button[type="submit"]',
+    '[class*="send-btn"]',
+    '[class*="sendBtn"]',
+    '[class*="send-button"]'
+  ],
+  // 响应容器选择器
+  responseContainer: [
+    '[class*="answerBox"]',
+    '[class*="response"]',
+    '[class*="ai-message"]',
+    '[class*="assistant"]',
+    '[class*="markdown"]',
+    '[class*="message-content"]',
+    '[class*="chat-message"]'
+  ],
+  // 停止生成按钮
+  stopButton: [
+    'button[aria-label*="\u505C\u6B62"]',
+    'button[aria-label*="Stop"]',
+    '[class*="stop-btn"]',
+    '[class*="stopBtn"]',
+    '[class*="stop-gen"]'
+  ]
+};
 function log(verbose, msg) {
   if (verbose) {
-    process.stderr.write(`[yiyan-agent] ${msg}
+    process.stderr.write(`[yiyan-browser-agent] ${msg}
 `);
   }
 }
@@ -85,326 +124,310 @@ async function closeBrowser(context, verbose) {
   }
 }
 async function navigateToYiyan(page, verbose = false) {
-  log(verbose, "\u5BFC\u822A\u5230\u6587\u5FC3\u4E00\u8A00\u804A\u5929\u9875\u9762...");
+  log(verbose, "\u5BFC\u822A\u5230\u6587\u5FC3\u4E00\u8A00...");
   try {
-    await page.goto(YIYAN_CHAT_URL, {
-      waitUntil: "domcontentloaded",
-      timeout: 3e4
-    });
+    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) {
-  const captchaIndicators = [
-    "\u8BF7\u5B8C\u6210\u4E0B\u5217\u9A8C\u8BC1\u540E\u7EE7\u7EED",
-    "\u6309\u4F4F\u5DE6\u8FB9\u6309\u94AE\u62D6\u52A8",
-    "\u6ED1\u52A8\u9A8C\u8BC1",
-    "\u70B9\u51FB\u9A8C\u8BC1",
-    "\u5B89\u5168\u9A8C\u8BC1",
-    "captcha",
-    "\u8BF7\u5B8C\u6210\u9A8C\u8BC1"
-  ];
-  const hasCaptcha = await page.evaluate((indicators) => {
-    const bodyText = document.body.innerText || "";
-    for (const indicator of indicators) {
-      if (bodyText.includes(indicator)) return true;
-    }
-    const dialogs = document.querySelectorAll('[role="dialog"], [class*="captcha"], [class*="verify"]');
-    if (dialogs.length > 0) return true;
-    return false;
-  }, captchaIndicators);
-  if (hasCaptcha) {
-    log(verbose, "\u26A0 \u68C0\u6D4B\u5230\u9A8C\u8BC1\u7801\u5F39\u7A97\uFF01");
-  }
-  return hasCaptcha;
-}
 async function checkLoggedIn(page, verbose = false) {
+  await page.waitForTimeout(2e3);
   const isLoggedIn = await page.evaluate(() => {
-    const bodyText = document.body.innerText || "";
-    if (bodyText.includes("\u672A\u767B\u5F55")) return false;
-    const loginButtons = document.querySelectorAll("button");
-    for (const btn of loginButtons) {
-      const text = btn.textContent?.trim() || "";
-      if (text === "\u767B\u5F55") {
-        return false;
-      }
-    }
+    const url = window.location.href;
+    const bodyText = document.body?.innerText || "";
+    if (url.includes("/auth") || url.includes("/login") || url.includes("/sign")) return false;
+    if (bodyText.includes("\u672A\u767B\u5F55") || bodyText.includes("\u8BF7\u767B\u5F55")) return false;
+    if (document.querySelector('input[type="password"]')) return false;
+    const loginBtn = document.querySelector("button");
+    if (loginBtn && loginBtn.textContent?.trim() === "\u767B\u5F55") return false;
     return true;
   });
-  if (isLoggedIn) {
-    log(verbose, "\u2713 \u5DF2\u68C0\u6D4B\u5230\u767B\u5F55\u72B6\u6001");
-  } else {
-    log(verbose, "\u26A0 \u672A\u68C0\u6D4B\u5230\u767B\u5F55\u72B6\u6001");
-  }
+  log(verbose, isLoggedIn ? "\u2713 \u5DF2\u767B\u5F55" : "\u26A0 \u672A\u767B\u5F55");
   return isLoggedIn;
 }
+async function checkCaptcha(page, verbose = false) {
+  const hasCaptcha = await page.evaluate(() => {
+    const bodyText = document.body.innerText || "";
+    const indicators = ["\u9A8C\u8BC1\u7801", "\u8BF7\u5B8C\u6210\u9A8C\u8BC1", "\u5B89\u5168\u9A8C\u8BC1", "\u6ED1\u52A8\u9A8C\u8BC1", "captcha"];
+    for (const kw of indicators) {
+      if (bodyText.includes(kw)) return true;
+    }
+    const dialogs = document.querySelectorAll('[role="dialog"], [class*="captcha"], [class*="verify"]');
+    return dialogs.length > 0;
+  });
+  if (hasCaptcha) log(verbose, "\u26A0 \u68C0\u6D4B\u5230\u9A8C\u8BC1\u7801\uFF01");
+  return hasCaptcha;
+}
 async function waitForUserAction(page, reason, verbose = false) {
-  const reasonText = reason === "captcha" ? "\u68C0\u6D4B\u5230\u9A8C\u8BC1\u7801\uFF0C\u8BF7\u5728\u6D4F\u89C8\u5668\u4E2D\u624B\u52A8\u5B8C\u6210\u9A8C\u8BC1" : reason === "login" ? "\u68C0\u6D4B\u5230\u672A\u767B\u5F55\uFF0C\u8BF7\u5728\u6D4F\u89C8\u5668\u4E2D\u624B\u52A8\u767B\u5F55" : "AI \u672A\u56DE\u590D\uFF0C\u8BF7\u68C0\u67E5\u6D4F\u89C8\u5668\u662F\u5426\u9700\u8981\u624B\u52A8\u64CD\u4F5C\uFF08\u9A8C\u8BC1\u7801/\u767B\u5F55\uFF09";
+  const messages = {
+    captcha: "\u68C0\u6D4B\u5230\u9A8C\u8BC1\u7801\uFF0C\u8BF7\u5728\u6D4F\u89C8\u5668\u4E2D\u624B\u52A8\u5B8C\u6210\u9A8C\u8BC1",
+    login: "\u68C0\u6D4B\u5230\u672A\u767B\u5F55\uFF0C\u8BF7\u5728\u6D4F\u89C8\u5668\u4E2D\u624B\u52A8\u767B\u5F55",
+    "no-reply": "AI \u672A\u56DE\u590D\uFF0C\u8BF7\u68C0\u67E5\u6D4F\u89C8\u5668\u662F\u5426\u9700\u8981\u624B\u52A8\u64CD\u4F5C"
+  };
   process.stderr.write(`
-[yiyan-agent] \u26A0 ${reasonText}
+[yiyan-browser-agent] \u26A0 ${messages[reason]}
 `);
-  process.stderr.write("[yiyan-agent] \u7B49\u5F85\u60A8\u64CD\u4F5C\u5B8C\u6210...\uFF08\u64CD\u4F5C\u5B8C\u6210\u540E\u4F1A\u81EA\u52A8\u7EE7\u7EED\uFF09\n\n");
+  process.stderr.write("[yiyan-browser-agent] \u7B49\u5F85\u60A8\u64CD\u4F5C\u5B8C\u6210...\uFF08\u64CD\u4F5C\u5B8C\u6210\u540E\u4F1A\u81EA\u52A8\u7EE7\u7EED\uFF09\n\n");
   const maxWait = 18e4;
-  const startTime = Date.now();
-  while (Date.now() - startTime < maxWait) {
+  const start = Date.now();
+  while (Date.now() - start < maxWait) {
     await page.waitForTimeout(2e3);
-    if (reason === "captcha") {
-      const stillHasCaptcha = await checkCaptcha(page, false);
-      if (!stillHasCaptcha) {
-        log(verbose, "\u2713 \u9A8C\u8BC1\u7801\u5DF2\u901A\u8FC7\uFF01");
-        return;
-      }
-    } else if (reason === "login") {
-      const loggedIn = await checkLoggedIn(page, false);
-      if (loggedIn) {
-        log(verbose, "\u2713 \u767B\u5F55\u6210\u529F\uFF01");
-        return;
-      }
-    } else {
-      const hasReply = await page.evaluate(() => {
-        const answerBox = document.querySelector('[class*="answerBox"]');
-        if (!answerBox) return false;
-        const text = answerBox.innerText?.trim() || "";
-        return text.length > 0;
-      });
-      if (hasReply) {
-        log(verbose, "\u2713 AI \u5DF2\u5F00\u59CB\u56DE\u590D\uFF01");
-        return;
-      }
+    if (reason === "captcha" && !await checkCaptcha(page, false)) {
+      log(verbose, "\u2713 \u9A8C\u8BC1\u7801\u5DF2\u901A\u8FC7\uFF01");
+      return;
+    }
+    if (reason === "login" && await checkLoggedIn(page, false)) {
+      log(verbose, "\u2713 \u767B\u5F55\u6210\u529F\uFF01");
+      return;
+    }
+    if (reason === "no-reply" && await getMessageCount(page) > 0) {
+      log(verbose, "\u2713 AI \u5DF2\u5F00\u59CB\u56DE\u590D\uFF01");
+      return;
     }
   }
   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 inputSelectors = [
-    '[contenteditable="true"][role="textbox"]',
-    '[contenteditable="true"]',
-    "textarea[placeholder]",
-    "textarea"
-  ];
-  log(verbose, "\u7B49\u5F85\u8F93\u5165\u6846\u51FA\u73B0...");
-  let inputElement = null;
-  for (const sel of inputSelectors) {
-    try {
-      inputElement = await page.waitForSelector(sel, { timeout: 4e3, state: "visible" });
-      if (inputElement) {
-        log(verbose, `\u2713 \u8F93\u5165\u6846\u5DF2\u627E\u5230: ${sel}`);
-        break;
-      }
-    } catch {
-    }
-  }
-  if (!inputElement) {
+  const input = await findInput(page, verbose);
+  if (!input) {
     throw new YiyanAgentError("NETWORK", "Cannot find the Yiyan chat input box");
   }
-  log(verbose, "\u70B9\u51FB\u8F93\u5165\u6846\u83B7\u53D6\u7126\u70B9...");
-  await inputElement.click({ force: true });
-  await page.waitForTimeout(500);
-  log(verbose, `\u8F93\u5165\u95EE\u9898: "${message}"`);
-  await inputElement.fill(message);
-  await page.waitForTimeout(500);
-  const filledValue = await inputElement.innerText();
-  if (!filledValue.includes(message)) {
-    log(verbose, "fill \u5931\u8D25\uFF0C\u5C1D\u8BD5 keyboard.type...");
-    await inputElement.click({ force: true });
-    await page.keyboard.press("Control+a");
-    await page.keyboard.type(message, { delay: 50 });
-    await page.waitForTimeout(500);
+  await input.click({ force: true });
+  await page.waitForTimeout(200);
+  await page.keyboard.press("Control+a");
+  await page.waitForTimeout(100);
+  const isTextarea = await input.evaluate((el) => el.tagName.toLowerCase() === "textarea");
+  if (isTextarea) {
+    await input.fill(message);
+  } else {
+    await input.evaluate((el, content) => {
+      el.focus();
+      document.execCommand("selectAll", false, null);
+      document.execCommand("delete", false, null);
+      document.execCommand("insertText", false, content);
+      el.dispatchEvent(new InputEvent("input", { bubbles: true, data: content }));
+    }, message);
+  }
+  await page.waitForTimeout(400);
+  log(verbose, `\u5DF2\u8F93\u5165\u95EE\u9898: "${message.substring(0, 50)}..."`);
+  const sent = await clickSendButton(page);
+  if (!sent) {
+    await page.keyboard.press("Enter");
   }
-  log(verbose, "\u6309 Enter \u53D1\u9001\u6D88\u606F...");
-  await page.keyboard.press("Enter");
-  await page.waitForTimeout(3e3);
+  await page.waitForTimeout(500);
   if (await checkCaptcha(page, verbose)) {
     if (headful) {
       await waitForUserAction(page, "captcha", verbose);
     } else {
-      throw new YiyanAgentError(
-        "CAPTCHA",
-        "Yiyan detected automation and triggered a captcha. Use --headful to manually solve it."
-      );
+      throw new YiyanAgentError("CAPTCHA", "Captcha detected. Use --headful to solve it.");
     }
   }
 }
-async function waitForReply(page, timeout, verbose = false, headful = false) {
-  const maxWait = Math.min(timeout, 6e4);
-  log(verbose, `\u7B49\u5F85 AI \u56DE\u590D\uFF08\u6700\u591A ${maxWait / 1e3} \u79D2\uFF09...`);
-  const startTime = Date.now();
-  let lastLen = 0;
-  let stableCount = 0;
-  let replyStarted = false;
-  while (Date.now() - startTime < maxWait) {
-    const state = await page.evaluate(() => {
-      const answerBox = document.querySelector('[class*="answerBox"]');
-      if (answerBox) {
-        const text = answerBox.innerText?.trim() || "";
-        return {
-          hasAiReply: text.length > 0,
-          aiTextLen: text.length,
-          aiPreview: text.substring(0, 100)
-        };
-      }
-      const cardList = document.querySelector('[class*="dialogueCardList"]');
-      if (cardList) {
-        const lastCard = cardList.lastElementChild;
-        if (lastCard) {
-          const text = lastCard.innerText?.trim() || "";
-          return {
-            hasAiReply: text.length > 0,
-            aiTextLen: text.length,
-            aiPreview: text.substring(0, 100)
-          };
-        }
+async function findInput(page, verbose = false) {
+  for (const sel of SELECTORS.chatInput) {
+    try {
+      const el = await page.waitForSelector(sel, { timeout: 4e3, state: "visible" });
+      if (el) {
+        log(verbose, `\u2713 \u8F93\u5165\u6846: ${sel}`);
+        return el;
       }
-      return { hasAiReply: false, aiTextLen: 0, aiPreview: "" };
-    });
-    if (state.hasAiReply && state.aiTextLen > 0) {
-      if (!replyStarted) {
-        log(verbose, `\u2713 AI \u56DE\u590D\u5DF2\u5F00\u59CB\u751F\u6210\uFF08${state.aiTextLen}\u5B57\uFF09`);
-        replyStarted = true;
+    } catch {
+    }
+  }
+  return null;
+}
+async function clickSendButton(page) {
+  for (const sel of SELECTORS.sendButton) {
+    try {
+      const el = await page.$(sel);
+      if (el && await el.isVisible() && await el.isEnabled()) {
+        await el.click();
+        return true;
       }
-      if (state.aiTextLen === lastLen) {
-        stableCount++;
-        if (stableCount >= 3) {
-          log(verbose, `\u2713 AI \u56DE\u590D\u5DF2\u5B8C\u6210\uFF08${state.aiTextLen}\u5B57\uFF09`);
+    } catch {
+    }
+  }
+  return false;
+}
+async function waitForReply(page, timeout, verbose = false, headful = false) {
+  const maxWait = Math.min(timeout, 12e4);
+  const stableDelay = 2500;
+  const start = Date.now();
+  const initialCount = await getMessageCount(page);
+  let appeared = false;
+  while (Date.now() - start < 12e3) {
+    const count = await getMessageCount(page);
+    if (count > initialCount) {
+      appeared = true;
+      log(verbose, "\u2713 AI \u56DE\u590D\u5DF2\u5F00\u59CB\u751F\u6210");
+      break;
+    }
+    await page.waitForTimeout(400);
+  }
+  if (!appeared) {
+    log(verbose, "\u56DE\u590D\u53EF\u80FD\u5EF6\u8FDF\uFF0C\u7EE7\u7EED\u7B49\u5F85...");
+  }
+  let lastText = "";
+  let stableStart = null;
+  while (Date.now() - start < maxWait) {
+    const text = await extractLastMessage(page);
+    if (text !== lastText) {
+      lastText = text;
+      stableStart = null;
+    } else if (text.length > 0) {
+      if (!stableStart) stableStart = Date.now();
+      else if (Date.now() - stableStart >= stableDelay) {
+        if (!await isGenerating(page)) {
+          log(verbose, `\u2713 AI \u56DE\u590D\u5DF2\u5B8C\u6210\uFF08${text.length}\u5B57\uFF09`);
           break;
         }
-      } else {
-        stableCount = 0;
-        lastLen = state.aiTextLen;
+        stableStart = null;
       }
     }
     await page.waitForTimeout(500);
   }
-  if (!replyStarted) {
+  if (lastText.length === 0) {
     if (await checkCaptcha(page, verbose)) {
       if (headful) {
         await waitForUserAction(page, "captcha", verbose);
       } else {
-        throw new YiyanAgentError(
-          "CAPTCHA",
-          "Yiyan detected automation and triggered a captcha."
-        );
+        throw new YiyanAgentError("CAPTCHA", "Captcha detected.");
       }
     } else if (headful) {
       await waitForUserAction(page, "no-reply", verbose);
     } else {
-      throw new YiyanAgentError(
-        "TIMEOUT",
-        "AI reply timeout in headless mode. Try running login first, or use headed mode."
-      );
+      throw new YiyanAgentError("TIMEOUT", "AI reply timeout.");
     }
   }
   await page.waitForTimeout(1e3);
 }
-async function extractReply(page, question, verbose = false) {
-  log(verbose, "\u63D0\u53D6 AI \u56DE\u590D\u5185\u5BB9...");
-  const reply = await page.evaluate((userQuestion) => {
-    const debugInfo = [];
-    const answerBox = document.querySelector('[class*="answerBox"]');
-    debugInfo.push(`answerBox found: ${!!answerBox}`);
-    if (answerBox) {
-      const clone = answerBox.cloneNode(true);
-      const processItems = clone.querySelectorAll('[class*="processItem"], [class*="processContent"]');
-      for (const item of processItems) item.remove();
-      const toolMessages = clone.querySelectorAll('[class*="toolMessage"]');
-      for (const msg of toolMessages) msg.remove();
-      const thinkHeaders = clone.querySelectorAll('[class*="headerMask"], [class*="topHeader"]');
-      for (const h of thinkHeaders) h.remove();
-      const text = clone.innerText?.trim() || "";
-      const cleanedText = text.replace(/^参考\d+个网页\s*/, "").replace(/^深度思考已完成\s*/, "").replace(/^思考完成[:：]\s*/, "").replace(/^准备输出结果\s*/, "").trim();
-      if (cleanedText.length > 0) {
-        debugInfo.push(`method1 success: ${cleanedText.length} chars`);
-        return JSON.stringify({ text: cleanedText, debug: debugInfo });
+async function getMessageCount(page) {
+  return await page.evaluate(() => {
+    for (const sel of [
+      '[class*="answerBox"]',
+      '[class*="assistant"]',
+      '[class*="ai-message"]',
+      '[class*="response"]'
+    ]) {
+      const els = document.querySelectorAll(sel);
+      if (els.length > 0) return els.length;
+    }
+    return document.querySelectorAll('[class*="message"]').length;
+  });
+}
+async function isGenerating(page) {
+  return await page.evaluate(() => {
+    for (const sel of ['button[aria-label*="\u505C\u6B62"]', '[class*="stop"]', '[class*="generating"]']) {
+      const el = document.querySelector(sel);
+      if (el) {
+        const s = window.getComputedStyle(el);
+        if (s.display !== "none" && s.visibility !== "hidden") return true;
       }
     }
-    const allMdContainers = document.querySelectorAll('[class*="mdRenderContainer"]');
-    debugInfo.push(`mdRenderContainer count: ${allMdContainers.length}`);
-    for (const container of allMdContainers) {
-      const parent = container.parentElement;
-      const parentClass = parent?.className?.toString() || "";
-      if (parentClass.includes("toolMessage") || parentClass.includes("process")) continue;
-      const text = container.innerText?.trim() || "";
-      if (text.length > 10) {
-        debugInfo.push(`method2 success: ${text.length} chars`);
-        return JSON.stringify({ text, debug: debugInfo });
+    for (const sel of ['[class*="loading"]', '[class*="typing"]', '[class*="spinner"]']) {
+      const el = document.querySelector(sel);
+      if (el) {
+        const s = window.getComputedStyle(el);
+        if (s.display !== "none" && s.visibility !== "hidden") return true;
       }
     }
-    const cardList = document.querySelector('[class*="dialogueCardList"]');
-    debugInfo.push(`dialogueCardList found: ${!!cardList}`);
-    if (cardList && cardList.children.length > 0) {
-      const lastCard = cardList.lastElementChild;
-      if (lastCard) {
-        const clone = lastCard.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 > 0) {
-          debugInfo.push(`method3 success: ${text.length} chars`);
-          return JSON.stringify({ text, debug: debugInfo });
+    return false;
+  });
+}
+async function extractLastMessage(page) {
+  return await page.evaluate(() => {
+    function getFullText(el) {
+      if (!el) return "";
+      let result = "";
+      function walk(node) {
+        if (node.nodeType === Node.TEXT_NODE) {
+          result += node.textContent || "";
+          return;
+        }
+        if (node.nodeType !== Node.ELEMENT_NODE) return;
+        const tag = node.tagName.toLowerCase();
+        if (tag === "pre") {
+          const codeEl = node.querySelector("code");
+          if (codeEl) {
+            const cls = codeEl.className || "";
+            const lang = (cls.match(/language-(\S+)/) || [])[1] || "";
+            result += "\n```" + lang + "\n" + (codeEl.textContent || "") + "\n```\n";
+          } else {
+            result += "\n```\n" + (node.textContent || "") + "\n```\n";
+          }
+          return;
+        }
+        for (const child of node.childNodes) walk(child);
+        if (["p", "div", "li", "br", "h1", "h2", "h3", "h4", "h5", "h6"].includes(tag)) {
+          result += "\n";
         }
       }
+      walk(el);
+      return result.trim();
     }
-    const fullText = document.body.innerText;
-    const lines = fullText.split("\n").map((l) => l.trim()).filter((l) => l.length > 0);
-    const uiWords = /* @__PURE__ */ new Set([
-      "\u6587\u5FC3\u4E00\u8A00",
-      "\u65B0\u5BF9\u8BDD",
-      "\u521B\u610F\u5199\u4F5C",
-      "\u667A\u6167\u7ED8\u56FE",
-      "\u8D85\u7EA7\u667A\u80FD\u4F53",
-      "\u66F4\u591A",
-      "\u6211\u7684\u6536\u85CF",
-      "\u9879\u76EE",
-      "\u5BF9\u8BDD",
-      "\u6682\u65E0\u8BB0\u5F55",
-      "\u672A\u767B\u5F55",
-      "\u767B\u5F55",
-      "\u5185\u5BB9\u7531AI\u751F\u6210\uFF0C\u4EC5\u4F9B\u53C2\u8003\uFF0C\u8BF7\u4ED4\u7EC6\u7504\u522B",
-      "\u53C2\u8003"
-    ]);
-    const thinkingKeywords = [
-      "\u6DF1\u5EA6\u601D\u8003\u5DF2\u5B8C\u6210",
-      "\u601D\u8003\u5B8C\u6210",
-      "\u51C6\u5907\u8F93\u51FA\u7ED3\u679C"
-    ];
-    const userQIdx = lines.findIndex((l) => l === userQuestion || l.includes(userQuestion));
-    debugInfo.push(`userQ idx: ${userQIdx}, lines: ${lines.length}`);
-    if (userQIdx >= 0) {
-      const replyLines = [];
-      for (let i = userQIdx + 1; i < lines.length; i++) {
-        const line = lines[i];
-        if (uiWords.has(line)) continue;
-        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 === "\u5FEB\u901F" || line === "\u66F4\u591A" || line.includes("\u5185\u5BB9\u7531AI\u751F\u6210")) break;
-      }
-      if (replyLines.length > 0) {
-        debugInfo.push(`method4 success: ${replyLines.length} lines`);
-        return JSON.stringify({ text: replyLines.join("\n"), debug: debugInfo });
+    for (const sel of SELECTORS.responseContainer) {
+      const els = document.querySelectorAll(sel);
+      if (els.length > 0) {
+        const text = getFullText(els[els.length - 1]);
+        if (text.length > 10) return text;
       }
     }
-    debugInfo.push("ALL METHODS FAILED");
-    return JSON.stringify({ text: "", debug: debugInfo });
-  }, question);
-  let parsed;
-  try {
-    parsed = JSON.parse(reply);
-  } catch {
-    parsed = { text: reply, debug: [] };
-  }
-  for (const line of parsed.debug) {
-    log(verbose, `  [extract] ${line}`);
-  }
-  if (parsed.text && parsed.text.length > 0) {
-    log(verbose, `\u63D0\u53D6\u6210\u529F\uFF0C\u56DE\u590D\u957F\u5EA6: ${parsed.text.length} \u5B57\u7B26`);
-    return parsed.text;
-  }
-  throw new YiyanAgentError("TIMEOUT", "Failed to extract reply content");
+    const mdEls = document.querySelectorAll('[class*="markdown"], [class*="prose"]');
+    if (mdEls.length > 0) {
+      const text = getFullText(mdEls[mdEls.length - 1]);
+      if (text.length > 10) return text;
+    }
+    return "";
+  });
+}
+async function extractReply(page, question, verbose = false) {
+  log(verbose, "\u63D0\u53D6 AI \u56DE\u590D...");
+  const text = await extractLastMessage(page);
+  if (text.length > 0) {
+    const cleaned = text.replace(/^参考\d+个网页\s*/, "").replace(/^深度思考已完成\s*/, "").replace(/^思考完成[:：]\s*/, "").replace(/\n{3,}/g, "\n\n").trim();
+    log(verbose, `\u63D0\u53D6\u6210\u529F: ${cleaned.length} \u5B57\u7B26`);
+    return cleaned;
+  }
+  throw new YiyanAgentError("TIMEOUT", "Failed to extract reply");
+}
+async function dumpDebugInfo(page) {
+  const info = await page.evaluate(() => {
+    const classFreq = {};
+    document.querySelectorAll("*").forEach((el) => {
+      el.classList.forEach((c) => {
+        if (c.match(/message|chat|input|send|stop|markdown|content|assistant|user|bot|answer/i)) {
+          classFreq[c] = (classFreq[c] || 0) + 1;
+        }
+      });
+    });
+    const inputs = Array.from(document.querySelectorAll("textarea, [contenteditable]")).map((e) => ({
+      tag: e.tagName,
+      id: e.id || null,
+      class: (e.className?.toString() || "").substring(0, 80),
+      placeholder: e.placeholder || null,
+      editable: e.isContentEditable,
+      visible: e.offsetParent !== null
+    }));
+    return {
+      url: window.location.href,
+      title: document.title,
+      classes: Object.entries(classFreq).sort((a, b) => b[1] - a[1]).slice(0, 40),
+      inputs
+    };
+  });
+  console.log("\n" + "\u2550".repeat(60));
+  console.log("  DOM DEBUG INFO");
+  console.log("\u2550".repeat(60));
+  console.log("URL   :", info.url);
+  console.log("Title :", info.title);
+  console.log("\nInput elements:");
+  info.inputs.forEach((i) => console.log(" ", JSON.stringify(i)));
+  console.log("\nMatching CSS classes (by frequency):");
+  info.classes.forEach(([cls, count]) => console.log(`  ${String(count).padStart(3)}x  .${cls}`));
+  console.log("\u2550".repeat(60) + "\n");
 }
 
 // src/agent.ts
@@ -448,7 +471,7 @@ var YiyanAgent = class {
         } else {
           throw new YiyanAgentError(
             "CAPTCHA",
-            'Not logged in. Please run "yiyan-agent login" first, or use headed mode.'
+            'Not logged in. Please run "yiyan-browser-agent login" first, or use headed mode.'
           );
         }
       }
@@ -473,14 +496,14 @@ var YiyanAgent = class {
     });
     try {
       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");
+      process.stderr.write("\n[yiyan-browser-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-browser-agent] \u2551  \u{1F510}  LOGIN REQUIRED                          \u2551\n");
+      process.stderr.write("[yiyan-browser-agent] \u2551                                              \u2551\n");
+      process.stderr.write("[yiyan-browser-agent] \u2551  1. Log in to Yiyan in the browser window     \u2551\n");
+      process.stderr.write("[yiyan-browser-agent] \u2551  2. Return here and press ENTER to continue  \u2551\n");
+      process.stderr.write("[yiyan-browser-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");
+      process.stderr.write("[yiyan-browser-agent] \u2713 Login complete, saving session...\n");
       await new Promise((resolve) => setTimeout(resolve, 3e3));
     } finally {
       await closeBrowser(context, this.verbose);
@@ -522,24 +545,52 @@ var YiyanAgent = class {
       sessionPath: this.sessionDir
     };
   }
+  /**
+   * Debug 模式：启动浏览器并输出 DOM 信息
+   */
+  async debug() {
+    const { context, page } = await launchBrowser({
+      sessionDir: this.sessionDir,
+      headless: false,
+      timeout: this.options.timeout,
+      verbose: true
+    });
+    try {
+      await navigateToYiyan(page, true);
+      await page.waitForTimeout(3e3);
+      await dumpDebugInfo(page);
+      process.stderr.write("\n[yiyan-browser-agent] \u6D4F\u89C8\u5668\u4FDD\u6301\u6253\u5F00\uFF0C\u53EF\u624B\u52A8\u6D4B\u8BD5\u3002\n");
+      process.stderr.write("[yiyan-browser-agent] \u5173\u95ED\u6D4F\u89C8\u5668\u7A97\u53E3\u7ED3\u675F debug \u6A21\u5F0F...\n");
+      await new Promise((resolve) => {
+        context.on("close", () => resolve());
+      });
+    } finally {
+      try {
+        await context.close();
+      } catch {
+      }
+    }
+  }
 };
 
 // src/cli.ts
 function printHelp() {
   console.log(`
-yiyan-agent - \u6587\u5FC3\u4E00\u8A00\u6D4F\u89C8\u5668\u4EE3\u7406 CLI
+yiyan-browser-agent - \u6587\u5FC3\u4E00\u8A00\u6D4F\u89C8\u5668\u4EE3\u7406 CLI
 
 \u7528\u6CD5:
-  yiyan-agent login                    \u9996\u6B21\u767B\u5F55\u6587\u5FC3\u4E00\u8A00\uFF08\u4F1A\u6253\u5F00\u6D4F\u89C8\u5668\u7A97\u53E3\uFF09
-  yiyan-agent ask "\u95EE\u9898" [--timeout ms] [--headful] [--verbose]
-  yiyan-agent status                   \u68C0\u67E5\u767B\u5F55\u72B6\u6001
-  yiyan-agent reset                    \u6E05\u9664\u4FDD\u5B58\u7684 session
+  yiyan-browser-agent login                    \u9996\u6B21\u767B\u5F55\u6587\u5FC3\u4E00\u8A00\uFF08\u4F1A\u6253\u5F00\u6D4F\u89C8\u5668\u7A97\u53E3\uFF09
+  yiyan-browser-agent ask "\u95EE\u9898" [--timeout ms] [--headful] [--verbose]
+  yiyan-browser-agent status                   \u68C0\u67E5\u767B\u5F55\u72B6\u6001
+  yiyan-browser-agent reset                    \u6E05\u9664\u4FDD\u5B58\u7684 session
+  yiyan-browser-agent debug                    \u8C03\u8BD5\u6A21\u5F0F\uFF08\u8F93\u51FA DOM \u4FE1\u606F\uFF09
 
 \u547D\u4EE4:
   login      \u6253\u5F00\u6D4F\u89C8\u5668\u624B\u52A8\u767B\u5F55\u6587\u5FC3\u4E00\u8A00\uFF08\u9996\u6B21\u4F7F\u7528\u5FC5\u987B\u5148\u767B\u5F55\uFF09
   ask        \u53D1\u9001\u95EE\u9898\u5E76\u83B7\u53D6\u7B54\u6848\uFF08\u9ED8\u8BA4\u65E0\u5934\u6A21\u5F0F\uFF0C\u4E0D\u5F39\u7A97\uFF1B\u767B\u5F55\u540E\u76F4\u63A5\u4F7F\u7528\uFF09
   status     \u68C0\u67E5\u767B\u5F55\u72B6\u6001
   reset      \u6E05\u9664\u4FDD\u5B58\u7684 session
+  debug      \u8C03\u8BD5\u6A21\u5F0F\uFF1A\u542F\u52A8\u6D4F\u89C8\u5668\u8F93\u51FA DOM \u4FE1\u606F\uFF0C\u7528\u4E8E\u6392\u67E5\u9009\u62E9\u5668\u95EE\u9898
 
 \u9009\u9879:
   --timeout <ms>    \u8D85\u65F6\u65F6\u95F4\uFF08\u6BEB\u79D2\uFF09\uFF0C\u9ED8\u8BA4 120000
@@ -548,16 +599,17 @@ yiyan-agent - \u6587\u5FC3\u4E00\u8A00\u6D4F\u89C8\u5668\u4EE3\u7406 CLI
   --help            \u663E\u793A\u5E2E\u52A9\u4FE1\u606F
 
 \u793A\u4F8B:
-  yiyan-agent login                           # \u9996\u6B21\u4F7F\u7528\uFF1A\u767B\u5F55\u6587\u5FC3\u4E00\u8A00
-  yiyan-agent ask "\u4EC0\u4E48\u662F TypeScript\uFF1F"         # \u63D0\u95EE\uFF08\u9759\u9ED8\u6A21\u5F0F\uFF09
-  yiyan-agent ask "\u89E3\u91CA Promise" --verbose    # \u663E\u793A\u8BE6\u7EC6\u65E5\u5FD7
-  yiyan-agent ask "30+30=" --headful         # \u6709\u5934\u6A21\u5F0F\uFF08\u53EF\u624B\u52A8\u8FC7\u9A8C\u8BC1\u7801\uFF09
-  yiyan-agent status
-  yiyan-agent reset
+  yiyan-browser-agent login                           # \u9996\u6B21\u4F7F\u7528\uFF1A\u767B\u5F55\u6587\u5FC3\u4E00\u8A00
+  yiyan-browser-agent ask "\u4EC0\u4E48\u662F TypeScript\uFF1F"         # \u63D0\u95EE\uFF08\u9759\u9ED8\u6A21\u5F0F\uFF09
+  yiyan-browser-agent ask "\u89E3\u91CA Promise" --verbose    # \u663E\u793A\u8BE6\u7EC6\u65E5\u5FD7
+  yiyan-browser-agent ask "30+30=" --headful         # \u6709\u5934\u6A21\u5F0F\uFF08\u53EF\u624B\u52A8\u8FC7\u9A8C\u8BC1\u7801\uFF09
+  yiyan-browser-agent debug                          # \u8C03\u8BD5\u6A21\u5F0F\uFF0C\u8F93\u51FA DOM \u4FE1\u606F
+  yiyan-browser-agent status
+  yiyan-browser-agent reset
 
 \u6D41\u7A0B:
-  1. \u5148\u8FD0\u884C yiyan-agent login \u767B\u5F55\uFF08\u53EA\u9700\u4E00\u6B21\uFF0C\u767B\u5F55\u72B6\u6001\u4F1A\u4FDD\u5B58\uFF09
-  2. \u4E4B\u540E\u76F4\u63A5 yiyan-agent ask "\u95EE\u9898" \u5373\u53EF\uFF0C\u65E0\u9700\u518D\u767B\u5F55
+  1. \u5148\u8FD0\u884C yiyan-browser-agent login \u767B\u5F55\uFF08\u53EA\u9700\u4E00\u6B21\uFF0C\u767B\u5F55\u72B6\u6001\u4F1A\u4FDD\u5B58\uFF09
+  2. \u4E4B\u540E\u76F4\u63A5 yiyan-browser-agent ask "\u95EE\u9898" \u5373\u53EF\uFF0C\u65E0\u9700\u518D\u767B\u5F55
   3. \u5982\u679C\u9047\u5230\u9A8C\u8BC1\u7801\uFF0C\u52A0 --headful \u5207\u6362\u6709\u5934\u6A21\u5F0F\u624B\u52A8\u5904\u7406
 
 \u63D0\u793A:
@@ -591,6 +643,9 @@ function parseCliArgs(args) {
     if (arg === "login") {
       result.command = "login";
     }
+    if (arg === "debug") {
+      result.command = "debug";
+    }
     if (arg === "--timeout") {
       if (i + 1 < args.length) {
         result.timeout = parseInt(args[i + 1], 10);
@@ -663,6 +718,26 @@ async function runCli(args) {
     }));
     return;
   }
+  if (parsed.command === "debug") {
+    try {
+      await agent.debug();
+      console.log(formatCliOutput({
+        success: true,
+        question: "debug",
+        answer: "Debug session completed",
+        duration: 0
+      }));
+    } catch (error) {
+      const errorMessage = error instanceof Error ? error.message : String(error);
+      console.log(formatCliOutput({
+        success: false,
+        question: "debug",
+        error: errorMessage,
+        duration: 0
+      }));
+    }
+    return;
+  }
   if (parsed.command === "ask") {
     if (!parsed.question) {
       console.log(formatCliOutput({