diff --git a/packages/types/src/providers/fireworks.ts b/packages/types/src/providers/fireworks.ts index c9017c54cde..d2a7da291a6 100644 --- a/packages/types/src/providers/fireworks.ts +++ b/packages/types/src/providers/fireworks.ts @@ -17,6 +17,7 @@ export type FireworksModelId = | "accounts/fireworks/models/glm-4p5-air" | "accounts/fireworks/models/glm-4p6" | "accounts/fireworks/models/glm-4p7" + | "accounts/fireworks/models/glm-5" | "accounts/fireworks/models/gpt-oss-20b" | "accounts/fireworks/models/gpt-oss-120b" | "accounts/fireworks/models/llama-v3p3-70b-instruct" @@ -210,6 +211,16 @@ export const fireworksModels = { description: "Z.ai GLM-4.7 is the latest coding model with exceptional performance on complex programming tasks. Features improved reasoning capabilities and enhanced code generation quality.", }, + "accounts/fireworks/models/glm-5": { + maxTokens: 25344, + contextWindow: 198000, + supportsImages: false, + supportsPromptCache: false, + inputPrice: 0.55, + outputPrice: 2.19, + description: + "Z.ai GLM-5 is the newest generation model from the GLM family with advanced reasoning, coding, and intelligent agent capabilities.", + }, "accounts/fireworks/models/llama-v3p3-70b-instruct": { maxTokens: 16384, contextWindow: 131072, diff --git a/src/api/providers/__tests__/fireworks.spec.ts b/src/api/providers/__tests__/fireworks.spec.ts index 79f69f868b1..5d5e6042929 100644 --- a/src/api/providers/__tests__/fireworks.spec.ts +++ b/src/api/providers/__tests__/fireworks.spec.ts @@ -308,6 +308,27 @@ describe("FireworksHandler", () => { ) }) + it("should return GLM-5 model with correct configuration", () => { + const testModelId: FireworksModelId = "accounts/fireworks/models/glm-5" + 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-5 is the newest generation model"), + }), + ) + }) + it("should return gpt-oss-20b model with correct configuration", () => { const testModelId: FireworksModelId = "accounts/fireworks/models/gpt-oss-20b" const handlerWithModel = new FireworksHandler({