From 11150a532801a1884281973bbcb85daad94da3bc Mon Sep 17 00:00:00 2001 From: Peter Steinberger Date: Mon, 11 May 2026 07:14:13 +0100 Subject: [PATCH] test: tighten fireworks provider assertions --- extensions/fireworks/index.test.ts | 56 ++++++++++++------------------ 1 file changed, 22 insertions(+), 34 deletions(-) diff --git a/extensions/fireworks/index.test.ts b/extensions/fireworks/index.test.ts index 7e03deebdf5..0654c2eebae 100644 --- a/extensions/fireworks/index.test.ts +++ b/extensions/fireworks/index.test.ts @@ -69,18 +69,14 @@ describe("fireworks provider plugin", () => { FIREWORKS_K2_6_MODEL_ID, FIREWORKS_DEFAULT_MODEL_ID, ]); - expect(models[0]).toMatchObject({ - reasoning: false, - input: ["text", "image"], - contextWindow: FIREWORKS_K2_6_CONTEXT_WINDOW, - maxTokens: FIREWORKS_K2_6_MAX_TOKENS, - }); - expect(models[1]).toMatchObject({ - reasoning: false, - input: ["text", "image"], - contextWindow: FIREWORKS_DEFAULT_CONTEXT_WINDOW, - maxTokens: FIREWORKS_DEFAULT_MAX_TOKENS, - }); + expect(models[0]?.reasoning).toBe(false); + expect(models[0]?.input).toEqual(["text", "image"]); + expect(models[0]?.contextWindow).toBe(FIREWORKS_K2_6_CONTEXT_WINDOW); + expect(models[0]?.maxTokens).toBe(FIREWORKS_K2_6_MAX_TOKENS); + expect(models[1]?.reasoning).toBe(false); + expect(models[1]?.input).toEqual(["text", "image"]); + expect(models[1]?.contextWindow).toBe(FIREWORKS_DEFAULT_CONTEXT_WINDOW); + expect(models[1]?.maxTokens).toBe(FIREWORKS_DEFAULT_MAX_TOKENS); }); it("resolves forward-compat Fireworks model ids from the default template", async () => { @@ -93,13 +89,11 @@ describe("fireworks provider plugin", () => { }), ); - expect(resolved).toMatchObject({ - provider: "fireworks", - id: "accounts/fireworks/models/qwen3.6-plus", - api: "openai-completions", - baseUrl: FIREWORKS_BASE_URL, - reasoning: true, - }); + expect(resolved?.provider).toBe("fireworks"); + expect(resolved?.id).toBe("accounts/fireworks/models/qwen3.6-plus"); + expect(resolved?.api).toBe("openai-completions"); + expect(resolved?.baseUrl).toBe(FIREWORKS_BASE_URL); + expect(resolved?.reasoning).toBe(true); }); it("disables reasoning metadata for Fireworks Kimi dynamic models", async () => { @@ -112,11 +106,9 @@ describe("fireworks provider plugin", () => { }), ); - expect(resolved).toMatchObject({ - provider: "fireworks", - id: "accounts/fireworks/models/kimi-k2p5", - reasoning: false, - }); + expect(resolved?.provider).toBe("fireworks"); + expect(resolved?.id).toBe("accounts/fireworks/models/kimi-k2p5"); + expect(resolved?.reasoning).toBe(false); }); it("disables reasoning metadata for Fireworks Kimi k2.5 aliases", async () => { @@ -129,11 +121,9 @@ describe("fireworks provider plugin", () => { }), ); - expect(resolved).toMatchObject({ - provider: "fireworks", - id: "accounts/fireworks/routers/kimi-k2.5-turbo", - reasoning: false, - }); + expect(resolved?.provider).toBe("fireworks"); + expect(resolved?.id).toBe("accounts/fireworks/routers/kimi-k2.5-turbo"); + expect(resolved?.reasoning).toBe(false); }); it("disables reasoning metadata for Fireworks Kimi k2.6 dynamic models", async () => { @@ -146,11 +136,9 @@ describe("fireworks provider plugin", () => { }), ); - expect(resolved).toMatchObject({ - provider: "fireworks", - id: "accounts/fireworks/models/kimi-k2p6", - reasoning: false, - }); + expect(resolved?.provider).toBe("fireworks"); + expect(resolved?.id).toBe("accounts/fireworks/models/kimi-k2p6"); + expect(resolved?.reasoning).toBe(false); }); it("exposes off-only thinking policy for Fireworks Kimi models", async () => {