Skip to content

Commit b7a20db

Browse files
committed
test(webview): cover Kimi Code model selection
1 parent f1ca5fe commit b7a20db

1 file changed

Lines changed: 28 additions & 0 deletions

File tree

webview-ui/src/components/ui/hooks/__tests__/useSelectedModel.spec.ts

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)