muonroi-cli 1.3.4 → 1.4.1

package/src/router/classifier/index.ts

@@ -1,21 +0,0 @@
-// FORBIDDEN imports (enforced by tests/arch/no-network-in-classifier.test.ts):
-//   node:net | node:http(s) | undici | axios | ../ee/* | global-fetch
-import type { ClassifierResult } from "../types.js";
-import { matchRegex } from "./regex.js";
-import { initTreeSitter, lazyTreeSitter } from "./tree-sitter.js";
-
-export function classify(prompt: string, threshold = 0.55): ClassifierResult {
-  const r = matchRegex(prompt);
-  if (r.confidence >= threshold) return r;
-  const t = lazyTreeSitter(prompt);
-  if (t.confidence >= threshold) return t;
-  return {
-    tier: "abstain",
-    confidence: Math.max(r.confidence, t.confidence),
-    reason: "low-confidence",
-  };
-}
-
-export async function warm(): Promise<void> {
-  await initTreeSitter(["typescript", "python"]);
-}

package/src/router/classifier/regex.test.ts

@@ -1,49 +0,0 @@
-import { describe, expect, it } from "vitest";
-import { matchRegex } from "./regex.js";
-
-describe("matchRegex", () => {
-  it('classifies "create file foo.ts with hello world" with confidence >= 0.55', () => {
-    const result = matchRegex("create file foo.ts with hello world");
-    expect(result.confidence).toBeGreaterThanOrEqual(0.55);
-    expect(result.reason).toMatch(/^regex:/);
-  });
-
-  it('classifies "hi how are you" as short message with moderate confidence', () => {
-    const result = matchRegex("hi how are you");
-    expect(result.confidence).toBe(0.6);
-    expect(result.reason).toBe("regex:short-message");
-    expect(result.tierHint).toBe("fast");
-  });
-
-  describe("seed intents (>= 7 patterns at >= 0.55 confidence)", () => {
-    const seeds = [
-      { prompt: "create a file called app.ts", intent: "create-file" },
-      { prompt: "edit the main.ts file to fix the bug", intent: "edit" },
-      { prompt: "run the test command", intent: "run-command" },
-      { prompt: "explain what this code does", intent: "explain" },
-      { prompt: "refactor the auth module", intent: "refactor" },
-      { prompt: "search for the function declaration", intent: "search" },
-      { prompt: "install package express", intent: "install" },
-    ];
-
-    for (const { prompt, intent } of seeds) {
-      it(`matches "${prompt}" as regex:${intent} with >= 0.55`, () => {
-        const result = matchRegex(prompt);
-        expect(result.confidence).toBeGreaterThanOrEqual(0.55);
-        expect(result.reason).toBe(`regex:${intent}`);
-      });
-    }
-  });
-
-  it('returns tier "hot" for high confidence matches', () => {
-    const result = matchRegex("create a file called foo.ts");
-    expect(result.tier).toBe("hot");
-  });
-
-  it('returns tier "abstain" for long complex input', () => {
-    const result = matchRegex(
-      "I need you to analyze the distributed system architecture across all microservices and identify race conditions in the payment processing pipeline",
-    );
-    expect(result.tier).toBe("abstain");
-  });
-});

package/src/router/classifier/regex.ts

@@ -1,113 +0,0 @@
-import type { ClassifierResult } from "../types.js";
-
-const PATTERNS: Array<{
-  re: RegExp;
-  intent: string;
-  confidence: number;
-  tierHint: "fast" | "balanced" | "premium";
-}> = [
-  // English patterns
-  {
-    re: /\b(create|new|make|generate)\s+(a\s+)?(file|component|module|class|function)\b/i,
-    intent: "create-file",
-    confidence: 0.85,
-    tierHint: "fast",
-  },
-  { re: /\b(edit|modify|update|change|fix|patch)\s+(the\s+)?\S+/i, intent: "edit", confidence: 0.8, tierHint: "fast" },
-  {
-    re: /\b(run|execute|exec)\s+(the\s+)?(command|script|npm|bun|tsc|test|build)\b/i,
-    intent: "run-command",
-    confidence: 0.85,
-    tierHint: "fast",
-  },
-  { re: /\b(explain|what\s+does|describe|how\s+does)\b/i, intent: "explain", confidence: 0.7, tierHint: "fast" },
-  { re: /\b(search|find|grep|look\s+for)\b/i, intent: "search", confidence: 0.8, tierHint: "fast" },
-  { re: /\b(install|add)\s+(package|dep|dependency|module)\b/i, intent: "install", confidence: 0.85, tierHint: "fast" },
-  { re: /\b(list|show|ls|cat|read|head|tail)\b/i, intent: "read", confidence: 0.8, tierHint: "fast" },
-  {
-    re: /\bgit\s+(status|log|diff|add|commit|push|pull|branch|checkout)\b/i,
-    intent: "git",
-    confidence: 0.85,
-    tierHint: "fast",
-  },
-  { re: /\brefactor\b/i, intent: "refactor", confidence: 0.75, tierHint: "balanced" },
-  { re: /\b(architect|design|plan|strategy)\b/i, intent: "design", confidence: 0.7, tierHint: "premium" },
-  // Vietnamese patterns — higher tiers first to avoid premature fast-match.
-  // Debug-specific patterns sit ABOVE the broad "sửa/fix" edit pattern so
-  // bug-fix prompts ("sửa lỗi …", "fix bug …") don't get miscategorised as
-  // generic edits (which downstream maps to taskType=generate).
-  {
-    re: /(sửa lỗi|sua loi|báo lỗi|bao loi|fix bug|fix the bug|debug|traceback|stack trace|exception|không chạy|khong chay)/i,
-    intent: "debug",
-    confidence: 0.75,
-    tierHint: "balanced",
-  },
-  {
-    re: /(thiết kế|kiến trúc|\barchitect\b|\bdesign\b|xây dựng hệ thống|chiến lược)/i,
-    intent: "design",
-    confidence: 0.7,
-    tierHint: "premium",
-  },
-  {
-    re: /(tái cấu trúc|refactor|cấu trúc lại|tổ chức lại)/i,
-    intent: "refactor",
-    confidence: 0.75,
-    tierHint: "balanced",
-  },
-  {
-    re: /(thêm tính năng|thêm chức năng|bổ sung|implement)/i,
-    intent: "add-feature",
-    confidence: 0.75,
-    tierHint: "balanced",
-  },
-  {
-    re: /(tạo|tạo mới|sinh|generate)\s+.*(file|component|module|class|hàm|function)/i,
-    intent: "create-file",
-    confidence: 0.85,
-    tierHint: "fast",
-  },
-  { re: /(sửa|fix|chỉnh|update|cập nhật|patch)\s+\S+/i, intent: "edit", confidence: 0.8, tierHint: "fast" },
-  {
-    re: /(chạy|run|thực thi)\s+(lệnh|command|script|test|build)/i,
-    intent: "run-command",
-    confidence: 0.85,
-    tierHint: "fast",
-  },
-  { re: /(giải thích|explain|mô tả|describe)\s+/i, intent: "explain", confidence: 0.7, tierHint: "fast" },
-  { re: /(tìm|tìm kiếm|search|grep)\s+/i, intent: "search", confidence: 0.8, tierHint: "fast" },
-  {
-    re: /(cài|cài đặt|install|thêm)\s+(package|gói|thư viện|dep)/i,
-    intent: "install",
-    confidence: 0.85,
-    tierHint: "fast",
-  },
-  {
-    re: /(liệt kê|hiển thị|show|đọc|xem)\s+(file|thư mục|danh sách)/i,
-    intent: "read",
-    confidence: 0.8,
-    tierHint: "fast",
-  },
-];
-
-export function matchRegex(prompt: string): ClassifierResult {
-  // Hot-path: catch obvious English CLI commands without brain call
-  for (const p of PATTERNS) {
-    if (p.re.test(prompt)) {
-      return {
-        tier: "hot",
-        confidence: p.confidence,
-        reason: `regex:${p.intent}`,
-        tierHint: p.tierHint,
-      };
-    }
-  }
-
-  // Short messages in any language — delegate tier choice to brain via warm path
-  // but mark as "hot" so we skip the cold path (expensive LLM)
-  const trimmed = prompt.trim();
-  if (trimmed.length <= 80 && trimmed.split(/\s+/).length <= 10) {
-    return { tier: "hot", confidence: 0.6, reason: "regex:short-message", tierHint: "fast" };
-  }
-
-  return { tier: "abstain", confidence: 0.0, reason: "regex:no-match" };
-}

package/src/router/classifier/tree-sitter.test.ts

@@ -1,26 +0,0 @@
-import { beforeAll, describe, expect, it } from "vitest";
-import { initTreeSitter, lazyTreeSitter } from "./tree-sitter.js";
-
-describe("tree-sitter classifier", () => {
-  beforeAll(async () => {
-    await initTreeSitter(["typescript", "python"]);
-  }, 30_000);
-
-  it("parses TypeScript fenced code with confidence >= 0.55", () => {
-    const result = lazyTreeSitter("```ts\nconst x: number = 1\n```");
-    expect(result.confidence).toBeGreaterThanOrEqual(0.55);
-    expect(result.reason).toMatch(/^tree-sitter:typescript/);
-  });
-
-  it("parses Python fenced code with confidence >= 0.55", () => {
-    const result = lazyTreeSitter("```python\ndef f(x): return x\n```");
-    expect(result.confidence).toBeGreaterThanOrEqual(0.55);
-    expect(result.reason).toMatch(/^tree-sitter:python/);
-  });
-
-  it("returns abstain for prompts without fenced code", () => {
-    const result = lazyTreeSitter("hello world no code here");
-    expect(result.tier).toBe("abstain");
-    expect(result.reason).toBe("tree-sitter:no-fenced-code");
-  });
-});

package/src/router/classifier/tree-sitter.ts

@@ -1,87 +0,0 @@
-import type { ClassifierResult } from "../types.js";
-import { type GrammarId, loadGrammarBytes } from "./grammars.js";
-
-type ParserHandle = { parser: any; lang: any };
-const cache = new Map<GrammarId, ParserHandle>();
-let initOnce: Promise<void> | null = null;
-
-// Resolve web-tree-sitter classes from dynamic import (CJS/ESM compat)
-function resolveModule(mod: any): { Parser: any; Language: any } {
-  // Named exports take priority, then nested under default
-  const Parser = mod.Parser ?? mod.default?.Parser;
-  const Language = mod.Language ?? mod.default?.Language;
-  return { Parser, Language };
-}
-
-async function initParser(): Promise<void> {
-  if (initOnce) return initOnce;
-  initOnce = (async () => {
-    const ts = await import("web-tree-sitter");
-    const { Parser } = resolveModule(ts);
-    await Parser.init();
-  })();
-  return initOnce;
-}
-
-export async function initTreeSitter(grammars: GrammarId[] = ["typescript", "python"]): Promise<void> {
-  await initParser();
-  const ts = await import("web-tree-sitter");
-  const { Parser, Language } = resolveModule(ts);
-  for (const id of grammars) {
-    if (cache.has(id)) continue;
-    const bytes = await loadGrammarBytes(id);
-    const lang = await Language.load(bytes);
-    const parser = new Parser();
-    parser.setLanguage(lang);
-    cache.set(id, { parser, lang });
-  }
-}
-
-// Fire-and-forget warm at boot
-export function warmTreeSitter(): void {
-  void initTreeSitter();
-}
-
-function detectLang(prompt: string): GrammarId | null {
-  if (/```(ts|tsx|typescript)\b/i.test(prompt)) return "typescript";
-  if (/```(py|python)\b/i.test(prompt)) return "python";
-  return null;
-}
-
-export function lazyTreeSitter(prompt: string): ClassifierResult {
-  const id = detectLang(prompt);
-  if (!id) {
-    return {
-      tier: "abstain",
-      confidence: 0.0,
-      reason: "tree-sitter:no-fenced-code",
-    };
-  }
-  const handle = cache.get(id);
-  if (!handle) {
-    return { tier: "abstain", confidence: 0.3, reason: "tree-sitter:cold" };
-  }
-  // Extract fenced code body (best-effort)
-  const m = prompt.match(/```\w+\n([\s\S]*?)```/);
-  const body = m ? m[1] : prompt;
-  try {
-    const tree = handle.parser.parse(body);
-    // Confidence based on syntactically meaningful node count
-    const root = tree.rootNode;
-    const named = root.namedChildCount;
-    const errs = root.hasError ? 1 : 0;
-    const conf = named >= 1 && !errs ? 0.8 : named >= 1 ? 0.55 : 0.3;
-    return {
-      tier: conf >= 0.55 ? "hot" : "abstain",
-      confidence: conf,
-      reason: `tree-sitter:${id}`,
-      tierHint: "balanced",
-    };
-  } catch {
-    return {
-      tier: "abstain",
-      confidence: 0.0,
-      reason: `tree-sitter:${id}-parse-error`,
-    };
-  }
-}

package/src/router/cold.test.ts

@@ -1,62 +0,0 @@
-import { afterAll, beforeAll, describe, expect, it } from "vitest";
-import { type StubHandle, startStubEEServer } from "../__test-stubs__/ee-server.js";
-import { createEEClient } from "../ee/client.js";
-import { setDefaultEEClient } from "../ee/intercept.js";
-import { loadCatalog } from "../models/registry.js";
-import { callColdRoute } from "./cold.js";
-
-describe("callColdRoute", () => {
-  let stub: StubHandle;
-
-  beforeAll(async () => {
-    await loadCatalog();
-    stub = await startStubEEServer({
-      coldRoute: (_req) => ({
-        model: "deepseek-v4-flash",
-        tier: "premium" as const,
-        reason: "fallback",
-        taskHash: "test-hash",
-      }),
-    });
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${stub.port}` }));
-  });
-
-  afterAll(async () => {
-    await stub.stop();
-  });
-
-  it("returns RouteDecision when stub responds successfully", async () => {
-    const result = await callColdRoute("write a function", {
-      tenantId: "default",
-      cwd: "/tmp",
-    });
-    expect(result).not.toBeNull();
-    expect(result!.tier).toBe("cold");
-    expect(result!.model).toBe("deepseek-v4-flash");
-    // provider is resolved by detectProviderForModel; catalog maps deepseek-v4-flash → provider: deepseek (native api.deepseek.com).
-    expect(result!.provider).toBe("deepseek");
-    expect(result!.reason).toContain("cold:");
-  });
-
-  it("returns null when cold path times out (>1000ms)", async () => {
-    const slowStub = await startStubEEServer({
-      latencyMs: 1500,
-      coldRoute: () => ({
-        model: "deepseek-v4-flash",
-        tier: "premium" as const,
-        reason: "fallback",
-        taskHash: "test-hash",
-      }),
-    });
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${slowStub.port}` }));
-
-    const result = await callColdRoute("write a function", {
-      tenantId: "default",
-      cwd: "/tmp",
-    });
-    expect(result).toBeNull();
-
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${stub.port}` }));
-    await slowStub.stop();
-  });
-});

package/src/router/cold.ts

@@ -1,27 +0,0 @@
-/**
- * Cold-path router tier.
- *
- * Calls EE /api/cold-route with a hard 1s timeout.
- * Returns null on timeout/5xx/network error (graceful degradation).
- */
-import { getDefaultEEClient } from "../ee/intercept.js";
-import { detectProviderForModel } from "../providers/runtime.js";
-import type { RouteDecision } from "./types.js";
-
-export async function callColdRoute(
-  prompt: string,
-  opts: { tenantId: string; cwd: string; signal?: AbortSignal; context?: Record<string, unknown> },
-): Promise<RouteDecision | null> {
-  const r = await getDefaultEEClient().coldRoute(
-    { task: prompt, tenantId: opts.tenantId, cwd: opts.cwd, context: opts.context as never },
-    opts.signal,
-  );
-  if (!r) return null;
-  return {
-    tier: "cold",
-    model: r.model,
-    provider: detectProviderForModel(r.model),
-    reason: `cold:${r.reason}`,
-    taskHash: r.taskHash,
-  };
-}

package/src/router/decide.test.ts

@@ -1,147 +0,0 @@
-import { afterAll, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
-import { type StubHandle, startStubEEServer } from "../__test-stubs__/ee-server.js";
-import { createEEClient } from "../ee/client.js";
-import { setDefaultEEClient } from "../ee/intercept.js";
-import { type DecideOpts, decide } from "./decide.js";
-import { routerStore } from "./store.js";
-
-// Mock bridge to always return null so tests go through HTTP path
-vi.mock("../ee/bridge.js", () => ({
-  routeModel: vi.fn().mockResolvedValue(null),
-  classifyViaBrain: vi.fn().mockResolvedValue(null),
-  searchCollection: vi.fn().mockResolvedValue([]),
-  getEmbeddingRaw: vi.fn().mockResolvedValue(null),
-  routeTask: vi.fn().mockResolvedValue(null),
-}));
-
-const BASE_OPTS: DecideOpts = {
-  tenantId: "default",
-  cwd: "/tmp",
-  defaultModel: "claude-sonnet-4-20250514",
-  defaultProvider: "anthropic",
-  threshold: 0.55,
-};
-
-describe("decide()", () => {
-  let stub: StubHandle;
-
-  beforeAll(async () => {
-    stub = await startStubEEServer({
-      routeModel: (_req) => ({
-        model: "qwen2.5-coder",
-        tier: "balanced" as const,
-        confidence: 0.7,
-        reason: "ee-warm",
-        source: "brain",
-        taskHash: "test-hash",
-      }),
-      coldRoute: (_req) => ({
-        model: "deepseek-v4-flash",
-        tier: "premium" as const,
-        reason: "ee-cold",
-        taskHash: "test-hash",
-      }),
-    });
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${stub.port}` }));
-  });
-
-  afterAll(async () => {
-    await stub.stop();
-  });
-
-  beforeEach(() => {
-    routerStore.setState({
-      tier: "hot",
-      degraded: false,
-      lastDecision: null,
-      lastHealthCheckAtMs: 0,
-    });
-  });
-
-  it("falls through to warm when classifier abstains (stub classifier always abstains)", async () => {
-    const result = await decide(
-      "I need to analyze and restructure the payment processing module with proper error boundaries and retry logic across multiple services",
-      BASE_OPTS,
-    );
-    expect(result.tier).toBe("warm");
-    expect(result.model).toBe("qwen2.5-coder");
-    expect(routerStore.getState().lastDecision).toEqual(result);
-  });
-
-  it("falls through to cold when warm returns null", async () => {
-    // Stub that returns null for warm but succeeds for cold.
-    // Use defaultModel as cold-route model to avoid cap-driven downgrade for unknown models.
-    const coldOnlyStub = await startStubEEServer({
-      routeModel: undefined, // 500 -> null
-      coldRoute: () => ({
-        model: BASE_OPTS.defaultModel,
-        tier: "cold" as const,
-        reason: "ee-cold",
-        taskHash: "test-hash",
-      }),
-    });
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${coldOnlyStub.port}` }));
-
-    const result = await decide(
-      "I need to analyze and restructure the payment processing module with proper error boundaries and retry logic across multiple services",
-      BASE_OPTS,
-    );
-    expect(result.tier).toBe("cold");
-    expect(result.model).toBe(BASE_OPTS.defaultModel);
-
-    // Restore
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${stub.port}` }));
-    await coldOnlyStub.stop();
-  });
-
-  it("returns fallback when both warm and cold are unreachable", async () => {
-    // Stub with no handlers -> both return 500 -> null
-    const deadStub = await startStubEEServer({});
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${deadStub.port}` }));
-
-    const result = await decide(
-      "I need to analyze and restructure the payment processing module with proper error boundaries and retry logic across multiple services",
-      BASE_OPTS,
-    );
-    expect(result.model).toBe("claude-sonnet-4-20250514");
-    expect(result.reason).toBe("fallback:ee-unreachable");
-
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${stub.port}` }));
-    await deadStub.stop();
-  });
-
-  it("returns degraded tier in fallback when store.degraded is true", async () => {
-    const deadStub = await startStubEEServer({});
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${deadStub.port}` }));
-    routerStore.setState({ degraded: true });
-
-    const result = await decide(
-      "I need to analyze and restructure the payment processing module with proper error boundaries and retry logic across multiple services",
-      BASE_OPTS,
-    );
-    expect(result.tier).toBe("degraded");
-    expect(result.reason).toBe("fallback:ee-unreachable");
-
-    setDefaultEEClient(createEEClient({ baseUrl: `http://localhost:${stub.port}` }));
-    await deadStub.stop();
-  });
-});
-
-describe("routerStore", () => {
-  it("exposes subscribe/getState/setState and emits on changes", () => {
-    const changes: any[] = [];
-    const unsub = routerStore.subscribe((s) => changes.push({ ...s }));
-
-    routerStore.setState({ tier: "warm" });
-    expect(changes.length).toBe(1);
-    expect(changes[0].tier).toBe("warm");
-
-    routerStore.setState({ tier: "cold" });
-    expect(changes.length).toBe(2);
-    expect(changes[1].tier).toBe("cold");
-
-    unsub();
-    routerStore.setState({ tier: "hot" });
-    expect(changes.length).toBe(2); // No more notifications after unsub
-  });
-});