android-midscene-automation 0.1.23 → 0.1.25

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # 更新记录
 
+## v0.1.25
+
+- Android 设备预览恢复使用 scrcpy 实时流，优化实时流参数，减少 H.264 丢帧导致的花屏；刷新画面时会重启底层 Playground 预览流并等待会话就绪。
+- 测试脚本生成新增提示词模板，可在通用提示词、广告测试、表单填写、聊天对话、播放器测试、搜索列表、订单支付、权限弹窗、个人设置和 H5/WebView 等场景间切换，并支持编辑或上传当前原始 Prompt。
+- 优化 AI 生成流程，支持后台处理和终止生成；生成中锁定原始 Prompt，再次点击“AI 生成”可恢复进度弹窗，终止后不再弹出额外提示。
+- 优化脚本生成提示词策略，区分广告、播放器、弹窗等被测目标与无关遮挡内容，登录流程增加登录后遮挡弹窗处理和无遮挡首页成功判定。
+
+## v0.1.24
+
+- 参数配置的运行配置新增保存校验，Android SDK 路径必须包含 `platform-tools/adb`，回放报告目录必须填写已存在且可写入的绝对路径。
+- Midscene 自定义提供方的 `Model Name` 和 `Model Family` 改为输入框，`Base URL` 匹配已知提供方时自动填充模型名称和模型系列，并在字段旁新增官方配置说明入口。
+
 ## v0.1.23
 
 - 常见问题拆分为独立 `FAQ.md` 文档，并在 README 的“功能文档”中增加跳转入口。
@@ -17,10 +29,6 @@
 - 连接脚本回放时会跳过子脚本开始后的重复“启动 App”和“清除 App”操作，减少跨脚本连接时的重复初始化。
 - 移除“连接到下一节点”“取消连接”“继续主流程”等冗余连接操作，保留复制节点作为主要复用方式。
 - 修复放大面板后点击操作按钮或节点配置会重置缩放的问题，保持当前画布大小和位置。
-- Android Playground 改为后台启动 5800 服务，不再自动打开 `http://localhost:5800/` 浏览器窗口。
-- CLI 启动时输出 GitHub 和 Issues 地址，方便用户反馈问题和提交建议。
-- 常见问题新增桌面点击 App 录制说明：部分厂商系统无法获取桌面图标稳定 id，建议使用开始节点后的“启动 App”操作替代。
-- 补充 GitHub 仓库、问题反馈和项目主页信息。
 
 ## v0.1.20
 

package/README.md

@@ -52,7 +52,7 @@ npx android-midscene-automation --port 5174
 ## 功能文档
 
 - [Appium 录制器使用说明](#appium-录制器使用说明)
-- [常见问题](./FAQ.md)
+- [常见问题](#常见问题)
 - [项目更新记录](#项目更新记录)
 
 ## 源码开发
@@ -165,7 +165,7 @@ midscene_run/                  # Midscene 执行报告和运行产物
 
 本文档说明 Appium 组件树录制器的完整使用方式，包括每个操作按钮的用途、录制方法和示例。
 
-本文档随 npm 包发布，也可以从 README 的“功能文档”直接打开。常见问题请查看 [FAQ.md](./FAQ.md)。
+本文档随 npm 包发布，也可以从 README 的“功能文档”直接打开。常见问题请查看 [常见问题](#常见问题)。
 
 ## 使用前准备
 
@@ -1106,9 +1106,6 @@ output/2026-08-21_17-13-42-831-登录流程.html
 - 校验页面返回路径。
 - 校验返回后列表仍可见。
 
-## 常见问题
-
-常见问题已独立整理到 [FAQ.md](./FAQ.md)。
 
 ## 推荐录制规范
 
@@ -1120,8 +1117,112 @@ output/2026-08-21_17-13-42-831-登录流程.html
 - 给关键节点填写“登录按钮”“账号输入框”等备注，方便查看流程和报告。
 - 回放失败后优先查看“回放输出”中的失败节点、Activity 和 selector，再打开 `output` 中的 Markdown 报告查看完整配置。
 
+# 常见问题
+
+## 同一个 id 定位到错误输入框怎么办
+
+如果账号框和密码框都是 `id/tg_edit`，只按 id 回放可能输入到第一个输入框。
+
+建议：
+
+- 直接分别选择账号框和密码框录制输入。
+- 在节点详情里确认 selector 是否显示“重复 N”，以及“推荐定位”是否显示父级 + 子级。
+- 如果推荐定位不准确，展开录制步骤，手动修改父级上下文 selector。
+- 坐标点击只作为最后兜底。
+
+## 点击后没有检测到页面跳转怎么办
+
+可能原因：
+
+- 按钮当前禁用。
+- 账号密码不正确。
+- App 使用单 Activity，Activity 不变化。
+- 页面跳转依赖网络。
+
+推荐：
+
+- 使用“断言存在”判断下个页面核心元素。
+- 多 Activity 页面可以拆成两个脚本，在判断分支中用“连接脚本”串起来；回放会等待目标 Activity 出现。
+- 单 Activity App 不要依赖 Activity 变化，应使用页面核心元素判断是否已进入目标状态。
+
+## 为什么连接脚本提示入口 Activity 不匹配
+
+- 主流程连接要求目标脚本入口 Activity 与当前插入点一致，适合复用同一页面上的公共流程。
+- 登录后进入首页等跨 Activity 场景，应先添加“判断存在”，再在“是”或“否”分支中添加“连接脚本”。
+- 分支连接允许选择同一 App 的其他 Activity，但不会主动跳转页面；前面的点击或输入必须真正触发跳转。
+- 如果页面未跳转，回放会在等待目标 Activity 超时后失败，这是为了防止在错误页面执行目标脚本。
+
+## 为什么在桌面点击 App 后回放无效
+
+部分厂商系统会限制桌面或启动器页面的控件信息，Appium 可能无法获取桌面图标的稳定 id。此时在桌面录制“点击 App 图标”可能无法回放。
+
+建议在开始节点后添加“启动 App”操作，由系统按包名打开目标 App，替代桌面点击图标。需要重置登录态或缓存时，可以先添加“清理 App 缓存”，再添加“启动 App”。
+
+## 弹窗只出现一次怎么办
+
+使用“判断存在”。
+
+```text
+判断存在 确认按钮
+  是 -> 点击 确认按钮
+  否 -> 不添加操作
+```
+
+不要直接把弹窗确认按钮作为普通必选点击，否则弹窗不出现时脚本会失败。
+
+## 什么时候用等待 Activity
+
+适合多 Activity App。
+
+如果 App 是单 Activity 架构，优先用：
+
+```text
+断言存在 页面核心元素
+```
+
+## 什么时候用可选步骤
+
+可选步骤适合非主流程阻塞项：
+
+- 权限弹窗
+- 协议弹窗
+- 活动弹窗
+- 首次引导
+
+不建议把登录按钮、提交按钮、核心断言设置为可选。
+
+## 回放时出现 UiAutomation not connected 怎么办
+
+组件树刷新和 Appium 回放不能同时占用 UiAutomation。当前版本会在回放期间暂停组件树自动刷新，并在断开时清理残留抓取进程后重试一次。
+
+如果仍然失败：
+
+1. 确认手机保持解锁，USB 调试授权没有失效。
+2. 确认 Appium 和 UiAutomator2 Driver 已正常启动。
+3. 停止其他正在抓取同一设备组件树的工具。
+4. 重新连接设备后再次回放。
+
+## 设备预览或组件树没有更新怎么办
+
+1. 先确认设备仍显示在设备下拉框中。
+2. 点击设备预览刷新按钮重新获取画面。
+3. 点击“刷新组件树”重新抓取页面结构。
+4. 回放期间组件树自动刷新会暂停，回放完成后会自动恢复。
+
 # 项目更新记录
 
+## v0.1.25
+
+- Android 设备预览恢复使用 scrcpy 实时流，优化实时流参数，减少 H.264 丢帧导致的花屏；刷新画面时会重启底层 Playground 预览流并等待会话就绪。
+- 测试脚本生成新增提示词模板，可在通用提示词、广告测试、表单填写、聊天对话、播放器测试、搜索列表、订单支付、权限弹窗、个人设置和 H5/WebView 等场景间切换，并支持编辑或上传当前原始 Prompt。
+- 优化 AI 生成流程，支持后台处理和终止生成；生成中锁定原始 Prompt，再次点击“AI 生成”可恢复进度弹窗，终止后不再弹出额外提示。
+- 优化脚本生成提示词策略，区分广告、播放器、弹窗等被测目标与无关遮挡内容，登录流程增加登录后遮挡弹窗处理和无遮挡首页成功判定。
+
+## v0.1.24
+
+- 参数配置的运行配置新增保存校验，Android SDK 路径必须包含 `platform-tools/adb`，回放报告目录必须填写已存在且可写入的绝对路径。
+- Midscene 自定义提供方的 `Model Name` 和 `Model Family` 改为输入框，`Base URL` 匹配已知提供方时自动填充模型名称和模型系列，并在字段旁新增官方配置说明入口。
+
 ## v0.1.23
 
 - 常见问题拆分为独立 `FAQ.md` 文档，并在 README 的“功能文档”中增加跳转入口。
@@ -1139,10 +1240,6 @@ output/2026-08-21_17-13-42-831-登录流程.html
 - 连接脚本回放时会跳过子脚本开始后的重复“启动 App”和“清除 App”操作，减少跨脚本连接时的重复初始化。
 - 移除“连接到下一节点”“取消连接”“继续主流程”等冗余连接操作，保留复制节点作为主要复用方式。
 - 修复放大面板后点击操作按钮或节点配置会重置缩放的问题，保持当前画布大小和位置。
-- Android Playground 改为后台启动 5800 服务，不再自动打开 `http://localhost:5800/` 浏览器窗口。
-- CLI 启动时输出 GitHub 和 Issues 地址，方便用户反馈问题和提交建议。
-- 常见问题新增桌面点击 App 录制说明：部分厂商系统无法获取桌面图标稳定 id，建议使用开始节点后的“启动 App”操作替代。
-- 补充 GitHub 仓库、问题反馈和项目主页信息。
 
 ## v0.1.20
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "android-midscene-automation",
-  "version": "0.1.23",
+  "version": "0.1.25",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/oooooooko/android-midscene-automation.git"

package/server/config.ts

@@ -1,4 +1,5 @@
 import fs from 'node:fs';
+import os from 'node:os';
 import path from 'node:path';
 import YAML from 'yaml';
 import { appDataPath } from './paths';
@@ -31,6 +32,14 @@ export type AppConfig = {
 let cachedConfig: AppConfig | null = null;
 const configPath = appDataPath('config.json');
 const legacyConfigPath = appDataPath('config.yaml');
+const adbFileName = process.platform === 'win32' ? 'adb.exe' : 'adb';
+
+export class ConfigValidationError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = 'ConfigValidationError';
+  }
+}
 
 function defaultConfig(): AppConfig {
   return {
@@ -95,6 +104,65 @@ function normalizeConfig(config: Partial<AppConfig> | null | undefined): AppConf
   };
 }
 
+function expandRuntimePath(value: string) {
+  const trimmed = value.trim().replace(/^(["'])(.*)\1$/, '$2');
+  if (trimmed === '~') return os.homedir();
+  if (trimmed.startsWith(`~${path.sep}`) || trimmed.startsWith('~/')) {
+    return path.join(os.homedir(), trimmed.slice(2));
+  }
+  return path.isAbsolute(trimmed) ? path.normalize(trimmed) : appDataPath(trimmed);
+}
+
+function isAbsoluteRuntimePath(value: string) {
+  const trimmed = value.trim().replace(/^(["'])(.*)\1$/, '$2');
+  return trimmed === '~'
+    || trimmed.startsWith(`~${path.sep}`)
+    || trimmed.startsWith('~/')
+    || path.isAbsolute(trimmed);
+}
+
+function validateAndroidSdkPath(value: string) {
+  if (!value) return;
+  const candidate = expandRuntimePath(value);
+  const lowerBaseName = path.basename(candidate).toLowerCase();
+  const root = lowerBaseName === adbFileName.toLowerCase()
+    ? path.dirname(path.dirname(candidate))
+    : lowerBaseName === 'platform-tools'
+      ? path.dirname(candidate)
+      : candidate;
+  const adbPath = path.join(root, 'platform-tools', adbFileName);
+  if (!fs.existsSync(adbPath)) {
+    throw new ConfigValidationError(
+      `Android SDK 路径无效：${value}。请选择包含 platform-tools/${adbFileName} 的 SDK 目录。`,
+    );
+  }
+}
+
+function validateReportOutputPath(value: string) {
+  if (!value) return;
+  if (!isAbsoluteRuntimePath(value)) {
+    throw new ConfigValidationError(`回放报告目录无效：${value}。请填写绝对路径，或留空使用默认 output 目录。`);
+  }
+  const outputDir = expandRuntimePath(value);
+  try {
+    if (!fs.existsSync(outputDir)) {
+      throw new Error('目录不存在');
+    }
+    if (!fs.statSync(outputDir).isDirectory()) {
+      throw new Error('不是目录');
+    }
+    fs.accessSync(outputDir, fs.constants.W_OK);
+  } catch (error) {
+    const detail = error instanceof Error && error.message ? `（${error.message}）` : '';
+    throw new ConfigValidationError(`回放报告目录无效：${value}。请选择已经存在并可写入的目录${detail}。`);
+  }
+}
+
+function validateRuntimeConfig(config: AppConfig) {
+  validateAndroidSdkPath(config.runtime.androidSdkPath);
+  validateReportOutputPath(config.runtime.reportOutputPath);
+}
+
 export function loadConfig(): AppConfig {
   if (cachedConfig) {
     return cachedConfig;
@@ -126,7 +194,9 @@ export function loadConfig(): AppConfig {
 }
 
 export function saveConfig(config: AppConfig) {
-  cachedConfig = normalizeConfig(config);
+  const normalized = normalizeConfig(config);
+  validateRuntimeConfig(normalized);
+  cachedConfig = normalized;
   saveModelConfigToDb(cachedConfig);
 }
 

package/server/http-api.ts

@@ -9,12 +9,10 @@ type NextHandleFunction = (req: IncomingMessage, res: ServerResponse, next: Next
 
 import { execFile, spawn } from 'node:child_process';
 import { appPath } from './paths';
-import { loadConfig, saveConfig, type AppConfig } from './config';
+import { ConfigValidationError, loadConfig, saveConfig, type AppConfig } from './config';
 import { listAppPresetRecords, removeAppPresetRecord, saveAppPresetRecord } from './config-store';
 import { generatePlan } from './script-agent';
-import { importMidsceneModelUsage } from './model-call-usage-importer';
 import { testModelConnection } from './model-tester';
-import { listModelUsageRecords, saveModelUsageRecord } from './model-usage-repository';
 import { DeviceLockConflictError } from './device-locks/types';
 import { acquireDeviceLock, listActiveDeviceLocks, releaseDeviceLock } from './device-locks/service';
 import { ensureDeviceSession, listDeviceSessions } from './device-sessions/service';
@@ -95,12 +93,38 @@ function writeNdjson(res: ServerResponse, event: RunGeneratedScriptEvent | { typ
   res.write(`${JSON.stringify(event)}\n`);
 }
 
+function delay(ms: number) {
+  return new Promise<void>((resolve) => {
+    setTimeout(resolve, ms);
+  });
+}
+
+// Work around @midscene/android-playground v1.12.x ESM __dirname usage on Node 24.
 const START_ANDROID_PLAYGROUND_WITHOUT_BROWSER = [
-  "import { androidPlaygroundPlatform, ScrcpyServer } from '@midscene/android-playground';",
-  "import { launchPreparedPlaygroundPlatform } from '@midscene/playground';",
-  'const scrcpyServer = new ScrcpyServer();',
-  'const prepared = await androidPlaygroundPlatform.prepare({ scrcpyServer });',
-  'await launchPreparedPlaygroundPlatform(prepared);',
+  "import { createRequire } from 'node:module';",
+  'const require = createRequire(import.meta.url);',
+  "const { androidPlaygroundPlatform, ScrcpyServer } = require('@midscene/android-playground');",
+  "const { launchPreparedPlaygroundPlatform } = require('@midscene/playground');",
+  'const STABLE_PREVIEW_OPTIONS = {',
+  '  maxFps: 15,',
+  '  maxSize: 1024,',
+  '  videoBitRate: 3000000,',
+  "  videoCodecOptions: 'i-frame-interval=0',",
+  '};',
+  'async function main() {',
+  '  const scrcpyServer = new ScrcpyServer();',
+  '  const startScrcpy = scrcpyServer.startScrcpy.bind(scrcpyServer);',
+  '  scrcpyServer.startScrcpy = (adb, options = {}, onProgress) => startScrcpy(adb, {',
+  '    ...options,',
+  '    ...STABLE_PREVIEW_OPTIONS,',
+  '  }, onProgress);',
+  '  const prepared = await androidPlaygroundPlatform.prepare({ scrcpyServer });',
+  '  await launchPreparedPlaygroundPlatform(prepared);',
+  '}',
+  "main().catch((error) => {",
+  "  console.error(error);",
+  "  process.exit(1);",
+  "});",
 ].join('\n');
 
 type AndroidDevice = {
@@ -458,6 +482,24 @@ async function proxyPlaygroundRequest(req: IncomingMessage, res: ServerResponse,
     return;
   }
 
+  if (targetPath === '/mjpeg' && response.body) {
+    const reader = response.body.getReader();
+    req.once('close', () => {
+      reader.cancel().catch(() => undefined);
+    });
+    try {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        if (value) res.write(Buffer.from(value));
+      }
+      res.end();
+    } catch {
+      if (!res.writableEnded) res.end();
+    }
+    return;
+  }
+
   const buffer = Buffer.from(await response.arrayBuffer());
   res.end(buffer);
 }
@@ -557,6 +599,39 @@ export function createApiMiddleware() {
       });
   };
 
+  const stopAndroidPlayground = async () => {
+    const processToStop = playgroundProcess;
+    if (!processToStop || processToStop.killed) return;
+
+    await new Promise<void>((resolve) => {
+      const timeout = setTimeout(resolve, 1500);
+      processToStop.once('exit', () => {
+        clearTimeout(timeout);
+        resolve();
+      });
+      processToStop.kill('SIGTERM');
+    });
+    if (playgroundProcess === processToStop) {
+      playgroundProcess = null;
+    }
+  };
+
+  const waitForUsablePlaygroundTarget = async (selectedDeviceId: string, timeoutMs = 30000) => {
+    const deadline = Date.now() + timeoutMs;
+    let resolved: PlaygroundTarget = null;
+    while (Date.now() < deadline) {
+      resolved = await resolvePlaygroundUrl(selectedDeviceId);
+      if (resolved && selectedDeviceId && !isUsablePlaygroundTarget(resolved, selectedDeviceId) && !resolved.sessionConnected) {
+        resolved = await createPlaygroundSession(resolved, selectedDeviceId) || resolved;
+      }
+      if (isUsablePlaygroundTarget(resolved, selectedDeviceId)) {
+        return resolved;
+      }
+      await delay(500);
+    }
+    return resolved;
+  };
+
   const handler: NextHandleFunction = async (req, res, next) => {
     const requestUrl = new URL(req.url || '/', 'http://localhost');
     const pathname = requestUrl.pathname;
@@ -648,7 +723,7 @@ export function createApiMiddleware() {
         res.setHeader('Content-Type', 'application/json; charset=utf-8');
         res.end(JSON.stringify({ success: true }));
       } catch (error) {
-        res.statusCode = 500;
+        res.statusCode = error instanceof ConfigValidationError ? 400 : 500;
         res.setHeader('Content-Type', 'application/json; charset=utf-8');
         res.end(JSON.stringify({ message: error instanceof Error ? error.message : 'Unknown error' }));
       }
@@ -947,16 +1022,53 @@ export function createApiMiddleware() {
       return;
     }
 
+    if (req.url === '/api/playground-restart' && req.method === 'POST') {
+      try {
+        const payload = await readBody<{ deviceId?: string }>(req);
+        const deviceId = payload.deviceId || selectedDeviceId;
+        if (!deviceId) {
+          throw new Error('请先选择 Android 设备');
+        }
+        if (isRemoteDeviceId(deviceId)) {
+          throw new Error('远程设备不支持重启本地实时预览');
+        }
+
+        setSelectedDeviceId(req, res, deviceId);
+        selectedDeviceId = deviceId;
+        playgroundEnsurePromise = null;
+        lastPlaygroundEnsureAt = 0;
+        lastPlaygroundEnsureDeviceId = '';
+        await stopAndroidPlayground();
+        const result = await ensureAndroidPlayground(deviceId);
+        const target = result.target || await waitForUsablePlaygroundTarget(deviceId);
+
+        ensureDeviceSession({
+          deviceId,
+          patch: {
+            playgroundUrl: target?.url || '',
+            previewKind: target?.previewKind || '',
+            sessionConnected: target?.sessionConnected || false,
+            setupState: target?.setupState || result.skippedReason || '',
+            previewError: target?.previewError || '',
+          },
+        });
+        res.setHeader('Content-Type', 'application/json; charset=utf-8');
+        res.end(JSON.stringify({
+          success: true,
+          started: result.started,
+          available: isUsablePlaygroundTarget(target, deviceId),
+          url: target?.url || '',
+          skippedReason: result.skippedReason,
+        }));
+      } catch (error) {
+        res.statusCode = 500;
+        res.setHeader('Content-Type', 'application/json; charset=utf-8');
+        res.end(JSON.stringify({ message: error instanceof Error ? error.message : '重启 Android Playground 失败' }));
+      }
+      return;
+    }
+
     if (req.url === '/api/test-model' && req.method === 'POST') {
-      const startedAt = Date.now();
-      let modelKey: 'midscene' | 'scriptOptimizer' = 'midscene';
-      let model: {
-        provider?: 'custom' | 'codex';
-        baseUrl?: string;
-        apiKey?: string;
-        name?: string;
-        family?: string;
-      } = {};
       try {
         const parsed = await readBody<{
           modelKey?: 'midscene' | 'scriptOptimizer';
@@ -968,8 +1080,7 @@ export function createApiMiddleware() {
             family?: string;
           };
         }>(req);
-        model = parsed.model || {};
-        modelKey = parsed.modelKey === 'scriptOptimizer' ? 'scriptOptimizer' : 'midscene';
+        const model = parsed.model || {};
 
         const result = await testModelConnection({
           provider: model.provider,
@@ -978,42 +1089,8 @@ export function createApiMiddleware() {
           name: model.name || '',
           family: model.family || '',
         });
-        saveModelUsageRecord({
-          modelKey,
-          provider: modelKey === 'midscene' ? model.provider || 'custom' : 'custom',
-          modelName: model.name || '',
-          family: model.family || '',
-          success: true,
-          durationMs: result.durationMs ?? Date.now() - startedAt,
-          promptTokens: result.usage?.promptTokens,
-          completionTokens: result.usage?.completionTokens,
-          totalTokens: result.usage?.totalTokens,
-        });
         res.setHeader('Content-Type', 'application/json; charset=utf-8');
         res.end(JSON.stringify(result));
-      } catch (error) {
-        saveModelUsageRecord({
-          modelKey,
-          provider: modelKey === 'midscene' ? model.provider || 'custom' : 'custom',
-          modelName: model.name || '',
-          family: model.family || '',
-          success: false,
-          durationMs: Date.now() - startedAt,
-          errorMessage: error instanceof Error ? error.message : 'Unknown error',
-        });
-        res.statusCode = 500;
-        res.setHeader('Content-Type', 'application/json; charset=utf-8');
-        res.end(JSON.stringify({ message: error instanceof Error ? error.message : 'Unknown error' }));
-      }
-      return;
-    }
-
-    if (req.url?.startsWith('/api/model-usage-records') && req.method === 'GET') {
-      try {
-        const requestUrl = new URL(req.url, 'http://localhost');
-        const limit = Number(requestUrl.searchParams.get('limit') || 50);
-        res.setHeader('Content-Type', 'application/json; charset=utf-8');
-        res.end(JSON.stringify({ records: listModelUsageRecords(limit) }));
       } catch (error) {
         res.statusCode = 500;
         res.setHeader('Content-Type', 'application/json; charset=utf-8');
@@ -1175,43 +1252,23 @@ export function createApiMiddleware() {
 
     try {
       if (req.url === '/api/generate' && req.method === 'POST') {
-        const startedAt = Date.now();
-        const config = loadConfig();
         const parsed = await readBody<{
           prompt?: string;
         }>(req);
+        const generateAbortController = new AbortController();
+        res.on('close', () => {
+          if (!res.writableEnded) {
+            generateAbortController.abort();
+          }
+        });
 
-        try {
-          const result = await generatePlan({
-            prompt: parsed.prompt || '',
-          });
-
-          saveModelUsageRecord({
-            modelKey: 'scriptOptimizer',
-            provider: 'custom',
-            modelName: config.scriptOptimizer.model.name || '',
-            family: '',
-            success: true,
-            durationMs: result.durationMs || Date.now() - startedAt,
-            promptTokens: result.usage?.promptTokens,
-            completionTokens: result.usage?.completionTokens,
-            totalTokens: result.usage?.totalTokens,
-          });
+        const result = await generatePlan({
+          prompt: parsed.prompt || '',
+          signal: generateAbortController.signal,
+        });
 
-          res.setHeader('Content-Type', 'application/json; charset=utf-8');
-          res.end(JSON.stringify(result));
-        } catch (error) {
-          saveModelUsageRecord({
-            modelKey: 'scriptOptimizer',
-            provider: 'custom',
-            modelName: config.scriptOptimizer.model.name || '',
-            family: '',
-            success: false,
-            durationMs: Date.now() - startedAt,
-            errorMessage: error instanceof Error ? error.message : 'AI 生成失败',
-          });
-          throw error;
-        }
+        res.setHeader('Content-Type', 'application/json; charset=utf-8');
+        res.end(JSON.stringify(result));
         return;
       }
 
@@ -1281,7 +1338,6 @@ export function createApiMiddleware() {
         res.setHeader('X-Accel-Buffering', 'no');
         writeNdjson(res, { type: 'operation', operation });
 
-        const usageImportSinceMs = Date.now();
         try {
           const result = await runGeneratedScript({
             code: parsed.code || '',
@@ -1316,26 +1372,6 @@ export function createApiMiddleware() {
               writeNdjson(res, event);
             },
           });
-          try {
-            const usageImportResult = importMidsceneModelUsage({
-              sinceMs: usageImportSinceMs,
-              model: loadConfig().midscene.model,
-              success: result.success,
-            });
-            if (usageImportResult.imported > 0) {
-              recordOperationEvent({
-                operationId: operation.id,
-                eventType: 'model_usage_imported',
-                data: usageImportResult,
-              });
-            }
-          } catch (error) {
-            recordOperationEvent({
-              operationId: operation.id,
-              eventType: 'model_usage_import_failed',
-              message: error instanceof Error ? error.message : '模型消耗导入失败',
-            });
-          }
           finishScriptRunOperation({
             operationId: operation.id,
             status: result.success ? 'succeeded' : runAbortController.signal.aborted ? 'cancelled' : 'failed',