mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-21 22:32:22 +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.
43 lines
1.6 KiB
TypeScript
43 lines
1.6 KiB
TypeScript
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import fs from "node:fs";
|
|
import path from "node:path";
|
|
|
|
import { resolveStreamFlag } from "../../open-sse/utils/aiSdkCompat.ts";
|
|
|
|
const ROUTES = [
|
|
{
|
|
name: "responses",
|
|
sourceFormat: "openai-responses",
|
|
bodyExpression: "resolvedBody?.stream",
|
|
source: fs.readFileSync(path.join("src", "app", "api", "v1", "responses", "route.ts"), "utf8"),
|
|
},
|
|
{
|
|
name: "messages",
|
|
sourceFormat: "claude",
|
|
bodyExpression: "body?.stream",
|
|
source: fs.readFileSync(path.join("src", "app", "api", "v1", "messages", "route.ts"), "utf8"),
|
|
},
|
|
] as const;
|
|
|
|
for (const route of ROUTES) {
|
|
test(`${route.name} early-heartbeat gate uses the real stream resolver`, () => {
|
|
const escapedBodyExpression = route.bodyExpression.replace(/[.?\\]/g, "\\$&");
|
|
assert.match(
|
|
route.source,
|
|
new RegExp(
|
|
`resolveStreamFlag\\(\\s*${escapedBodyExpression},\\s*accept,\\s*"${route.sourceFormat}"\\s*\\)`
|
|
)
|
|
);
|
|
assert.match(route.source, /if \(wantsStreaming\) \{[\s\S]*withEarlyStreamKeepalive\(/);
|
|
assert.doesNotMatch(route.source, /accept\.includes\(["']text\/event-stream["']\)/);
|
|
});
|
|
|
|
test(`${route.name} stream intent covers body and Accept without overriding stream:false`, () => {
|
|
assert.equal(resolveStreamFlag(true, "application/json", route.sourceFormat), true);
|
|
assert.equal(resolveStreamFlag(false, "text/event-stream", route.sourceFormat), false);
|
|
assert.equal(resolveStreamFlag(undefined, "text/event-stream", route.sourceFormat), true);
|
|
assert.equal(resolveStreamFlag(undefined, "application/json", route.sourceFormat), false);
|
|
});
|
|
}
|