auto-model-router 0.4.14 → 0.6.0

package/test/team-client.test.ts

@@ -0,0 +1,97 @@
+import { describe, expect, test } from "bun:test";
+import { existsSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+
+import { addExtensions, codexBlock, joinTeam, setDotenv, type JoinOptions } from "../src/cli/join.ts";
+import { parseTeamClient, readTeamClient, teamProviderRegistration } from "../omp-extension/team-logic.ts";
+
+/**
+ * Team-client mode: team.json puts the omp extensions on the team endpoint,
+ * and `join` configures every harness it finds without touching anything it
+ * does not recognise.
+ */
+
+describe("team-logic", () => {
+	test("team.json is parsed defensively and turned into omp's provider registration", () => {
+		expect(parseTeamClient("nope")).toBeNull();
+		expect(parseTeamClient(JSON.stringify({ url: "https://t/", key: "" }))).toBeNull();
+		const t = parseTeamClient(JSON.stringify({ url: "https://team.example/", key: "amrt_k", userId: "u_1", name: "Ada" }))!;
+		expect(t.url).toBe("https://team.example");
+		const reg = teamProviderRegistration(t, "sess-1", true, { inputPerMtok: 1, outputPerMtok: 4 });
+		expect(reg).toMatchObject({ baseUrl: "https://team.example/v1", api: "openai-completions", apiKey: "amrt_k", headers: { "X-Omp-Session": "sess-1", "X-Omp-Subagent": "1" } });
+		expect(reg.models.map((m) => m.id)).toEqual(["auto", "auto-cheap", "auto-max"]);
+		expect(reg.models[0]!.cost).toEqual({ input: 1, output: 4, cacheRead: 0.1, cacheWrite: 1.25 });
+		const dir = mkdtempSync(join(tmpdir(), "amr-team-client-"));
+		expect(readTeamClient(dir)).toBeNull();
+		writeFileSync(join(dir, "team.json"), JSON.stringify({ url: "https://t", key: "k" }));
+		expect(readTeamClient(dir)?.key).toBe("k");
+		rmSync(dir, { recursive: true, force: true });
+	});
+
+	test("text edits keep files byte-identical apart from the lines they add", () => {
+		expect(addExtensions("", ["/a.ts"])).toBe("extensions:\n  - /a.ts\n");
+		expect(addExtensions("foo: 1\nextensions:\n  - /x.ts\nbar: 2\n", ["/x.ts", "/a.ts"])).toBe("foo: 1\nextensions:\n  - /a.ts\n  - /x.ts\nbar: 2\n");
+		expect(addExtensions("foo: 1\r\n", ["/a.ts"])).toBe("foo: 1\r\nextensions:\r\n  - /a.ts\r\n");
+		expect(setDotenv("A=1\nB=2\n", { B: "3", C: "4" })).toBe("A=1\nB=3\nC=4\n");
+		expect(setDotenv("", { A: "1" })).toBe("A=1\n");
+		expect(codexBlock("https://t")).toContain('base_url = "https://t/v1"');
+	});
+});
+
+describe("join", () => {
+	function scenario(extra: Partial<JoinOptions> = {}): { home: string; o: JoinOptions } {
+		const home = mkdtempSync(join(tmpdir(), "amr-join-"));
+		const agent = join(home, ".omp", "agent");
+		mkdirSync(agent, { recursive: true });
+		writeFileSync(join(agent, "config.yml"), "extensions:\n  - E:/other/ext.ts\nsetupVersion: 2\n");
+		mkdirSync(join(home, ".hermes"), { recursive: true });
+		writeFileSync(join(home, ".hermes", ".env"), "OPENAI_API_KEY=x\n");
+		mkdirSync(join(home, ".codex"), { recursive: true });
+		writeFileSync(join(home, ".codex", "config.toml"), 'model = "gpt-5"\n');
+		const o: JoinOptions = { url: "https://team.example", key: "amrt_key", userId: "u_ada", name: "Ada", profile: false, dryRun: false, only: [], env: { HOME: home, HERMES_HOME: join(home, ".hermes"), PI_CODING_AGENT_DIR: agent, AUTO_MODEL_ROUTER_HOME: join(home, ".auto-model-router") }, home, packageDir: process.cwd(), platform: "linux", pathHas: (b) => b === "claude" || b === "aider", ...extra };
+		return { home, o };
+	}
+
+	test("writes team.json and configures omp, Hermes, Codex, Aider and Claude Code idempotently", () => {
+		const { home, o } = scenario();
+		const r1 = joinTeam(o);
+		expect(existsSync(r1.teamFile)).toBe(true);
+		expect(JSON.parse(readFileSync(r1.teamFile, "utf8"))).toMatchObject({ url: "https://team.example", key: "amrt_key", userId: "u_ada", name: "Ada" });
+		const ompCfg = readFileSync(join(home, ".omp", "agent", "config.yml"), "utf8");
+		expect(ompCfg).toContain("extensions:\n  - ");
+		expect(ompCfg).toContain("omp-extension/router-embed.ts");
+		expect(ompCfg).toContain("E:/other/ext.ts");
+		expect(ompCfg).toContain("setupVersion: 2");
+		expect(existsSync(join(home, ".hermes", "plugins", "model-providers", "auto-model-router", "__init__.py"))).toBe(true);
+		expect(existsSync(join(home, ".hermes", "plugins", "auto-model-router", "plugin.yaml"))).toBe(true);
+		expect(readFileSync(join(home, ".hermes", ".env"), "utf8")).toBe("OPENAI_API_KEY=x\nAUTO_MODEL_ROUTER_URL=https://team.example\nAUTO_MODEL_ROUTER_API_KEY=amrt_key\n");
+		expect(readFileSync(join(home, ".codex", "config.toml"), "utf8")).toContain("[model_providers.auto-model-router]");
+		expect(readFileSync(join(home, ".aider.conf.yml"), "utf8")).toContain("openai-api-base: https://team.example/v1");
+		expect(r1.configured.join("\n")).toMatch(/omp[\s\S]*Hermes[\s\S]*Codex[\s\S]*Aider[\s\S]*Claude Code/);
+		expect(r1.envLines).toEqual(["AUTO_MODEL_ROUTER_URL=https://team.example", "AUTO_MODEL_ROUTER_API_KEY=amrt_key", "ANTHROPIC_BASE_URL=https://team.example", "ANTHROPIC_API_KEY=amrt_key"]);
+		// Running again changes nothing.
+		const snapshot = [ompCfg, readFileSync(join(home, ".codex", "config.toml"), "utf8"), readFileSync(join(home, ".aider.conf.yml"), "utf8")];
+		joinTeam(o);
+		expect([readFileSync(join(home, ".omp", "agent", "config.yml"), "utf8"), readFileSync(join(home, ".codex", "config.toml"), "utf8"), readFileSync(join(home, ".aider.conf.yml"), "utf8")]).toEqual(snapshot);
+		rmSync(home, { recursive: true, force: true });
+	});
+
+	test("--harness restricts, --dry-run writes nothing, --profile appends once to the shell rc", () => {
+		const { home, o } = scenario({ only: ["omp"], dryRun: true });
+		const r = joinTeam(o);
+		expect(existsSync(r.teamFile)).toBe(false);
+		expect(r.configured.some((c) => c.startsWith("omp"))).toBe(true);
+		expect(r.skipped.some((s) => s.startsWith("Hermes"))).toBe(true);
+		expect(existsSync(join(home, ".hermes", "plugins"))).toBe(false);
+		const { home: h2, o: o2 } = scenario({ profile: true, env: { SHELL: "/bin/zsh" } });
+		o2.env = { ...o.env, HOME: h2, HERMES_HOME: join(h2, ".hermes"), PI_CODING_AGENT_DIR: join(h2, ".omp", "agent"), AUTO_MODEL_ROUTER_HOME: join(h2, ".auto-model-router"), SHELL: "/bin/zsh" };
+		joinTeam(o2);
+		joinTeam(o2);
+		const rc = readFileSync(join(h2, ".zshrc"), "utf8");
+		expect(rc.split("# auto-model-router team").length).toBe(2);
+		expect(rc).toContain("export ANTHROPIC_BASE_URL=https://team.example");
+		rmSync(home, { recursive: true, force: true });
+		rmSync(h2, { recursive: true, force: true });
+	});
+});

package/test/turn.test.ts

@@ -77,6 +77,7 @@ function mkConfig(escalation: Partial<EscalationConfig> = {}): RouterConfig {
 		compaction: { enabled: false, budgetTokens: 40_000, floorRatio: 1, fitToWindow: true, protectRecentTurns: 4, maxToolResultBytes: 4_096, keepHeadBytes: 512, keepTailBytes: 512, elideSupersededReads: true, collapseDuplicateResults: true, replanGrowthRatio: 1, digestToolResults: false, digestMaxPerTurn: 2 },
 		budget: { onExceeded: "downgrade" },
 		report: { baselines: [], dailySummary: false },
+		anthropic: { models: { "*haiku*": "auto-cheap", "claude-*": "auto" } },
 		harnessSwitch: { enabled: false, models: {}, minConfidence: 0.6 },
 		digest: { enabled: false, minBytes: 12_000, maxBytes: 400_000, tools: ["read"], fromTier: "moderate", tier: "simple", model: "", maxOutputTokens: 700, maxCostUsd: 0.02, timeoutMs: 25_000, toolAliases: {} },
 		profiles: [],