auto-model-router 0.3.4 → 0.4.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (48) hide show
  1. package/.omp-plugin/marketplace.json +2 -2
  2. package/README.md +29 -4
  3. package/omp-extension/report-logic.ts +93 -0
  4. package/omp-extension/router-configure.ts +128 -2
  5. package/omp-extension/router-embed.ts +9 -6
  6. package/package.json +1 -1
  7. package/src/catalog/ollama-catalog.ts +30 -2
  8. package/src/cli/config-wizard.ts +11 -0
  9. package/src/cli/report.ts +7 -2
  10. package/src/config/defaults.ts +17 -0
  11. package/src/config/schema.ts +8 -0
  12. package/src/config/types.ts +65 -0
  13. package/src/cost/feedback.ts +81 -0
  14. package/src/cost/ledger.ts +110 -5
  15. package/src/cost/report.ts +118 -2
  16. package/src/cost/types.ts +32 -1
  17. package/src/router/candidates.ts +13 -4
  18. package/src/router/classify.ts +13 -0
  19. package/src/router/features.ts +59 -1
  20. package/src/router/index.ts +23 -5
  21. package/src/router/learned.ts +202 -0
  22. package/src/router/select.ts +64 -8
  23. package/src/router/types.ts +39 -1
  24. package/src/server/http.ts +82 -5
  25. package/src/server/overrides.ts +83 -0
  26. package/src/server/providers.ts +10 -2
  27. package/src/server/turn.ts +16 -1
  28. package/src/upstream/ollama-usage.ts +79 -2
  29. package/src/util/sqlite.ts +30 -0
  30. package/src/wire/openai/request.ts +4 -0
  31. package/src/wire/types.ts +2 -0
  32. package/test/classify.test.ts +13 -0
  33. package/test/config-wizard.test.ts +8 -6
  34. package/test/controls.test.ts +238 -0
  35. package/test/escalate.test.ts +1 -0
  36. package/test/failover.test.ts +6 -4
  37. package/test/features.test.ts +63 -0
  38. package/test/http-resilience.test.ts +1 -1
  39. package/test/learned.test.ts +61 -0
  40. package/test/ollama.test.ts +74 -2
  41. package/test/report-hub.test.ts +6 -2
  42. package/test/report-logic.test.ts +3 -0
  43. package/test/report.test.ts +57 -0
  44. package/test/select.test.ts +126 -1
  45. package/test/trust-attribution.test.ts +95 -0
  46. package/test/turn.test.ts +36 -4
  47. package/tools/replay.ts +267 -156
  48. package/tools/train-classifier.ts +111 -0
@@ -29,7 +29,7 @@ import type {
29
29
 
30
30
  function mkConfig(escalation: Partial<EscalationConfig> = {}): RouterConfig {
31
31
  return {
32
- server: { host: "127.0.0.1", port: 8787, maxConcurrentTurns: 24 },
32
+ server: { host: "127.0.0.1", port: 8787, maxConcurrentTurns: 24, subagentProfile: "auto-sub" },
33
33
  openrouter: { baseUrl: "https://openrouter.ai/api/v1", apiKey: "", title: "test", timeoutMs: 30_000, catalogTtlMs: 3_600_000, catalogRefreshMs: 0 },
34
34
  ollama: { enabled: false, baseUrl: "http://127.0.0.1:11434/v1", apiKey: "", timeoutMs: 30_000, catalogTtlMs: 300_000, includeLocal: false, prices: {}, twins: {}, costBias: 1, biasUntilUsage: 0.9, usagePollMs: 0, quotaCooldownMs: 0, rateLimitCooldownMs: 0, planCreditsUsd: 0 },
35
35
  benchmarks: { enabled: false, artificialAnalysisApiKey: "", benchlm: true, refreshMs: 86_400_000, timeoutMs: 30_000, useLocalScores: false },
@@ -46,10 +46,10 @@ function mkConfig(escalation: Partial<EscalationConfig> = {}): RouterConfig {
46
46
  data: { axis: "intelligence", minQuality: 0 },
47
47
  chat: { axis: "intelligence", minQuality: 0 },
48
48
  },
49
- filters: { allow: [], deny: [], includeFree: false, requireToolSupport: true, minTrust: 0.6, minTrustSamples: 5, trustScopedByHarness: false, trustWindowDays: 0, contextHeadroom: 1.2, latencyWeight: 0, latencyReferenceMs: 5000, latencyReferenceTokensPerSec: 30, latencyMinSamples: 20, escalationCostWeight: 0 },
49
+ filters: { allow: [], deny: [], includeFree: false, requireToolSupport: true, minTrust: 0.6, feedbackWeight: 0, minTrustSamples: 5, trustScopedByHarness: false, trustWindowDays: 0, contextHeadroom: 1.2, latencyWeight: 0, latencyReferenceMs: 5000, latencyReferenceTokensPerSec: 30, cacheReliabilityMinSamples: 10, latencyMinSamples: 20, escalationCostWeight: 0 },
50
50
  classifier: {
51
51
  ambiguityThreshold: 0,
52
- model: "test/adjudicator",
52
+ model: "test/adjudicator", learnedModelPath: "",
53
53
  maxCostFraction: 0.1,
54
54
  maxCostUsd: 0.01,
55
55
  timeoutMs: 5000,
@@ -57,7 +57,7 @@ function mkConfig(escalation: Partial<EscalationConfig> = {}): RouterConfig {
57
57
  toolAxis: "coding",
58
58
  chatAxis: "intelligence",
59
59
  agenticLoopDepth: 3,
60
- mechanicalRetryFactor: 0.2,
60
+ mechanicalRetryFactor: 0.2, readOnlyToolWeight: 0,
61
61
  reasoningWeights: { medium: 0.14, high: 0.24, xhigh: 0.3, max: 0.34 },
62
62
  },
63
63
  escalation: {
@@ -76,6 +76,7 @@ function mkConfig(escalation: Partial<EscalationConfig> = {}): RouterConfig {
76
76
  context: { enabled: false, baseUrl: "", token: "", defaultScope: "", timeoutMs: 3_000, maxStalenessMs: 900_000, maxBlockChars: 24_000, memoryLimit: 8, docsLimit: 2, sessionLimit: 6, briefChars: 0, recordTurns: false, maxQueue: 64 },
77
77
  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 },
78
78
  budget: { onExceeded: "downgrade" },
79
+ report: { baselines: [] },
79
80
  profiles: [],
80
81
  ledger: { path: ":memory:", blendWindowDays: 7, blendMinSamples: 20, fallbackBlend: { inputPerMtok: 1, outputPerMtok: 4 }, conversationTtlMs: 86_400_000 },
81
82
  adaptiveTierFloors: true,
@@ -91,6 +92,7 @@ function mkReq(): NormRequest {
91
92
  harnessId: "",
92
93
  ompSessionId: "",
93
94
  agentdoxScope: "",
95
+ isSubagent: false,
94
96
  requestedModel: "auto",
95
97
  messages: [{ role: "user", text: "hi", images: 0, textBytes: 2, toolCalls: [] }],
96
98
  tools: [],
@@ -306,3 +306,66 @@ describe("user-visible tool failure (review 2026-09-05)", () => {
306
306
  expect(f.lastToolFailed).toBe(false);
307
307
  });
308
308
  });
309
+
310
+ describe("prompt anatomy", () => {
311
+ test("splits prompt bytes by role and marks the older half and stale tool results", () => {
312
+ // 24 non-system messages: 12 tool-call/result pairs. The newest 20
313
+ // non-system messages are "fresh"; the 4 before them hold 2 stale tool results.
314
+ const messages: unknown[] = [SYSTEM];
315
+ for (let i = 0; i < 12; i++) {
316
+ messages.push(toolCall(`c${i}`, "bash", `{"command":"ls ${i}"}`));
317
+ messages.push({ role: "tool", tool_call_id: `c${i}`, content: "x".repeat(100) });
318
+ }
319
+ messages.push({ role: "user", content: "y".repeat(50) });
320
+ const f = extractFeatures(req(messages), 5000);
321
+ const a = f.anatomy!;
322
+ expect(a.messages).toBe(26);
323
+ expect(a.systemBytes).toBe("You are a coding agent.".length);
324
+ expect(a.userBytes).toBe(50);
325
+ expect(a.toolBytes).toBe(1200);
326
+ // 25 non-system messages; the older half is the first 12 (6 pairs ⇒ 6 tool results).
327
+ expect(a.olderHalfBytes).toBeGreaterThanOrEqual(600);
328
+ expect(a.olderHalfBytes).toBeLessThan(1200);
329
+ // Stale: tool results among the first 25-20 = 5 non-system messages ⇒ results at index 1 and 3.
330
+ expect(a.staleToolBytes).toBe(200);
331
+ });
332
+
333
+ test("a bare chat request has no stale tool bytes", () => {
334
+ const a = extractFeatures(req([SYSTEM, { role: "user", content: "hi" }]), 20).anatomy!;
335
+ expect(a.toolBytes).toBe(0);
336
+ expect(a.staleToolBytes).toBe(0);
337
+ expect(a.olderHalfBytes).toBe(0);
338
+ });
339
+ });
340
+
341
+ describe("subagent and read-only tool loop signals", () => {
342
+ test("a tool-result tail behind read-only calls is flagged; a write call clears it", () => {
343
+ const reads = req([
344
+ SYSTEM,
345
+ { role: "user", content: "find the retry helper" },
346
+ { role: "assistant", content: null, tool_calls: [
347
+ { id: "a", type: "function", function: { name: "grep", arguments: "{\"pattern\":\"retry\"}" } },
348
+ { id: "b", type: "function", function: { name: "read", arguments: "{\"path\":\"x.ts\"}" } },
349
+ ] },
350
+ { role: "tool", tool_call_id: "a", content: "x.ts:12" },
351
+ { role: "tool", tool_call_id: "b", content: "export function retry() {}" },
352
+ ]);
353
+ expect(extractFeatures(reads, 1000).readOnlyToolTail).toBe(true);
354
+ const write = req([
355
+ SYSTEM,
356
+ { role: "user", content: "fix it" },
357
+ toolCall("c", "edit", "{\"path\":\"x.ts\"}"),
358
+ { role: "tool", tool_call_id: "c", content: "ok" },
359
+ ]);
360
+ expect(extractFeatures(write, 1000).readOnlyToolTail).toBe(false);
361
+ // A fresh user turn is never a read-only tail, whatever came before.
362
+ expect(extractFeatures(req([SYSTEM, { role: "user", content: "now what?" }]), 100).readOnlyToolTail).toBe(false);
363
+ });
364
+
365
+ test("the subagent marker rides on the request", () => {
366
+ const r = parseChatRequest({ model: "auto", messages: [SYSTEM, { role: "user", content: "hi" }], tools: TOOLS }, new Headers({ "x-omp-subagent": "1" }));
367
+ expect(r.isSubagent).toBe(true);
368
+ expect(extractFeatures(r, 100).isSubagent).toBe(true);
369
+ expect(req([SYSTEM, { role: "user", content: "hi" }]).isSubagent).toBe(false);
370
+ });
371
+ });
@@ -10,7 +10,7 @@ describe("HTTP server resilience against dead streams", () => {
10
10
  beforeAll(() => {
11
11
  const cfg: RouterConfig = {
12
12
  ...DEFAULT_CONFIG,
13
- server: { host: "127.0.0.1", port: 0, maxConcurrentTurns: 24 },
13
+ server: { host: "127.0.0.1", port: 0, maxConcurrentTurns: 24, subagentProfile: "auto-sub" },
14
14
  ledger: { ...DEFAULT_CONFIG.ledger, path: ":memory:" },
15
15
  context: { ...DEFAULT_CONFIG.context, enabled: false },
16
16
  logLevel: "silent",
@@ -0,0 +1,61 @@
1
+ import { describe, expect, test } from "bun:test";
2
+
3
+ import { auc, FEATURE_NAMES, learnedVector, predictRisk, trainLogistic, type LearnedModel } from "../src/router/learned.ts";
4
+
5
+ /**
6
+ * The learned escalation-risk model: a deterministic logistic regression that
7
+ * must separate a synthetic dataset, score by rank correctly, and tolerate
8
+ * old ledger rows missing fields.
9
+ */
10
+
11
+ describe("learnedVector", () => {
12
+ test("matches FEATURE_NAMES in length and tolerates missing fields", () => {
13
+ const v = learnedVector({});
14
+ expect(v).toHaveLength(FEATURE_NAMES.length);
15
+ expect(v.every((x) => x === 0)).toBe(true);
16
+ const w = learnedVector({ promptTokens: 1000, lastToolFailed: true, requestedReasoning: "high", complexityKeywords: ["refactor", "migrate"] });
17
+ expect(w[FEATURE_NAMES.indexOf("log_prompt_tokens")]).toBeCloseTo(Math.log1p(1000), 6);
18
+ expect(w[FEATURE_NAMES.indexOf("last_tool_failed")]).toBe(1);
19
+ expect(w[FEATURE_NAMES.indexOf("requested_reasoning")]).toBe(3);
20
+ expect(w[FEATURE_NAMES.indexOf("complexity_keywords")]).toBe(2);
21
+ });
22
+ });
23
+
24
+ describe("auc", () => {
25
+ test("perfect ranking is 1, inverted is 0, ties count half", () => {
26
+ expect(auc([0.9, 0.8, 0.1, 0.2], [1, 1, 0, 0])).toBe(1);
27
+ expect(auc([0.1, 0.2, 0.9, 0.8], [1, 1, 0, 0])).toBe(0);
28
+ expect(auc([0.5, 0.5], [1, 0])).toBe(0.5);
29
+ expect(auc([0.3], [1])).toBe(0.5);
30
+ });
31
+ });
32
+
33
+ describe("trainLogistic", () => {
34
+ test("separates a dataset where escalation follows failed tools and long prompts", () => {
35
+ const xs: number[][] = [];
36
+ const ys: number[] = [];
37
+ let seed = 7;
38
+ const rnd = () => ((seed = (seed * 1103515245 + 12345) % 2147483648) / 2147483648);
39
+ for (let i = 0; i < 2000; i++) {
40
+ const failed = rnd() < 0.1;
41
+ const prompt = Math.floor(rnd() * 150_000);
42
+ const y = failed && prompt > 60_000 ? (rnd() < 0.9 ? 1 : 0) : rnd() < 0.02 ? 1 : 0;
43
+ xs.push(learnedVector({ promptTokens: prompt, lastToolFailed: failed, turnDepth: Math.floor(rnd() * 100), toolCount: 12 }));
44
+ ys.push(y);
45
+ }
46
+ const fit = trainLogistic(xs.slice(0, 1600), ys.slice(0, 1600), { epochs: 300 });
47
+ const model: LearnedModel = { version: 1, trainedAtMs: 0, rows: 1600, positives: 0, names: [...FEATURE_NAMES], means: fit.means, stds: fit.stds, weights: fit.weights, bias: fit.bias, auc: 0 };
48
+ const scores = xs.slice(1600).map((x) => {
49
+ // predictRisk takes features; rebuild the same vector through it for parity.
50
+ const f = { promptTokens: Math.expm1(x[0]!), lastToolFailed: x[8] === 1, turnDepth: x[2]!, toolCount: x[3]! };
51
+ return predictRisk(model, f);
52
+ });
53
+ expect(auc(scores, ys.slice(1600))).toBeGreaterThan(0.85);
54
+ expect(fit.weights[FEATURE_NAMES.indexOf("last_tool_failed")]!).toBeGreaterThan(0);
55
+ expect(fit.weights[FEATURE_NAMES.indexOf("log_prompt_tokens")]!).toBeGreaterThan(0);
56
+ });
57
+
58
+ test("refuses an empty dataset", () => {
59
+ expect(() => trainLogistic([], [])).toThrow();
60
+ });
61
+ });
@@ -12,17 +12,30 @@ import {
12
12
  parseOllamaListing,
13
13
  parseOllamaShow,
14
14
  type OllamaListing,
15
+ loadOllamaCatalogCache,
15
16
  } from "../src/catalog/ollama-catalog.ts";
16
17
  import { bareCloudName, ollamaRateFor } from "../src/catalog/ollama-prices.ts";
17
18
  import { normalizeCatalogModel } from "../src/catalog/openrouter-catalog.ts";
18
19
  import type { CatalogModel, CatalogSnapshot, CatalogSource } from "../src/catalog/types.ts";
19
20
  import { DEFAULT_CONFIG } from "../src/config/defaults.ts";
21
+ import { openDb } from "../src/util/sqlite.ts";
20
22
  import type { OllamaConfig, RouterConfig } from "../src/config/types.ts";
21
23
  import { buildCandidates } from "../src/router/candidates.ts";
22
24
  import { extractFeatures } from "../src/router/features.ts";
23
25
  import { createMultiUpstream } from "../src/upstream/multi.ts";
24
26
  import { classifyOllamaStatus, createOllamaClient, toOllamaBody } from "../src/upstream/ollama.ts";
25
- import { createOllamaUsageSource, effectiveOllamaBias, NO_USAGE, ollamaMeter, parseOllamaPlan, parseOllamaUsage, usageFraction } from "../src/upstream/ollama-usage.ts";
27
+ import {
28
+ CALIBRATION_MAX_FACTOR,
29
+ CALIBRATION_MIN_FACTOR,
30
+ calibrationFrom,
31
+ createOllamaUsageSource,
32
+ effectiveOllamaBias,
33
+ NO_USAGE,
34
+ ollamaMeter,
35
+ parseOllamaPlan,
36
+ parseOllamaUsage,
37
+ usageFraction,
38
+ } from "../src/upstream/ollama-usage.ts";
26
39
  import type { Dispatch, DispatchOptions, UpstreamClient } from "../src/upstream/types.ts";
27
40
  import { createLogger } from "../src/util/log.ts";
28
41
  import { parseChatRequest } from "../src/wire/openai/request.ts";
@@ -195,6 +208,20 @@ describe("createOllamaCatalog", () => {
195
208
  expect(src.peek()).toHaveLength(1);
196
209
  });
197
210
 
211
+ test("a built set is persisted, and a fresh source hydrates it before its first listing", async () => {
212
+ const db = openDb(":memory:");
213
+ const fetchImpl = async (url: string): Promise<Response> => Response.json({ models: url.endsWith("/api/tags") ? DAEMON_TAGS : [] });
214
+ const src = createOllamaCatalog(OLLAMA, log, fetchImpl, db);
215
+ const built = await src.get(OR_MODELS);
216
+ expect(built.length).toBeGreaterThan(0);
217
+ expect(loadOllamaCatalogCache(db).models.map((m) => m.slug)).toEqual(built.map((m) => m.slug));
218
+ // A new process: peek() answers from disk with no network at all.
219
+ const dead = async (): Promise<Response> => { throw new Error("offline"); };
220
+ const again = createOllamaCatalog(OLLAMA, log, dead, db);
221
+ expect(again.peek().map((m) => m.slug)).toEqual(built.map((m) => m.slug));
222
+ db.close();
223
+ });
224
+
198
225
  test("a failed listing keeps the previous set", async () => {
199
226
  let fail = false;
200
227
  const fetchImpl = async (url: string): Promise<Response> => {
@@ -488,7 +515,7 @@ describe("ollama plan usage (credit-aware bias)", () => {
488
515
  const source = { get: async () => ollamaModels, peek: () => ollamaModels, invalidate: () => {} };
489
516
  const breaker = { available: () => true, cooldownUntilMs: () => null, lastTrip: () => null };
490
517
  let used = 0.2;
491
- const usage = { get: async () => ({ monthlyUsedFraction: used, monthlyUsageRaw: used * 100, activityCostUsd: null, requestsThisMonth: 0, plan: null, fetchedAtMs: 0 }), peek: () => ({ monthlyUsedFraction: used, monthlyUsageRaw: used * 100, activityCostUsd: null, requestsThisMonth: 0, plan: null, fetchedAtMs: 0 }) };
518
+ const usage = { get: async () => ({ monthlyUsedFraction: used, monthlyUsageRaw: used * 100, activityCostUsd: null, requestsThisMonth: 0, plan: null, fetchedAtMs: 0 }), peek: () => ({ monthlyUsedFraction: used, monthlyUsageRaw: used * 100, activityCostUsd: null, requestsThisMonth: 0, plan: null, fetchedAtMs: 0 }), calibration: () => null };
492
519
  const catalog = createCompositeCatalog(openrouter, source, breaker, { costBias: 0.1, biasUntilUsage: 0.9, usage });
493
520
 
494
521
  const a = await catalog.get();
@@ -538,3 +565,48 @@ describe("ollamaMeter", () => {
538
565
  expect(parseOllamaPlan("nope")).toBeNull();
539
566
  });
540
567
  });
568
+
569
+ describe("ollama calibration", () => {
570
+ const s = (h: number, meterUsd: number, ledgerUsd: number) => ({ atMs: h * 3_600_000, meterUsd, ledgerUsd });
571
+
572
+ test("compares the newest reading with the oldest since the last meter reset", () => {
573
+ // Ledger estimated $4 over the span; the meter moved $5 ⇒ estimates run 20% low.
574
+ const c = calibrationFrom([s(0, 10, 20), s(12, 12.5, 22), s(24, 15, 24)])!;
575
+ expect(c.factor).toBeCloseTo(1.25, 6);
576
+ expect(c.spanHours).toBe(24);
577
+ expect(c.samples).toBe(3);
578
+ // A billing-cycle reset (meter drops) ends the span.
579
+ const reset = calibrationFrom([s(0, 50, 40), s(12, 0.5, 41), s(24, 1.5, 42)])!;
580
+ expect(reset.samples).toBe(2);
581
+ expect(reset.factor).toBeCloseTo(1, 6);
582
+ });
583
+
584
+ test("needs enough metered spend to mean anything, and clamps to 0.5–2×", () => {
585
+ expect(calibrationFrom([s(0, 1, 1), s(1, 1.03, 1.2)])).toBeNull(); // meter moved less than its resolution
586
+ expect(calibrationFrom([s(0, 1, 1), s(1, 2, 1.3)])).toBeNull(); // ledger moved less than $0.50
587
+ expect(calibrationFrom([s(0, 0, 0), s(1, 10, 1)])!.factor).toBe(CALIBRATION_MAX_FACTOR);
588
+ expect(calibrationFrom([s(0, 0, 0), s(1, 0.1, 10)])!.factor).toBe(CALIBRATION_MIN_FACTOR);
589
+ expect(calibrationFrom([s(0, 1, 1)])).toBeNull();
590
+ });
591
+
592
+ test("the source records a sample per poll and exposes the calibration", async () => {
593
+ const db = openDb(":memory:");
594
+ let ledgerUsd = 1;
595
+ let frac = 0.1;
596
+ const fetchImpl = async (url: string): Promise<Response> => {
597
+ if (url.endsWith("/api/me")) return Response.json({ Plan: "pro" });
598
+ return Response.json({ limits: { monthly: { usage: frac, models: [] } } });
599
+ };
600
+ const src = createOllamaUsageSource({ apiKey: "k", pollMs: 5, timeoutMs: 1000, log, fetchImpl, calibration: { db, ledgerUsd: () => ledgerUsd, planCreditsOverrideUsd: 0 } });
601
+ await src.get(); // meter $6 (10% of $60), ledger $1
602
+ expect(src.calibration()).toBeNull(); // one sample
603
+ await new Promise((r) => setTimeout(r, 20));
604
+ ledgerUsd = 3; // ledger +$2
605
+ frac = 0.15; // meter +$3
606
+ await src.get();
607
+ const c = src.calibration()!;
608
+ expect(c.factor).toBeCloseTo(1.5, 6);
609
+ expect((db.query("SELECT COUNT(*) n FROM ollama_meter_samples").get() as { n: number }).n).toBe(2);
610
+ db.close();
611
+ });
612
+ });
@@ -76,17 +76,21 @@ function report(over: Partial<UsageReport> = {}): UsageReport {
76
76
  aborted: 0,
77
77
  modelSwitches: 1,
78
78
  cacheEstimated: false,
79
+ subagentDispatches: 0,
80
+ subagentSpendUsd: 0,
79
81
  },
80
82
  providers: [row("openrouter", 2), row("ollama", 1)],
81
83
  models: [
82
- { ...row("z-ai/glm", 2), provider: "openrouter", tiers: { simple: 6, moderate: 4 } },
83
- { ...row("ollama/kimi", 1), provider: "ollama", tiers: { hard: 10 } },
84
+ { ...row("z-ai/glm", 2), provider: "openrouter", tiers: { simple: 6, moderate: 4 }, feedback: { good: 0, bad: 0 } },
85
+ { ...row("ollama/kimi", 1), provider: "ollama", tiers: { hard: 10 }, feedback: { good: 0, bad: 0 } },
84
86
  ],
85
87
  tiers: [row("simple", 2), row("hard", 1)],
86
88
  days: [
87
89
  { day: "2026-09-05", dispatches: 10, spendUsd: 1.5, cacheHitRate: 0.8 },
88
90
  { day: "2026-09-06", dispatches: 10, spendUsd: 1.5, cacheHitRate: 0.8 },
89
91
  ],
92
+ anatomy: null,
93
+ baselines: [],
90
94
  ...over,
91
95
  };
92
96
  }
@@ -66,6 +66,8 @@ describe("renderStatus", () => {
66
66
  lastTrip: { kind: "quota", atMs: now - 120_000, message: "402" },
67
67
  usage: { monthlyUsedFraction: 0.42, activityCostUsd: 3.1, fetchedAtMs: now },
68
68
  meter: { usedUsd: 25.2, creditsUsd: 60, plan: "pro" },
69
+ calibration: { factor: 1.25, meterDeltaUsd: 5, ledgerDeltaUsd: 4, spanHours: 48 },
70
+ runway: { dailyBurnUsd: 1.5, creditsLeftUsd: 34.8, days: 23.2 },
69
71
  costBias: { configured: 0.1, effective: 0.1, biasUntilUsage: 0.9 },
70
72
  },
71
73
  catalog: { models: 240, ageMs: 5 * 60_000, keyScoped: true, shrink: { fromModels: 300, toModels: 120, atMs: now } },
@@ -77,6 +79,7 @@ describe("renderStatus", () => {
77
79
  expect(text).toContain("SHRANK 300 -> 120");
78
80
  expect(text).toContain("COOLING DOWN");
79
81
  expect(text).toContain("pro plan usage 42.0% ($25.20 of $60)");
82
+ expect(text).toContain("ollama billing: ledger estimate ×1.25 to match the meter (48h span) · burn $1.50/day · ~23 days of credits left");
80
83
  expect(text).toContain("cost bias ×0.1 (until 90%)");
81
84
  expect(text).toContain("last trip quota 2m ago");
82
85
  expect(text).toContain("scope omp-router");
@@ -156,6 +156,59 @@ describe("buildUsageReport", () => {
156
156
  db.close();
157
157
  });
158
158
 
159
+ test("prompt anatomy averages the recorded byte shares", () => {
160
+ const { db, ledger } = seeded();
161
+ const feat = (tool: number, older: number, stale: number) => ({ toolSchemaBytes: 1000, anatomy: { messages: 30, systemBytes: 1000, userBytes: 500, assistantBytes: 500, toolBytes: tool, olderHalfBytes: older, staleToolBytes: stale } });
162
+ ledger.record(entry({ features: feat(8000, 5000, 4000) }));
163
+ ledger.record(entry({ features: feat(6000, 4000, 2000) }));
164
+ ledger.record(entry({ features: null })); // pre-anatomy row: ignored
165
+ const r = buildUsageReport(db, { windowDays: 7, nowMs: NOW });
166
+ const a = r.anatomy!;
167
+ expect(a.rows).toBe(2);
168
+ // mean bytes: system 1000, user 500, assistant 500, tool 7000 ⇒ total 9000
169
+ expect(a.tool).toBeCloseTo(7000 / 9000, 6);
170
+ expect(a.system).toBeCloseTo(1000 / 9000, 6);
171
+ expect(a.schemas).toBeCloseTo(1000 / 9000, 6);
172
+ expect(a.olderHalf).toBeCloseTo(4500 / 9000, 6);
173
+ expect(a.staleTool).toBeCloseTo(3000 / 9000, 6);
174
+ expect(renderUsageReport(r)).toContain("prompt anatomy (mean of 2): tool results 78%");
175
+ db.close();
176
+ });
177
+
178
+ test("baselines price the window on one model with its own cache hit rate", () => {
179
+ const { db, ledger } = seeded();
180
+ ledger.record(entry({ reportedUsd: 0.5, usage: { promptTokens: 100_000, cachedTokens: 80_000, cacheWriteTokens: 0, completionTokens: 1_000, reasoningTokens: 0, images: 0 } }));
181
+ const r = buildUsageReport(db, {
182
+ windowDays: 7,
183
+ nowMs: NOW,
184
+ baselines: [
185
+ { slug: "big/model", prompt: 15 / 1e6, completion: 75 / 1e6, cacheRead: 1.5 / 1e6 },
186
+ { slug: "nocache/model", prompt: 3 / 1e6, completion: 15 / 1e6 },
187
+ ],
188
+ });
189
+ // 20k fresh × $15/M + 80k cached × $1.5/M + 1k completion × $75/M = 0.30 + 0.12 + 0.075
190
+ expect(r.baselines[0]!.usd).toBeCloseTo(0.495, 6);
191
+ expect(r.baselines[0]!.savedShare).toBeCloseTo(1 - 0.5 / 0.495, 6);
192
+ // No cache rate published: every prompt token at list price.
193
+ expect(r.baselines[1]!.usd).toBeCloseTo(0.3 + 0.015, 6);
194
+ const text = renderUsageReport(r);
195
+ expect(text).toContain("same traffic on one model: big/model $0.4950 (router cost extra 1%)");
196
+ expect(text).toContain("nocache/model $0.3150 (router cost extra 59%)");
197
+ db.close();
198
+ });
199
+
200
+ test("subagent turns are counted with their spend", () => {
201
+ const { db, ledger } = seeded();
202
+ ledger.record(entry({ reportedUsd: 0.01, features: { isSubagent: true } }));
203
+ ledger.record(entry({ reportedUsd: 0.03, features: { isSubagent: false } }));
204
+ ledger.record(entry({ reportedUsd: 0.06 }));
205
+ const r = buildUsageReport(db, { windowDays: 7, nowMs: NOW });
206
+ expect(r.totals.subagentDispatches).toBe(1);
207
+ expect(r.totals.subagentSpendUsd).toBeCloseTo(0.01, 6);
208
+ expect(renderUsageReport(r)).toContain("subagents: 1 dispatches, $0.0100 (10% of spend)");
209
+ db.close();
210
+ });
211
+
159
212
  test("empty ledger yields zeroed totals and null speeds", () => {
160
213
  const { db } = seeded();
161
214
  const r = buildUsageReport(db, { windowDays: 7, nowMs: NOW });
@@ -172,9 +225,13 @@ describe("buildUsageReport", () => {
172
225
  aborted: 0,
173
226
  modelSwitches: 0,
174
227
  cacheEstimated: false,
228
+ subagentDispatches: 0,
229
+ subagentSpendUsd: 0,
175
230
  });
176
231
  expect(r.providers).toEqual([]);
177
232
  expect(r.models).toEqual([]);
233
+ expect(r.anatomy).toBeNull();
234
+ expect(r.baselines).toEqual([]);
178
235
  db.close();
179
236
  });
180
237
 
@@ -8,7 +8,8 @@ import type { ProfileConfig, RouterConfig } from "../src/config/types.ts";
8
8
  import type { Ledger } from "../src/cost/types.ts";
9
9
  import { extractFeatures } from "../src/router/features.ts";
10
10
  import { scoreHeuristic } from "../src/router/classify.ts";
11
- import { BudgetExceededError, select } from "../src/router/select.ts";
11
+ import { latencyWeightFor } from "../src/router/candidates.ts";
12
+ import { BudgetExceededError, monthPace, monthStartMs, select } from "../src/router/select.ts";
12
13
  import type { ConversationState, Tier } from "../src/router/types.ts";
13
14
  import { parseChatRequest } from "../src/wire/openai/request.ts";
14
15
  import type { NormRequest } from "../src/wire/types.ts";
@@ -1050,3 +1051,127 @@ describe("hysteresis.confirmUpgradesBelowConfidence", () => {
1050
1051
  expect(d.upgradeDeferred).toBeNull();
1051
1052
  });
1052
1053
  });
1054
+
1055
+ describe("cache reliability in the stay/switch comparison", () => {
1056
+ const warmSlug = "x-ai/grok-4.6";
1057
+ function ledgerWithReliability(rate: number | null, samples = 50): Ledger {
1058
+ return {
1059
+ record: () => {},
1060
+ conversationSpend: () => 0,
1061
+ spendSince: () => 0,
1062
+ blendedRate: () => null,
1063
+ trust: () => null,
1064
+ allTrust: () => [],
1065
+ latency: () => null,
1066
+ tokenRatio: () => null,
1067
+ recentEntries: () => [],
1068
+ cacheReliability: (slug) => (rate === null || slug !== warmSlug ? null : { slug, samples, hitRate: rate }),
1069
+ };
1070
+ }
1071
+ const stayCostOf = (ledger: Ledger, cfg: RouterConfig = BASE): number => {
1072
+ const d = run({
1073
+ tier: "hard",
1074
+ promptTokens: 80_000,
1075
+ cfg,
1076
+ ledger,
1077
+ st: state({ currentSlug: warmSlug, currentTier: "hard", cacheWarmSlug: warmSlug, cacheWarmAtMs: Date.now(), lastPromptTokens: 80_000 }),
1078
+ });
1079
+ const m = /cache: (?:keeping warm|switch) .*?stay \$([0-9.]+)/.exec(d.reasons.join("\n"));
1080
+ if (m === null) throw new Error(`no stay/switch reason in ${d.reasons.join(" | ")}`);
1081
+ return Number(m[1]);
1082
+ };
1083
+
1084
+ test("an unreliable cache prices staying at the fresh rate, a reliable one at the cached rate", () => {
1085
+ const reliable = stayCostOf(ledgerWithReliability(1));
1086
+ const flaky = stayCostOf(ledgerWithReliability(0));
1087
+ const unknown = stayCostOf(ledgerWithReliability(null));
1088
+ expect(flaky).toBeGreaterThan(reliable);
1089
+ expect(unknown).toBeCloseTo(reliable, 6);
1090
+ });
1091
+
1092
+ test("too few samples, or the feature off, assume a reliable cache", () => {
1093
+ const reliable = stayCostOf(ledgerWithReliability(1));
1094
+ expect(stayCostOf(ledgerWithReliability(0, 3))).toBeCloseTo(reliable, 6);
1095
+ const off: RouterConfig = { ...BASE, filters: { ...BASE.filters, cacheReliabilityMinSamples: 0 } };
1096
+ expect(stayCostOf(ledgerWithReliability(0), off)).toBeCloseTo(reliable, 6);
1097
+ });
1098
+
1099
+ test("the reason names the measured hit rate", () => {
1100
+ const d = run({
1101
+ tier: "hard",
1102
+ promptTokens: 80_000,
1103
+ ledger: ledgerWithReliability(0.5, 40),
1104
+ st: state({ currentSlug: warmSlug, currentTier: "hard", cacheWarmSlug: warmSlug, cacheWarmAtMs: Date.now(), lastPromptTokens: 80_000 }),
1105
+ });
1106
+ expect(d.reasons.some((r) => r.includes("warm hit 50% over 40"))).toBe(true);
1107
+ });
1108
+ });
1109
+
1110
+ describe("session pin (forceSlug)", () => {
1111
+ test("a pinned catalog model wins over ranking and the warm model; an unknown pin is ignored with a reason", () => {
1112
+ const warmSlug = run({ tier: "moderate" }).slug;
1113
+ const pinSlug = run({ tier: "hard" }).slug; // a real, differently-ranked model
1114
+ const req = request("tidy the retry helper");
1115
+ const features = extractFeatures(req, 50_000);
1116
+ const base = {
1117
+ req,
1118
+ features,
1119
+ classification: scoreHeuristic(features, BASE),
1120
+ profile: PROFILE,
1121
+ state: state({ currentSlug: warmSlug, currentTier: "moderate", cacheWarmSlug: warmSlug, cacheWarmAtMs: Date.now(), lastPromptTokens: 50_000 }),
1122
+ snapshot: SNAPSHOT,
1123
+ ledger: null,
1124
+ cfg: { ...BASE, hysteresis: { ...BASE.hysteresis, switchMargin: 1e6 } },
1125
+ nowMs: Date.now(),
1126
+ };
1127
+ const pinned = select({ ...base, forceSlug: pinSlug });
1128
+ expect(pinned.slug).toBe(pinSlug);
1129
+ expect(pinned.sticky).toBe(false);
1130
+ expect(pinned.reasons.some((r) => r.includes(`pinned to ${pinSlug} by session override`))).toBe(true);
1131
+ const unknown = select({ ...base, forceSlug: "nope/model" });
1132
+ expect(unknown.slug).toBe(warmSlug); // the huge switch margin keeps the warm model
1133
+ expect(unknown.reasons.some((r) => r.includes("pin nope/model ignored"))).toBe(true);
1134
+ });
1135
+ });
1136
+
1137
+ describe("budget.perMonthUsd pacing", () => {
1138
+ test("monthPace spreads what is left over the days left, today included", () => {
1139
+ const sep7 = Date.UTC(2026, 8, 7, 12);
1140
+ expect(monthStartMs(sep7)).toBe(Date.UTC(2026, 8, 1));
1141
+ const p = monthPace(sep7, 60, 30);
1142
+ expect(p.daysLeft).toBe(24); // 7th..30th
1143
+ expect(p.dailyCapUsd).toBeCloseTo(30 / 24, 6);
1144
+ expect(monthPace(sep7, 60, 70).dailyCapUsd).toBe(0);
1145
+ expect(monthPace(Date.UTC(2026, 8, 30, 12), 60, 0).daysLeft).toBe(1);
1146
+ });
1147
+
1148
+ test("a month running ahead of pace tightens the daily cap and says so", () => {
1149
+ const ledger: Ledger = {
1150
+ record: () => {},
1151
+ conversationSpend: () => 0,
1152
+ spendSince: (sinceMs) => (sinceMs <= monthStartMs(Date.now()) + 1 ? 59.99 : 0), // month-to-date $59.99, last 24h $0
1153
+ blendedRate: () => null,
1154
+ trust: () => null,
1155
+ allTrust: () => [],
1156
+ latency: () => null,
1157
+ tokenRatio: () => null,
1158
+ recentEntries: () => [],
1159
+ };
1160
+ const cfg: RouterConfig = { ...BASE, budget: { ...BASE.budget, perMonthUsd: 60, onExceeded: "reject" } };
1161
+ expect(() => run({ tier: "hard", promptTokens: 50_000, cfg, ledger })).toThrow(/month pacing: \$59\.99 of \$60 spent/);
1162
+ // Under pace: the cap is generous and nothing breaches.
1163
+ const easy: Ledger = { ...ledger, spendSince: () => 1 };
1164
+ expect(run({ tier: "hard", promptTokens: 50_000, cfg, ledger: easy }).budgetDowngraded).toBe(false);
1165
+ });
1166
+ });
1167
+
1168
+ describe("filters.latencyWeightContinuation", () => {
1169
+ test("applies only to tool-result continuations, and only when set", () => {
1170
+ const f = { ...BASE.filters, latencyWeight: 0.75 };
1171
+ expect(latencyWeightFor(f, false)).toBe(0.75);
1172
+ expect(latencyWeightFor(f, true)).toBe(0.75);
1173
+ const g = { ...f, latencyWeightContinuation: 0.1 };
1174
+ expect(latencyWeightFor(g, false)).toBe(0.75);
1175
+ expect(latencyWeightFor(g, true)).toBe(0.1);
1176
+ });
1177
+ });
@@ -1,4 +1,5 @@
1
1
  import { describe, expect, test } from "bun:test";
2
+ import { createFeedbackStore } from "../src/cost/feedback.ts";
2
3
 
3
4
  import { loadConfig } from "../src/config/load.ts";
4
5
  import { createLedger, LATENCY_WINDOW_ROWS } from "../src/cost/ledger.ts";
@@ -368,3 +369,97 @@ describe("v6 classifier instrumentation", () => {
368
369
  }
369
370
  });
370
371
  });
372
+
373
+ describe("cache reliability signal", () => {
374
+ // Observed hit rate when a warm cache was expected: the previous kept turn
375
+ // of the conversation was on the same model within the warm TTL.
376
+ function seed(rows: Array<Partial<LedgerEntry>>) {
377
+ const db = openDb(":memory:");
378
+ const ledger = createLedger(db, cfg);
379
+ for (const r of rows) ledger.record(entry(r));
380
+ return { db, ledger };
381
+ }
382
+ const t0 = Date.UTC(2026, 8, 7, 12);
383
+ const usage = (prompt: number, cached: number, estimated = false) => ({ ...EMPTY_USAGE, promptTokens: prompt, cachedTokens: cached, ...(estimated ? { cachedEstimated: true } : {}) });
384
+
385
+ test("a model that hits when warm scores 1; one that misses scores 0; the first turn never counts", () => {
386
+ const { db, ledger } = seed([
387
+ { conversationKey: "a", slug: "good/m", servedSlug: "good/m", createdAtMs: t0, usage: usage(50_000, 0) }, // first turn: no expectation
388
+ { conversationKey: "a", slug: "good/m", servedSlug: "good/m", createdAtMs: t0 + 60_000, usage: usage(60_000, 50_000) },
389
+ { conversationKey: "a", slug: "good/m", servedSlug: "good/m", createdAtMs: t0 + 120_000, usage: usage(70_000, 60_000) },
390
+ { conversationKey: "b", slug: "flaky/m", servedSlug: "flaky/m", createdAtMs: t0, usage: usage(50_000, 0) },
391
+ { conversationKey: "b", slug: "flaky/m", servedSlug: "flaky/m", createdAtMs: t0 + 60_000, usage: usage(60_000, 0) },
392
+ { conversationKey: "b", slug: "flaky/m", servedSlug: "flaky/m", createdAtMs: t0 + 120_000, usage: usage(70_000, 30_000) },
393
+ ]);
394
+ expect(ledger.cacheReliability?.("good/m")).toEqual({ slug: "good/m", samples: 2, hitRate: 1 });
395
+ const flaky = ledger.cacheReliability?.("flaky/m");
396
+ expect(flaky?.samples).toBe(2);
397
+ expect(flaky?.hitRate).toBeCloseTo(0.25, 6); // (0 + 30k/60k) / 2
398
+ expect(ledger.cacheReliability?.("never/m")).toBeNull();
399
+ db.close();
400
+ });
401
+
402
+ test("a switch, an idle gap past the TTL, or a router-estimated count is not a warm-expected sample", () => {
403
+ const { db, ledger } = seed([
404
+ { conversationKey: "a", slug: "x/m", servedSlug: "x/m", createdAtMs: t0, usage: usage(50_000, 0) },
405
+ { conversationKey: "a", slug: "y/m", servedSlug: "y/m", createdAtMs: t0 + 60_000, usage: usage(60_000, 0) }, // switch
406
+ { conversationKey: "a", slug: "y/m", servedSlug: "y/m", createdAtMs: t0 + 60_000 + cfg.hysteresis.cacheWarmTtlMs + 1, usage: usage(70_000, 0) }, // gap
407
+ { conversationKey: "a", slug: "y/m", servedSlug: "y/m", createdAtMs: t0 + 60_000 + cfg.hysteresis.cacheWarmTtlMs + 2, usage: usage(80_000, 70_000, true) }, // estimated
408
+ ]);
409
+ expect(ledger.cacheReliability?.("x/m")).toBeNull();
410
+ expect(ledger.cacheReliability?.("y/m")).toBeNull();
411
+ db.close();
412
+ });
413
+ });
414
+
415
+ describe("feedback in trust", () => {
416
+ // A user verdict counts as filters.feedbackWeight attempts of that outcome.
417
+ function trustWith(weight: number, verdicts: Array<"good" | "bad">): { rate: number; good: number; bad: number } {
418
+ const db = openDb(":memory:");
419
+ try {
420
+ const c = structuredClone(cfg);
421
+ c.filters.feedbackWeight = weight;
422
+ const ledger = createLedger(db, c);
423
+ const fb = createFeedbackStore(db);
424
+ let last = "";
425
+ for (let i = 0; i < 10; i++) {
426
+ const e = entry({ error: null });
427
+ last = e.id;
428
+ ledger.record(e);
429
+ }
430
+ for (const v of verdicts) fb.record({ ledgerId: last, ompSessionId: "s", slug: "vendor/model", tier: "simple", verdict: v, note: "" });
431
+ const t = ledger.trust("vendor/model")!;
432
+ return { rate: t.successRate, good: t.feedbackGood ?? -1, bad: t.feedbackBad ?? -1 };
433
+ } finally {
434
+ db.close();
435
+ }
436
+ }
437
+
438
+ test("weight 0 records verdicts without moving the rate", () => {
439
+ const base = trustWith(0, []);
440
+ expect(base.rate).toBeCloseTo(11 / 12, 6); // (10 - 0 + 1) / (10 + 2)
441
+ expect(trustWith(0, ["bad", "bad"]).rate).toBeCloseTo(base.rate, 6);
442
+ });
443
+
444
+ test("a bad verdict counts as `weight` failures, a good one as `weight` successes", () => {
445
+ // 10 clean attempts + one bad verdict at weight 3: attempts 13, failures 3.
446
+ const bad = trustWith(3, ["bad"]);
447
+ expect(bad.rate).toBeCloseTo((13 - 3 + 1) / (13 + 2), 6);
448
+ expect(bad.bad).toBe(1);
449
+ const good = trustWith(3, ["good"]);
450
+ expect(good.rate).toBeCloseTo((13 - 0 + 1) / (13 + 2), 6);
451
+ expect(good.good).toBe(1);
452
+ // allTrust and signals agree with trust().
453
+ const db = openDb(":memory:");
454
+ const c = structuredClone(cfg);
455
+ c.filters.feedbackWeight = 3;
456
+ const ledger = createLedger(db, c);
457
+ const fb = createFeedbackStore(db);
458
+ const e = entry({ error: null });
459
+ ledger.record(e);
460
+ fb.record({ ledgerId: e.id, ompSessionId: "s", slug: "vendor/model", tier: "simple", verdict: "bad", note: "" });
461
+ expect(ledger.allTrust()[0]?.successRate).toBeCloseTo(ledger.trust("vendor/model")!.successRate, 9);
462
+ expect(ledger.signals?.(["vendor/model"]).get("vendor/model")?.trust?.successRate).toBeCloseTo(ledger.trust("vendor/model")!.successRate, 9);
463
+ db.close();
464
+ });
465
+ });