mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-05-05 09:46:30 +00:00
dc6d9e2e4b
Introduce runtime-configurable payload mutation/filter rules with file reload support and a settings API so upstream request bodies can be customized per model and protocol without restarts. Expand search support with Google PSE, Linkup, SearchAPI, and SearXNG, including validation, routing, analytics costing, MCP schema updates, and search-type-aware provider selection. Update Pollinations to support anonymous access, endpoint failover, and the latest public model lineup. Add OmniRoute response metadata headers/SSE comments, per-connection model exclusion rules, combo tag-based routing, buffered spend writes, and scheduled daily/weekly/monthly budget resets. Update model catalog and dashboard UIs to surface source labels and hide models excluded by all active connections.
68 lines
2.1 KiB
TypeScript
68 lines
2.1 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import {
|
|
buildOmniRouteResponseMetaHeaders,
|
|
buildOmniRouteSseMetadataComment,
|
|
formatOmniRouteCost,
|
|
getOmniRouteTokenCounts,
|
|
} from "../../src/domain/omnirouteResponseMeta.ts";
|
|
|
|
test("getOmniRouteTokenCounts normalizes common usage shapes", () => {
|
|
assert.deepEqual(
|
|
getOmniRouteTokenCounts({
|
|
prompt_tokens: 12,
|
|
completion_tokens: 5,
|
|
}),
|
|
{ input: 12, output: 5 }
|
|
);
|
|
assert.deepEqual(
|
|
getOmniRouteTokenCounts({
|
|
input_tokens: "9",
|
|
output_tokens: "4",
|
|
}),
|
|
{ input: 9, output: 4 }
|
|
);
|
|
});
|
|
|
|
test("buildOmniRouteResponseMetaHeaders formats provider alias, tokens, latency, and cost", () => {
|
|
const headers = buildOmniRouteResponseMetaHeaders({
|
|
provider: "claude",
|
|
model: "claude-sonnet-4-6",
|
|
cacheHit: true,
|
|
latencyMs: 1234.6,
|
|
usage: {
|
|
prompt_tokens: 11,
|
|
completion_tokens: 7,
|
|
},
|
|
costUsd: 0.00123456789,
|
|
});
|
|
|
|
assert.equal(headers["X-OmniRoute-Provider"], "cc");
|
|
assert.equal(headers["X-OmniRoute-Model"], "claude-sonnet-4-6");
|
|
assert.equal(headers["X-OmniRoute-Cache-Hit"], "true");
|
|
assert.equal(headers["X-OmniRoute-Latency-Ms"], "1235");
|
|
assert.equal(headers["X-OmniRoute-Tokens-In"], "11");
|
|
assert.equal(headers["X-OmniRoute-Tokens-Out"], "7");
|
|
assert.equal(headers["X-OmniRoute-Response-Cost"], "0.0012345679");
|
|
});
|
|
|
|
test("buildOmniRouteSseMetadataComment emits comment lines compatible with SSE", () => {
|
|
const comment = buildOmniRouteSseMetadataComment({
|
|
provider: "openai",
|
|
model: "gpt-4o-mini",
|
|
usage: {
|
|
prompt_tokens: 4,
|
|
completion_tokens: 2,
|
|
},
|
|
latencyMs: 50,
|
|
costUsd: formatOmniRouteCost(0),
|
|
});
|
|
|
|
assert.match(comment, /^: x-omniroute-cache-hit=false/m);
|
|
assert.match(comment, /^: x-omniroute-provider=openai/m);
|
|
assert.match(comment, /^: x-omniroute-model=gpt-4o-mini/m);
|
|
assert.match(comment, /^: x-omniroute-tokens-in=4/m);
|
|
assert.match(comment, /^: x-omniroute-tokens-out=2/m);
|
|
assert.match(comment, /^: x-omniroute-response-cost=0\.0000000000/m);
|
|
});
|