test: tighten fireworks provider assertions

This commit is contained in:
Peter Steinberger
2026-05-11 07:14:13 +01:00
parent 23709c5481
commit 11150a5328

View File

@@ -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 () => {