Luke The Dev
107 lines
2.8 KiB
TypeScript
107 lines
2.8 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import {
|
|
buildAllowedModelKeys,
|
|
buildGatewayModelChoices,
|
|
resolveConfiguredModelKey,
|
|
type GatewayModelPolicySnapshot,
|
|
} from "@/lib/gateway/models";
|
|
|
|
describe("gateway model policy helpers", () => {
|
|
it("resolves configured aliases and shorthand ids", () => {
|
|
const modelAliases = {
|
|
"anthropic/claude-sonnet-4-5": { alias: "sonnet" },
|
|
"openai/gpt-4o": { alias: "omni" },
|
|
};
|
|
|
|
expect(resolveConfiguredModelKey("sonnet", modelAliases)).toBe(
|
|
"anthropic/claude-sonnet-4-5"
|
|
);
|
|
expect(resolveConfiguredModelKey("omni", modelAliases)).toBe("openai/gpt-4o");
|
|
expect(resolveConfiguredModelKey("claude-opus-4", modelAliases)).toBe(
|
|
"anthropic/claude-opus-4"
|
|
);
|
|
expect(resolveConfiguredModelKey("openai/o3", modelAliases)).toBe("openai/o3");
|
|
expect(resolveConfiguredModelKey(" ", modelAliases)).toBeNull();
|
|
});
|
|
|
|
it("builds deduped allowlist keys from defaults and aliases", () => {
|
|
const snapshot: GatewayModelPolicySnapshot = {
|
|
config: {
|
|
agents: {
|
|
defaults: {
|
|
model: {
|
|
primary: "sonnet",
|
|
fallbacks: ["omni", "anthropic/claude-sonnet-4-5", ""],
|
|
},
|
|
models: {
|
|
"anthropic/claude-sonnet-4-5": { alias: "sonnet" },
|
|
"openai/gpt-4o": { alias: "omni" },
|
|
"openai/o3": {},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
};
|
|
|
|
expect(buildAllowedModelKeys(snapshot)).toEqual([
|
|
"anthropic/claude-sonnet-4-5",
|
|
"openai/gpt-4o",
|
|
"openai/o3",
|
|
]);
|
|
});
|
|
|
|
it("filters model catalog and appends configured extras", () => {
|
|
const snapshot: GatewayModelPolicySnapshot = {
|
|
config: {
|
|
agents: {
|
|
defaults: {
|
|
model: "sonnet",
|
|
models: {
|
|
"anthropic/claude-sonnet-4-5": { alias: "sonnet" },
|
|
"openai/o3": {},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
};
|
|
const catalog = [
|
|
{
|
|
provider: "anthropic",
|
|
id: "claude-sonnet-4-5",
|
|
name: "Claude Sonnet 4.5",
|
|
},
|
|
{
|
|
provider: "openai",
|
|
id: "gpt-4o",
|
|
name: "GPT-4o",
|
|
},
|
|
];
|
|
|
|
expect(buildGatewayModelChoices(catalog, snapshot)).toEqual([
|
|
{
|
|
provider: "anthropic",
|
|
id: "claude-sonnet-4-5",
|
|
name: "Claude Sonnet 4.5",
|
|
},
|
|
{
|
|
provider: "openai",
|
|
id: "o3",
|
|
name: "openai/o3",
|
|
},
|
|
]);
|
|
});
|
|
|
|
it("returns the catalog unchanged when no config allowlist is present", () => {
|
|
const catalog = [
|
|
{
|
|
provider: "anthropic",
|
|
id: "claude-sonnet-4-5",
|
|
name: "Claude Sonnet 4.5",
|
|
},
|
|
];
|
|
|
|
expect(buildGatewayModelChoices(catalog, null)).toEqual(catalog);
|
|
});
|
|
});
|