ai-spec-dev 0.41.0 → 0.46.0

package/dist/index.d.mts

@@ -81,6 +81,8 @@ interface ProviderMeta {
     models: string[];
     /** Environment variable name for the API key */
     envKey: string;
+    /** Fallback env var names checked if envKey is not set */
+    fallbackEnvKeys?: string[];
     /**
      * Base URL for OpenAI-compatible providers.
      * Undefined means the provider has its own SDK (Gemini / Claude).
@@ -126,10 +128,9 @@ declare class OpenAICompatibleProvider implements AIProvider {
     generate(prompt: string, systemInstruction?: string): Promise<string>;
 }
 declare class MiMoProvider implements AIProvider {
+    private client;
     readonly providerName = "mimo";
     readonly modelName: string;
-    private apiKey;
-    private readonly baseUrl;
     constructor(apiKey: string, modelName?: string);
     generate(prompt: string, systemInstruction?: string): Promise<string>;
 }

package/dist/index.d.ts

@@ -81,6 +81,8 @@ interface ProviderMeta {
     models: string[];
     /** Environment variable name for the API key */
     envKey: string;
+    /** Fallback env var names checked if envKey is not set */
+    fallbackEnvKeys?: string[];
     /**
      * Base URL for OpenAI-compatible providers.
      * Undefined means the provider has its own SDK (Gemini / Claude).
@@ -126,10 +128,9 @@ declare class OpenAICompatibleProvider implements AIProvider {
     generate(prompt: string, systemInstruction?: string): Promise<string>;
 }
 declare class MiMoProvider implements AIProvider {
+    private client;
     readonly providerName = "mimo";
     readonly modelName: string;
-    private apiKey;
-    private readonly baseUrl;
     constructor(apiKey: string, modelName?: string);
     generate(prompt: string, systemInstruction?: string): Promise<string>;
 }

package/dist/index.js

@@ -68,7 +68,6 @@ module.exports = __toCommonJS(index_exports);
 var import_generative_ai = require("@google/generative-ai");
 var import_sdk = __toESM(require("@anthropic-ai/sdk"));
 var import_openai = __toESM(require("openai"));
-var import_axios = __toESM(require("axios"));
 var import_undici = require("undici");
 
 // prompts/spec.prompt.ts
@@ -182,9 +181,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
 var import_chalk = __toESM(require("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 = import_chalk.default.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(import_chalk.default.green(`\u2714 ${successText}`));
+    },
+    fail(failText) {
+      this.stop(import_chalk.default.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(import_chalk.default.yellow(`  \u250C\u2500 Retry ${attempt}/${maxAttempts} `) + import_chalk.default.gray(`[${label}]`) + import_chalk.default.yellow(` ${"\u2500".repeat(Math.max(1, 40 - label.length))}
+`));
+  process.stderr.write(import_chalk.default.yellow(`  \u2502 `) + import_chalk.default.white(shortErr) + "\n");
+  process.stderr.write(import_chalk.default.yellow(`  \u2502 `) + import_chalk.default.gray(`Waiting before retry...`) + "\n");
+  const totalSeconds = Math.ceil(waitMs / 1e3);
+  for (let s = totalSeconds; s > 0; s--) {
+    const bar = import_chalk.default.green("\u2588".repeat(totalSeconds - s)) + import_chalk.default.gray("\u2591".repeat(s));
+    const line = import_chalk.default.yellow(`  \u2502 `) + `${bar} ${import_chalk.default.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(import_chalk.default.yellow(`  \u2514\u2500 `) + import_chalk.default.cyan(`Retrying now...`) + "\n\n");
+}
+
+// core/provider-utils.ts
 var ProviderError = class extends Error {
   constructor(message, kind, originalError) {
     super(message);
@@ -268,11 +335,14 @@ async function withReliability(fn, opts) {
         throw classifyError(err, label);
       }
       const waitMs = attempt === 0 ? 2e3 : 6e3;
-      console.warn(
-        import_chalk.default.yellow(`  \u26A0 ${label} failed (attempt ${attempt + 1}/${retries + 1}), retrying in ${waitMs / 1e3}s`) + import_chalk.default.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");
@@ -290,7 +360,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: {
@@ -459,8 +531,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}` }
@@ -505,43 +577,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 import_sdk.default({ 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 import_axios.default.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}` }
     );
@@ -5146,13 +5204,11 @@ function typeLabel(v2) {
 
 // core/token-budget.ts
 var import_chalk6 = __toESM(require("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,
@@ -5160,8 +5216,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
@@ -6463,6 +6529,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 ${import_chalk10.default.bold(item.file)}...`);
       try {
         const raw = await this.provider.generate(codePrompt, systemPrompt);
         const fileContent = stripCodeFences(raw);
@@ -6470,11 +6537,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 ? import_chalk10.default.yellow("~") : import_chalk10.default.green("+")} ${import_chalk10.default.bold(item.file)} ${import_chalk10.default.green("\u2714")}`);
+        fileSpinner.succeed(`${existingContent ? import_chalk10.default.yellow("~") : import_chalk10.default.green("+")} ${import_chalk10.default.bold(item.file)}`);
         successCount++;
         writtenFiles.push(item.file);
       } catch (err) {
-        console.log(`${prefix}${import_chalk10.default.red("\u2718")} ${import_chalk10.default.bold(item.file)} \u2014 ${import_chalk10.default.red(err.message)}`);
+        fileSpinner.fail(`${import_chalk10.default.bold(item.file)} \u2014 ${err.message}`);
       }
     }
     if (!taskLabel) {
@@ -6621,7 +6688,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) {
@@ -6685,7 +6752,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 {
@@ -6816,15 +6883,13 @@ ${specContent || "(No spec \u2014 review for general code quality)"}
 ${codeContext}`;
     const archReview = await this.provider.generate(archPrompt, reviewArchitectureSystemPrompt);
     console.log(import_chalk12.default.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}
@@ -6833,11 +6898,8 @@ ${historyContext}`;
     console.log(import_chalk12.default.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}
@@ -6911,8 +6973,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 += `