@@ -231,6 +231,50 @@ describe("ChutesHandler", () => {
231231 )
232232 } )
233233
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+
234278 it ( "completePrompt method should return text from Chutes API" , async ( ) => {
235279 const expectedResponse = "This is a test response from Chutes"
236280 mockCreate . mockResolvedValueOnce ( { choices : [ { message : { content : expectedResponse } } ] } )
0 commit comments