Skip to content

Commit 8367b1a

Browse files
feat: add Deepseek v3.1 to Fireworks AI provider (#7375)
Co-authored-by: Roo Code <[email protected]>
1 parent 1322e9a commit 8367b1a

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
@@ -6,6 +6,7 @@ export type FireworksModelId =
66
| "accounts/fireworks/models/qwen3-coder-480b-a35b-instruct"
77
| "accounts/fireworks/models/deepseek-r1-0528"
88
| "accounts/fireworks/models/deepseek-v3"
9+
| "accounts/fireworks/models/deepseek-v3p1"
910
| "accounts/fireworks/models/glm-4p5"
1011
| "accounts/fireworks/models/glm-4p5-air"
1112
| "accounts/fireworks/models/gpt-oss-20b"
@@ -62,6 +63,16 @@ export const fireworksModels = {
6263
description:
6364
"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.",
6465
},
66+
"accounts/fireworks/models/deepseek-v3p1": {
67+
maxTokens: 16384,
68+
contextWindow: 163840,
69+
supportsImages: false,
70+
supportsPromptCache: false,
71+
inputPrice: 0.56,
72+
outputPrice: 1.68,
73+
description:
74+
"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.",
75+
},
6576
"accounts/fireworks/models/glm-4p5": {
6677
maxTokens: 16384,
6778
contextWindow: 128000,

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

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,27 @@ describe("FireworksHandler", () => {
179179
)
180180
})
181181

182+
it("should return DeepSeek V3.1 model with correct configuration", () => {
183+
const testModelId: FireworksModelId = "accounts/fireworks/models/deepseek-v3p1"
184+
const handlerWithModel = new FireworksHandler({
185+
apiModelId: testModelId,
186+
fireworksApiKey: "test-fireworks-api-key",
187+
})
188+
const model = handlerWithModel.getModel()
189+
expect(model.id).toBe(testModelId)
190+
expect(model.info).toEqual(
191+
expect.objectContaining({
192+
maxTokens: 16384,
193+
contextWindow: 163840,
194+
supportsImages: false,
195+
supportsPromptCache: false,
196+
inputPrice: 0.56,
197+
outputPrice: 1.68,
198+
description: expect.stringContaining("DeepSeek v3.1 is an improved version"),
199+
}),
200+
)
201+
})
202+
182203
it("should return GLM-4.5 model with correct configuration", () => {
183204
const testModelId: FireworksModelId = "accounts/fireworks/models/glm-4p5"
184205
const handlerWithModel = new FireworksHandler({

0 commit comments

Comments
 (0)