File tree Expand file tree Collapse file tree 2 files changed +32
-0
lines changed
packages/types/src/providers
src/api/providers/__tests__ Expand file tree Collapse file tree 2 files changed +32
-0
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 ( {
You can’t perform that action at this time.
0 commit comments