ai-spec-dev 0.42.0 → 0.46.0

package/core/config-defaults.ts

@@ -0,0 +1,44 @@
+/**
+ * config-defaults.ts — Centralized default values for all configurable constants.
+ *
+ * Modules import their defaults from here instead of defining local magic numbers.
+ * The pipeline can override any value via AiSpecConfig at runtime.
+ */
+
+// ─── Directory & File Names ─────────────────────────────────────────────────
+
+export const DEFAULT_LOG_DIR = ".ai-spec-logs";
+export const DEFAULT_VCR_DIR = ".ai-spec-vcr";
+export const DEFAULT_BACKUP_DIR = ".ai-spec-backup";
+export const DEFAULT_REVIEW_HISTORY_FILE = ".ai-spec-reviews.json";
+
+// ─── URLs ───────────────────────────────────────────────────────────────────
+
+export const DEFAULT_OPENAPI_SERVER_URL = "http://localhost:3000";
+
+// ─── Numeric Limits ─────────────────────────────────────────────────────────
+
+/** Max chars captured from build/test/lint command output before parsing. */
+export const DEFAULT_MAX_COMMAND_OUTPUT_CHARS = 30_000;
+
+/** Max chars of an existing file sent to the AI for auto-fix. */
+export const DEFAULT_MAX_FIX_FILE_CHARS = 60_000;
+
+/** Max DSL extraction retries on parse failure. */
+export const DEFAULT_DSL_MAX_RETRIES = 2;
+
+/** Max constitution chars in codegen prompts (trimmed if exceeded). */
+export const DEFAULT_MAX_CONSTITUTION_CHARS = 4_000;
+
+/** Max chars of file content sent per file in review (reviewFiles mode). */
+export const DEFAULT_MAX_REVIEW_FILE_CHARS = 3_000;
+
+// ─── Token Budgets ──────────────────────────────────────────────────────────
+
+export const DEFAULT_TOKEN_BUDGETS: Record<string, number> = {
+  gemini: 900_000,
+  claude: 180_000,
+  openai: 120_000,
+  deepseek: 60_000,
+  default: 100_000,
+};

package/core/constitution-generator.ts

@@ -4,6 +4,7 @@ import * as path from "path";
 import { AIProvider } from "./spec-generator";
 import { ContextLoader, ProjectContext } from "./context-loader";
 import { constitutionSystemPrompt } from "../prompts/constitution.prompt";
+import { DEFAULT_MAX_CONSTITUTION_CHARS } from "./config-defaults";
 
 export const CONSTITUTION_FILE = ".ai-spec-constitution.md";
 
@@ -41,7 +42,7 @@ function buildConstitutionPrompt(context: ProjectContext, projectRoot: string):
   }
 
   if (context.schema) {
-    parts.push(`=== Prisma Schema ===\n${context.schema.slice(0, 4000)}\n`);
+    parts.push(`=== Prisma Schema ===\n${context.schema.slice(0, DEFAULT_MAX_CONSTITUTION_CHARS)}\n`);
   }
 
   if (context.errorPatterns) {

package/core/dsl-extractor.ts

@@ -13,6 +13,7 @@ import {
 } from "../prompts/dsl.prompt";
 import { estimateTokens, getDefaultBudget } from "./token-budget";
 import { parseJsonFromAiOutput } from "./safe-json";
+import { DEFAULT_DSL_MAX_RETRIES } from "./config-defaults";
 
 // ─── DSL Sanitizer ───────────────────────────────────────────────────────────
 
@@ -50,7 +51,7 @@ function sanitizeDsl(raw: unknown): unknown {
 // ─── Constants ────────────────────────────────────────────────────────────────
 
 /** Maximum AI attempts (1 initial + up to this many retries). */
-const MAX_RETRIES = 2;
+const MAX_RETRIES = DEFAULT_DSL_MAX_RETRIES;
 
 /** Default maximum spec length passed to AI. Overridden by token budget when provider is known. */
 const DEFAULT_MAX_SPEC_CHARS = 12_000;

package/core/error-feedback.ts

@@ -8,6 +8,7 @@ import { SpecDSL } from "./dsl-types";
 import { buildDslContextSection } from "./dsl-extractor";
 import { getActiveSnapshot } from "./run-snapshot";
 import { startSpinner } from "./cli-ui";
+import { DEFAULT_MAX_COMMAND_OUTPUT_CHARS, DEFAULT_MAX_FIX_FILE_CHARS } from "./config-defaults";
 
 // ─── Types ──────────────────────────────────────────────────────────────────────
 
@@ -30,14 +31,14 @@ interface FixResult {
  * ~10K tokens — enough for any realistic error listing; prevents a pathological
  * build output (e.g. 10MB of warnings) from ballooning the AI context.
  */
-const MAX_COMMAND_OUTPUT_CHARS = 50_000;
+const MAX_COMMAND_OUTPUT_CHARS = DEFAULT_MAX_COMMAND_OUTPUT_CHARS;
 
 /**
  * Maximum characters of an existing file sent to the AI for auto-fix.
  * ~12K tokens — covers large files; content beyond this is truncated with a
  * notice so the AI knows it may be seeing an incomplete file.
  */
-const MAX_FIX_FILE_CHARS = 60_000;
+const MAX_FIX_FILE_CHARS = DEFAULT_MAX_FIX_FILE_CHARS;
 
 // ─── Error Detection ────────────────────────────────────────────────────────────
 

package/core/openapi-exporter.ts

@@ -1,6 +1,7 @@
 import * as path from "path";
 import * as fs from "fs-extra";
 import { SpecDSL, ApiEndpoint, DataModel, ModelField, FieldMap } from "./dsl-types";
+import { DEFAULT_OPENAPI_SERVER_URL } from "./config-defaults";
 
 // ─── Types ────────────────────────────────────────────────────────────────────
 
@@ -272,7 +273,7 @@ function buildYamlDoc(obj: Record<string, unknown>): string {
  * Convert a SpecDSL to an OpenAPI 3.1.0 document.
  * Returns the document as a plain JS object (can be serialised to YAML or JSON).
  */
-export function dslToOpenApi(dsl: SpecDSL, serverUrl = "http://localhost:3000"): Record<string, unknown> {
+export function dslToOpenApi(dsl: SpecDSL, serverUrl = DEFAULT_OPENAPI_SERVER_URL): Record<string, unknown> {
   // ── Info ──────────────────────────────────────────────────────────────────
   const info = {
     title: dsl.feature.title,
@@ -338,7 +339,7 @@ export async function exportOpenApi(
   opts: OpenApiExportOptions = {}
 ): Promise<string> {
   const format = opts.format ?? "yaml";
-  const serverUrl = opts.serverUrl ?? "http://localhost:3000";
+  const serverUrl = opts.serverUrl ?? DEFAULT_OPENAPI_SERVER_URL;
   const defaultName = `openapi.${format}`;
   const outputPath = opts.outputPath
     ? path.isAbsolute(opts.outputPath)

package/core/repo-store.ts

@@ -0,0 +1,95 @@
+import * as fs from "fs-extra";
+import * as path from "path";
+import * as os from "os";
+import { RepoType, RepoRole } from "./workspace-loader";
+
+const REPO_STORE_FILE = path.join(os.homedir(), ".ai-spec-repos.json");
+
+// ─── Types ───────────────────────────────────────────────────────────────────
+
+export interface RegisteredRepo {
+  /** Display name (defaults to directory basename) */
+  name: string;
+  /** Absolute path to the repo root */
+  path: string;
+  /** Auto-detected repo type */
+  type: RepoType;
+  /** Auto-detected repo role */
+  role: RepoRole;
+  /** Whether a project constitution has been generated */
+  hasConstitution: boolean;
+  /** ISO timestamp of registration */
+  registeredAt: string;
+}
+
+interface RepoStoreData {
+  repos: RegisteredRepo[];
+}
+
+// ─── Read / Write ────────────────────────────────────────────────────────────
+
+async function readStore(): Promise<RepoStoreData> {
+  try {
+    if (await fs.pathExists(REPO_STORE_FILE)) {
+      return await fs.readJson(REPO_STORE_FILE);
+    }
+  } catch (err) {
+    console.warn(`Warning: Could not read repo store at ${REPO_STORE_FILE}: ${(err as Error).message}.`);
+  }
+  return { repos: [] };
+}
+
+async function writeStore(store: RepoStoreData): Promise<void> {
+  await fs.ensureFile(REPO_STORE_FILE);
+  await fs.writeJson(REPO_STORE_FILE, store, { spaces: 2 });
+}
+
+// ─── Public API ──────────────────────────────────────────────────────────────
+
+/** Get all registered repos. */
+export async function getRegisteredRepos(): Promise<RegisteredRepo[]> {
+  const store = await readStore();
+  return store.repos;
+}
+
+/** Find a registered repo by its absolute path. */
+export async function getRepoByPath(absPath: string): Promise<RegisteredRepo | undefined> {
+  const store = await readStore();
+  return store.repos.find((r) => r.path === absPath);
+}
+
+/** Register a new repo. If already registered (by path), update it. */
+export async function registerRepo(repo: RegisteredRepo): Promise<void> {
+  const store = await readStore();
+  const idx = store.repos.findIndex((r) => r.path === repo.path);
+  if (idx >= 0) {
+    store.repos[idx] = repo;
+  } else {
+    store.repos.push(repo);
+  }
+  await writeStore(store);
+}
+
+/** Remove a registered repo by path. */
+export async function unregisterRepo(absPath: string): Promise<boolean> {
+  const store = await readStore();
+  const before = store.repos.length;
+  store.repos = store.repos.filter((r) => r.path !== absPath);
+  if (store.repos.length < before) {
+    await writeStore(store);
+    return true;
+  }
+  return false;
+}
+
+/** Update the hasConstitution flag for a repo. */
+export async function markRepoConstitution(absPath: string, has: boolean): Promise<void> {
+  const store = await readStore();
+  const repo = store.repos.find((r) => r.path === absPath);
+  if (repo) {
+    repo.hasConstitution = has;
+    await writeStore(store);
+  }
+}
+
+export { REPO_STORE_FILE };

package/core/reviewer.ts

@@ -10,6 +10,7 @@ import {
   reviewImpactComplexitySystemPrompt,
 } from "../prompts/codegen.prompt";
 import { CONSTITUTION_FILE } from "./constitution-generator";
+import { DEFAULT_REVIEW_HISTORY_FILE, DEFAULT_MAX_REVIEW_FILE_CHARS } from "./config-defaults";
 
 // ─── Constitution Lessons Helper ──────────────────────────────────────────────
 
@@ -48,7 +49,7 @@ interface ReviewHistoryEntry {
   complexityLevel?: "低" | "中" | "高";
 }
 
-const REVIEW_HISTORY_FILE = ".ai-spec-reviews.json";
+const REVIEW_HISTORY_FILE = DEFAULT_REVIEW_HISTORY_FILE;
 
 async function loadReviewHistory(projectRoot: string): Promise<ReviewHistoryEntry[]> {
   const historyPath = path.join(projectRoot, REVIEW_HISTORY_FILE);
@@ -225,16 +226,19 @@ ${codeContext}`;
     console.log(chalk.gray("  Pass 2/3: Implementation review..."));
 
     // ── Pass 2: Implementation + History ─────────────────────────────────────
+    // Token savings: Pass 2/3 receive a spec digest instead of the full spec,
+    // and omit the raw code context (Pass 1 already analyzed it).
+    const specDigest = specContent && specContent.length > 600
+      ? specContent.slice(0, 600) + "\n... [spec truncated — 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 — review for general code quality)"}
-
-=== Code ===
-${codeContext}
+=== Feature Spec (digest — full spec was provided in Pass 0/1) ===
+${specDigest}
 
 === Architecture Review (Pass 1 — do NOT repeat these findings) ===
 ${archReview}
@@ -246,11 +250,8 @@ ${historyContext}`;
     // ── Pass 3: Impact & Complexity ───────────────────────────────────────────
     const impactPrompt = `Assess the impact and complexity of this change.
 
-=== Feature Spec ===
-${specContent || "(No spec — review for general code quality)"}
-
-=== Code ===
-${codeContext}
+=== Feature Spec (digest) ===
+${specDigest}
 
 === Architecture Review (Pass 1 — do NOT repeat) ===
 ${archReview}
@@ -338,8 +339,8 @@ ${implReview}`;
       const fullPath = path.join(workingDir, filePath);
       try {
         const content = await fs.readFile(fullPath, "utf-8");
-        filesSection += `\n\n=== ${filePath} ===\n${content.slice(0, 3000)}`;
-        if (content.length > 3000) filesSection += `\n... (truncated, ${content.length} chars total)`;
+        filesSection += `\n\n=== ${filePath} ===\n${content.slice(0, DEFAULT_MAX_REVIEW_FILE_CHARS)}`;
+        if (content.length > DEFAULT_MAX_REVIEW_FILE_CHARS) filesSection += `\n... (truncated, ${content.length} chars total)`;
       } catch {
         filesSection += `\n\n=== ${filePath} ===\n(file not found)`;
       }

package/core/run-logger.ts

@@ -1,8 +1,7 @@
 import * as fs from "fs-extra";
 import * as path from "path";
 import chalk from "chalk";
-
-const LOG_DIR = ".ai-spec-logs";
+import { DEFAULT_LOG_DIR } from "./config-defaults";
 
 // ─── JSONL helpers ────────────────────────────────────────────────────────────
 // Each event is synchronously appended as one JSON line to a `.jsonl` shadow
@@ -131,8 +130,8 @@ export class RunLogger {
     meta?: { provider?: string; model?: string; specPath?: string }
   ) {
     this.startMs = Date.now();
-    this.logPath  = path.join(workingDir, LOG_DIR, `${runId}.json`);
-    this.jsonlPath = path.join(workingDir, LOG_DIR, `${runId}.jsonl`);
+    this.logPath  = path.join(workingDir, DEFAULT_LOG_DIR, `${runId}.json`);
+    this.jsonlPath = path.join(workingDir, DEFAULT_LOG_DIR, `${runId}.jsonl`);
     this.log = {
       runId,
       startedAt: new Date().toISOString(),

package/core/run-snapshot.ts

@@ -1,7 +1,6 @@
 import * as fs from "fs-extra";
 import * as path from "path";
-
-const BACKUP_DIR = ".ai-spec-backup";
+import { DEFAULT_BACKUP_DIR } from "./config-defaults";
 
 // ─── RunSnapshot ──────────────────────────────────────────────────────────────
 /**
@@ -18,7 +17,7 @@ export class RunSnapshot {
     private readonly workingDir: string,
     readonly runId: string
   ) {
-    this.backupRoot = path.join(workingDir, BACKUP_DIR, runId);
+    this.backupRoot = path.join(workingDir, DEFAULT_BACKUP_DIR, runId);
   }
 
   /**

package/core/run-trend.ts

@@ -2,8 +2,7 @@ import * as fs from "fs-extra";
 import * as path from "path";
 import chalk from "chalk";
 import { RunLog, reconstructRunLogFromJsonl } from "./run-logger";
-
-const LOG_DIR = ".ai-spec-logs";
+import { DEFAULT_LOG_DIR } from "./config-defaults";
 
 // ─── Types ────────────────────────────────────────────────────────────────────
 
@@ -45,7 +44,7 @@ export interface TrendReport {
  * Silently skips unreadable / corrupt files.
  */
 export async function loadRunLogs(workingDir: string): Promise<RunLog[]> {
-  const logDir = path.join(workingDir, LOG_DIR);
+  const logDir = path.join(workingDir, DEFAULT_LOG_DIR);
   if (!(await fs.pathExists(logDir))) return [];
 
   const files = await fs.readdir(logDir);
@@ -255,7 +254,7 @@ export function printTrendReport(report: TrendReport, workingDir: string): void
   }
 
   // ── Footer ────────────────────────────────────────────────────────
-  const logRelDir = path.relative(workingDir, path.join(workingDir, LOG_DIR));
+  const logRelDir = path.relative(workingDir, path.join(workingDir, DEFAULT_LOG_DIR));
   console.log(chalk.gray(`\n  ${entries.length} run(s) shown  ·  logs: ${logRelDir}/`));
   console.log(chalk.cyan("─".repeat(63)));
 }

package/core/spec-generator.ts

@@ -1,16 +1,15 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 import Anthropic from "@anthropic-ai/sdk";
 import OpenAI from "openai";
-import axios from "axios";
 import { ProxyAgent } from "undici";
 import { specPrompt } from "../prompts/spec.prompt";
 import { ProjectContext } from "./context-loader";
 import { withReliability } from "./provider-utils";
 
 // ─── Proxy Helper ─────────────────────────────────────────────────────────────
-// 仅用于 Gemini：其他 SDK（Anthropic / OpenAI）会自动读取 HTTPS_PROXY。
 // Gemini SDK 使用 Node.js 原生 fetch（undici），不会自动读代理环境变量，
 // 需要手动创建 ProxyAgent 并通过 fetchOptions 注入。
+// Anthropic SDK (node-fetch) 也不会自动读代理环境变量。
 // 这是 in-process 级别的配置，完全不影响 execSync 启动的子进程（如 claude CLI）。
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -47,6 +46,8 @@ export 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).
@@ -72,6 +73,8 @@ export const PROVIDER_CATALOG: Record<string, ProviderMeta> = {
     description: "小米 MiMo — mimo-v2-pro (Anthropic-compatible API)",
     models: ["mimo-v2-pro"],
     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: {
@@ -244,8 +247,8 @@ export class ClaudeProvider implements AIProvider {
           ...(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}` }
@@ -307,58 +310,40 @@ export class OpenAICompatibleProvider implements AIProvider {
 // ─── MiMo Provider ─────────────────────────────────────────────────────────────
 // MiMo uses the Anthropic messages format but with a different base URL
 // and a custom "api-key" auth header (not "x-api-key" / "Authorization: Bearer").
-// The Anthropic SDK does not support custom auth headers, so we call the API
-// directly via axios.
+// MiMo's token-plan API is Anthropic-compatible — we reuse the Anthropic SDK
+// directly, reading base URL from env (MIMO_BASE_URL / ANTHROPIC_BASE_URL).
 
 export class MiMoProvider implements AIProvider {
+  private client: Anthropic;
   readonly providerName = "mimo";
   readonly modelName: string;
-  private apiKey: string;
-  private readonly baseUrl = "https://api.xiaomimimo.com/anthropic/v1/messages";
 
   constructor(apiKey: string, 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: string, systemInstruction?: string): Promise<string> {
     return withReliability(
       async () => {
-        const body: Record<string, unknown> = {
+        // Use streaming to avoid timeout errors with large max_tokens
+        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.0,
-          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 }],
         });
-
-        // Response follows Anthropic format: { content: [{ type: "text"|"thinking", ... }] }
-        // MiMo may return a "thinking" block before the actual "text" block — skip it.
-        const data = response.data as { stop_reason?: string; content?: Array<{ type: string; text?: string; thinking?: string }> };
-        const blocks = data?.content ?? [];
-
-        const textBlock = blocks.find((b) => b.type === "text");
-        if (textBlock?.text) return textBlock.text;
-
-        // If stop_reason is max_tokens, the model was cut off mid-generation (thinking block only)
-        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();
+        // MiMo may return "thinking" blocks before or instead of "text" blocks.
+        // Extract the first text block; fall back to thinking content; last resort: concatenate all.
+        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 as unknown as { thinking: string }).thinking;
+        return message.content.map((b: { type: string; text?: string }) => b.text ?? "").join("");
       },
       { label: `${this.providerName}/${this.modelName}` }
     );

package/core/token-budget.ts

@@ -6,6 +6,7 @@
  */
 
 import chalk from "chalk";
+import { DEFAULT_TOKEN_BUDGETS as _CONFIG_BUDGETS } from "./config-defaults";
 
 // ─── Token Estimation ────────────────────────────────────────────────────────
 
@@ -110,14 +111,8 @@ export function assembleSections(
 
 // ─── Default Budgets ─────────────────────────────────────────────────────────
 
-/** Default context token budgets per provider. */
-export const DEFAULT_TOKEN_BUDGETS: Record<string, number> = {
-  gemini: 900_000,
-  claude: 180_000,
-  openai: 120_000,
-  deepseek: 60_000,
-  default: 100_000,
-};
+/** Default context token budgets per provider (sourced from config-defaults). */
+export const DEFAULT_TOKEN_BUDGETS = _CONFIG_BUDGETS;
 
 export function getDefaultBudget(providerName: string): number {
   return DEFAULT_TOKEN_BUDGETS[providerName] ?? DEFAULT_TOKEN_BUDGETS.default;

package/core/vcr.ts

@@ -27,7 +27,9 @@ import * as fs from "fs-extra";
 import * as path from "path";
 import { AIProvider } from "./spec-generator";
 
-export const VCR_DIR = ".ai-spec-vcr";
+import { DEFAULT_VCR_DIR } from "./config-defaults";
+
+export const VCR_DIR = DEFAULT_VCR_DIR;
 
 // ─── Types ────────────────────────────────────────────────────────────────────