@@ -23,6 +23,7 @@ import {
2323 vscodeLlmDefaultModelId ,
2424 moonshotDefaultModelId ,
2525 moonshotModels ,
26+ kimiCodeDefaultModelInfo ,
2627 providerIdentifiers ,
2728} from "@roo-code/types"
2829
@@ -1079,6 +1080,33 @@ describe("useSelectedModel", () => {
10791080 } )
10801081 } )
10811082
1083+ describe ( "Kimi Code provider" , ( ) => {
1084+ it ( "should resolve the configured model from router models" , ( ) => {
1085+ const modelInfo : ModelInfo = {
1086+ ...kimiCodeDefaultModelInfo ,
1087+ description : "Configured Kimi Code model" ,
1088+ }
1089+
1090+ mockUseRouterModels . mockReturnValue ( {
1091+ data : { "kimi-code" : { "kimi-for-coding" : modelInfo } } ,
1092+ isLoading : false ,
1093+ isError : false ,
1094+ } as any )
1095+
1096+ const apiConfiguration : ProviderSettings = {
1097+ apiProvider : providerIdentifiers . kimiCode ,
1098+ apiModelId : "kimi-for-coding" ,
1099+ }
1100+
1101+ const wrapper = createWrapper ( )
1102+ const { result } = renderHook ( ( ) => useSelectedModel ( apiConfiguration ) , { wrapper } )
1103+
1104+ expect ( result . current . provider ) . toBe ( providerIdentifiers . kimiCode )
1105+ expect ( result . current . id ) . toBe ( "kimi-for-coding" )
1106+ expect ( result . current . info ) . toEqual ( modelInfo )
1107+ } )
1108+ } )
1109+
10821110 describe ( "moonshot provider" , ( ) => {
10831111 beforeEach ( ( ) => {
10841112 mockUseRouterModels . mockReturnValue ( {
0 commit comments