mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-08-26 17:12:27 +03:00
65e81158ab
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.
41 lines
1.3 KiB
TypeScript
41 lines
1.3 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { estimateFinalInputTokens } from "../../open-sse/handlers/chatCore/contextEstimation.ts";
|
|
import { estimateTokens } from "../../open-sse/services/contextManager.ts";
|
|
|
|
test("estimateFinalInputTokens counts nested request contents and auxiliary fields", () => {
|
|
const contents = [{ role: "user", parts: [{ text: "nested Gemini request" }] }];
|
|
const tools = [{ name: "lookup", description: "find a record" }];
|
|
const body = {
|
|
request: { contents },
|
|
tools,
|
|
system: "system prompt",
|
|
instructions: 42,
|
|
};
|
|
|
|
assert.equal(
|
|
estimateFinalInputTokens(body),
|
|
estimateTokens(contents) +
|
|
estimateTokens(tools) +
|
|
estimateTokens(body.system) +
|
|
estimateTokens(body.instructions)
|
|
);
|
|
});
|
|
|
|
test("estimateFinalInputTokens ignores a malformed nested request and uses input", () => {
|
|
const input = [{ role: "user", content: "fallback input" }];
|
|
const body = {
|
|
request: "not an object",
|
|
input,
|
|
};
|
|
|
|
assert.equal(estimateFinalInputTokens(body), estimateTokens(input));
|
|
});
|
|
|
|
test("estimateTokens accepts the unknown JSON values it already serializes", () => {
|
|
assert.equal(estimateTokens(true), 1);
|
|
assert.equal(estimateTokens(42), 1);
|
|
assert.equal(estimateTokens(undefined), 0);
|
|
});
|