test: tighten fireworks provider assertions

This commit is contained in:
Shakker
2026-05-08 18:57:27 +01:00
parent 7f5df0b97d
commit db883ec26a

View File

@@ -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,