yiyan-browser-agent 1.10.2 → 1.11.0

package/README.md

@@ -9,7 +9,7 @@
 
 **An autonomous AI coding agent that runs entirely for free — no API key required.**
 
-It drives a real browser to talk to [Yiyan (文心一言)](https://yiyan.baidu.com), giving you a Claude Code / Cursor-style coding agent powered by Baidu's AI models at zero cost.
+It drives a real browser to talk to [Yiyan (文心一言)](https://chat.baidu.com), giving you a Claude Code / Cursor-style coding agent powered by Baidu's AI models at zero cost.
 
 [Installation](#-installation) · [Quick Start](#-quick-start) · [Usage](#-usage) · [HTTP API](#-http-api) · [Configuration](#-configuration) · [Tools](#-available-tools) · [Contributing](#-contributing)
 
@@ -26,7 +26,7 @@ It drives a real browser to talk to [Yiyan (文心一言)](https://yiyan.baidu.c
 
 Most AI coding agents talk to a paid API. This one doesn't.
 
-Instead, it uses **Playwright** to control a real Chromium browser, navigates to `yiyan.baidu.com`, sends your task, waits for the response, and parses it to extract tool calls — all automatically. Your local files and terminal are wired up as tools the AI can use, so it can read code, write files, run commands, and and build complete projects step by step.
+Instead, it uses **Playwright** to control a real Chromium browser, navigates to `chat.baidu.com`, sends your task, waits for the response, and parses it to extract tool calls — all automatically. Your local files and terminal are wired up as tools the AI can use, so it can read code, write files, run commands, and and build complete projects step by step.
 
 ```
 Your Terminal
@@ -34,7 +34,7 @@ Your Terminal
      ▼
  Agent Core          ← orchestrates the loop
      │
-     ├──► Browser (Playwright)  ← talks to yiyan.baidu.com
+     ├──► Browser (Playwright)  ← talks to chat.baidu.com
      │         │
      │    Yiyan AI (文心一言)  ← thinks, decides what tool to use
      │         │
@@ -551,7 +551,7 @@ yiyan-browser-agent/
 ├── src/
 │   ├── index.js          ← CLI entry point and argument parsing
 │   ├── agent.js          ← Core agent loop (send → wait → parse → execute)
-│   ├── browser.js        ← Playwright controller for yiyan.baidu.com
+│   ├── browser.js        ← Playwright controller for chat.baidu.com
 │   ├── server.js         ← HTTP server for process communication (v1.5.0+)
 │   ├── client.js         ← HTTP client to forward tasks (v1.5.0+)
 │   ├── tools.js          ← All 15 filesystem and shell tools
@@ -611,7 +611,7 @@ Open an issue on GitHub with:
 
 ## ⚠️ Disclaimer
 
-This project automates a web browser to interact with yiyan.baidu.com. Automating web UIs may violate the terms of service of the website being automated. Use this tool for **personal and development purposes only**. The authors take no responsibility for account suspensions or other consequences of use.
+This project automates a web browser to interact with chat.baidu.com. Automating web UIs may violate the terms of service of the website being automated. Use this tool for **personal and development purposes only**. The authors take no responsibility for account suspensions or other consequences of use.
 
 ---
 

package/docs/USAGE_GUIDE.md

@@ -0,0 +1,366 @@
+# Yiyan Browser Agent 使用指南
+
+## 📖 完整使用流程
+
+### 第一步：首次登录（有头模式）
+
+**首次使用需要登录百度账号，必须使用有头模式：**
+
+```bash
+# 启动交互模式（有头，浏览器可见）
+yiyan-agent -i
+```
+
+**操作步骤：**
+
+1. 浏览器窗口自动打开 Yiyan 网站 (chat.baidu.com)
+2. 在浏览器中手动登录你的百度账号
+3. 登录成功后回到终端按 Enter 继续
+4. 会话会自动保存到 `~/.yiyan-agent/session/`
+5. **只需登录一次，后续自动复用**
+
+**为什么需要有头模式登录？**
+- 需要手动输入账号密码
+- 可能需要验证码
+- 需要看到登录界面操作
+
+---
+
+### 第二步：切换到后台服务（无头模式）
+
+**登录成功后，切换到无头模式后台运行：**
+
+```bash
+# 关闭当前有头模式服务器（Ctrl+C）
+
+# 启动无头交互服务器（后台运行）
+yiyan-agent -i --headless
+
+# 或使用 nohup 真正后台运行
+nohup yiyan-agent -i --headless > yiyan.log 2>&1 &
+```
+
+**无头模式优势：**
+- ✅ **无浏览器窗口**（节省资源）
+- ✅ **性能最优**（30-40% 加速）
+- ✅ **后台服务**（适合服务器部署）
+- ✅ **自动复用登录会话**（无需重复登录）
+
+---
+
+### 第三步：提交任务（客户端）
+
+**后台服务启动后，客户端提交任务：**
+
+#### 方式 1：命令行客户端
+
+```bash
+# 自动检测后台服务器并转发任务
+yiyan-agent "创建 Express REST API"
+
+# 无需浏览器，任务自动转发到后台服务器
+# 后台服务器使用保存的登录会话访问 Yiyan
+```
+
+#### 方式 2：HTTP API
+
+```bash
+# 直接调用后台服务器 HTTP 接口
+curl -X POST http://localhost:9527/task \
+  -H "Content-Type: application/json" \
+  -d '{"task":"创建基础 Express 服务器"}'
+```
+
+---
+
+## 🔄 模式切换流程图
+
+```
+┌─────────────────────────────────────────────────────────────┐
+│                     完整使用流程                              │
+└─────────────────────────────────────────────────────────────┘
+
+首次使用：
+  步骤 1: yiyan-agent -i                 (有头模式)
+          │
+          ├─ 浏览器打开 Yiyan 网站
+          ├─ 手动登录百度账号
+          ├─ 会话保存到 ~/.yiyan-agent/session/
+          │
+  步骤 2: Ctrl+C 关闭                    (停止有头服务器)
+          │
+  步骤 3: yiyan-agent -i --headless      (无头模式)
+          │
+          ├─ 后台运行，无浏览器窗口
+          ├─ 自动复用登录会话
+          ├─ 监听端口 9527
+          │
+  步骤 4: yiyan-agent "任务"              (客户端提交)
+          │
+          ├─ 自动转发到后台服务器
+          ├─ 后台服务器处理任务
+          └─ 返回结果到客户端
+
+后续使用：
+  yiyan-agent -i --headless              (直接启动无头服务)
+  yiyan-agent "任务"                      (提交任务)
+```
+
+---
+
+## 📊 模式对比表
+
+| 模式 | 命令 | 浏览器 | 性能 | 适用场景 |
+|------|------|--------|------|---------|
+| **有头交互** | `yiyan-agent -i` | ✅ 显示 | 标准 | 首次登录、调试 |
+| **无头交互** | `yiyan-agent -i --headless` | ❌ 隐藏 | ⚡ **最优** | 后台服务、生产环境 |
+| **单任务** | `yiyan-agent "任务"` | ❌ 隐藏 | ⚡ 快 | 自动转发到服务器 |
+
+---
+
+## 🚀 生产环境部署
+
+### Docker 部署
+
+```bash
+# 1. 创建 Dockerfile
+FROM node:18
+RUN npm install -g yiyan-browser-agent
+RUN npx playwright install chromium
+
+# 2. 启动容器（无头模式）
+docker run -d \
+  -p 9527:9527 \
+  -v ~/.yiyan-agent:/root/.yiyan-agent \
+  --name yiyan-agent \
+  yiyan-agent -i --headless
+
+# 3. 客户端提交任务
+curl -X POST http://localhost:9527/task \
+  -d '{"task":"创建项目"}'
+```
+
+### systemd 服务部署
+
+```bash
+# 1. 创建服务文件
+sudo nano /etc/systemd/system/yiyan-agent.service
+
+[Unit]
+Description=Yiyan Browser Agent Service
+After=network.target
+
+[Service]
+Type=simple
+User=your-user
+ExecStart=/usr/bin/yiyan-agent -i --headless
+Restart=on-failure
+RestartSec=10
+
+[Install]
+WantedBy=multi-user.target
+
+# 2. 启动服务
+sudo systemctl start yiyan-agent
+sudo systemctl enable yiyan-agent
+
+# 3. 查看日志
+sudo journalctl -u yiyan-agent -f
+```
+
+---
+
+## ⚙️ 配置优化
+
+### 性能配置（已默认启用）
+
+```json
+// ~/.yiyan-agent/config.json
+{
+  "STABLE_DELAY": 1500,      // 快速响应检测
+  "SEND_DELAY": 100,         // 快速发送
+  "RESPONSE_TIMEOUT": 120000, // 120s 超时
+  "HEADLESS": true           // 无头模式（性能优先）
+}
+```
+
+### 安全配置（已默认启用）
+
+```json
+{
+  "COMMAND_SECURITY_ENABLED": true,
+  "PATH_TRAVERSAL_PROTECTION": true,
+  "FILE_OVERWRITE_PROTECTION": true,
+  "FILE_BACKUP_ENABLED": true
+}
+```
+
+---
+
+## 🔧 常见问题
+
+### Q1: 如何重新登录？
+
+```bash
+# 1. 停止后台服务
+yiyan-agent --stop  # 或 Ctrl+C
+
+# 2. 删除旧会话
+rm -rf ~/.yiyan-agent/session/
+
+# 3. 有头模式重新登录
+yiyan-agent -i  # 浏览器打开，手动登录
+
+# 4. 登录后切换无头
+yiyan-agent -i --headless
+```
+
+### Q2: 后台服务如何调试？
+
+```bash
+# 临时使用有头模式查看问题
+yiyan-agent -i  # 查看浏览器操作
+
+# 或启用调试日志
+yiyan-agent -i --headless --debug
+# 查看 yiyan.log 日志文件
+```
+
+### Q3: 如何查看后台服务状态？
+
+```bash
+# 查看服务是否运行
+ps aux | grep yiyan-agent
+
+# 查看端口占用
+netstat -tunlp | grep 9527
+
+# 查看锁文件
+ls ~/.yiyan-agent/server.lock
+```
+
+### Q4: 如何停止后台服务？
+
+```bash
+# 方式 1: 找到进程并杀死
+ps aux | grep yiyan-agent
+kill <PID>
+
+# 方式 2: 删除锁文件（自动停止）
+rm ~/.yiyan-agent/server.lock
+
+# 方式 3: systemd 服务
+sudo systemctl stop yiyan-agent
+```
+
+---
+
+## 📁 文件位置
+
+### 会话文件
+
+```
+~/.yiyan-agent/
+├── session/                     ← 登录会话（cookies）
+│   └── Default/                 ← Chromium 会话数据
+├── logs/                        ← 日志文件
+│   ├── command-security.log     ← 命令执行审计
+│   ├── file-security.log        ← 文件操作审计
+│   └── yiyan.log                ← 服务日志（自定义）
+├── backups/                     ← 文件备份
+│   └── filename.timestamp.bak   ← 被覆盖的文件备份
+├── server.lock                  ← 后台服务器锁文件
+└── config.json                  ← 全局配置（可选）
+```
+
+---
+
+## 💡 最佳实践
+
+### 开发环境
+
+```bash
+# 调试模式（有头）
+yiyan-agent -i --debug
+
+# 可看到浏览器操作和详细日志
+```
+
+### 生产环境
+
+```bash
+# 后台服务（无头）
+nohup yiyan-agent -i --headless > /var/log/yiyan.log 2>&1 &
+
+# 客户端提交任务
+yiyan-agent "自动化任务"
+
+# 定期检查日志
+tail -f /var/log/yiyan.log
+```
+
+### 安全建议
+
+```bash
+# 1. 定期审查安全日志
+tail -100 ~/.yiyan-agent/logs/command-security.log
+
+# 2. 定期清理旧备份
+find ~/.yiyan-agent/backups -mtime +30 -delete
+
+# 3. 监控后台服务
+watch -n 5 'ps aux | grep yiyan-agent'
+```
+
+---
+
+## 🎯 快速参考
+
+```bash
+# 首次登录流程
+yiyan-agent -i              # 有头模式登录
+Ctrl+C                      # 关闭
+yiyan-agent -i --headless   # 无头后台服务
+
+# 日常使用
+yiyan-agent -i --headless   # 启动后台服务
+yiyan-agent "任务"          # 提交任务
+
+# 调试模式
+yiyan-agent -i              # 有头可视化
+yiyan-agent -i --debug      # 详细日志
+
+# 参数说明
+--headless      无头模式（强制）
+--show-browser  有头模式（强制）
+--debug         调试模式
+-i              交互模式
+```
+
+---
+
+## 📚 相关文档
+
+- [安全解决方案文档](SECURITY_SOLUTION.md) - 完整安全配置说明
+- [CHANGELOG.md](../CHANGELOG.md) - 版本更新历史
+- [README.md](../README.md) - 项目概述
+
+---
+
+## 🆘 获取帮助
+
+```bash
+# 命令行帮助
+yiyan-agent --help
+
+# GitHub Issues
+https://github.com/readfor/yiyan-browser-agent/issues
+
+# 日志位置
+~/.yiyan-agent/logs/
+```
+
+---
+
+**版本: 1.10.2**
+**最后更新: 2026-06-02**

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "yiyan-browser-agent",
-  "version": "1.10.2",
-  "description": "AI coding agent powered by Yiyan (文心一言) via browser automation — no API key needed. Performance-optimized (30-40% faster). New --headless parameter for interactive mode. Enhanced with comprehensive security.",
+  "version": "1.11.0",
+  "description": "AI coding agent powered by Yiyan (文心一言) via browser automation (chat.baidu.com) — no API key needed. Performance-optimized. Enhanced with comprehensive security.",
   "main": "src/index.js",
   "bin": {
     "yiyan-agent": "src/index.js",

package/src/browser.js

@@ -1,4 +1,4 @@
-// src/browser.js — Playwright controller for Yiyan (yiyan.baidu.com)
+// src/browser.js — Playwright controller for Yiyan (chat.baidu.com)
 // Performance optimized: smart waits, paste input, MutationObserver
 'use strict';
 
@@ -14,12 +14,11 @@ const CrashHandler = require('./stability/crash-handler');
 
 const SEL = {
   chatInput: [
-    '.editable__T7WAW4uW',
-    '[role="textbox"]',
-    '.editable',
-    '#chat-input',
+    '#chat-textarea',
+    'textarea.ci-textarea',
     'textarea[placeholder]',
     'textarea',
+    '[role="textbox"]',
     '[contenteditable="true"][role="textbox"]',
     '[contenteditable="true"]',
     '[class*="input-box"]',
@@ -46,12 +45,16 @@ const SEL = {
     'button[aria-label*="停止"]',
     '[aria-label*="stop generating" i]',
     '[data-testid="stop-button"]',
+    '[class*="stop-gen"]',
+    '[class*="stopGen"]',
     '[class*="stop-btn"]',
     '[class*="stopBtn"]',
     '[class*="abort"]',
   ],
 
   newChat: [
+    '.new-dialog-container-button',
+    '[class*="new-dialog-container"]',
     'button[aria-label*="New chat" i]',
     'button[aria-label*="新对话"]',
     'button[aria-label*="New conversation" i]',
@@ -61,15 +64,19 @@ const SEL = {
     '[class*="newChat"]',
   ],
 
-  // Response ready indicators
+  // Response ready indicators (chat.baidu.com)
   responseReady: [
+    '.ai-entry-block.ai-markdown',
+    '.answer-container',
+    '.cs-answer-container',
+    '.answer-box',
     '[class*="answer"]',
-    '[class*="response"]',
     '[class*="markdown"]',
-    '.ds-markdown',
   ],
 
   messageContainer: [
+    '#chat-container-main',
+    '[class*="chat-container"]',
     '[class*="chat-content"]',
     '[class*="message-list"]',
     '[class*="conversation"]',
@@ -178,17 +185,22 @@ class YiyanBrowser {
 
     const needsLogin = await this.page.evaluate(() => {
       const url = window.location.href;
+      // URL 中包含登录/认证路径 → 需要登录
+      if (url.includes('/auth') || url.includes('/login') || url.includes('/sign')) {
+        return true;
+      }
+      // 存在密码输入框 → 需要登录
+      if (document.querySelector('input[type="password"]')) {
+        return true;
+      }
+      // 页面主体区域被登录界面遮挡（排除侧栏的"请登录"提示）
       const bodyText = document.body?.innerText || '';
-      return (
-        url.includes('/auth') ||
-        url.includes('/login') ||
-        url.includes('/sign') ||
-        bodyText.includes('Sign in') ||
-        bodyText.includes('Log in') ||
-        bodyText.includes('登录') ||
-        bodyText.includes('登 录') ||
-        !!document.querySelector('input[type="password"]')
-      );
+      const mainInput = document.querySelector('#chat-textarea, textarea.ci-textarea');
+      if (!mainInput) {
+        // 输入框不存在可能是需要登录
+        return bodyText.includes('Sign in') || bodyText.includes('Log in');
+      }
+      return false;
     });
 
     if (needsLogin) {
@@ -235,18 +247,24 @@ class YiyanBrowser {
   // ── Sending Messages (stable: keyboard.type) ────────────────────────────────
 
   async sendMessage(text) {
-    const { el } = await this._findInput();
-
-    // Focus and select all existing content
-    await el.click({ clickCount: 3, force: true });
-    await this.page.waitForTimeout(100);
-
-    // Clear by pressing Delete
-    await this.page.keyboard.press('Delete');
-    await this.page.waitForTimeout(50);
-
-    // Type text character by character (stable, works reliably)
-    await this.page.keyboard.type(text, { delay: 10 });
+    const { el, isTextarea } = await this._findInput();
+
+    if (isTextarea) {
+      // textarea 元素：用 fill() 快速填充
+      await el.click({ force: true });
+      await this.page.waitForTimeout(100);
+      await el.fill('');
+      await this.page.waitForTimeout(50);
+      await el.fill(text);
+      await this.page.waitForTimeout(100);
+    } else {
+      // contenteditable 元素：用键盘输入
+      await el.click({ clickCount: 3, force: true });
+      await this.page.waitForTimeout(100);
+      await this.page.keyboard.press('Delete');
+      await this.page.waitForTimeout(50);
+      await this.page.keyboard.type(text, { delay: 10 });
+    }
 
     // Press Enter to send
     await this.page.keyboard.press('Enter');
@@ -308,9 +326,21 @@ class YiyanBrowser {
     while (Date.now() - start < timeout) {
       const text = await this._extractLastMessage();
 
-      // ── 思考区域内容检测 ──
+      // ── 思考区域内容检测 (chat.baidu.com) ──
       const thinkingInfo = await this.page.evaluate(() => {
-        const thinkingEl = document.querySelector('.container__SPpahQHm, [class*="container__SPpah"]');
+        // chat.baidu.com 深度思考区域
+        const thinkingSelectors = [
+          '[class*="deep-think"]',
+          '[class*="deepThink"]',
+          '[class*="deep-search"]',
+          '[class*="thinking-container"]',
+          '[class*="container__SPpah"]',  // 旧版兼容
+        ];
+        let thinkingEl = null;
+        for (const sel of thinkingSelectors) {
+          thinkingEl = document.querySelector(sel);
+          if (thinkingEl) break;
+        }
         if (!thinkingEl) return { text: '', exists: false };
 
         const s = window.getComputedStyle(thinkingEl);
@@ -320,17 +350,22 @@ class YiyanBrowser {
         return { text, exists: true };
       });
 
-      // ── 统一稳定性检测：两者都不变才算稳定 ──
+      // ── 统一稳定性检测：答案不变即算稳定 ──
       const thinkingStable = thinkingInfo.text === lastThinkingText;
       const answerStable   = text === lastText;
-      const hasContent     = thinkingInfo.text.length > 5 && text.length > 0;
-
-      if (thinkingStable && answerStable && hasContent) {
-        // 两者都稳定且有内容，检查稳定时间
+      // 如果有思考区域，需要两者都稳定；否则只看答案
+      const hasThinking    = thinkingInfo.exists && thinkingInfo.text.length > 0;
+      const hasContent     = text.length > 0;
+      const stable         = hasThinking
+        ? (thinkingStable && answerStable && hasContent)
+        : (answerStable && hasContent);
+
+      if (stable) {
+        // 稳定且有内容，检查稳定时间
         if (Date.now() - lastStableTime >= stableDelay) {
           stableCount++;
           lastStableTime = Date.now();
-          logger.dim(`Both stable: ${stableCount}/2 (thinking: ${thinkingInfo.text.length} chars, answer: ${text.length} chars)`);
+          logger.dim(`Stable: ${stableCount}/2 (answer: ${text.length} chars${hasThinking ? ', thinking: ' + thinkingInfo.text.length + ' chars' : ''})`);
         }
       } else {
         // 任一变化，重置
@@ -340,13 +375,19 @@ class YiyanBrowser {
         lastStableTime = Date.now();
       }
 
-      // ── 完成标记检测 ──
+      // ── 完成标记检测 (chat.baidu.com) ──
       const detected = await this.page.evaluate(() => {
         const selectors = [
+          '.cos-icon.cos-icon-copy',
+          '.cos-icon-copy',
+          '.feedback-hover-show',
+          '[class*="feedback-wrapper"]',
+          '.cos-icon.cos-icon-share1',
+          '.cos-icon.cos-icon-feedback',
+          // 旧版兼容
           '.dialogCardBottom__qoXjps3z',
           '[class*="dialogCardBottom"]',
           '[class*="dialog-bottom"]',
-          '[class*="response-footer"]',
         ];
         for (const sel of selectors) {
           const el = document.querySelector(sel);
@@ -400,16 +441,16 @@ class YiyanBrowser {
   async _getMessageCount() {
     return await this.page.evaluate(() => {
       const candidates = [
-        '[class*="answer"]',
-        '[class*="response"]',
-        '[class*="content"]',
-        '[class*="markdown"]',
-        '[class*="assistant"][class*="message"]',
-        '[data-role="assistant"]',
-        '[class*="markdown-content"]',
-        '.ds-markdown',
-        '[class*="chat-message"]',
-        '[class*="message-bubble"]',
+        '.ai-entry-block.ai-markdown',
+        '.answer-box',
+        '.cs-answer-container',
+        '.answer-container',
+        '.chat-search-answer-generate-item',
+        '[class*="answer-container"]',
+        '[class*="answer-box"]',
+        '.ai-markdown',
+        '.cs-question-bubble',
+        '[class*="question-bubble"]',
       ];
       for (const sel of candidates) {
         const els = document.querySelectorAll(sel);
@@ -435,10 +476,15 @@ class YiyanBrowser {
           if (node.nodeType !== Node.ELEMENT_NODE) return;
           const tag = node.tagName.toLowerCase();
 
-          // 排除思考过程区域
-          const cls = node.className || '';
-          if (cls.includes('container__SPpahQHm') || cls.includes('thinking') || cls.includes('Thinking')) {
-            return;  // 跳过思考区域
+          // 排除思考过程区域和 UI 建议区域
+          const cls = (typeof node.className === 'string') ? node.className : '';
+          if (cls.includes('thinking') || cls.includes('Thinking') ||
+              cls.includes('deep-search') || cls.includes('deep_think') ||
+              cls.includes('suggestion') || cls.includes('follow-up') ||
+              cls.includes('question-container') || cls.includes('question-bubble') ||
+              cls.includes('quick-entrance') || cls.includes('feedback-wrapper') ||
+              cls.includes('hover-menu') || cls.includes('action-bar')) {
+            return;  // 跳过思考区域和 UI 元素
           }
 
           if (tag === 'pre') {
@@ -474,62 +520,88 @@ class YiyanBrowser {
         return result.trim();
       }
 
-      // ── 从 answer_text_id 提取内容 ──
+      // ── chat.baidu.com 新选择器 ──
+
+      // 优先: 获取最后一个回答块的内容
+      const answerBlocks = document.querySelectorAll(
+        '.ai-entry-block.ai-markdown, .answer-container.cs-enable-selection, .cs-answer-container'
+      );
+      if (answerBlocks.length > 0) {
+        const lastBlock = answerBlocks[answerBlocks.length - 1];
+        const text = getFullText(lastBlock);
+        if (text.length > 0) return text;
+      }
+
+      // 回退 1: 通过 answer-box 获取
+      const answerBoxes = document.querySelectorAll('.answer-box, .last-answer-box');
+      if (answerBoxes.length > 0) {
+        const lastBox = answerBoxes[answerBoxes.length - 1];
+        const text = getFullText(lastBox);
+        if (text.length > 0) return text;
+      }
+
+      // 回退 2: 旧版 #answer_text_id (兼容)
       const answerEl = document.querySelector('#answer_text_id');
       if (answerEl) {
         return getFullText(answerEl);
       }
 
-      // answer_text_id 不存在，返回空字符串
+      // 回退 3: 通用 markdown 容器
+      const markdowns = document.querySelectorAll('.ai-markdown, [class*="markdown-content"]');
+      if (markdowns.length > 0) {
+        const lastMd = markdowns[markdowns.length - 1];
+        const text = getFullText(lastMd);
+        if (text.length > 0) return text;
+      }
+
       return '';
     });
   }
 
   async _isGenerating() {
     return await this.page.evaluate(() => {
-      // ── 1. 检测停止按钮/生成状态 UI ──
-      const stopSelectors = [
-        'button[aria-label*="Stop" i]',
-        'button[aria-label*="停止"]',
-        '[class*="stop-gen"]',
-        '[class*="stopGen"]',
+      // ── 1. 检测 typing/generating 指示器 (chat.baidu.com) ──
+      const typingSelectors = [
+        '.cosd-markdown-content-typingall',
+        '.markdown-typing-all',
+        '[class*="typing"]',
         '[class*="generating"]',
-        '[class*=" Generating"]',
+        '[class*="loading-indicator"]',
+        'svg[class*="loading"]',
+        'svg[class*="spinner"]',
+        '[class*="blink"]',
+        '[class*="cursor-blink"]',
+        '[class*="pulsing"]',
       ];
-      for (const sel of stopSelectors) {
+      for (const sel of typingSelectors) {
         const el = document.querySelector(sel);
         if (el) {
           const s = window.getComputedStyle(el);
-          if (s.display !== 'none' && s.visibility !== 'hidden' && s.opacity !== '0') return true;
+          if (s.display !== 'none' && s.visibility !== 'hidden') return true;
         }
       }
 
-      // ── 2. 检测加载动画/typing指示器 ──
-      const loaderSelectors = [
-        '[class*="typing"]',
-        '[class*="loading"]',
-        '[class*="spinner"]',
-        '[class*="blink"]',
-        '[class*="cursor"]',
-        '[class*="pulsing"]',
-        '[class*="thinking"]',
-        'svg[class*="loading"]',
-        'svg[class*="spinner"]',
-        '.loading-indicator',
-        '.generating-indicator',
+      // ── 2. 检测停止按钮 ──
+      const stopSelectors = [
+        'button[aria-label*="Stop" i]',
+        'button[aria-label*="停止"]',
+        '[class*="stop-gen"]',
+        '[class*="stopGen"]',
       ];
-      for (const sel of loaderSelectors) {
+      for (const sel of stopSelectors) {
         const el = document.querySelector(sel);
         if (el) {
           const s = window.getComputedStyle(el);
-          if (s.display !== 'none' && s.visibility !== 'hidden') return true;
+          if (s.display !== 'none' && s.visibility !== 'hidden' && s.opacity !== '0') return true;
         }
       }
 
-      // ── 3. 检测是否缺少完成标记元素 ──
+      // ── 3. 检测完成标记 — 如果有回答但没有完成标记 → 还在生成 ──
       const completionMarkers = [
-        '[class*="dialogCardBottom"]',
-        '.dialogCardBottom__qoXjps3z',
+        '.cos-icon.cos-icon-copy',
+        '.cos-icon-copy',
+        '.cos-icon.cos-icon-share1',
+        '.cos-icon-feedback',
         '[class*="copy-btn"]',
         '[class*="copyBtn"]',
         '[aria-label*="Copy" i]',
@@ -537,6 +609,7 @@ class YiyanBrowser {
         '[class*="regenerate"]',
         '[class*="retry"]',
         '[class*="action-btn"]',
+        '.feedback-hover-show',
       ];
       let hasCompletionMarker = false;
       for (const sel of completionMarkers) {
@@ -547,7 +620,9 @@ class YiyanBrowser {
       }
 
       // 如果有响应内容但没有完成标记 → 还在生成
-      const answerArea = document.querySelector('[class*="answer"], [class*="response"], [class*="markdown"]');
+      const answerArea = document.querySelector(
+        '.ai-entry-block.ai-markdown, .answer-container, .cs-answer-container, .answer-box'
+      );
       if (answerArea && answerArea.innerText && answerArea.innerText.length > 5) {
         if (!hasCompletionMarker) {
           return true;
@@ -568,29 +643,34 @@ class YiyanBrowser {
       '正在思考中',
       '正在思考',
       '思考过程',
-      '我来',
-      '我需要',
+      '深度思考',
       '根据搜索结果',
       '参考',
-      'picaole需要',
     ];
 
-    // Remove lines that contain thinking markers
-    for (const marker of thinkingMarkers) {
-      // 如果整行包含思考标记，移除该行
-      const lines = text.split('\n');
-      text = lines.filter(line => !line.includes(marker)).join('\n');
-    }
-
-    // Remove everything before "准备输出结果" (Yiyan's thinking process end marker)
+    // Remove lines that are standalone thinking/UI markers
+    const lines = text.split('\n');
+    text = lines.filter(line => {
+      const trimmed = line.trim();
+      // 移除纯 UI 标记行
+      if (thinkingMarkers.some(m => trimmed === m)) return false;
+      // 移除建议追问行（新 UI 的 follow-up 按钮）
+      if (/^(能否|能再|能帮|可以|请用|用一句|用一段)/.test(trimmed) && trimmed.length < 50) return false;
+      return true;
+    }).join('\n');
+
+    // Remove everything before "准备输出结果" (thinking process end marker)
     const outputMarker = '准备输出结果';
     const markerIndex = text.indexOf(outputMarker);
     if (markerIndex !== -1) {
       text = text.slice(markerIndex + outputMarker.length).trim();
     }
 
-    // Remove everything after regenerate/suggestion markers (Yiyan's UI elements)
-    const cutMarkers = ['重新生成', '重新生成的', '换个回答', '输出更详细的', '再多提供'];
+    // Remove everything after regenerate/suggestion markers (UI elements)
+    const cutMarkers = [
+      '重新生成', '重新生成的', '换个回答', '输出更详细的', '再多提供',
+      '内容由AI生成', '查看使用规则',
+    ];
     for (const marker of cutMarkers) {
       const cutIndex = text.indexOf(marker);
       if (cutIndex !== -1) {

package/src/config.js

@@ -8,7 +8,7 @@ const os   = require('os');
 // ─────────────────────────────────────────────
 const defaults = {
   // Browser
-  YIYAN_URL      : 'https://yiyan.baidu.com/',
+  YIYAN_URL      : 'https://chat.baidu.com/',
   SESSION_DIR    : path.join(os.homedir(), '.yiyan-agent', 'session'),
   HEADLESS       : true,  // Performance: 无头模式减少渲染开销
 

package/src/logger.js

@@ -44,7 +44,7 @@ const logger = {
 ${c('cyan','╔══════════════════════════════════════════════════╗')}
 ${c('cyan','║')}   ${cb('lcyan','🤖  Yiyan Browser Agent (文心一言)')}              ${c('cyan','║')}
 ${c('cyan','║')}   ${c('gray','AI Coding Agent via Browser Automation')}         ${c('cyan','║')}
-${c('cyan','║')}   ${c('gray','No API key needed — uses yiyan.baidu.com')}       ${c('cyan','║')}
+${c('cyan','║')}   ${c('gray','No API key needed — uses chat.baidu.com')}       ${c('cyan','║')}
 ${c('cyan','╚══════════════════════════════════════════════════╝')}
 `);
   },