From db883ec26a24a99cd8524af2a71522fc30bb0cd9 Mon Sep 17 00:00:00 2001 From: Shakker Date: Fri, 8 May 2026 18:57:27 +0100 Subject: [PATCH] test: tighten fireworks provider assertions --- extensions/fireworks/index.test.ts | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/extensions/fireworks/index.test.ts b/extensions/fireworks/index.test.ts index 920fd69a587..52e655f8442 100644 --- a/extensions/fireworks/index.test.ts +++ b/extensions/fireworks/index.test.ts @@ -48,8 +48,12 @@ describe("fireworks provider plugin", () => { expect(provider.aliases).toEqual(["fireworks-ai"]); expect(provider.envVars).toEqual(["FIREWORKS_API_KEY"]); expect(provider.auth).toHaveLength(1); - expect(resolved?.provider.id).toBe("fireworks"); - expect(resolved?.method.id).toBe("api-key"); + expect(resolved).toBeDefined(); + if (!resolved) { + throw new Error("expected Fireworks api-key auth choice"); + } + expect(resolved.provider.id).toBe("fireworks"); + expect(resolved.method.id).toBe("api-key"); }); it("builds the Fireworks catalog", async () => { @@ -58,17 +62,22 @@ describe("fireworks provider plugin", () => { expect(catalogProvider.api).toBe("openai-completions"); expect(catalogProvider.baseUrl).toBe(FIREWORKS_BASE_URL); - expect(catalogProvider.models?.map((model) => model.id)).toEqual([ + const models = catalogProvider.models; + expect(models).toBeDefined(); + if (!models) { + throw new Error("expected Fireworks catalog models"); + } + expect(models.map((model) => model.id)).toEqual([ FIREWORKS_K2_6_MODEL_ID, FIREWORKS_DEFAULT_MODEL_ID, ]); - expect(catalogProvider.models?.[0]).toMatchObject({ + expect(models[0]).toMatchObject({ reasoning: false, input: ["text", "image"], contextWindow: FIREWORKS_K2_6_CONTEXT_WINDOW, maxTokens: FIREWORKS_K2_6_MAX_TOKENS, }); - expect(catalogProvider.models?.[1]).toMatchObject({ + expect(models[1]).toMatchObject({ reasoning: false, input: ["text", "image"], contextWindow: FIREWORKS_DEFAULT_CONTEXT_WINDOW,