mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-08-28 05:27:24 +00:00
feat: add DeepSeek V3.1 model to Chutes AI provider (#7295)
Co-authored-by: Roo Code <roomote@roocode.com>
This commit is contained in:
parent
7137c1902f
commit
262033ddd0
2 changed files with 32 additions and 0 deletions
|
|
@ -5,6 +5,7 @@ export type ChutesModelId =
|
|||
| "deepseek-ai/DeepSeek-R1-0528"
|
||||
| "deepseek-ai/DeepSeek-R1"
|
||||
| "deepseek-ai/DeepSeek-V3"
|
||||
| "deepseek-ai/DeepSeek-V3.1"
|
||||
| "unsloth/Llama-3.3-70B-Instruct"
|
||||
| "chutesai/Llama-4-Scout-17B-16E-Instruct"
|
||||
| "unsloth/Mistral-Nemo-Instruct-2407"
|
||||
|
|
@ -60,6 +61,15 @@ export const chutesModels = {
|
|||
outputPrice: 0,
|
||||
description: "DeepSeek V3 model.",
|
||||
},
|
||||
"deepseek-ai/DeepSeek-V3.1": {
|
||||
maxTokens: 32768,
|
||||
contextWindow: 163840,
|
||||
supportsImages: false,
|
||||
supportsPromptCache: false,
|
||||
inputPrice: 0,
|
||||
outputPrice: 0,
|
||||
description: "DeepSeek V3.1 model.",
|
||||
},
|
||||
"unsloth/Llama-3.3-70B-Instruct": {
|
||||
maxTokens: 32768, // From Groq
|
||||
contextWindow: 131072, // From Groq
|
||||
|
|
|
|||
|
|
@ -163,6 +163,28 @@ describe("ChutesHandler", () => {
|
|||
expect(model.info).toEqual(expect.objectContaining(chutesModels[testModelId]))
|
||||
})
|
||||
|
||||
it("should return DeepSeek V3.1 model with correct configuration", () => {
|
||||
const testModelId: ChutesModelId = "deepseek-ai/DeepSeek-V3.1"
|
||||
const handlerWithModel = new ChutesHandler({
|
||||
apiModelId: testModelId,
|
||||
chutesApiKey: "test-chutes-api-key",
|
||||
})
|
||||
const model = handlerWithModel.getModel()
|
||||
expect(model.id).toBe(testModelId)
|
||||
expect(model.info).toEqual(
|
||||
expect.objectContaining({
|
||||
maxTokens: 32768,
|
||||
contextWindow: 163840,
|
||||
supportsImages: false,
|
||||
supportsPromptCache: false,
|
||||
inputPrice: 0,
|
||||
outputPrice: 0,
|
||||
description: "DeepSeek V3.1 model.",
|
||||
temperature: 0.5, // Non-R1 DeepSeek models use default temperature
|
||||
}),
|
||||
)
|
||||
})
|
||||
|
||||
it("should return Qwen3-235B-A22B-Instruct-2507 model with correct configuration", () => {
|
||||
const testModelId: ChutesModelId = "Qwen/Qwen3-235B-A22B-Instruct-2507"
|
||||
const handlerWithModel = new ChutesHandler({
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue