ai-spec-dev 0.41.0 → 0.46.0

package/dist/index.mjs

@@ -2,7 +2,6 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import Anthropic from "@anthropic-ai/sdk";
 import OpenAI from "openai";
-import axios from "axios";
 import { ProxyAgent } from "undici";
 
 // prompts/spec.prompt.ts
@@ -116,9 +115,77 @@ CRITICAL \u2014 \u5386\u53F2\u6559\u8BAD\u5E94\u7528\uFF08Accumulated Lessons\uF
 3. \u5BF9\u4E8E\u6BCF\u6761\u76F4\u63A5\u76F8\u5173\u7684\u6559\u8BAD\uFF0C\u5728 \xA78 \u5B9E\u65BD\u8981\u70B9\u672B\u5C3E\u8FFD\u52A0\u4E00\u884C\uFF1A\u300C\u26A0 \u57FA\u4E8E\u5386\u53F2\u6559\u8BAD\uFF1A[\u7B80\u8FF0\u672C\u6B21 spec \u5982\u4F55\u89C4\u907F\u8BE5\u95EE\u9898]\u300D
 4. \u5982\u65E0\u76F8\u5173\u6559\u8BAD\uFF0C\xA78 \u4E0D\u5FC5\u8FFD\u52A0\u4EFB\u4F55\u5185\u5BB9`;
 
-// core/provider-utils.ts
+// core/cli-ui.ts
 import chalk from "chalk";
-var sleep = (ms2) => new Promise((r) => setTimeout(r, ms2));
+var SPINNER_FRAMES = ["\u280B", "\u2819", "\u2839", "\u2838", "\u283C", "\u2834", "\u2826", "\u2827", "\u2807", "\u280F"];
+function startSpinner(text) {
+  const isTTY = process.stderr.isTTY;
+  let frame = 0;
+  let currentText = text;
+  let stopped = false;
+  function render() {
+    if (stopped) return;
+    const symbol = chalk.cyan(SPINNER_FRAMES[frame % SPINNER_FRAMES.length]);
+    if (isTTY) {
+      process.stderr.write(`\r  ${symbol} ${currentText}${" ".repeat(10)}`);
+    }
+    frame++;
+  }
+  if (!isTTY) {
+    process.stderr.write(`  \u2026 ${currentText}
+`);
+  }
+  const timer = setInterval(render, 80);
+  render();
+  return {
+    update(newText) {
+      currentText = newText;
+    },
+    stop(finalText) {
+      if (stopped) return;
+      stopped = true;
+      clearInterval(timer);
+      if (isTTY) {
+        process.stderr.write(`\r${" ".repeat(currentText.length + 20)}\r`);
+      }
+      if (finalText) {
+        process.stderr.write(`  ${finalText}
+`);
+      }
+    },
+    succeed(successText) {
+      this.stop(chalk.green(`\u2714 ${successText}`));
+    },
+    fail(failText) {
+      this.stop(chalk.red(`\u2718 ${failText}`));
+    }
+  };
+}
+async function retryCountdown(opts) {
+  const { attempt, maxAttempts, waitMs, errorMessage, label } = opts;
+  const isTTY = process.stderr.isTTY;
+  const shortErr = errorMessage.length > 120 ? errorMessage.slice(0, 117) + "..." : errorMessage;
+  process.stderr.write("\n");
+  process.stderr.write(chalk.yellow(`  \u250C\u2500 Retry ${attempt}/${maxAttempts} `) + chalk.gray(`[${label}]`) + chalk.yellow(` ${"\u2500".repeat(Math.max(1, 40 - label.length))}
+`));
+  process.stderr.write(chalk.yellow(`  \u2502 `) + chalk.white(shortErr) + "\n");
+  process.stderr.write(chalk.yellow(`  \u2502 `) + chalk.gray(`Waiting before retry...`) + "\n");
+  const totalSeconds = Math.ceil(waitMs / 1e3);
+  for (let s = totalSeconds; s > 0; s--) {
+    const bar = chalk.green("\u2588".repeat(totalSeconds - s)) + chalk.gray("\u2591".repeat(s));
+    const line = chalk.yellow(`  \u2502 `) + `${bar} ${chalk.bold.white(`${s}s`)}`;
+    if (isTTY) {
+      process.stderr.write(`\r${line}${" ".repeat(10)}`);
+    }
+    await new Promise((r) => setTimeout(r, 1e3));
+  }
+  if (isTTY) {
+    process.stderr.write(`\r${" ".repeat(70)}\r`);
+  }
+  process.stderr.write(chalk.yellow(`  \u2514\u2500 `) + chalk.cyan(`Retrying now...`) + "\n\n");
+}
+
+// core/provider-utils.ts
 var ProviderError = class extends Error {
   constructor(message, kind, originalError) {
     super(message);
@@ -202,11 +269,14 @@ async function withReliability(fn, opts) {
         throw classifyError(err, label);
       }
       const waitMs = attempt === 0 ? 2e3 : 6e3;
-      console.warn(
-        chalk.yellow(`  \u26A0 ${label} failed (attempt ${attempt + 1}/${retries + 1}), retrying in ${waitMs / 1e3}s`) + chalk.gray(` \u2014 ${err.message}`)
-      );
       onRetry?.(attempt + 1, err);
-      await sleep(waitMs);
+      await retryCountdown({
+        attempt: attempt + 1,
+        maxAttempts: retries + 1,
+        waitMs,
+        errorMessage: err.message ?? String(err),
+        label
+      });
     }
   }
   throw new Error("unreachable");
@@ -224,7 +294,9 @@ var PROVIDER_CATALOG = {
     displayName: "MiMo (Xiaomi)",
     description: "\u5C0F\u7C73 MiMo \u2014 mimo-v2-pro (Anthropic-compatible API)",
     models: ["mimo-v2-pro"],
-    envKey: "MIMO_API_KEY"
+    envKey: "MIMO_API_KEY",
+    // Fallback env var — MiMo's token plan uses ANTHROPIC_AUTH_TOKEN
+    fallbackEnvKeys: ["ANTHROPIC_AUTH_TOKEN"]
     // baseURL not used — MiMo has a dedicated provider class
   },
   gemini: {
@@ -393,8 +465,8 @@ var ClaudeProvider = class {
           ...systemInstruction ? { system: systemInstruction } : {},
           messages: [{ role: "user", content: prompt }]
         });
-        const block = message.content[0];
-        if (block.type === "text") return block.text;
+        const textBlock = message.content.find((b) => b.type === "text");
+        if (textBlock) return textBlock.text;
         throw new Error("Unexpected response type from Claude API");
       },
       { label: `${this.providerName}/${this.modelName}` }
@@ -439,43 +511,29 @@ var OpenAICompatibleProvider = class {
   }
 };
 var MiMoProvider = class {
+  client;
   providerName = "mimo";
   modelName;
-  apiKey;
-  baseUrl = "https://api.xiaomimimo.com/anthropic/v1/messages";
   constructor(apiKey, modelName = PROVIDER_CATALOG.mimo.models[0]) {
-    this.apiKey = apiKey;
+    const baseURL = process.env["MIMO_BASE_URL"] || process.env["ANTHROPIC_BASE_URL"] || "https://token-plan-cn.xiaomimimo.com/anthropic";
+    this.client = new Anthropic({ apiKey, baseURL });
     this.modelName = modelName;
   }
   async generate(prompt, systemInstruction) {
     return withReliability(
       async () => {
-        const body = {
+        const stream = this.client.messages.stream({
           model: this.modelName,
-          max_tokens: 16384,
-          messages: [{ role: "user", content: [{ type: "text", text: prompt }] }],
-          top_p: 0.95,
-          stream: false,
-          temperature: 1,
-          stop_sequences: null
-        };
-        if (systemInstruction) {
-          body.system = systemInstruction;
-        }
-        const response = await axios.post(this.baseUrl, body, {
-          headers: {
-            "api-key": this.apiKey,
-            "Content-Type": "application/json"
-          }
+          max_tokens: 65536,
+          ...systemInstruction ? { system: systemInstruction } : {},
+          messages: [{ role: "user", content: prompt }]
         });
-        const data = response.data;
-        const blocks = data?.content ?? [];
-        const textBlock = blocks.find((b) => b.type === "text");
-        if (textBlock?.text) return textBlock.text;
-        if (data?.stop_reason === "max_tokens") {
-          throw new Error(`MiMo response truncated (max_tokens reached). The prompt may be too long. Try a shorter spec or switch to a model with larger context.`);
-        }
-        throw new Error(`Unexpected MiMo response: ${JSON.stringify(response.data).slice(0, 200)}`);
+        const message = await stream.finalMessage();
+        const textBlock = message.content.find((b) => b.type === "text");
+        if (textBlock) return textBlock.text;
+        const thinkBlock = message.content.find((b) => b.type === "thinking");
+        if (thinkBlock) return thinkBlock.thinking;
+        return message.content.map((b) => b.text ?? "").join("");
       },
       { label: `${this.providerName}/${this.modelName}` }
     );
@@ -5080,13 +5138,11 @@ function typeLabel(v2) {
 
 // core/token-budget.ts
 import chalk6 from "chalk";
-var CJK_RANGE = /[\u4e00-\u9fff\u3400-\u4dbf\u3000-\u303f\uff00-\uffef]/g;
-function estimateTokens(text) {
-  if (!text) return 0;
-  const cjkCount = (text.match(CJK_RANGE) ?? []).length;
-  const nonCjkLength = text.length - cjkCount;
-  return Math.ceil(cjkCount + nonCjkLength / 4);
-}
+
+// core/config-defaults.ts
+var DEFAULT_REVIEW_HISTORY_FILE = ".ai-spec-reviews.json";
+var DEFAULT_MAX_CONSTITUTION_CHARS = 4e3;
+var DEFAULT_MAX_REVIEW_FILE_CHARS = 3e3;
 var DEFAULT_TOKEN_BUDGETS = {
   gemini: 9e5,
   claude: 18e4,
@@ -5094,8 +5150,18 @@ var DEFAULT_TOKEN_BUDGETS = {
   deepseek: 6e4,
   default: 1e5
 };
+
+// core/token-budget.ts
+var CJK_RANGE = /[\u4e00-\u9fff\u3400-\u4dbf\u3000-\u303f\uff00-\uffef]/g;
+function estimateTokens(text) {
+  if (!text) return 0;
+  const cjkCount = (text.match(CJK_RANGE) ?? []).length;
+  const nonCjkLength = text.length - cjkCount;
+  return Math.ceil(cjkCount + nonCjkLength / 4);
+}
+var DEFAULT_TOKEN_BUDGETS2 = DEFAULT_TOKEN_BUDGETS;
 function getDefaultBudget(providerName) {
-  return DEFAULT_TOKEN_BUDGETS[providerName] ?? DEFAULT_TOKEN_BUDGETS.default;
+  return DEFAULT_TOKEN_BUDGETS2[providerName] ?? DEFAULT_TOKEN_BUDGETS2.default;
 }
 
 // core/dsl-extractor.ts
@@ -6397,6 +6463,7 @@ ${spec}
 ${constitutionSection}
 === ${existingContent ? "Existing content (modify and return the complete file)" : "Create this file from scratch"} ===
 ${existingContent || "Output only the complete file content."}`;
+      const fileSpinner = startSpinner(`${prefix}Generating ${chalk10.bold(item.file)}...`);
       try {
         const raw = await this.provider.generate(codePrompt, systemPrompt);
         const fileContent = stripCodeFences(raw);
@@ -6404,11 +6471,11 @@ ${existingContent || "Output only the complete file content."}`;
         await fs10.ensureDir(path6.dirname(fullPath));
         await fs10.writeFile(fullPath, fileContent, "utf-8");
         getActiveLogger()?.fileWritten(item.file);
-        console.log(`${prefix}${existingContent ? chalk10.yellow("~") : chalk10.green("+")} ${chalk10.bold(item.file)} ${chalk10.green("\u2714")}`);
+        fileSpinner.succeed(`${existingContent ? chalk10.yellow("~") : chalk10.green("+")} ${chalk10.bold(item.file)}`);
         successCount++;
         writtenFiles.push(item.file);
       } catch (err) {
-        console.log(`${prefix}${chalk10.red("\u2718")} ${chalk10.bold(item.file)} \u2014 ${chalk10.red(err.message)}`);
+        fileSpinner.fail(`${chalk10.bold(item.file)} \u2014 ${err.message}`);
       }
     }
     if (!taskLabel) {
@@ -6555,7 +6622,7 @@ ${context.routeSummary}
   }
   if (context.schema) {
     parts.push(`=== Prisma Schema ===
-${context.schema.slice(0, 4e3)}
+${context.schema.slice(0, DEFAULT_MAX_CONSTITUTION_CHARS)}
 `);
   }
   if (context.errorPatterns) {
@@ -6619,7 +6686,7 @@ async function loadAccumulatedLessons(projectRoot) {
   const nextSection = section.slice(marker.length).match(/\n## \d/);
   return nextSection ? section.slice(0, marker.length + nextSection.index) : section;
 }
-var REVIEW_HISTORY_FILE = ".ai-spec-reviews.json";
+var REVIEW_HISTORY_FILE = DEFAULT_REVIEW_HISTORY_FILE;
 async function loadReviewHistory(projectRoot) {
   const historyPath = path8.join(projectRoot, REVIEW_HISTORY_FILE);
   try {
@@ -6750,15 +6817,13 @@ ${specContent || "(No spec \u2014 review for general code quality)"}
 ${codeContext}`;
     const archReview = await this.provider.generate(archPrompt, reviewArchitectureSystemPrompt);
     console.log(chalk12.gray("  Pass 2/3: Implementation review..."));
+    const specDigest = specContent && specContent.length > 600 ? specContent.slice(0, 600) + "\n... [spec truncated \u2014 see Pass 0/1 for full text]" : specContent || "(No spec)";
     const history = await loadReviewHistory(this.projectRoot);
     const historyContext = buildHistoryContext(history);
     const implPrompt = `Review the implementation details of this change.
 
-=== Feature Spec ===
-${specContent || "(No spec \u2014 review for general code quality)"}
-
-=== Code ===
-${codeContext}
+=== Feature Spec (digest \u2014 full spec was provided in Pass 0/1) ===
+${specDigest}
 
 === Architecture Review (Pass 1 \u2014 do NOT repeat these findings) ===
 ${archReview}
@@ -6767,11 +6832,8 @@ ${historyContext}`;
     console.log(chalk12.gray("  Pass 3/3: Impact & complexity assessment..."));
     const impactPrompt = `Assess the impact and complexity of this change.
 
-=== Feature Spec ===
-${specContent || "(No spec \u2014 review for general code quality)"}
-
-=== Code ===
-${codeContext}
+=== Feature Spec (digest) ===
+${specDigest}
 
 === Architecture Review (Pass 1 \u2014 do NOT repeat) ===
 ${archReview}
@@ -6845,8 +6907,8 @@ ${sep}
         filesSection += `
 
 === ${filePath} ===
-${content.slice(0, 3e3)}`;
-        if (content.length > 3e3) filesSection += `
+${content.slice(0, DEFAULT_MAX_REVIEW_FILE_CHARS)}`;
+        if (content.length > DEFAULT_MAX_REVIEW_FILE_CHARS) filesSection += `
 ... (truncated, ${content.length} chars total)`;
       } catch {
         filesSection += `