qat-cli 0.3.3 → 0.3.5

package/dist/index.js

@@ -2524,6 +2524,15 @@ var NoopAIProvider = class {
 };
 
 // src/ai/openai-provider.ts
+import chalk2 from "chalk";
+function isDebug() {
+  return process.env.QAT_DEBUG === "true";
+}
+function debugLog(tag, ...args) {
+  if (!isDebug()) return;
+  const timestamp = (/* @__PURE__ */ new Date()).toISOString().slice(11, 19);
+  console.error(chalk2.gray(`[DEBUG ${timestamp}] [${tag}]`), ...args);
+}
 var OpenAICompatibleProvider = class {
   constructor(config) {
     this.capabilities = {
@@ -2537,25 +2546,20 @@ var OpenAICompatibleProvider = class {
     this.baseUrl = config.baseUrl || this.getDefaultBaseUrl(config.provider);
   }
   async generateTest(req) {
+    debugLog("GENERATE", `type=${req.type} target=${req.target}`);
     const systemPrompt = this.buildGenerateTestSystemPrompt(req);
     const userPrompt = this.buildGenerateTestUserPrompt(req);
     const content = await this.chat(systemPrompt, userPrompt);
-    return this.parseGenerateTestResponse(content);
+    const result = this.parseGenerateTestResponse(content);
+    debugLog("GENERATE", `done code=${result.code.length}chars confidence=${result.confidence}`);
+    return result;
   }
   async analyzeResult(req) {
-    const systemPrompt = `\u4F60\u662F\u4E00\u4E2A\u4E13\u4E1A\u7684\u6D4B\u8BD5\u5206\u6790\u4E13\u5BB6\u3002\u5206\u6790\u6D4B\u8BD5\u8FD0\u884C\u7ED3\u679C\uFF0C\u627E\u51FA\u95EE\u9898\u6839\u56E0\uFF0C\u7ED9\u51FA\u5177\u4F53\u53EF\u64CD\u4F5C\u7684\u6539\u8FDB\u5EFA\u8BAE\u3002
-\u8F93\u51FA\u683C\u5F0F\uFF1A
-1. \u5206\u6790\u6458\u8981\uFF081-3\u53E5\u8BDD\uFF09
-2. \u6539\u8FDB\u5EFA\u8BAE\u5217\u8868\uFF08\u6BCF\u6761\u5EFA\u8BAE\u5177\u4F53\u3001\u53EF\u64CD\u4F5C\uFF09`;
-    const resultSummary = req.testResults.map((r) => {
-      const failed = r.suites.flatMap((s) => s.tests.filter((t) => t.status === "failed"));
-      return `\u7C7B\u578B: ${r.type}, \u72B6\u6001: ${r.status}, \u8017\u65F6: ${r.duration}ms, \u5931\u8D25\u7528\u4F8B: ${failed.length}`;
-    }).join("\n");
-    const errorDetails = req.errorLogs?.join("\n") || req.testResults.flatMap((r) => r.suites.flatMap((s) => s.tests.filter((t) => t.status === "failed" && t.error))).map((t) => `[${t.name}] ${t.error?.message}`).join("\n") || "\u65E0\u9519\u8BEF\u8BE6\u60C5";
-    const userPrompt = `\u6D4B\u8BD5\u7ED3\u679C:
-${resultSummary}
-
-\u9519\u8BEF\u8BE6\u60C5:
+    const systemPrompt = `\u6D4B\u8BD5\u5206\u6790\u4E13\u5BB6\u3002\u627E\u95EE\u9898\u6839\u56E0\uFF0C\u7ED9\u53EF\u64CD\u4F5C\u5EFA\u8BAE\u3002
+\u8F93\u51FA:1.\u6458\u8981(1-3\u53E5) 2.\u5EFA\u8BAE\u5217\u8868`;
+    const failed = req.testResults.flatMap((r) => r.suites.flatMap((s) => s.tests.filter((t) => t.status === "failed")));
+    const errorDetails = req.errorLogs?.join("\n") || failed.map((t) => `[${t.name}] ${t.error?.message}`).join("\n") || "\u65E0";
+    const userPrompt = `\u5931\u8D25:${failed.length}
 ${errorDetails}`;
     const content = await this.chat(systemPrompt, userPrompt);
     return {
@@ -2565,23 +2569,22 @@ ${errorDetails}`;
     };
   }
   async suggestFix(error) {
-    const systemPrompt = `\u4F60\u662F\u4E00\u4E2A\u4E13\u4E1A\u7684\u4EE3\u7801\u4FEE\u590D\u4E13\u5BB6\u3002\u6839\u636E\u6D4B\u8BD5\u9519\u8BEF\u4FE1\u606F\uFF0C\u7ED9\u51FA\u5177\u4F53\u7684\u4FEE\u590D\u5EFA\u8BAE\u3002
-\u6BCF\u6761\u5EFA\u8BAE\u5E94\u8BE5\u5305\u542B\uFF1A
-1. \u95EE\u9898\u5B9A\u4F4D
-2. \u4FEE\u590D\u65B9\u6848
-3. \u793A\u4F8B\u4EE3\u7801\uFF08\u5982\u679C\u9002\u7528\uFF09`;
-    const userPrompt = `\u9519\u8BEF\u4FE1\u606F: ${error.message}
-${error.stack ? `\u5806\u6808: ${error.stack}` : ""}
-${error.expected ? `\u671F\u671B\u503C: ${error.expected}` : ""}
-${error.actual ? `\u5B9E\u9645\u503C: ${error.actual}` : ""}`;
+    const systemPrompt = `\u4EE3\u7801\u4FEE\u590D\u4E13\u5BB6\u3002\u7ED9\u51FA:1.\u95EE\u9898\u5B9A\u4F4D 2.\u4FEE\u590D\u65B9\u6848 3.\u793A\u4F8B\u4EE3\u7801`;
+    const userPrompt = `\u9519\u8BEF:${error.message}${error.stack ? `
+\u5806\u6808:${error.stack}` : ""}${error.expected ? `
+\u671F\u671B:${error.expected}` : ""}${error.actual ? `
+\u5B9E\u9645:${error.actual}` : ""}`;
     const content = await this.chat(systemPrompt, userPrompt);
     return content.split("\n").filter((l) => l.trim().startsWith("-") || l.trim().startsWith("\u2022") || l.trim().match(/^\d+\./)).map((l) => l.replace(/^[-•\d.]+\s*/, "").trim()).filter(Boolean);
   }
   async reviewTest(req) {
+    debugLog("REVIEW", `target=${req.target} type=${req.testType}`);
     const systemPrompt = this.buildReviewTestSystemPrompt(req);
     const userPrompt = this.buildReviewTestUserPrompt(req);
     const content = await this.chat(systemPrompt, userPrompt);
-    return this.parseReviewTestResponse(content);
+    const result = this.parseReviewTestResponse(content);
+    debugLog("REVIEW", `approved=${result.approved} score=${result.score} feedback=${result.feedback}`);
+    return result;
   }
   // ─── 内部方法 ──────────────────────────────────────────────
   /**
@@ -2645,8 +2648,9 @@ ${error.actual ? `\u5B9E\u9645\u503C: ${error.actual}` : ""}`;
       return { ok: false, message: `\u8FDE\u63A5\u5931\u8D25: ${error instanceof Error ? error.message : String(error)}`, latencyMs };
     }
   }
-  async chat(systemPrompt, userPrompt) {
+  async chat(systemPrompt, userPrompt, retries = 2) {
     const url = `${this.baseUrl}/chat/completions`;
+    const useStream = isDebug();
     const body = {
       model: this.model,
       messages: [
@@ -2656,101 +2660,202 @@ ${error.actual ? `\u5B9E\u9645\u503C: ${error.actual}` : ""}`;
       temperature: 0.3,
       max_tokens: 4096
     };
+    if (useStream) {
+      body.stream = true;
+    }
     const headers = {
       "Content-Type": "application/json"
     };
     if (this.apiKey) {
       headers["Authorization"] = `Bearer ${this.apiKey}`;
     }
-    const response = await fetch(url, {
-      method: "POST",
-      headers,
-      body: JSON.stringify(body),
-      signal: AbortSignal.timeout(6e4)
-      // 60s timeout
-    });
-    if (!response.ok) {
-      const text = await response.text().catch(() => "");
-      throw new Error(`AI API \u8BF7\u6C42\u5931\u8D25 (${response.status}): ${text.slice(0, 500)}`);
+    debugLog("REQUEST", `POST ${url}`);
+    debugLog("REQUEST", `model=${this.model} stream=${useStream}`);
+    debugLog("SYSTEM", systemPrompt.length > 500 ? `${systemPrompt.slice(0, 500)}...` : systemPrompt);
+    debugLog("USER", userPrompt.length > 1e3 ? `${userPrompt.slice(0, 1e3)}...` : userPrompt);
+    let lastError = null;
+    for (let attempt = 0; attempt <= retries; attempt++) {
+      try {
+        if (attempt > 0) {
+          debugLog("RETRY", `\u7B2C${attempt}\u6B21\u91CD\u8BD5...`);
+        }
+        const response = await fetch(url, {
+          method: "POST",
+          headers,
+          body: JSON.stringify(body),
+          signal: AbortSignal.timeout(12e4)
+          // 120s timeout
+        });
+        if (!response.ok) {
+          const text = await response.text().catch(() => "");
+          if ((response.status === 429 || response.status >= 500) && attempt < retries) {
+            const delay = Math.min(1e3 * Math.pow(2, attempt), 8e3);
+            debugLog("RETRY", `HTTP ${response.status}, ${delay}ms\u540E\u91CD\u8BD5`);
+            await new Promise((r) => setTimeout(r, delay));
+            lastError = new Error(`AI API \u8BF7\u6C42\u5931\u8D25 (${response.status}): ${text.slice(0, 200)}`);
+            continue;
+          }
+          throw new Error(`AI API \u8BF7\u6C42\u5931\u8D25 (${response.status}): ${text.slice(0, 500)}`);
+        }
+        if (useStream && response.body) {
+          const content = await this.readStream(response.body);
+          debugLog("RESPONSE", content.length > 500 ? `${content.slice(0, 500)}...` : content);
+          return content;
+        }
+        const data = await response.json();
+        if (!data.choices?.[0]?.message?.content) {
+          throw new Error("AI API \u8FD4\u56DE\u7A7A\u54CD\u5E94");
+        }
+        debugLog("RESPONSE", `tokens: prompt=${data.usage?.prompt_tokens} completion=${data.usage?.completion_tokens} total=${data.usage?.total_tokens}`);
+        debugLog("RESPONSE", data.choices[0].message.content.length > 500 ? `${data.choices[0].message.content.slice(0, 500)}...` : data.choices[0].message.content);
+        return data.choices[0].message.content;
+      } catch (error) {
+        if (error instanceof Error && error.name === "TimeoutError" && attempt < retries) {
+          debugLog("TIMEOUT", `\u7B2C${attempt}\u6B21\u8D85\u65F6\uFF0C\u91CD\u8BD5\u4E2D...`);
+          lastError = error;
+          continue;
+        }
+        throw error;
+      }
     }
-    const data = await response.json();
-    if (!data.choices?.[0]?.message?.content) {
-      throw new Error("AI API \u8FD4\u56DE\u7A7A\u54CD\u5E94");
+    throw lastError || new Error("AI API \u8BF7\u6C42\u5931\u8D25");
+  }
+  /**
+   * 读取 SSE 流式响应，实时输出内容
+   */
+  async readStream(body) {
+    const chunks = [];
+    const decoder = new TextDecoder();
+    const reader = body.getReader();
+    let buffer = "";
+    let lineCount = 0;
+    try {
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        buffer += decoder.decode(value, { stream: true });
+        const lines = buffer.split("\n");
+        buffer = lines.pop() || "";
+        for (const line of lines) {
+          const trimmed = line.trim();
+          if (!trimmed || trimmed === "data: [DONE]") continue;
+          if (!trimmed.startsWith("data: ")) continue;
+          try {
+            const json = JSON.parse(trimmed.slice(6));
+            const delta = json.choices?.[0]?.delta?.content;
+            if (delta) {
+              chunks.push(delta);
+              lineCount++;
+              process.stderr.write(chalk2.gray(delta));
+              if (lineCount % 20 === 0) {
+                process.stderr.write("\n");
+              }
+            }
+            if (json.choices?.[0]?.finish_reason === "stop") {
+              debugLog("STREAM", "\u5B8C\u6210");
+            }
+          } catch {
+          }
+        }
+      }
+    } finally {
+      reader.releaseLock();
+    }
+    if (chunks.length > 0) {
+      process.stderr.write("\n");
+    }
+    return chunks.join("");
+  }
+  /**
+   * 压缩源码：保留签名和关键逻辑，剔除注释、空行、样式块
+   * 目标：在保留准确性的前提下减少 token 消耗
+   * @param code 源码内容
+   * @param maxLength 最大长度
+   * @param importPathRewrites import 路径重写映射（原路径→正确路径）
+   */
+  compressSourceCode(code, maxLength = 3e3, importPathRewrites) {
+    let compressed = code;
+    if (importPathRewrites && importPathRewrites.size > 0) {
+      compressed = this.rewriteImportPaths(compressed, importPathRewrites);
     }
-    return data.choices[0].message.content;
+    compressed = compressed.replace(/<style[^>]*>[\s\S]*?<\/style>/gi, "");
+    compressed = compressed.replace(/<template[^>]*>([\s\S]*?)<\/template>/gi, (_match, content) => {
+      return `<template>${content.replace(/\s*(?:class|style)\s*=\s*["'][^"']*["']/gi, "")}</template>`;
+    });
+    compressed = compressed.replace(/\/\*[\s\S]*?\*\//g, "");
+    compressed = compressed.replace(/(^|[^:])(\/\/.*$)/gm, "$1");
+    compressed = compressed.replace(/\n\s*\n\s*\n/g, "\n\n");
+    compressed = compressed.split("\n").map((line) => line.trimEnd()).join("\n").trim();
+    if (compressed.length > maxLength) {
+      const scriptMatch = compressed.match(/<script[^>]*>([\s\S]*?)<\/script>/i);
+      const templateMatch = compressed.match(/<template[^>]*>([\s\S]*?)<\/template>/i);
+      if (scriptMatch) {
+        let scriptPart = scriptMatch[1].trim();
+        scriptPart = compressFunctionBodies(scriptPart, maxLength * 0.7);
+        const templatePart = templateMatch ? `<template>${templateMatch[1].replace(/\s+/g, " ").trim().slice(0, 300)}...</template>` : "";
+        compressed = `${templatePart}
+<script${scriptMatch[0].match(/<script[^>]*>/)?.[0]?.slice(7) || ">"}>${scriptPart}</script>`;
+      } else {
+        compressed = compressFunctionBodies(compressed, maxLength);
+      }
+    }
+    return compressed;
   }
   buildGenerateTestSystemPrompt(req) {
     const typeMap = {
-      unit: "\u5355\u5143\u6D4B\u8BD5\uFF08Vitest + @vue/test-utils\uFF09",
-      component: "\u7EC4\u4EF6\u6D4B\u8BD5\uFF08Vitest + @vue/test-utils + mount\uFF09",
-      e2e: "E2E\u7AEF\u5230\u7AEF\u6D4B\u8BD5\uFF08Playwright\uFF09",
-      api: "API\u63A5\u53E3\u6D4B\u8BD5\uFF08Vitest + fetch\uFF09",
-      visual: "\u89C6\u89C9\u56DE\u5F52\u6D4B\u8BD5\uFF08Playwright screenshot\uFF09",
-      performance: "\u6027\u80FD\u6D4B\u8BD5\uFF08Playwright + performance metrics\uFF09"
+      unit: "\u5355\u5143\u6D4B\u8BD5(Vitest)",
+      component: "\u7EC4\u4EF6\u6D4B\u8BD5(Vitest+@vue/test-utils)",
+      e2e: "E2E\u6D4B\u8BD5(Playwright)",
+      api: "API\u6D4B\u8BD5(Vitest+fetch)",
+      visual: "\u89C6\u89C9\u56DE\u5F52\u6D4B\u8BD5(Playwright)",
+      performance: "\u6027\u80FD\u6D4B\u8BD5(Playwright)"
     };
-    return `\u4F60\u662F\u4E00\u4E2A\u4E13\u4E1A\u7684\u524D\u7AEF\u6D4B\u8BD5\u5DE5\u7A0B\u5E08\uFF0C\u64C5\u957F\u7F16\u5199\u9AD8\u8D28\u91CF\u7684${typeMap[req.type] || req.type}\u3002
-\u8981\u6C42\uFF1A
-1. \u53EA\u8F93\u51FA\u6D4B\u8BD5\u4EE3\u7801\uFF0C\u4E0D\u8981\u591A\u4F59\u7684\u89E3\u91CA
-2. \u4EE3\u7801\u5FC5\u987B\u53EF\u76F4\u63A5\u8FD0\u884C\uFF0C\u5305\u542B\u6240\u6709\u5FC5\u8981\u7684 import
-3. \u6D4B\u8BD5\u7528\u4F8B\u8986\u76D6\uFF1A\u6B63\u5E38\u8DEF\u5F84\u3001\u8FB9\u754C\u6761\u4EF6\u3001\u9519\u8BEF\u5904\u7406
-4. \u4F7F\u7528\u4E2D\u6587\u63CF\u8FF0 it/test \u5757\u540D\u79F0
-5. Vue \u7EC4\u4EF6\u6D4B\u8BD5\u4F7F\u7528 @vue/test-utils \u7684 mount
-6. \u5982\u679C\u6709 props/emits \u4FE1\u606F\uFF0C\u52A1\u5FC5\u9488\u5BF9\u6BCF\u4E2A prop \u548C emit \u751F\u6210\u6D4B\u8BD5`;
+    return `\u4F60\u662F\u524D\u7AEF\u6D4B\u8BD5\u5DE5\u7A0B\u5E08\uFF0C\u7F16\u5199${typeMap[req.type] || req.type}\u3002
+\u89C4\u5219:1.\u53EA\u8F93\u51FA\u6D4B\u8BD5\u4EE3\u7801 2.\u5305\u542B\u6240\u6709import 3.\u8986\u76D6\u6B63\u5E38/\u8FB9\u754C/\u9519\u8BEF 4.it\u540D\u7528\u4E2D\u6587 5.Vue\u7528mount 6.\u5FC5\u6D4B\u6BCF\u4E2Aprop\u548Cemit`;
   }
   buildGenerateTestUserPrompt(req) {
     const importPath = this.computeTestImportPath(req.type, req.target);
-    let prompt = `\u8BF7\u4E3A\u4EE5\u4E0B\u6587\u4EF6\u751F\u6210${req.type}\u6D4B\u8BD5\u4EE3\u7801:
-\u76EE\u6807\u6587\u4EF6: ${req.target}
-\u6D4B\u8BD5\u6587\u4EF6\u5C06\u653E\u5728: ${this.getTestOutputDir(req.type)}/
-\u6B63\u786E\u7684 import \u8DEF\u5F84: ${importPath}
-
-\u91CD\u8981\uFF1Aimport \u8BED\u53E5\u4E2D\u5FC5\u987B\u4F7F\u7528\u4E0A\u8FF0\u6B63\u786E\u7684\u76F8\u5BF9\u8DEF\u5F84 ${importPath}\uFF0C\u4E0D\u8981\u4F7F\u7528 ${req.target} \u6216\u5176\u4ED6\u8DEF\u5F84\uFF01
+    let prompt = `\u4E3A${req.target}\u751F\u6210${req.type}\u6D4B\u8BD5\u3002
+\u3010\u5F3A\u5236\u3011import\u88AB\u6D4B\u6A21\u5757\u5FC5\u987B\u7528: ${importPath}
 `;
     if (req.analysis) {
-      prompt += "\n\u6E90\u7801\u5206\u6790\u7ED3\u679C:\n";
+      const parts = [];
       if (req.analysis.exports?.length > 0) {
-        prompt += `\u5BFC\u51FA\u9879:
-${req.analysis.exports.map((e) => {
-          const params = e.params?.length ? `(${e.params.join(", ")})` : "";
-          const asyncFlag = e.isAsync ? "async " : "";
-          return `  - ${asyncFlag}${e.name}${params} [${e.kind}]`;
-        }).join("\n")}
-`;
+        parts.push(`\u5BFC\u51FA:${req.analysis.exports.map((e) => {
+          const p = e.params?.length ? `(${e.params.join(",")})` : "";
+          return `${e.isAsync ? "async " : ""}${e.name}${p}[${e.kind}]`;
+        }).join(",")}`);
       }
       if (req.analysis.props?.length) {
-        prompt += `Props:
-${req.analysis.props.map(
-          (p) => `  - ${p.name}: ${p.type}${p.required ? " (\u5FC5\u586B)" : " (\u53EF\u9009)"}`
-        ).join("\n")}
-`;
+        parts.push(`Props:${req.analysis.props.map((p) => `${p.name}:${p.type}${p.required ? "!" : "?"}`).join(",")}`);
       }
       if (req.analysis.emits?.length) {
-        prompt += `Emits:
-${req.analysis.emits.map(
-          (e) => `  - ${e.name}${e.params?.length ? `(${e.params.join(", ")})` : ""}`
-        ).join("\n")}
-`;
+        parts.push(`Emits:${req.analysis.emits.map((e) => `${e.name}(${e.params?.join(",") || ""})`).join(",")}`);
       }
       if (req.analysis.methods?.length) {
-        prompt += `Methods: ${req.analysis.methods.join(", ")}
-`;
+        parts.push(`Methods:${req.analysis.methods.join(",")}`);
       }
       if (req.analysis.computed?.length) {
-        prompt += `Computed: ${req.analysis.computed.join(", ")}
-`;
+        parts.push(`Computed:${req.analysis.computed.join(",")}`);
+      }
+      if (req.analysis.importSignatures?.length) {
+        parts.push(`\u4F9D\u8D56\u7B7E\u540D:${req.analysis.importSignatures.map(
+          (imp) => `${imp.source}{${Object.entries(imp.signatures).map(([k, v]) => `${k}:${v}`).join(",")}}`
+        ).join(";")}`);
       }
+      prompt += parts.join("\n") + "\n";
     }
     if (req.context) {
+      const importPathRewrites = this.buildImportPathRewrites(req.type, req.target);
       prompt += `
-\u6E90\u7801\u5185\u5BB9:
-\`\`\`typescript
-${req.context}
+\u6E90\u7801:
+\`\`\`
+${this.compressSourceCode(req.context, 3e3, importPathRewrites)}
 \`\`\`
 `;
     }
     if (req.framework) {
-      prompt += `
-\u6846\u67B6: ${req.framework}`;
+      prompt += `\u6846\u67B6:${req.framework}`;
     }
     return prompt;
   }
@@ -2781,18 +2886,54 @@ ${req.context}
     return `${prefix}${cleanPath}`;
   }
   /**
-   * 获取测试输出目录
+   * 构建 import 路径重写映射
+   * 将源码中可能引用自身的各种写法，映射到测试文件中正确的导入路径
    */
-  getTestOutputDir(testType) {
-    const dirMap = {
-      unit: "tests/unit",
-      component: "tests/component",
-      e2e: "tests/e2e",
-      api: "tests/api",
-      visual: "tests/visual",
-      performance: "tests/e2e"
-    };
-    return dirMap[testType] || "tests/unit";
+  buildImportPathRewrites(testType, targetPath) {
+    const rewrites = /* @__PURE__ */ new Map();
+    if (!targetPath) return rewrites;
+    const correctImportPath = this.computeTestImportPath(testType, targetPath);
+    const variations = /* @__PURE__ */ new Set();
+    variations.add(targetPath);
+    variations.add(targetPath.replace(/^\.\//, ""));
+    const withoutExt = targetPath.replace(/\.(ts|js|tsx|jsx)$/, "");
+    variations.add(withoutExt);
+    const srcDirMatch = targetPath.match(/^(?:\.\/)?(src\/.+)$/);
+    if (srcDirMatch) {
+      variations.add(`@/${srcDirMatch[1]}`);
+      variations.add(`@/${srcDirMatch[1].replace(/\.(ts|js|tsx|jsx)$/, "")}`);
+    }
+    const pathParts = targetPath.replace(/^\.\//, "").split("/");
+    const fileName = pathParts[pathParts.length - 1];
+    const fileNameNoExt = fileName.replace(/\.(ts|js|tsx|jsx|vue)$/, "");
+    variations.add(`./${fileName}`);
+    variations.add(`./${fileNameNoExt}`);
+    for (let i = 1; i < pathParts.length; i++) {
+      const relPath = "../".repeat(i) + pathParts.slice(i).join("/");
+      variations.add(relPath);
+      variations.add(relPath.replace(/\.(ts|js|tsx|jsx)$/, ""));
+    }
+    for (const variant of variations) {
+      if (variant && variant !== correctImportPath) {
+        rewrites.set(variant, correctImportPath);
+      }
+    }
+    return rewrites;
+  }
+  /**
+   * 重写源码中的 import 路径
+   * 将 from 'oldPath' / from "oldPath" 替换为正确的路径
+   */
+  rewriteImportPaths(code, rewrites) {
+    let result = code;
+    for (const [oldPath, newPath] of rewrites) {
+      const escaped = oldPath.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+      const singleQuoteRegex = new RegExp(`(from\\s+')${escaped}(')`, "g");
+      const doubleQuoteRegex = new RegExp(`(from\\s+")${escaped}(")`, "g");
+      result = result.replace(singleQuoteRegex, `$1${newPath}$2`);
+      result = result.replace(doubleQuoteRegex, `$1${newPath}$2`);
+    }
+    return result;
   }
   parseGenerateTestResponse(content) {
     const codeBlockMatch = content.match(/```(?:typescript|ts|javascript|js)?\s*\n([\s\S]*?)```/);
@@ -2805,69 +2946,56 @@ ${req.context}
     };
   }
   buildReviewTestSystemPrompt(_req) {
-    return `\u4F60\u662F\u4E00\u4F4D\u4E25\u8C28\u7684\u6D4B\u8BD5\u5BA1\u8BA1\u4E13\u5BB6\u3002\u4F60\u7684\u804C\u8D23\u662F\u5BA1\u67E5 AI \u751F\u6210\u7684\u6D4B\u8BD5\u7528\u4F8B\u662F\u5426\u4E0E\u6E90\u7801\u8D34\u5207\u4E14\u51C6\u786E\u3002
-
-\u5BA1\u67E5\u6807\u51C6\uFF1A
-1. **\u6D4B\u8BD5\u7C7B\u578B\u5339\u914D**\uFF1A\u6D4B\u8BD5\u7C7B\u578B\u662F\u5426\u4E0E\u6E90\u7801\u6587\u4EF6\u6027\u8D28\u5339\u914D\uFF08\u5982 .vue \u5E94\u4E3A\u7EC4\u4EF6\u6D4B\u8BD5\uFF0Cutils \u5E94\u4E3A\u5355\u5143\u6D4B\u8BD5\uFF09
-2. **\u8986\u76D6\u5B8C\u6574\u6027**\uFF1A\u662F\u5426\u8986\u76D6\u4E86\u6E90\u7801\u4E2D\u7684\u6838\u5FC3\u5BFC\u51FA\u9879\uFF08\u51FD\u6570\u3001\u7EC4\u4EF6\u7684 props/emits/methods\uFF09
-3. **\u65AD\u8A00\u6709\u6548\u6027**\uFF1A\u65AD\u8A00\u662F\u5426\u771F\u5B9E\u68C0\u9A8C\u4E86\u88AB\u6D4B\u884C\u4E3A\uFF0C\u800C\u975E\u7A7A\u65AD\u8A00\u6216\u6C38\u771F\u65AD\u8A00
-4. **\u5BFC\u5165\u6B63\u786E\u6027**\uFF1Aimport \u8DEF\u5F84\u548C\u6A21\u5757\u662F\u5426\u6B63\u786E
-5. **\u4EE3\u7801\u53EF\u8FD0\u884C\u6027**\uFF1A\u6D4B\u8BD5\u4EE3\u7801\u662F\u5426\u53EF\u76F4\u63A5\u8FD0\u884C\uFF0C\u65E0\u8BED\u6CD5\u9519\u8BEF
-
-\u8F93\u51FA\u683C\u5F0F\uFF08\u4E25\u683C\u9075\u5B88\uFF09\uFF1A
-APPROVED: true \u6216 false
-SCORE: 0.0 \u5230 1.0 \u4E4B\u95F4\u7684\u8BC4\u5206
-FEEDBACK: \u4E00\u53E5\u8BDD\u5BA1\u8BA1\u610F\u89C1
-ISSUES: \u95EE\u9898\u5217\u8868\uFF08\u6BCF\u884C\u4E00\u4E2A\uFF0C\u683C\u5F0F "- \u95EE\u9898\u63CF\u8FF0"\uFF09
-SUGGESTIONS: \u6539\u8FDB\u5EFA\u8BAE\u5217\u8868\uFF08\u6BCF\u884C\u4E00\u4E2A\uFF0C\u683C\u5F0F "- \u5EFA\u8BAE\u63CF\u8FF0"\uFF09`;
+    return `\u4F60\u662F\u6D4B\u8BD5\u5BA1\u8BA1\u4E13\u5BB6\u3002\u5BA1\u67E5\u6D4B\u8BD5\u4EE3\u7801\u662F\u5426\u51C6\u786E\u3002
+\u6807\u51C6:1.\u7C7B\u578B\u5339\u914D 2.\u8986\u76D6\u6838\u5FC3\u5BFC\u51FA/props/emits 3.\u65AD\u8A00\u6709\u6548(\u975E\u6C38\u771F) 4.import\u6B63\u786E 5.\u53EF\u8FD0\u884C
+\u8F93\u51FA:
+APPROVED:true\u6216false
+SCORE:0.0-1.0
+FEEDBACK:\u4E00\u53E5\u8BDD
+ISSUES:- \u95EE\u9898(\u6BCF\u884C\u4E00\u4E2A)
+SUGGESTIONS:- \u5EFA\u8BAE(\u6BCF\u884C\u4E00\u4E2A)`;
   }
   buildReviewTestUserPrompt(req) {
     const importPath = this.computeTestImportPath(req.testType, req.target);
-    let prompt = `\u8BF7\u5BA1\u67E5\u4EE5\u4E0B\u6D4B\u8BD5\u7528\u4F8B\u662F\u5426\u4E0E\u6E90\u7801\u8D34\u5207\u4E14\u51C6\u786E\u3002
-
-\u88AB\u6D4B\u6587\u4EF6: ${req.target}
-\u6D4B\u8BD5\u7C7B\u578B: ${req.testType}
-\u6B63\u786E\u7684 import \u8DEF\u5F84\u5E94\u4E3A: ${importPath}\uFF08\u6D4B\u8BD5\u6587\u4EF6\u4F4D\u4E8E ${this.getTestOutputDir(req.testType)}/\uFF09
-
---- \u6E90\u7801\u5185\u5BB9 ---
-\`\`\`typescript
-${req.sourceCode}
+    let prompt = `\u5BA1\u67E5\u6D4B\u8BD5\u4EE3\u7801\u3002\u88AB\u6D4B:${req.target} \u7C7B\u578B:${req.testType}
+\u3010\u5F3A\u5236\u3011import\u88AB\u6D4B\u6A21\u5757\u5FC5\u987B\u7528: ${importPath}
+`;
+    const importPathRewrites = this.buildImportPathRewrites(req.testType, req.target);
+    prompt += `
+\u6E90\u7801:
+\`\`\`
+${this.compressSourceCode(req.sourceCode, 2e3, importPathRewrites)}
+\`\`\`
+`;
+    prompt += `
+\u6D4B\u8BD5\u4EE3\u7801:
 \`\`\`
-
---- \u751F\u6210\u7684\u6D4B\u8BD5\u4EE3\u7801 ---
-\`\`\`typescript
 ${req.testCode}
 \`\`\``;
     if (req.analysis) {
-      prompt += "\n\n--- \u6E90\u7801\u5206\u6790 ---";
+      const parts = [];
       if (req.analysis.exports?.length > 0) {
-        prompt += `
-\u5BFC\u51FA\u9879:
-${req.analysis.exports.map((e) => {
-          const params = e.params?.length ? `(${e.params.join(", ")})` : "";
-          const asyncFlag = e.isAsync ? "async " : "";
-          return `  - ${asyncFlag}${e.name}${params} [${e.kind}]`;
-        }).join("\n")}`;
+        parts.push(`\u5BFC\u51FA:${req.analysis.exports.map((e) => `${e.isAsync ? "async " : ""}${e.name}(${e.params?.join(",") || ""})[${e.kind}]`).join(",")}`);
       }
       if (req.analysis.props?.length) {
-        prompt += `
-Props:
-${req.analysis.props.map(
-          (p) => `  - ${p.name}: ${p.type}${p.required ? " (\u5FC5\u586B)" : " (\u53EF\u9009)"}`
-        ).join("\n")}`;
+        parts.push(`Props:${req.analysis.props.map((p) => `${p.name}:${p.type}${p.required ? "!" : "?"}`).join(",")}`);
       }
       if (req.analysis.emits?.length) {
+        parts.push(`Emits:${req.analysis.emits.map((e) => `${e.name}(${e.params?.join(",") || ""})`).join(",")}`);
+      }
+      if (req.analysis.importSignatures?.length) {
+        parts.push(`\u4F9D\u8D56:${req.analysis.importSignatures.map(
+          (imp) => `${imp.source}{${Object.entries(imp.signatures).map(([k, v]) => `${k}:${v}`).join(",")}}`
+        ).join(";")}`);
+      }
+      if (parts.length > 0) {
         prompt += `
-Emits:
-${req.analysis.emits.map(
-          (e) => `  - ${e.name}${e.params?.length ? `(${e.params.join(", ")})` : ""}`
-        ).join("\n")}`;
+\u5206\u6790:${parts.join("|")}`;
       }
     }
     if (req.generationDescription) {
       prompt += `
-
-\u751F\u6210\u8005\u8BF4\u660E: ${req.generationDescription}`;
+\u8BF4\u660E:${req.generationDescription}`;
     }
     return prompt;
   }
@@ -2927,6 +3055,61 @@ ${req.analysis.emits.map(
     return urlMap[provider] || "https://api.openai.com/v1";
   }
 };
+function compressFunctionBodies(code, maxLength) {
+  const lines = code.split("\n");
+  const result = [];
+  let depth = 0;
+  let fnDepth = 0;
+  let inFunction = false;
+  let braceBalance = 0;
+  let capturedLines = 0;
+  for (const line of lines) {
+    const trimmed = line.trim();
+    for (const ch of trimmed) {
+      if (ch === "{") {
+        braceBalance++;
+        depth++;
+      }
+      if (ch === "}") {
+        braceBalance--;
+        depth = Math.max(0, depth - 1);
+      }
+    }
+    const isFnSignature = /^(export\s+)?(async\s+)?function\s|=>\s*\{|=>\s*$/m.test(trimmed) || /^(const|let|var)\s+\w+\s*=\s*(async\s+)?(\([^)]*\)|[^=])\s*=>/.test(trimmed);
+    if (isFnSignature && !inFunction) {
+      inFunction = true;
+      fnDepth = depth;
+      result.push(line);
+      capturedLines++;
+      continue;
+    }
+    if (inFunction) {
+      const isReturn = trimmed.startsWith("return ");
+      const isBranch = /^(if|else|switch|case|try|catch|finally|for|while)/.test(trimmed);
+      const isClosing = trimmed === "}" && depth <= fnDepth - 1;
+      if (isClosing) {
+        result.push(line);
+        inFunction = false;
+        capturedLines++;
+      } else if (isReturn || isBranch) {
+        result.push(trimmed.length > 120 ? `${trimmed.slice(0, 120)}...` : line);
+        capturedLines++;
+      } else if (trimmed.startsWith("//") || trimmed === "") {
+      } else if (capturedLines < maxLength / 30) {
+        result.push(trimmed.length > 100 ? `${trimmed.slice(0, 100)}...` : line);
+        capturedLines++;
+      }
+    } else {
+      result.push(line);
+      capturedLines++;
+    }
+    if (result.join("\n").length > maxLength) {
+      result.push("// ... (truncated)");
+      break;
+    }
+  }
+  return result.join("\n");
+}
 
 // src/ai/provider.ts
 var providerRegistry = /* @__PURE__ */ new Map();
@@ -3416,15 +3599,18 @@ import path9 from "path";
 function analyzeFile(filePath) {
   const absolutePath = path9.resolve(process.cwd(), filePath);
   if (!fs9.existsSync(absolutePath)) {
-    return { filePath, exports: [], apiCalls: [] };
+    return { filePath, exports: [], imports: [], importSignatures: [], apiCalls: [] };
   }
   const content = fs9.readFileSync(absolutePath, "utf-8");
   const ext = path9.extname(filePath);
   const result = {
     filePath,
     exports: extractExports(content, ext),
+    imports: extractImports(content),
+    importSignatures: [],
     apiCalls: extractAPICalls(content, filePath)
   };
+  result.importSignatures = analyzeImportSignatures(result.imports, path9.dirname(absolutePath));
   if (ext === ".vue") {
     result.vueAnalysis = analyzeVueComponent(content, path9.basename(filePath, ".vue"));
     const scriptMatch = content.match(/<script[^>]*>([\s\S]*?)<\/script>/);
@@ -3847,6 +4033,90 @@ function generateMockRoutesFromAPICalls(apiCalls) {
   }
   return routes;
 }
+function extractImports(content) {
+  const imports = [];
+  const namedImportRegex = /import\s*\{([^}]+)\}\s*from\s*['"]([^'"]+)['"]/g;
+  let match;
+  while ((match = namedImportRegex.exec(content)) !== null) {
+    const names = match[1].split(",").map((n) => {
+      const parts = n.trim().split(/\s+as\s+/);
+      return parts[0].trim();
+    }).filter(Boolean);
+    imports.push({ names, source: match[2], isDefault: false, isNamespace: false });
+  }
+  const defaultImportRegex = /import\s+(\w+)\s+from\s*['"]([^'"]+)['"]/g;
+  while ((match = defaultImportRegex.exec(content)) !== null) {
+    const fullLine = content.slice(Math.max(0, match.index - 5), match.index + match[0].length);
+    if (fullLine.includes("{")) continue;
+    imports.push({ names: [match[1]], source: match[2], isDefault: true, isNamespace: false });
+  }
+  const namespaceImportRegex = /import\s*\*\s*as\s+(\w+)\s*from\s*['"]([^'"]+)['"]/g;
+  while ((match = namespaceImportRegex.exec(content)) !== null) {
+    imports.push({ names: [match[1]], source: match[2], isDefault: false, isNamespace: true });
+  }
+  return imports;
+}
+function analyzeImportSignatures(imports, baseDir) {
+  const signatures = [];
+  for (const imp of imports) {
+    if (!imp.source.startsWith(".") && !imp.source.startsWith("@/") && !imp.source.startsWith("~")) {
+      continue;
+    }
+    const resolvedPath = resolveImportPath2(imp.source, baseDir);
+    if (!resolvedPath || !fs9.existsSync(resolvedPath)) continue;
+    try {
+      const content = fs9.readFileSync(resolvedPath, "utf-8");
+      const ext = path9.extname(resolvedPath);
+      const exports = extractExports(content, ext);
+      const sigMap = {};
+      for (const name of imp.names) {
+        const exp = exports.find((e) => e.name === name);
+        if (exp) {
+          const params = exp.params.length > 0 ? `(${exp.params.join(", ")})` : "";
+          const ret = exp.returnType ? `: ${exp.returnType}` : "";
+          const async = exp.isAsync ? "async " : "";
+          sigMap[name] = `${async}${exp.kind}${params}${ret}`;
+        } else if (imp.isDefault) {
+          const defaultExp = exports.find((e) => e.kind === "default");
+          if (defaultExp) {
+            sigMap[name] = `default[${defaultExp.name || "anonymous"}]`;
+          }
+        } else if (imp.isNamespace) {
+          sigMap[name] = `{${exports.map((e) => e.name).join(", ")}}`;
+        }
+      }
+      if (ext === ".vue") {
+        const vueAnalysis = analyzeVueComponent(content, path9.basename(resolvedPath, ".vue"));
+        if (vueAnalysis.props.length > 0) {
+          sigMap["__props__"] = vueAnalysis.props.map((p) => `${p.name}:${p.type}${p.required ? "!" : "?"}`).join(",");
+        }
+      }
+      if (Object.keys(sigMap).length > 0) {
+        signatures.push({ source: imp.source, signatures: sigMap });
+      }
+    } catch {
+    }
+  }
+  return signatures;
+}
+function resolveImportPath2(importSource, baseDir) {
+  let resolved;
+  if (importSource.startsWith("@/") || importSource.startsWith("~")) {
+    const relativePath = importSource.replace(/^[@~]\//, "src/");
+    resolved = path9.resolve(process.cwd(), relativePath);
+  } else if (importSource.startsWith(".")) {
+    resolved = path9.resolve(baseDir, importSource);
+  } else {
+    return null;
+  }
+  const extensions = [".ts", ".tsx", ".js", ".jsx", ".vue", "/index.ts", "/index.js"];
+  if (fs9.existsSync(resolved)) return resolved;
+  for (const ext of extensions) {
+    const withExt = resolved + ext;
+    if (fs9.existsSync(withExt)) return withExt;
+  }
+  return null;
+}
 export {
   AI_PRESET_PROVIDERS,
   DEFAULT_CONFIG,