Files
OmniRoute/tests/unit/usage-tracking-zero-input-tokens-10705.test.ts
Praveen K Palaniswamy 65e81158ab fix(ollama): route models by advertised capability (#11088)
Landed with the design call resolved per the owner's pick — **option 1**: the synced store is now endpoint-agnostic (persistDiscoveredModels and managedModelImport no longer drop non-chat models at write time), and chat selectability moved to read time (auto-pool expansion in autoStrategy applies filterChatSelectableModels; the models-route projection already had its chatOnly filter). Your discovery test now passes end-to-end (3/3): /api/show capabilities persist per connection and image/embedding requests route through the advertising host.

Reconciliation notes: conflicted areas merged onto the current tip (adobe discovery import, requestedModel preflight signature, resolvedProvider fast-path coexists with the synced-route override — explicit resolution wins); carried base-red drains (#10055 memoization, #11071 test variants) dropped as already-landed; the managed-model-import exclusion test was propagated to the new contract (image/video models persist; the read filter still hides them from chat pickers — pinned by a new assertion). Full battery: 205/206 focused (the one red is a confirmed periodic-timer timing flake on the loaded devbox — 20/20 isolated), autoCombo vitest 30/30, combo suites 46/46, gates + typecheck clean.

Thank you @yourspraveen — the capability probe + routing design was right; it just needed the store contract opened up. Fixes #11087.
2026-08-23 11:45:01 -03:00

27 lines
1.3 KiB
TypeScript

import test from "node:test";
import assert from "node:assert/strict";
const { sanitizeProviderUsageForRequest } = await import("../../open-sse/utils/usageTracking.ts");
const { FORMATS } = await import("../../open-sse/translator/formats.ts");
test("issue #10705: provider-reported input_tokens: 0 on a non-trivial request must be repaired, not passed through", () => {
const body = {
model: "claude-fable-5",
messages: [
{ role: "user", content: "你好,我想测试一下这个模型,请问你能帮我写一段代码吗?" },
{ role: "assistant", content: "你好!继续测试也行" },
{ role: "user", content: "帮我写一个 quicksort 的 python 实现,并解释一下时间复杂度。" },
],
};
const usage = { input_tokens: 0, output_tokens: 43 };
const result = sanitizeProviderUsageForRequest(usage, body, FORMATS.CLAUDE);
assert.ok(result && result.input_tokens > 0, "input_tokens: 0 for a real body must be repaired");
});
test("issue #10705: input_tokens: 0 for a genuinely empty/no-body request stays 0", () => {
const usage = { input_tokens: 0, output_tokens: 5 };
const result = sanitizeProviderUsageForRequest(usage, undefined, FORMATS.CLAUDE);
assert.equal(result, usage, "no body to estimate from — must pass through unchanged");
});