From d80bcc367ee78d629b4dd8b001a851ef1884115e Mon Sep 17 00:00:00 2001 From: Armin Ronacher Date: Fri, 22 May 2026 22:59:10 +0200 Subject: [PATCH] test(ai): avoid hardcoded Fireworks router id --- packages/ai/test/fireworks-models.test.ts | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/packages/ai/test/fireworks-models.test.ts b/packages/ai/test/fireworks-models.test.ts index 0260a910..6d9bebc6 100644 --- a/packages/ai/test/fireworks-models.test.ts +++ b/packages/ai/test/fireworks-models.test.ts @@ -3,7 +3,7 @@ import type { AddressInfo } from "node:net"; import { Type } from "typebox"; import { afterEach, describe, expect, it } from "vitest"; import { findEnvKeys, getEnvApiKey } from "../src/env-api-keys.ts"; -import { getModel } from "../src/models.ts"; +import { getModel, getModels } from "../src/models.ts"; import { streamAnthropic } from "../src/providers/anthropic.ts"; import type { Context, Model, Tool } from "../src/types.ts"; @@ -38,12 +38,14 @@ describe("Fireworks models", () => { }); it("registers the Fire Pass turbo router model", () => { - const model = getModel("fireworks", "accounts/fireworks/routers/kimi-k2p5-turbo"); + const model = getModels("fireworks").find( + (candidate) => candidate.id.startsWith("accounts/fireworks/routers/") && candidate.id.endsWith("-turbo"), + ); expect(model).toBeDefined(); - expect(model.api).toBe("anthropic-messages"); - expect(model.baseUrl).toBe("https://api.fireworks.ai/inference"); - expect(model.input).toEqual(["text", "image"]); + expect(model?.api).toBe("anthropic-messages"); + expect(model?.baseUrl).toBe("https://api.fireworks.ai/inference"); + expect(model?.input).toEqual(["text", "image"]); }); it("resolves FIREWORKS_API_KEY from the environment", () => {