mirror of https://github.com/openclaw/openclaw.git
81 lines
2.3 KiB
TypeScript
81 lines
2.3 KiB
TypeScript
import type { ModelRegistry } from "@mariozechner/pi-coding-agent";
|
|
import { describe, expect, it } from "vitest";
|
|
import { cloneFirstTemplateModel, matchesExactOrPrefix } from "./provider-model-helpers.js";
|
|
import type { ProviderResolveDynamicModelContext, ProviderRuntimeModel } from "./types.js";
|
|
|
|
function createContext(models: ProviderRuntimeModel[]): ProviderResolveDynamicModelContext {
|
|
return {
|
|
provider: "test-provider",
|
|
modelId: "next-model",
|
|
modelRegistry: {
|
|
find(providerId: string, modelId: string) {
|
|
return (
|
|
models.find((model) => model.provider === providerId && model.id === modelId) ?? null
|
|
);
|
|
},
|
|
} as ModelRegistry,
|
|
};
|
|
}
|
|
|
|
function createTemplateModel(
|
|
id: string,
|
|
overrides: Partial<ProviderRuntimeModel> = {},
|
|
): ProviderRuntimeModel {
|
|
return {
|
|
id,
|
|
name: id,
|
|
provider: "test-provider",
|
|
api: "openai-completions",
|
|
...overrides,
|
|
} as ProviderRuntimeModel;
|
|
}
|
|
|
|
describe("cloneFirstTemplateModel", () => {
|
|
it.each([
|
|
{
|
|
name: "clones the first matching template and applies patches",
|
|
params: {
|
|
providerId: "test-provider",
|
|
modelId: " next-model ",
|
|
templateIds: ["missing", "template-a", "template-b"],
|
|
ctx: createContext([createTemplateModel("template-a", { name: "Template A" })]),
|
|
patch: { reasoning: true },
|
|
},
|
|
expected: {
|
|
id: "next-model",
|
|
name: "next-model",
|
|
provider: "test-provider",
|
|
api: "openai-completions",
|
|
reasoning: true,
|
|
},
|
|
},
|
|
{
|
|
name: "returns undefined when no template exists",
|
|
params: {
|
|
providerId: "test-provider",
|
|
modelId: "next-model",
|
|
templateIds: ["missing"],
|
|
ctx: createContext([]),
|
|
},
|
|
expected: undefined,
|
|
},
|
|
] as const)("$name", ({ params, expected }) => {
|
|
const model = cloneFirstTemplateModel(params);
|
|
if (expected == null) {
|
|
expect(model).toBeUndefined();
|
|
return;
|
|
}
|
|
expect(model).toMatchObject(expected);
|
|
});
|
|
});
|
|
|
|
describe("matchesExactOrPrefix", () => {
|
|
it.each([
|
|
["MiniMax-M2.7", ["minimax-m2.7"], true],
|
|
["minimax-m2.7-highspeed", ["MiniMax-M2.7"], true],
|
|
["glm-5", ["minimax-m2.7"], false],
|
|
] as const)("matches %s against prefixes", (id, candidates, expected) => {
|
|
expect(matchesExactOrPrefix(id, candidates)).toBe(expected);
|
|
});
|
|
});
|