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,