rules.test.ts

import { describe, expect, it } from "vitest";

import { classifyByRules } from "./rules.js";
import { DEFAULT_ROUTING_CONFIG } from "./config.js";

const scoring = DEFAULT_ROUTING_CONFIG.scoring;
const signalsFor = (prompt: string) =>
  classifyByRules(prompt, undefined, Math.ceil(prompt.length / 4), scoring).signals;

describe("classifyByRules keyword matching", () => {
  it("does not count a keyword that only appears inside a longer word", () => {
    // "classic" is not `class`, "history" is not `story`, "credit" is not `edit`,
    // "pipeline" is not `pip`, "prefix" is not `fix`.
    expect(signalsFor("I need a recipe for a classic dinner party")).not.toContainEqual(
      expect.stringContaining("code"),
    );
    expect(signalsFor("Tell me the history of the Roman Empire")).not.toContainEqual(
      expect.stringContaining("creative"),
    );
    expect(
      signalsFor("Why does my credit card pipeline report a prefix mismatch?"),
    ).not.toContainEqual(expect.stringContaining("agentic"));
  });

  it("still matches ordinary English inflections of a keyword", () => {
    expect(signalsFor("I imported the classes and defined two functions")).toContainEqual(
      expect.stringContaining("code"),
    );
    expect(signalsFor("Keep editing until the tests pass, then deploying")).toContainEqual(
      expect.stringContaining("agentic"),
    );
  });

  it("keeps substring matching for CJK keywords, which have no word boundaries", () => {
    expect(signalsFor("请帮我定义一个函数来处理输入")).toContainEqual(
      expect.stringContaining("code"),
    );
  });

  it("keeps inflections of the ASCII keywords of other European languages", () => {
    // `beweis` and `mathematisch` are German reasoning markers spelled in
    // ASCII; the end boundary must not require English morphology of them.
    const result = classifyByRules("Beweise den mathematischen Satz.", undefined, 8, scoring);
    expect(result.tier).toBe("REASONING");
  });

  it("matches doubled-consonant English inflections", () => {
    const result = classifyByRules(
      "Keep debugging and editing until it works.",
      undefined,
      11,
      scoring,
    );
    expect(result.agenticScore).toBe(0.6);
  });

  it("does not let a short keyword plus a suffix spell a different word", () => {
    // "pipes" is not `pip`, "defer" is not `def`, "important" is not `import`,
    // "basic" is not `asic`, "semantic" is not `sem`.
    const prompt = "It is important to defer the basic semantic check for the pipes";
    const result = classifyByRules(prompt, undefined, 16, scoring);
    expect(result.signals.filter((s) => /code|agentic|domain|negation/.test(s))).toEqual([]);
    expect(signalsFor("Apply the fixes and rerun")).toContainEqual(expect.stringContaining("agentic"));
  });

  it("treats a CJK character next to a Latin keyword as a word boundary", () => {
    // Mixed-script prompts carry no spaces between scripts.
    const result = classifyByRules("请step by step证明这个结论。", undefined, 8, scoring);
    expect(result.signals).toContainEqual(expect.stringContaining("step by step"));
    expect(result.tier).toBe("REASONING");
  });

  it("keeps stem matching for Cyrillic keywords", () => {
    // "оптимизаци" is a deliberate stem that must still hit "оптимизацию".
    expect(signalsFor("Проведи оптимизацию алгоритма для базы данных")).toContainEqual(
      expect.stringContaining("technical"),
    );
  });
});