openclaw/src/agents/models-config.openrouter-routing.test.ts

118 lines
3.1 KiB
TypeScript

import { describe, expect, it } from "vitest";
import type { MoltbotConfig } from "../config/config.js";
describe("OpenRouter routing", () => {
it("config accepts openRouterRouting with 'only' field", () => {
const cfg: MoltbotConfig = {
env: {
OPENROUTER_API_KEY: "sk-or-test-key",
},
agents: {
defaults: {
model: { primary: "openrouter/anthropic/claude-sonnet-4-5" },
models: {
"openrouter/anthropic/claude-sonnet-4-5": {
alias: "Claude Sonnet",
compat: {
openRouterRouting: {
only: ["anthropic"],
},
},
},
},
},
},
};
const modelKey = "openrouter/anthropic/claude-sonnet-4-5";
const modelConfig = cfg.agents?.defaults?.models?.[modelKey];
expect(modelConfig?.compat?.openRouterRouting).toEqual({
only: ["anthropic"],
});
});
it("config accepts openRouterRouting with 'order' field", () => {
const cfg: MoltbotConfig = {
env: {
OPENROUTER_API_KEY: "sk-or-test-key",
},
agents: {
defaults: {
model: { primary: "openrouter/openai/gpt-5.2" },
models: {
"openrouter/openai/gpt-5.2": {
alias: "GPT-5.2",
compat: {
openRouterRouting: {
order: ["anthropic", "openai"],
},
},
},
},
},
},
};
const modelKey = "openrouter/openai/gpt-5.2";
const modelConfig = cfg.agents?.defaults?.models?.[modelKey];
expect(modelConfig?.compat?.openRouterRouting).toEqual({
order: ["anthropic", "openai"],
});
});
it("validates openRouterRouting config shape", () => {
const testCases = [
{
name: "only with single provider",
config: { only: ["anthropic"] },
},
{
name: "only with multiple providers",
config: { only: ["anthropic", "openai"] },
},
{
name: "order with single provider",
config: { order: ["anthropic"] },
},
{
name: "order with multiple providers",
config: { order: ["anthropic", "openai"] },
},
{
name: "empty routing",
config: {},
},
{
name: "both only and order",
config: { only: ["anthropic"], order: ["openai"] },
},
];
for (const testCase of testCases) {
const cfg: MoltbotConfig = {
env: { OPENROUTER_API_KEY: "sk-or-test-key" },
agents: {
defaults: {
model: { primary: "openrouter/anthropic/claude-sonnet-4-5" },
models: {
"openrouter/anthropic/claude-sonnet-4-5": {
compat: {
openRouterRouting: testCase.config,
},
},
},
},
},
};
expect(() => {
const modelConfig =
cfg.agents?.defaults?.models?.["openrouter/anthropic/claude-sonnet-4-5"];
expect(modelConfig?.compat?.openRouterRouting).toEqual(testCase.config);
}).not.toThrow();
}
});
});