feat: add Deepseek v3.1 to Fireworks AI provider (#7375)

Co-authored-by: Roo Code <roomote@roocode.com>
This commit is contained in:
roomote[bot] 2025-08-24 18:43:06 -04:00 committed by GitHub
parent 1322e9a981
commit 8367b1ac92
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 32 additions and 0 deletions

View file

@ -6,6 +6,7 @@ export type FireworksModelId =
| "accounts/fireworks/models/qwen3-coder-480b-a35b-instruct"
| "accounts/fireworks/models/deepseek-r1-0528"
| "accounts/fireworks/models/deepseek-v3"
| "accounts/fireworks/models/deepseek-v3p1"
| "accounts/fireworks/models/glm-4p5"
| "accounts/fireworks/models/glm-4p5-air"
| "accounts/fireworks/models/gpt-oss-20b"
@ -62,6 +63,16 @@ export const fireworksModels = {
description:
"A strong Mixture-of-Experts (MoE) language model with 671B total parameters with 37B activated for each token from Deepseek. Note that fine-tuning for this model is only available through contacting fireworks at https://fireworks.ai/company/contact-us.",
},
"accounts/fireworks/models/deepseek-v3p1": {
maxTokens: 16384,
contextWindow: 163840,
supportsImages: false,
supportsPromptCache: false,
inputPrice: 0.56,
outputPrice: 1.68,
description:
"DeepSeek v3.1 is an improved version of the v3 model with enhanced performance, better reasoning capabilities, and improved code generation. This Mixture-of-Experts (MoE) model maintains the same 671B total parameters with 37B activated per token.",
},
"accounts/fireworks/models/glm-4p5": {
maxTokens: 16384,
contextWindow: 128000,

View file

@ -179,6 +179,27 @@ describe("FireworksHandler", () => {
)
})
it("should return DeepSeek V3.1 model with correct configuration", () => {
const testModelId: FireworksModelId = "accounts/fireworks/models/deepseek-v3p1"
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: 16384,
contextWindow: 163840,
supportsImages: false,
supportsPromptCache: false,
inputPrice: 0.56,
outputPrice: 1.68,
description: expect.stringContaining("DeepSeek v3.1 is an improved version"),
}),
)
})
it("should return GLM-4.5 model with correct configuration", () => {
const testModelId: FireworksModelId = "accounts/fireworks/models/glm-4p5"
const handlerWithModel = new FireworksHandler({