From ff3c1c9a422c2ede94a01c6c5fb3fb19260243d5 Mon Sep 17 00:00:00 2001 From: Roo Code Date: Sun, 24 Aug 2025 19:13:44 +0000 Subject: [PATCH 1/2] feat: add Deepseek v3.1 to Fireworks AI provider - Added accounts/fireworks/models/deepseek-v3p1 to FireworksModelId type - Added configuration for Deepseek v3.1 model with specs matching v3 - Added test case to verify Deepseek v3.1 model configuration - Resolves #7374 --- packages/types/src/providers/fireworks.ts | 11 ++++++++++ src/api/providers/__tests__/fireworks.spec.ts | 21 +++++++++++++++++++ 2 files changed, 32 insertions(+) diff --git a/packages/types/src/providers/fireworks.ts b/packages/types/src/providers/fireworks.ts index 79c1d314cf..7b134b2644 100644 --- a/packages/types/src/providers/fireworks.ts +++ b/packages/types/src/providers/fireworks.ts @@ -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: 128000, + supportsImages: false, + supportsPromptCache: false, + inputPrice: 0.9, + outputPrice: 0.9, + 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, diff --git a/src/api/providers/__tests__/fireworks.spec.ts b/src/api/providers/__tests__/fireworks.spec.ts index e6f0467e08..65fbc06a76 100644 --- a/src/api/providers/__tests__/fireworks.spec.ts +++ b/src/api/providers/__tests__/fireworks.spec.ts @@ -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: 128000, + supportsImages: false, + supportsPromptCache: false, + inputPrice: 0.9, + outputPrice: 0.9, + 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({ From d68c8e542e4b6edccca8a874efec002572d18e0c Mon Sep 17 00:00:00 2001 From: Roo Code Date: Sun, 24 Aug 2025 22:30:06 +0000 Subject: [PATCH 2/2] fix: update Deepseek v3.1 configuration with correct context length and pricing - Updated context length from 128,000 to 163,840 tokens - Updated input pricing from $0.90 to $0.56 per 1M tokens - Updated output pricing from $0.90 to $1.68 per 1M tokens - Updated test expectations to match new configuration --- packages/types/src/providers/fireworks.ts | 6 +++--- src/api/providers/__tests__/fireworks.spec.ts | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/packages/types/src/providers/fireworks.ts b/packages/types/src/providers/fireworks.ts index 7b134b2644..45144b666f 100644 --- a/packages/types/src/providers/fireworks.ts +++ b/packages/types/src/providers/fireworks.ts @@ -65,11 +65,11 @@ export const fireworksModels = { }, "accounts/fireworks/models/deepseek-v3p1": { maxTokens: 16384, - contextWindow: 128000, + contextWindow: 163840, supportsImages: false, supportsPromptCache: false, - inputPrice: 0.9, - outputPrice: 0.9, + 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.", }, diff --git a/src/api/providers/__tests__/fireworks.spec.ts b/src/api/providers/__tests__/fireworks.spec.ts index 65fbc06a76..ed1e119a99 100644 --- a/src/api/providers/__tests__/fireworks.spec.ts +++ b/src/api/providers/__tests__/fireworks.spec.ts @@ -190,11 +190,11 @@ describe("FireworksHandler", () => { expect(model.info).toEqual( expect.objectContaining({ maxTokens: 16384, - contextWindow: 128000, + contextWindow: 163840, supportsImages: false, supportsPromptCache: false, - inputPrice: 0.9, - outputPrice: 0.9, + inputPrice: 0.56, + outputPrice: 1.68, description: expect.stringContaining("DeepSeek v3.1 is an improved version"), }), )