Skip to content

Commit 262033d

Browse files
feat: add DeepSeek V3.1 model to Chutes AI provider (#7295)
Co-authored-by: Roo Code <[email protected]>
1 parent 7137c19 commit 262033d

File tree

2 files changed

+32
-0
lines changed

2 files changed

+32
-0
lines changed

packages/types/src/providers/chutes.ts

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ export type ChutesModelId =
55
| "deepseek-ai/DeepSeek-R1-0528"
66
| "deepseek-ai/DeepSeek-R1"
77
| "deepseek-ai/DeepSeek-V3"
8+
| "deepseek-ai/DeepSeek-V3.1"
89
| "unsloth/Llama-3.3-70B-Instruct"
910
| "chutesai/Llama-4-Scout-17B-16E-Instruct"
1011
| "unsloth/Mistral-Nemo-Instruct-2407"
@@ -60,6 +61,15 @@ export const chutesModels = {
6061
outputPrice: 0,
6162
description: "DeepSeek V3 model.",
6263
},
64+
"deepseek-ai/DeepSeek-V3.1": {
65+
maxTokens: 32768,
66+
contextWindow: 163840,
67+
supportsImages: false,
68+
supportsPromptCache: false,
69+
inputPrice: 0,
70+
outputPrice: 0,
71+
description: "DeepSeek V3.1 model.",
72+
},
6373
"unsloth/Llama-3.3-70B-Instruct": {
6474
maxTokens: 32768, // From Groq
6575
contextWindow: 131072, // From Groq

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

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,28 @@ describe("ChutesHandler", () => {
163163
expect(model.info).toEqual(expect.objectContaining(chutesModels[testModelId]))
164164
})
165165

166+
it("should return DeepSeek V3.1 model with correct configuration", () => {
167+
const testModelId: ChutesModelId = "deepseek-ai/DeepSeek-V3.1"
168+
const handlerWithModel = new ChutesHandler({
169+
apiModelId: testModelId,
170+
chutesApiKey: "test-chutes-api-key",
171+
})
172+
const model = handlerWithModel.getModel()
173+
expect(model.id).toBe(testModelId)
174+
expect(model.info).toEqual(
175+
expect.objectContaining({
176+
maxTokens: 32768,
177+
contextWindow: 163840,
178+
supportsImages: false,
179+
supportsPromptCache: false,
180+
inputPrice: 0,
181+
outputPrice: 0,
182+
description: "DeepSeek V3.1 model.",
183+
temperature: 0.5, // Non-R1 DeepSeek models use default temperature
184+
}),
185+
)
186+
})
187+
166188
it("should return Qwen3-235B-A22B-Instruct-2507 model with correct configuration", () => {
167189
const testModelId: ChutesModelId = "Qwen/Qwen3-235B-A22B-Instruct-2507"
168190
const handlerWithModel = new ChutesHandler({

0 commit comments

Comments
 (0)