mirror of
https://github.com/diegosouzapw/OmniRoute.git
synced 2026-09-13 18:32:12 +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.
91 lines
2.9 KiB
TypeScript
91 lines
2.9 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert";
|
|
import {
|
|
ADOBE_FIREFLY_VIDEO_MODELS,
|
|
extractAdobeSourceImageReferences,
|
|
normalizeAdobeReferenceBlobs,
|
|
} from "../../open-sse/services/adobeFireflyClient.ts";
|
|
import { getAdobeModels } from "../../src/app/api/providers/[id]/models/adobeFireflyDiscovery.ts";
|
|
|
|
function userImsJwt(): string {
|
|
const payload = Buffer.from(
|
|
JSON.stringify({
|
|
user_id: "test@AdobeID",
|
|
type: "access_token",
|
|
client_id: "clio-playground-web",
|
|
})
|
|
).toString("base64url");
|
|
return `eyJhbGciOiJSUzI1NiJ9.${payload}.${"sig".padEnd(40, "x")}`;
|
|
}
|
|
|
|
test("reference validation enforces discovered roles, counts, and frame order", () => {
|
|
const kling = ADOBE_FIREFLY_VIDEO_MODELS["kling-3"];
|
|
assert.deepEqual(
|
|
normalizeAdobeReferenceBlobs(kling, [
|
|
{ id: "frame-a", mediaType: "image", usage: "frame" },
|
|
{ id: "frame-b", mediaType: "image", usage: "frame" },
|
|
]),
|
|
[
|
|
{ id: "frame-a", usage: "frame", order: 1 },
|
|
{ id: "frame-b", usage: "frame", order: 2 },
|
|
]
|
|
);
|
|
assert.throws(
|
|
() => normalizeAdobeReferenceBlobs(kling, [{ id: "bad", mediaType: "image", usage: "mask" }]),
|
|
/does not support image references with usage 'mask'/
|
|
);
|
|
assert.throws(
|
|
() =>
|
|
normalizeAdobeReferenceBlobs(kling, [
|
|
{ id: "frame-a", usage: "frame" },
|
|
{ id: "frame-b", usage: "frame" },
|
|
{ id: "frame-c", usage: "frame" },
|
|
]),
|
|
/at most 2 frame image reference/
|
|
);
|
|
});
|
|
|
|
test("structured references skip malformed entries and preserve explicit roles", () => {
|
|
assert.deepEqual(
|
|
extractAdobeSourceImageReferences({
|
|
adobe_reference_inputs: [
|
|
null,
|
|
{ media_type: "video", source: "ignored" },
|
|
{ media_type: "image", source: "data:image/png;base64,AAAA", usage: "frame", order: 2 },
|
|
],
|
|
}),
|
|
[{ source: "data:image/png;base64,AAAA", usage: "frame", order: 2 }]
|
|
);
|
|
});
|
|
|
|
test("provider discovery adapter returns live capabilities and verified fallback", async () => {
|
|
const live = await getAdobeModels(undefined, userImsJwt(), {}, async () =>
|
|
Response.json({
|
|
models: [
|
|
{
|
|
modelId: "firefly-image",
|
|
acModelFamilyProviderDisplayName: "Adobe",
|
|
modelVersions: {
|
|
image5: {
|
|
enabled: true,
|
|
outputModality: ["image"],
|
|
modelDisplayName: "Firefly Image 5",
|
|
requestSchema: { type: "object", properties: { prompt: { type: "string" } } },
|
|
},
|
|
},
|
|
},
|
|
],
|
|
})
|
|
);
|
|
assert.equal(live.source, "api");
|
|
assert.equal(live.models[0].id, "firefly-image-image5");
|
|
assert.ok(live.models[0].media_capabilities);
|
|
|
|
const fallback = await getAdobeModels(undefined, userImsJwt(), {}, async () => {
|
|
throw new Error("offline");
|
|
});
|
|
assert.equal(fallback.source, "local_catalog");
|
|
assert.equal(fallback.models.length, 52);
|
|
assert.match(fallback.warning || "", /discovery unavailable/);
|
|
});
|