Skip to content

Commit 78d3dac

Browse files
feat(fireworks): add glm-5.1, kimi-k2.6 and deepseek-v4-pro models (#198) (#231)
* feat(fireworks): add glm-5.1, kimi-k2.6 and deepseek-v4-pro models (#198) Adds the three models requested in #198 to the Fireworks provider: - accounts/fireworks/models/glm-5p1 - accounts/fireworks/models/kimi-k2p6 - accounts/fireworks/models/deepseek-v4-pro Metadata mirrors the closest existing sibling in each family (glm-4p7, kimi-k2p5, deepseek-v3p2). Adds a parameterized test asserting the new models are exposed and selectable. * fix(fireworks): correct context window and pricing for deepseek-v4-pro, glm-5.1, kimi-k2.6 (#198) * Update src/api/providers/__tests__/fireworks.spec.ts --------- Co-authored-by: Armando Vaquera <263793884+proyectoauraorg@users.noreply.github.com> Co-authored-by: edelauna <54631123+edelauna@users.noreply.github.com>
1 parent dacc2d4 commit 78d3dac

2 files changed

Lines changed: 57 additions & 0 deletions

File tree

packages/types/src/providers/fireworks.ts

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ export type FireworksModelId =
55
| "accounts/fireworks/models/kimi-k2-instruct-0905"
66
| "accounts/fireworks/models/kimi-k2-thinking"
77
| "accounts/fireworks/models/kimi-k2p5"
8+
| "accounts/fireworks/models/kimi-k2p6"
89
| "accounts/fireworks/models/minimax-m2"
910
| "accounts/fireworks/models/minimax-m2p1"
1011
| "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507"
@@ -13,10 +14,12 @@ export type FireworksModelId =
1314
| "accounts/fireworks/models/deepseek-v3"
1415
| "accounts/fireworks/models/deepseek-v3p1"
1516
| "accounts/fireworks/models/deepseek-v3p2"
17+
| "accounts/fireworks/models/deepseek-v4-pro"
1618
| "accounts/fireworks/models/glm-4p5"
1719
| "accounts/fireworks/models/glm-4p5-air"
1820
| "accounts/fireworks/models/glm-4p6"
1921
| "accounts/fireworks/models/glm-4p7"
22+
| "accounts/fireworks/models/glm-5p1"
2023
| "accounts/fireworks/models/gpt-oss-20b"
2124
| "accounts/fireworks/models/gpt-oss-120b"
2225
| "accounts/fireworks/models/llama-v3p3-70b-instruct"
@@ -240,4 +243,37 @@ export const fireworksModels = {
240243
description:
241244
"Llama 4 Scout is a smaller, faster variant of Llama 4 with multimodal capabilities, ideal for quick iterations and cost-effective deployments.",
242245
},
246+
"accounts/fireworks/models/kimi-k2p6": {
247+
maxTokens: 16384,
248+
contextWindow: 262144,
249+
supportsImages: true,
250+
supportsPromptCache: true,
251+
inputPrice: 0.95,
252+
outputPrice: 4.0,
253+
cacheReadsPrice: 0.16,
254+
description:
255+
"Kimi K2.6 is Moonshot AI's latest flagship agentic model, building on K2.5 with stronger long-horizon reasoning, multi-step tool use, and unified vision/text understanding.",
256+
},
257+
"accounts/fireworks/models/deepseek-v4-pro": {
258+
maxTokens: 16384,
259+
contextWindow: 1048576,
260+
supportsImages: false,
261+
supportsPromptCache: true,
262+
inputPrice: 1.74,
263+
outputPrice: 3.48,
264+
cacheReadsPrice: 0.14,
265+
description:
266+
"DeepSeek V4 Pro is the latest iteration of the DeepSeek model family, with improved reasoning, code generation, and instruction following over the V3 series.",
267+
},
268+
"accounts/fireworks/models/glm-5p1": {
269+
maxTokens: 25344,
270+
contextWindow: 202752,
271+
supportsImages: false,
272+
supportsPromptCache: true,
273+
inputPrice: 1.4,
274+
outputPrice: 4.4,
275+
cacheReadsPrice: 0.26,
276+
description:
277+
"Z.ai GLM-5.1 is the latest coding-focused model in the GLM family, with exceptional performance on complex programming tasks and enhanced reasoning and code generation quality.",
278+
},
243279
} as const satisfies Record<string, ModelInfo>

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

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,27 @@ describe("FireworksHandler", () => {
9494
expect(model.info).toEqual(expect.objectContaining(fireworksModels[testModelId]))
9595
})
9696

97+
it.each([
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 }) => {
102+
expect(fireworksModels[modelId]).toBeDefined()
103+
const info = fireworksModels[modelId]
104+
expect(info.maxTokens).toBeGreaterThan(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)
109+
expect(info.description).toBeTruthy()
110+
111+
const handlerWithModel = new FireworksHandler({
112+
apiModelId: modelId,
113+
fireworksApiKey: "test-fireworks-api-key",
114+
})
115+
expect(handlerWithModel.getModel().id).toBe(modelId)
116+
})
117+
97118
it("should return Kimi K2 Instruct model with correct configuration", () => {
98119
const testModelId: FireworksModelId = "accounts/fireworks/models/kimi-k2-instruct"
99120
const handlerWithModel = new FireworksHandler({

0 commit comments

Comments
 (0)