Skip to content

Commit 26dbac0

Browse files
committed
Fix more tests
1 parent 6941009 commit 26dbac0

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/core/webview/__tests__/webviewMessageHandler.spec.ts

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
195195
litellm: mockModels,
196196
"kilocode-openrouter": mockModels,
197197
ollama: mockModels, // kilocode_change
198+
deepinfra: mockModels, // kilocode_change
198199
lmstudio: {},
199200
},
200201
})
@@ -283,6 +284,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
283284
litellm: {},
284285
"kilocode-openrouter": mockModels,
285286
ollama: mockModels, // kilocode_change
287+
deepinfra: mockModels, // kilocode_change
286288
lmstudio: {},
287289
},
288290
})
@@ -361,6 +363,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
361363
.mockRejectedValueOnce(new Error("Unbound API error")) // unbound
362364
.mockResolvedValueOnce({}) // kilocode-openrouter - Success
363365
.mockRejectedValueOnce(new Error("Ollama API error")) // kilocode_change
366+
.mockRejectedValueOnce({}) // kilocode_change deepinfra
364367
.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm
365368

366369
await webviewMessageHandler(mockClineProvider, {

0 commit comments

Comments
 (0)