Vegard Stikbakke
cc2db98002
Use the new provider-specific Xiaomi Token Plan catalogs from models.dev instead of cloning the API-billing Xiaomi catalog into every token-plan region. This removes API-billing-only models such as mimo-v2-omni from xiaomi-token-plan-{cn,ams,sgp}.
18 lines
648 B
TypeScript
18 lines
648 B
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { getModel, getModels } from "../src/compat.ts";
|
|
|
|
describe("Xiaomi MiMo models", () => {
|
|
it.each(["mimo-v2-flash", "mimo-v2-omni"] as const)("keeps %s on the API billing provider", (modelId) => {
|
|
expect(getModel("xiaomi", modelId)).toBeDefined();
|
|
});
|
|
|
|
it.each(["xiaomi-token-plan-cn", "xiaomi-token-plan-ams", "xiaomi-token-plan-sgp"] as const)(
|
|
"omits API-billing-only models from %s",
|
|
(provider) => {
|
|
const modelIds = getModels(provider).map((model) => model.id);
|
|
expect(modelIds).not.toContain("mimo-v2-flash");
|
|
expect(modelIds).not.toContain("mimo-v2-omni");
|
|
},
|
|
);
|
|
});
|