Skip to content

Commit 059b143

Browse files
roomote[bot]roomote
authored andcommitted
feat: add GLM-4.6 model to Fireworks provider (RooCodeInc#8754)
Co-authored-by: Roo Code <[email protected]>
1 parent ef698cd commit 059b143

File tree

2 files changed

+32
-0
lines changed

2 files changed

+32
-0
lines changed

packages/types/src/providers/fireworks.ts

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ export type FireworksModelId =
1010
| "accounts/fireworks/models/deepseek-v3p1"
1111
| "accounts/fireworks/models/glm-4p5"
1212
| "accounts/fireworks/models/glm-4p5-air"
13+
| "accounts/fireworks/models/glm-4p6"
1314
| "accounts/fireworks/models/gpt-oss-20b"
1415
| "accounts/fireworks/models/gpt-oss-120b"
1516

@@ -105,6 +106,16 @@ export const fireworksModels = {
105106
description:
106107
"Z.ai GLM-4.5-Air with 106B total parameters and 12B active parameters. Features unified reasoning, coding, and intelligent agent capabilities.",
107108
},
109+
"accounts/fireworks/models/glm-4p6": {
110+
maxTokens: 25344,
111+
contextWindow: 198000,
112+
supportsImages: false,
113+
supportsPromptCache: false,
114+
inputPrice: 0.55,
115+
outputPrice: 2.19,
116+
description:
117+
"Z.ai GLM-4.6 is an advanced coding model with exceptional performance on complex programming tasks. Features improved reasoning capabilities and enhanced code generation quality, making it ideal for software development workflows.",
118+
},
108119
"accounts/fireworks/models/gpt-oss-20b": {
109120
maxTokens: 16384,
110121
contextWindow: 128000,

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

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,27 @@ describe("FireworksHandler", () => {
242242
)
243243
})
244244

245+
it("should return GLM-4.6 model with correct configuration", () => {
246+
const testModelId: FireworksModelId = "accounts/fireworks/models/glm-4p6"
247+
const handlerWithModel = new FireworksHandler({
248+
apiModelId: testModelId,
249+
fireworksApiKey: "test-fireworks-api-key",
250+
})
251+
const model = handlerWithModel.getModel()
252+
expect(model.id).toBe(testModelId)
253+
expect(model.info).toEqual(
254+
expect.objectContaining({
255+
maxTokens: 25344,
256+
contextWindow: 198000,
257+
supportsImages: false,
258+
supportsPromptCache: false,
259+
inputPrice: 0.55,
260+
outputPrice: 2.19,
261+
description: expect.stringContaining("Z.ai GLM-4.6 is an advanced coding model"),
262+
}),
263+
)
264+
})
265+
245266
it("should return gpt-oss-20b model with correct configuration", () => {
246267
const testModelId: FireworksModelId = "accounts/fireworks/models/gpt-oss-20b"
247268
const handlerWithModel = new FireworksHandler({

0 commit comments

Comments
 (0)