Skip to content

Commit 0dbc2cb

Browse files
authored
Update src/api/providers/__tests__/fireworks.spec.ts
1 parent cb77a9b commit 0dbc2cb

1 file changed

Lines changed: 8 additions & 7 deletions

File tree

src/api/providers/__tests__/fireworks.spec.ts

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -95,16 +95,17 @@ describe("FireworksHandler", () => {
9595
})
9696

9797
it.each([
98-
"accounts/fireworks/models/glm-5p1",
99-
"accounts/fireworks/models/kimi-k2p6",
100-
"accounts/fireworks/models/deepseek-v4-pro",
101-
] as const satisfies readonly FireworksModelId[])("should expose newly added model %s", (modelId) => {
98+
{ modelId: "accounts/fireworks/models/glm-5p1" as const, contextWindow: 202752, inputPrice: 1.4, outputPrice: 4.4, cacheReadsPrice: 0.26 },
99+
{ modelId: "accounts/fireworks/models/kimi-k2p6" as const, contextWindow: 262144, inputPrice: 0.95, outputPrice: 4.0, cacheReadsPrice: 0.16 },
100+
{ modelId: "accounts/fireworks/models/deepseek-v4-pro" as const, contextWindow: 1048576, inputPrice: 1.74, outputPrice: 3.48, cacheReadsPrice: 0.14 },
101+
])("should expose newly added model $modelId", ({ modelId, contextWindow, inputPrice, outputPrice, cacheReadsPrice }) => {
102102
expect(fireworksModels[modelId]).toBeDefined()
103103
const info = fireworksModels[modelId]
104104
expect(info.maxTokens).toBeGreaterThan(0)
105-
expect(info.contextWindow).toBeGreaterThan(0)
106-
expect(info.inputPrice).toBeGreaterThanOrEqual(0)
107-
expect(info.outputPrice).toBeGreaterThanOrEqual(0)
105+
expect(info.contextWindow).toBe(contextWindow)
106+
expect(info.inputPrice).toBe(inputPrice)
107+
expect(info.outputPrice).toBe(outputPrice)
108+
expect(info.cacheReadsPrice).toBe(cacheReadsPrice)
108109
expect(info.description).toBeTruthy()
109110

110111
const handlerWithModel = new FireworksHandler({

0 commit comments

Comments
 (0)