mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-09-10 22:41:14 +00:00
feat: add GLM-4.6 model to Fireworks provider (#8754)
Co-authored-by: Roo Code <roomote@roocode.com>
This commit is contained in:
parent
34f45f1b28
commit
8dc8452123
2 changed files with 32 additions and 0 deletions
|
|
@ -10,6 +10,7 @@ export type FireworksModelId =
|
|||
| "accounts/fireworks/models/deepseek-v3p1"
|
||||
| "accounts/fireworks/models/glm-4p5"
|
||||
| "accounts/fireworks/models/glm-4p5-air"
|
||||
| "accounts/fireworks/models/glm-4p6"
|
||||
| "accounts/fireworks/models/gpt-oss-20b"
|
||||
| "accounts/fireworks/models/gpt-oss-120b"
|
||||
|
||||
|
|
@ -105,6 +106,16 @@ export const fireworksModels = {
|
|||
description:
|
||||
"Z.ai GLM-4.5-Air with 106B total parameters and 12B active parameters. Features unified reasoning, coding, and intelligent agent capabilities.",
|
||||
},
|
||||
"accounts/fireworks/models/glm-4p6": {
|
||||
maxTokens: 25344,
|
||||
contextWindow: 198000,
|
||||
supportsImages: false,
|
||||
supportsPromptCache: false,
|
||||
inputPrice: 0.55,
|
||||
outputPrice: 2.19,
|
||||
description:
|
||||
"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.",
|
||||
},
|
||||
"accounts/fireworks/models/gpt-oss-20b": {
|
||||
maxTokens: 16384,
|
||||
contextWindow: 128000,
|
||||
|
|
|
|||
|
|
@ -242,6 +242,27 @@ describe("FireworksHandler", () => {
|
|||
)
|
||||
})
|
||||
|
||||
it("should return GLM-4.6 model with correct configuration", () => {
|
||||
const testModelId: FireworksModelId = "accounts/fireworks/models/glm-4p6"
|
||||
const handlerWithModel = new FireworksHandler({
|
||||
apiModelId: testModelId,
|
||||
fireworksApiKey: "test-fireworks-api-key",
|
||||
})
|
||||
const model = handlerWithModel.getModel()
|
||||
expect(model.id).toBe(testModelId)
|
||||
expect(model.info).toEqual(
|
||||
expect.objectContaining({
|
||||
maxTokens: 25344,
|
||||
contextWindow: 198000,
|
||||
supportsImages: false,
|
||||
supportsPromptCache: false,
|
||||
inputPrice: 0.55,
|
||||
outputPrice: 2.19,
|
||||
description: expect.stringContaining("Z.ai GLM-4.6 is an advanced coding model"),
|
||||
}),
|
||||
)
|
||||
})
|
||||
|
||||
it("should return gpt-oss-20b model with correct configuration", () => {
|
||||
const testModelId: FireworksModelId = "accounts/fireworks/models/gpt-oss-20b"
|
||||
const handlerWithModel = new FireworksHandler({
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue