@@ -372,6 +372,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
372372 . mockResolvedValueOnce ( mockModels ) // vercel-ai-gateway
373373 . mockResolvedValueOnce ( mockModels ) // deepinfra
374374 . mockResolvedValueOnce ( mockModels ) // roo
375+ . mockRejectedValueOnce ( new Error ( "Cognima API error" ) ) // cognima
375376 . mockRejectedValueOnce ( new Error ( "LiteLLM connection failed" ) ) // litellm
376377
377378 await webviewMessageHandler ( mockClineProvider , {
@@ -413,6 +414,13 @@ describe("webviewMessageHandler - requestRouterModels", () => {
413414 values : { provider : "unbound" } ,
414415 } )
415416
417+ expect ( mockClineProvider . postMessageToWebview ) . toHaveBeenCalledWith ( {
418+ type : "singleRouterModelFetchResponse" ,
419+ success : false ,
420+ error : "Cognima API error" ,
421+ values : { provider : "cognima" } ,
422+ } )
423+
416424 expect ( mockClineProvider . postMessageToWebview ) . toHaveBeenCalledWith ( {
417425 type : "singleRouterModelFetchResponse" ,
418426 success : false ,
@@ -431,6 +439,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
431439 . mockRejectedValueOnce ( new Error ( "Vercel AI Gateway error" ) ) // vercel-ai-gateway
432440 . mockRejectedValueOnce ( new Error ( "DeepInfra API error" ) ) // deepinfra
433441 . mockRejectedValueOnce ( new Error ( "Roo API error" ) ) // roo
442+ . mockRejectedValueOnce ( new Error ( "Cognima API error" ) ) // cognima
434443 . mockRejectedValueOnce ( new Error ( "LiteLLM connection failed" ) ) // litellm
435444
436445 await webviewMessageHandler ( mockClineProvider , {
@@ -469,15 +478,15 @@ describe("webviewMessageHandler - requestRouterModels", () => {
469478 expect ( mockClineProvider . postMessageToWebview ) . toHaveBeenCalledWith ( {
470479 type : "singleRouterModelFetchResponse" ,
471480 success : false ,
472- error : "DeepInfra API error" ,
473- values : { provider : "deepinfra " } ,
481+ error : "Vercel AI Gateway error" ,
482+ values : { provider : "vercel-ai-gateway " } ,
474483 } )
475484
476485 expect ( mockClineProvider . postMessageToWebview ) . toHaveBeenCalledWith ( {
477486 type : "singleRouterModelFetchResponse" ,
478487 success : false ,
479- error : "Vercel AI Gateway error" ,
480- values : { provider : "vercel-ai-gateway " } ,
488+ error : "DeepInfra API error" ,
489+ values : { provider : "deepinfra " } ,
481490 } )
482491
483492 expect ( mockClineProvider . postMessageToWebview ) . toHaveBeenCalledWith ( {
@@ -487,6 +496,13 @@ describe("webviewMessageHandler - requestRouterModels", () => {
487496 values : { provider : "roo" } ,
488497 } )
489498
499+ expect ( mockClineProvider . postMessageToWebview ) . toHaveBeenCalledWith ( {
500+ type : "singleRouterModelFetchResponse" ,
501+ success : false ,
502+ error : "Cognima API error" ,
503+ values : { provider : "cognima" } ,
504+ } )
505+
490506 expect ( mockClineProvider . postMessageToWebview ) . toHaveBeenCalledWith ( {
491507 type : "singleRouterModelFetchResponse" ,
492508 success : false ,
0 commit comments