@@ -231,6 +231,50 @@ describe("ChutesHandler", () => {
231
231
)
232
232
} )
233
233
234
+ it ( "should return Qwen/Qwen3-Coder-480B-A35B-Instruct-FP8 model with correct configuration" , ( ) => {
235
+ const testModelId : ChutesModelId = "Qwen/Qwen3-Coder-480B-A35B-Instruct-FP8"
236
+ const handlerWithModel = new ChutesHandler ( {
237
+ apiModelId : testModelId ,
238
+ chutesApiKey : "test-chutes-api-key" ,
239
+ } )
240
+ const model = handlerWithModel . getModel ( )
241
+ expect ( model . id ) . toBe ( testModelId )
242
+ expect ( model . info ) . toEqual (
243
+ expect . objectContaining ( {
244
+ maxTokens : 32768 ,
245
+ contextWindow : 262144 ,
246
+ supportsImages : false ,
247
+ supportsPromptCache : false ,
248
+ inputPrice : 0 ,
249
+ outputPrice : 0 ,
250
+ description : "Qwen3 Coder 480B A35B Instruct FP8 model, optimized for coding tasks." ,
251
+ temperature : 0.5 , // Default temperature for non-DeepSeek models
252
+ } ) ,
253
+ )
254
+ } )
255
+
256
+ it ( "should return moonshotai/Kimi-K2-Instruct-75k model with correct configuration" , ( ) => {
257
+ const testModelId : ChutesModelId = "moonshotai/Kimi-K2-Instruct-75k"
258
+ const handlerWithModel = new ChutesHandler ( {
259
+ apiModelId : testModelId ,
260
+ chutesApiKey : "test-chutes-api-key" ,
261
+ } )
262
+ const model = handlerWithModel . getModel ( )
263
+ expect ( model . id ) . toBe ( testModelId )
264
+ expect ( model . info ) . toEqual (
265
+ expect . objectContaining ( {
266
+ maxTokens : 32768 ,
267
+ contextWindow : 75000 ,
268
+ supportsImages : false ,
269
+ supportsPromptCache : false ,
270
+ inputPrice : 0.1481 ,
271
+ outputPrice : 0.5926 ,
272
+ description : "Moonshot AI Kimi K2 Instruct model with 75k context window." ,
273
+ temperature : 0.5 , // Default temperature for non-DeepSeek models
274
+ } ) ,
275
+ )
276
+ } )
277
+
234
278
it ( "completePrompt method should return text from Chutes API" , async ( ) => {
235
279
const expectedResponse = "This is a test response from Chutes"
236
280
mockCreate . mockResolvedValueOnce ( { choices : [ { message : { content : expectedResponse } } ] } )
0 commit comments