diff --git a/extensions/vercel-ai-gateway/provider-catalog.test.ts b/extensions/vercel-ai-gateway/provider-catalog.test.ts index d51c8c2a58c..7cc09bb06b9 100644 --- a/extensions/vercel-ai-gateway/provider-catalog.test.ts +++ b/extensions/vercel-ai-gateway/provider-catalog.test.ts @@ -5,36 +5,35 @@ import { buildVercelAiGatewayProvider, } from "./provider-catalog.js"; +const STATIC_MODEL_IDS = [ + "anthropic/claude-opus-4.6", + "openai/gpt-5.4", + "openai/gpt-5.4-pro", + "moonshotai/kimi-k2.6", +]; + describe("vercel ai gateway provider catalog", () => { it("builds the bundled Vercel AI Gateway defaults", async () => { const provider = await buildVercelAiGatewayProvider(); - expect(provider.baseUrl).toBe(VERCEL_AI_GATEWAY_BASE_URL); - expect(provider.api).toBe("anthropic-messages"); - expect(provider.models?.map((model) => model.id)).toEqual( - expect.arrayContaining([ - "anthropic/claude-opus-4.6", - "openai/gpt-5.4", - "openai/gpt-5.4-pro", - "moonshotai/kimi-k2.6", - ]), - ); + expect(provider).toStrictEqual({ + baseUrl: VERCEL_AI_GATEWAY_BASE_URL, + api: "anthropic-messages", + models: getStaticVercelAiGatewayModelCatalog(), + }); }); it("exposes the static fallback model catalog", () => { - expect(getStaticVercelAiGatewayModelCatalog().map((model) => model.id)).toEqual( - expect.arrayContaining([ - "anthropic/claude-opus-4.6", - "openai/gpt-5.4", - "openai/gpt-5.4-pro", - "moonshotai/kimi-k2.6", - ]), + expect(getStaticVercelAiGatewayModelCatalog().map((model) => model.id)).toStrictEqual( + STATIC_MODEL_IDS, ); }); it("builds an offline static provider catalog", () => { - expect(buildStaticVercelAiGatewayProvider().models?.map((model) => model.id)).toEqual( - expect.arrayContaining(["moonshotai/kimi-k2.6"]), - ); + expect(buildStaticVercelAiGatewayProvider()).toStrictEqual({ + baseUrl: VERCEL_AI_GATEWAY_BASE_URL, + api: "anthropic-messages", + models: getStaticVercelAiGatewayModelCatalog(), + }); }); });